]> git.vomp.tv Git - vompclient-marten.git/blob - stream.cc
FFMPEG decoding in color at 7 fps
[vompclient-marten.git] / stream.cc
1 /*\r
2     Copyright 2005-2006 Mark Calderbank\r
3 \r
4     This file is part of VOMP.\r
5 \r
6     VOMP is free software; you can redistribute it and/or modify\r
7     it under the terms of the GNU General Public License as published by\r
8     the Free Software Foundation; either version 2 of the License, or\r
9     (at your option) any later version.\r
10 \r
11     VOMP is distributed in the hope that it will be useful,\r
12     but WITHOUT ANY WARRANTY; without even the implied warranty of\r
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
14     GNU General Public License for more details.\r
15 \r
16     You should have received a copy of the GNU General Public License\r
17     along with VOMP; if not, write to the Free Software\r
18     Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.\r
19 */\r
20 \r
21 #include "stream.h"\r
22 #include "log.h"\r
23 \r
24 Stream::Stream()\r
25 {\r
26   initted = 0;\r
27   draintarget = NULL;\r
28   cur_packet_pos = 0;\r
29 }\r
30 \r
31 Stream::~Stream()\r
32 {\r
33   shutdown();\r
34 }\r
35 \r
36 void Stream::shutdown()\r
37 {\r
38   if (initted)\r
39   {\r
40     free(outbuf);\r
41 #ifdef WIN32\r
42     CloseHandle(mutex);\r
43 #endif\r
44       \r
45   }\r
46   initted = 0;\r
47 }\r
48 \r
49 int Stream::init(DrainTarget* tdt, int bufsize)\r
50 {\r
51   outbuf = (UCHAR*) malloc(bufsize);\r
52   if (!outbuf) return 0;\r
53   draintarget = tdt;\r
54   bufferSize = bufsize;\r
55   initted = 1;\r
56 #ifndef WIN32\r
57   pthread_mutex_init(&mutex, NULL);\r
58 #else\r
59   mutex=CreateMutex(NULL,FALSE,NULL);\r
60 #endif\r
61   return 1;\r
62 }\r
63 \r
64 void Stream::flush()\r
65 {\r
66   lock();\r
67 \r
68   mediapackets.clear();\r
69   unLock();\r
70   if (draintarget) draintarget->ResetTimeOffsets();\r
71 }\r
72 \r
73 int Stream::put(const UCHAR* inbuf, int len, UCHAR type,unsigned int index)\r
74 {\r
75   int ret = 0;\r
76   if (!draintarget) return 0;\r
77   MediaPacket newPacket;\r
78   newPacket.length = len;\r
79   newPacket.pos_buffer = 0;\r
80   newPacket.type = type;\r
81   newPacket.pts=0;\r
82   newPacket.dts=0;\r
83   newPacket.synched=false;\r
84   newPacket.index=index;\r
85 #ifndef VOMP_PLATTFORM_MVP\r
86   newPacket.disconti=false;\r
87   newPacket.presentation_time=0;\r
88 #endif\r
89   if (type!=MPTYPE_MPEG_AUDIO_LAYER3) {//no PES\r
90     //Extract the pts...\r
91       bool hasdts=false;\r
92     if ((inbuf[7] & 0x80) && len>14 ) {\r
93         newPacket.synched=true;\r
94         newPacket.pts=((ULLONG)(inbuf[9] & 0x0E) << 29 ) |\r
95                     ( (ULLONG)(inbuf[10])        << 22 ) |\r
96                     ( (ULLONG)(inbuf[11] & 0xFE) << 14 ) |\r
97                      ( (ULLONG)(inbuf[12])        <<  7 ) |\r
98                      ( (ULLONG)(inbuf[13] & 0xFE) >>  1 );\r
99         if ((inbuf[7] & 0x40) && len>19) {\r
100             newPacket.dts=((ULLONG)(inbuf[14] & 0x0E) << 29 ) |\r
101                     ( (ULLONG)(inbuf[15])        << 22 ) |\r
102                     ( (ULLONG)(inbuf[16] & 0xFE) << 14 ) |\r
103                      ( (ULLONG)(inbuf[17])        <<  7 ) |\r
104                      ( (ULLONG)(inbuf[18] & 0xFE) >>  1 );\r
105             hasdts=true;\r
106         }\r
107 #ifndef VOMP_PLATTFORM_MVP\r
108         //ok we have the pts now convert it to a continously time code in 100ns units\r
109         if (hasdts && draintarget->dtsTimefix()) newPacket.presentation_time=(ULLONG)(newPacket.dts*10000LL/90LL);\r
110         else newPacket.presentation_time=(ULLONG)(newPacket.pts*10000LL/90LL);\r
111 \r
112         //newPacket.presentation_time-=draintarget->SetStartOffset((ULLONG)(newPacket.pts*10000LL/90LL),&newPacket.disconti);\r
113         newPacket.presentation_time-=draintarget->SetStartOffset((ULLONG)(newPacket.pts*10000LL/90LL),&newPacket.disconti);\r
114 #endif\r
115     }\r
116   }\r
117 \r
118   lock();\r
119   int front, back;\r
120   if (mediapackets.empty())\r
121   {\r
122     back = 0; front = bufferSize;\r
123   }\r
124   else\r
125   {\r
126     front = mediapackets.front().pos_buffer;\r
127     back = mediapackets.back().pos_buffer + mediapackets.back().length;\r
128     if (back == bufferSize) back = 0;\r
129   }\r
130   unLock();\r
131 \r
132   if (back <= front)\r
133   {\r
134     // The free space (if any) is in one continuous chunk.\r
135     if (len <= front - back) ret = len; // Is there enough of it?\r
136   }\r
137   else if (len <= bufferSize - back)\r
138   {\r
139     // There is enough space at the end of the buffer\r
140     ret = len;\r
141   }\r
142   else if (len <= front)\r
143   {\r
144     // There is enough space at the start of the buffer\r
145     back = 0;\r
146     ret = len;\r
147   }\r
148 \r
149   if (ret) // Nonzero if we managed to find room for the packet\r
150   {\r
151     memcpy(outbuf + back, inbuf, len);\r
152     newPacket.pos_buffer = back;\r
153     lock();\r
154     mediapackets.push_back(newPacket);\r
155     unLock();\r
156   } else {\r
157      // Log::getInstance()->log("Stream", Log::DEBUG, "We are full %d!",bufferSize);\r
158   }\r
159 \r
160   return ret;\r
161 }\r
162 \r
163 bool Stream::drain()\r
164 {\r
165   bool ret = false;\r
166   lock();\r
167   UINT listlength = mediapackets.size();\r
168   if (listlength != 0)\r
169   {\r
170     draintarget->PrepareMediaSample(mediapackets, cur_packet_pos);\r
171     unLock();\r
172     UINT consumed = draintarget->DeliverMediaSample(outbuf, &cur_packet_pos);\r
173     lock();\r
174     if (consumed != 0) ret = true;\r
175     if (consumed > listlength) consumed = listlength;\r
176     while (consumed--) \r
177     {\r
178         mediapackets.pop_front();\r
179     }\r
180   }\r
181   unLock();\r
182   return ret;\r
183 }\r
184 \r
185 void Stream::lock()\r
186 {\r
187 #ifndef WIN32\r
188   pthread_mutex_lock(&mutex);\r
189   //logger->log("Player", Log::DEBUG, "LOCKED");\r
190 \r
191 #else\r
192    WaitForSingleObject(mutex, INFINITE );\r
193 #endif\r
194 }\r
195 \r
196 void Stream::unLock()\r
197 {\r
198 #ifndef WIN32\r
199   //logger->log("Player", Log::DEBUG, "UNLOCKING");\r
200   pthread_mutex_unlock(&mutex);\r
201 #else\r
202    ReleaseMutex(mutex);\r
203 #endif\r
204 }\r