rt_benchs/communication_techniques/include/mcringbuffer_comm.h

70 lines
1.3 KiB
C

#ifndef _SPECIFIC_COMM_H_
#define _SPECIFIC_COMM_H_ 1
/* Non standard include */
#include <commtech.h>
#define SHARED_SPACE_SIZE (2 * BUF_SIZE)
#define SHARED_SPACE_VOIDPTR (SHARED_SPACE_SIZE / sizeof(void *))
struct control
{
volatile int read;
volatile int write;
};
struct cons
{
int localWrite;
int nextRead;
int rBatch;
};
struct prod
{
int localRead;
int nextWrite;
int wBatch;
};
struct comm
{
struct control ctrl __attribute__ ((aligned (CACHE_LINE_SIZE)));
struct prod prod __attribute__ ((aligned (CACHE_LINE_SIZE)));
struct cons cons __attribute__ ((aligned (CACHE_LINE_SIZE)));
void * volatile *shared_space __attribute__ ((aligned (CACHE_LINE_SIZE))); // Align only to isolate cons on its cache line
};
__BEGIN_DECLS
extern __thread struct comm *comm;
extern const int batchSize;
static inline void send(void **addr)
{
while (1)
{
int afterNextWrite = (comm->prod.nextWrite + 1) % SHARED_SPACE_VOIDPTR;
if (afterNextWrite == comm->prod.localRead)
{
if (afterNextWrite == comm->ctrl.read)
continue;
comm->prod.localRead = comm->ctrl.read;
}
comm->shared_space[comm->prod.nextWrite] = addr;
comm->prod.nextWrite = afterNextWrite;
comm->prod.wBatch++;
if (comm->prod.wBatch >= batchSize)
{
comm->ctrl.write = comm->prod.nextWrite;
comm->prod.wBatch = 0;
}
break;
}
}
__END_DECLS
#endif