csdr/tsmpool.cpp

56 lines
1.3 KiB
C++
Raw Normal View History

2016-06-05 04:54:17 +08:00
#include "tsmpool.h"
tsmpool::tsmpool(size_t size, int num)
2016-06-04 23:42:29 +08:00
{
this->threads_cntr = 0;
this->size = size;
2016-06-05 04:54:17 +08:00
this->num = num; //number of buffers of (size) to alloc
2016-06-04 23:42:29 +08:00
this->ok = 1;
this->lowest_read_index = -1;
if (pthread_mutex_init(&this->mutex, NULL) != 0) this->ok=0;
}
size_t tsmpool::get_size() { return this->size; }
2016-06-05 04:54:17 +08:00
void* tsmpool::get_write_buffer()
{
2016-06-04 23:42:29 +08:00
if(write_index==index_before(lowest_read_index)) return NULL;
void* to_return = buffers[write_index];
write_index=index_next(write_index);
}
tsmthread_t* tsmpool::register_thread()
{
2016-06-05 04:54:17 +08:00
if(!ok) return NULL;
2016-06-04 23:42:29 +08:00
pthread_mutex_lock(&this->mutex);
tsmthread_t* thread = new tsmthread_t;
thread->read_index = write_index;
threads.push_back(thread);
pthread_mutex_unlock(&this->mutex);
return thread;
}
int 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(i);
break;
}
pthread_mutex_unlock(&this->mutex);
}
void* tsmpool::get_read_buffer(tsmthread_t* thread)
{
if(thread->read_index==write_index) return NULL;
void* to_return = buffers[thread->read_index];
thread->read_index=index_next(thread->read_index);
}
void* tsmpool::set_read_index_distance(tsmthread_t* thread, int distance)
{
}