ha7ilm-csdr/tsmpool.cpp

#include "tsmpool.h"

tsmpool::tsmpool(size_t size, int num) :
	size(size), 
	num(num) //number of buffers of (size) to alloc
{
	this->threads_cntr = 0;
	this->ok = 1;
	this->lowest_read_index = -1;
	this->write_index = 0;
	this->my_read_index = index_before(0);
    if (pthread_mutex_init(&this->mutex, NULL) != 0) { this->ok = 0; return; }
	for(int i=0; i<num; i++) 
	{
		void* newptr = (void*)new char[size];
		if(!newptr) { this->ok = 0; return; }
		buffers.push_back(newptr);
	}
}

int tsmpool::is_ok() { return this->ok; }

void* tsmpool::get_write_buffer()
{
	//if(write_index==index_before(lowest_read_index)) return NULL;
	pthread_mutex_lock(&this->mutex);
	void* to_return = buffers[write_index];
	write_index = index_next(write_index);
	pthread_mutex_unlock(&this->mutex);
	if(TSM_DEBUG) fprintf(stderr, "gwb: write_index = %d\n", write_index);
	return to_return;
}

tsmthread_t* tsmpool::register_thread()
{
	if(!ok) return NULL;
	pthread_mutex_lock(&this->mutex);
	tsmthread_t* thread = new tsmthread_t();
	thread->read_index = index_before(write_index);
	threads.push_back(thread);
	pthread_mutex_unlock(&this->mutex);
	return thread;
}

void tsmpool::remove_thread(tsmthread_t* thread)
{
	pthread_mutex_lock(&this->mutex);
	for(int i=0;i<threads.size();i++)
		if(threads[i] == thread)
		{
			delete threads[i];
			threads.erase(threads.begin()+i);
			break;
		}
	pthread_mutex_unlock(&this->mutex);
}

void* tsmpool::get_read_buffer(tsmthread_t* thread)
{
	pthread_mutex_lock(&this->mutex);
	int* actual_read_index = (thread==NULL) ? &my_read_index : &thread->read_index;
	if(*actual_read_index==index_before(write_index)) 
	{
		if(TSM_DEBUG) fprintf(stderr, "grb: fail,"
			"read_index %d is just before write_index\n", *actual_read_index);
		pthread_mutex_unlock(&this->mutex);
		return NULL;
	}
	void* to_return = buffers[*actual_read_index];
	*actual_read_index=index_next(*actual_read_index);
	pthread_mutex_unlock(&this->mutex);
	if(TSM_DEBUG) fprintf(stderr, "grb: read_index = %d\n", *actual_read_index);
	return to_return;
}