summaryrefslogtreecommitdiffstats
path: root/core/Fifo.h
blob: d9240fb255454262e6e960e531f4688f7bb94d19 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
/* $Id: Fifo.h,v 1.10 2004/01/16 19:04:04 titer Exp $

   This file is part of the HandBrake source code.
   Homepage: <http://handbrake.m0k.org/>.
   It may be used under the terms of the GNU General Public License. */

#ifndef HB_FIFO_H
#define HB_FIFO_H

#include "Utils.h"
#include "Thread.h"

struct HBBuffer
{
    /* Members used everywhere */
    int       alloc;
    int       size;
    uint8_t * data;
    float     position;
    int       pass;

    /* Only used for PStoES */
    int       streamId;
    uint64_t  pts;

    /* NTSC suxx */
    int       repeat;

    /* Only used for raw audio buffers */
    int       samples; /* Number of samples for each track */
    float   * left;
    float   * right;

    /* Only used for MPEG-4, MP3 and AAC buffers */
    int       keyFrame;
};

HBBuffer * HBBufferInit( int size );
void       HBBufferReAlloc( HBBuffer *, int size );
void       HBBufferClose( HBBuffer ** );

struct HBFifo
{
    int         die;
    int         capacity;
    int         whereToPush;
    int         whereToPop;
    HBBuffer ** buffers;
    HBLock    * lock;
    HBCond    * cond;
};

HBFifo                 * HBFifoInit( int capacity );
static inline int        HBFifoSize( HBFifo * );
static inline int        HBFifoPush( HBFifo *, HBBuffer ** );
static inline HBBuffer * HBFifoPop( HBFifo * );
static inline int        HBFifoWait( HBFifo * );
static inline float      HBFifoPosition( HBFifo * );
void                     HBFifoDie( HBFifo * );
void                     HBFifoClose( HBFifo ** );

static inline int HBFifoSize( HBFifo * f )
{
    int size;
    HBLockLock( f->lock );
    size = ( f->capacity + 1 + f->whereToPush - f->whereToPop ) %
                 ( f->capacity + 1 );
    HBLockUnlock( f->lock );
    return size;
}

static inline int HBFifoPush( HBFifo * f, HBBuffer ** b )
{
    HBLockLock( f->lock );
    HBCondSignal( f->cond );
    if( ( f->capacity + 1 + f->whereToPush - f->whereToPop ) %
            ( f->capacity + 1 ) != f->capacity )
    {
        f->buffers[f->whereToPush] = *b;
        f->whereToPush++;
        f->whereToPush %= ( f->capacity + 1 );
        HBLockUnlock( f->lock );
        *b = NULL;
        return 1;
    }
    HBLockUnlock( f->lock );
    return 0;
}

static inline HBBuffer * HBFifoPop( HBFifo * f )
{
    HBLockLock( f->lock );
    if( f->whereToPush != f->whereToPop )
    {
        HBBuffer * b = f->buffers[f->whereToPop];
        f->whereToPop++;
        f->whereToPop %= ( f->capacity + 1 );
        HBLockUnlock( f->lock );
        return b;
    }
    HBLockUnlock( f->lock );
    return NULL;
}

static inline int HBFifoWait( HBFifo * f )
{
    HBLockLock( f->lock );
    if( f->whereToPush != f->whereToPop )
    {
        HBLockUnlock( f->lock );
        return 1;
    }
    if( f->die )
    {
        HBLockUnlock( f->lock );
        return 0;
    }
    HBCondWait( f->cond, f->lock );
    if( f->whereToPush != f->whereToPop )
    {
        HBLockUnlock( f->lock );
        return 1;
    }
    HBLockUnlock( f->lock );
    return 0;
}

static inline float HBFifoPosition( HBFifo * f )
{
    float pos;
    HBLockLock( f->lock );
    if( f->whereToPush != f->whereToPop )
    {
        pos = f->buffers[f->whereToPop]->position;
    }
    else
    {
        pos = 0.0;
    }
    HBLockUnlock( f->lock );
    return pos;
}

#endif