[RecordTimer] fix repeat timer issue.
[vuplus_dvbapp] / lib / dvb / tstools.h
1 #ifndef __lib_dvb_tstools_h
2 #define __lib_dvb_tstools_h
3
4 #include <sys/types.h>
5 #include <lib/dvb/pvrparse.h>
6 #include <lib/base/rawfile.h>
7 #include <lib/base/elock.h>
8
9 /*
10  * Note: we're interested in PTS values, not STC values.
11  * thus we're evaluating PES headers, not adaption fields.
12  */
13
14 typedef long long pts_t;
15
16 class eDVBTSTools
17 {
18 public:
19         eDVBTSTools();
20         ~eDVBTSTools();
21
22         void setSource(ePtr<iTsSource> &source, const char *streaminfo_filename=NULL);
23         void closeSource();
24
25         int openFile(const char *filename, int nostreaminfo = 0);
26         void closeFile();
27
28         void setSyncPID(int pid);
29         void setSearchRange(int maxrange);
30         
31                 /* get first PTS *after* the given offset. */
32                 /* pts values are zero-based. */
33         int getPTS(off_t &offset, pts_t &pts, int fixed=0);
34         
35                 /* this fixes up PTS to end up in a [0..len) range.
36                    discontinuities etc. are handled here.
37                 
38                   input:        
39                     offset - approximate offset in file to resolve ambiguities
40                     pts - video-pts (i.e. current STC of video decoder)
41                   output:
42                     pts - zero-based PTS value
43                 */
44         int fixupPTS(const off_t &offset, pts_t &pts);
45         
46                 /* get (approximate) offset corresponding to PTS */
47         int getOffset(off_t &offset, pts_t &pts, int marg=0);
48         
49         int getNextAccessPoint(pts_t &ts, const pts_t &start, int direction);
50         
51         void calcBegin();
52         void calcEnd();
53         
54         int calcLen(pts_t &len);
55         
56         int calcBitrate(); /* in bits/sec */
57         
58         void takeSamples();
59         int takeSample(off_t off, pts_t &p);
60         
61         int findPMT(int &pmt_pid, int &service_id);
62         
63         enum { 
64                 frametypeI = 1, 
65                 frametypeP = 2, 
66                 frametypeB = 4, 
67                 frametypeAll = frametypeI | frametypeP | frametypeB
68         };
69         /** findFrame: finds a specific frame at a given position
70         
71         findFrame will look for the specified frame type starting at the given position, moving forward
72         (when direction is >0) or backward (when direction is <0). (direction=0 is a special case and also moves
73         forward, but starts with the last frame.)
74         
75         return values are the new offset, the length of the found frame (both unaligned), and the (signed) 
76         number of frames skipped. */
77         int findFrame(off_t &_iframe_offset, off_t &offset, size_t &len, int &direction, int frame_types = frametypeI);
78         int findNextPicture(off_t &offset, size_t &len, int &distance, int frame_types = frametypeAll);
79 private:
80         int m_pid;
81         int m_maxrange;
82
83         ePtr<iTsSource> m_source;
84
85         int m_begin_valid, m_end_valid;
86         pts_t m_pts_begin, m_pts_end;
87         off_t m_offset_begin, m_offset_end;
88         
89                 /* for simple linear interpolation */
90         std::map<pts_t, off_t> m_samples;
91         int m_samples_taken;
92         
93         eMPEGStreamInformation m_streaminfo;
94         int m_use_streaminfo;
95         off_t m_last_filelength;
96         int m_futile;
97 };
98
99 #endif