1
0
mirror of https://github.com/VDR4Arch/vdr.git synced 2023-10-10 13:36:52 +02:00
vdr/ringbuffer.c

280 lines
5.6 KiB
C
Raw Normal View History

/*
* ringbuffer.c: A ring buffer
*
* See the main source file 'vdr.c' for copyright information and
* how to reach the author.
*
* Parts of this file were inspired by the 'ringbuffy.c' from the
* LinuxDVB driver (see linuxtv.org).
*
2002-07-28 12:48:44 +02:00
* $Id: ringbuffer.c 1.10 2002/07/28 12:47:32 kls Exp $
*/
#include "ringbuffer.h"
2002-07-28 12:48:44 +02:00
#include <stdlib.h>
#include <unistd.h>
#include "tools.h"
// --- cRingBuffer -----------------------------------------------------------
cRingBuffer::cRingBuffer(int Size, bool Statistics)
{
size = Size;
statistics = Statistics;
maxFill = 0;
lastPercent = 0;
2001-08-05 12:23:24 +02:00
}
cRingBuffer::~cRingBuffer()
{
if (statistics)
2002-05-13 16:35:49 +02:00
dsyslog("buffer stats: %d (%d%%) used", maxFill, maxFill * 100 / (size - 1));
2001-08-05 12:23:24 +02:00
}
void cRingBuffer::WaitForPut(void)
{
putMutex.Lock();
readyForPut.Wait(putMutex);
putMutex.Unlock();
}
void cRingBuffer::WaitForGet(void)
{
getMutex.Lock();
readyForGet.Wait(getMutex);
getMutex.Unlock();
}
void cRingBuffer::EnablePut(void)
{
readyForPut.Broadcast();
}
void cRingBuffer::EnableGet(void)
{
readyForGet.Broadcast();
}
// --- cRingBufferLinear -----------------------------------------------------
2001-08-05 12:23:24 +02:00
cRingBufferLinear::cRingBufferLinear(int Size, bool Statistics)
:cRingBuffer(Size, Statistics)
{
buffer = NULL;
getThreadPid = -1;
2001-08-05 12:23:24 +02:00
if (Size > 1) { // 'Size - 1' must not be 0!
buffer = new uchar[Size];
if (!buffer)
2002-05-13 16:35:49 +02:00
esyslog("ERROR: can't allocate ring buffer (size=%d)", Size);
Clear();
}
else
2002-05-13 16:35:49 +02:00
esyslog("ERROR: illegal size for ring buffer (%d)", Size);
}
2001-08-05 12:23:24 +02:00
cRingBufferLinear::~cRingBufferLinear()
{
delete buffer;
}
2001-08-05 12:23:24 +02:00
int cRingBufferLinear::Available(void)
{
2001-08-05 12:23:24 +02:00
Lock();
int diff = head - tail;
2001-08-05 12:23:24 +02:00
Unlock();
return (diff >= 0) ? diff : Size() + diff;
}
2001-08-05 12:23:24 +02:00
void cRingBufferLinear::Clear(void)
{
2001-08-05 12:23:24 +02:00
Lock();
head = tail = 0;
2001-08-05 12:23:24 +02:00
Unlock();
EnablePut();
EnableGet();
}
2001-08-05 12:23:24 +02:00
int cRingBufferLinear::Put(const uchar *Data, int Count)
{
if (Count > 0) {
2001-08-05 12:23:24 +02:00
Lock();
int rest = Size() - head;
int diff = tail - head;
2001-08-05 12:23:24 +02:00
int free = (diff > 0) ? diff - 1 : Size() + diff - 1;
if (statistics) {
2001-08-05 12:23:24 +02:00
int fill = Size() - free - 1 + Count;
if (fill >= Size())
fill = Size() - 1;
if (fill > maxFill)
maxFill = fill;
int percent = maxFill * 100 / (Size() - 1) / 5 * 5;
if (abs(lastPercent - percent) >= 5) {
if (percent > 75)
dsyslog("buffer usage: %d%% (pid=%d)", percent, getThreadPid);
lastPercent = percent;
}
}
if (free > 0) {
if (free < Count)
Count = free;
if (Count > maxFill)
maxFill = Count;
if (Count >= rest) {
memcpy(buffer + head, Data, rest);
if (Count - rest)
memcpy(buffer, Data + rest, Count - rest);
head = Count - rest;
}
else {
memcpy(buffer + head, Data, Count);
head += Count;
}
}
else
Count = 0;
Unlock();
EnableGet();
}
return Count;
}
2001-08-05 12:23:24 +02:00
int cRingBufferLinear::Get(uchar *Data, int Count)
{
if (Count > 0) {
2001-08-05 12:23:24 +02:00
Lock();
if (getThreadPid < 0)
getThreadPid = getpid();
2001-08-05 12:23:24 +02:00
int rest = Size() - tail;
int diff = head - tail;
2001-08-05 12:23:24 +02:00
int cont = (diff >= 0) ? diff : Size() + diff;
if (rest > 0) {
if (cont < Count)
Count = cont;
if (Count >= rest) {
memcpy(Data, buffer + tail, rest);
if (Count - rest)
memcpy(Data + rest, buffer, Count - rest);
tail = Count - rest;
}
else {
memcpy(Data, buffer + tail, Count);
tail += Count;
}
}
else
Count = 0;
Unlock();
if (Count == 0)
WaitForGet();
}
return Count;
}
2001-08-05 12:23:24 +02:00
// --- cFrame ----------------------------------------------------------------
cFrame::cFrame(const uchar *Data, int Count, eFrameType Type, int Index)
{
2001-08-05 12:23:24 +02:00
count = Count;
type = Type;
2001-08-05 12:23:24 +02:00
index = Index;
data = new uchar[count];
if (data)
memcpy(data, Data, count);
else
2002-05-13 16:35:49 +02:00
esyslog("ERROR: can't allocate frame buffer (count=%d)", count);
2001-08-05 12:23:24 +02:00
next = NULL;
}
cFrame::~cFrame()
{
delete data;
}
// --- cRingBufferFrame ------------------------------------------------------
cRingBufferFrame::cRingBufferFrame(int Size, bool Statistics)
2001-08-05 12:23:24 +02:00
:cRingBuffer(Size, Statistics)
{
head = NULL;
currentFill = 0;
}
cRingBufferFrame::~cRingBufferFrame()
{
Clear();
}
void cRingBufferFrame::Clear(void)
{
Lock();
const cFrame *p;
while ((p = Get()) != NULL)
2001-08-05 12:23:24 +02:00
Drop(p);
Unlock();
EnablePut();
EnableGet();
}
bool cRingBufferFrame::Put(cFrame *Frame)
{
if (Frame->Count() <= Free()) {
Lock();
if (head) {
Frame->next = head->next;
head->next = Frame;
head = Frame;
}
2001-08-05 12:23:24 +02:00
else {
head = Frame->next = Frame;
}
currentFill += Frame->Count();
Unlock();
EnableGet();
return true;
}
2001-08-05 12:23:24 +02:00
return false;
}
const cFrame *cRingBufferFrame::Get(void)
{
2001-08-05 12:23:24 +02:00
Lock();
cFrame *p = head ? head->next : NULL;
Unlock();
return p;
}
2001-08-05 12:23:24 +02:00
void cRingBufferFrame::Delete(const cFrame *Frame)
{
2001-08-05 12:23:24 +02:00
currentFill -= Frame->Count();
delete Frame;
}
void cRingBufferFrame::Drop(const cFrame *Frame)
{
Lock();
if (head) {
if (Frame == head->next) {
if (head->next != head) {
head->next = Frame->next;
Delete(Frame);
}
else {
Delete(head);
head = NULL;
}
}
else
2002-05-13 16:35:49 +02:00
esyslog("ERROR: attempt to drop wrong frame from ring buffer!");
2001-08-05 12:23:24 +02:00
}
Unlock();
EnablePut();
}
2001-08-05 12:23:24 +02:00
int cRingBufferFrame::Available(void)
{
Lock();
int av = currentFill;
Unlock();
return av;
}