tstools.h

Go to the documentation of this file.

#ifndef __lib_dvb_tstools_h
#define __lib_dvb_tstools_h
 
#include <sys/types.h>
#include <lib/dvb/pvrparse.h>
#include <lib/base/rawfile.h>
#include <lib/base/elock.h>
#include <lib/dvb/pmtparse.h>
#include <lib/dvb/idemux.h>
 
/*
 * Note: we're interested in PTS values, not STC values.
 * thus we're evaluating PES headers, not adaption fields.
 */
 
typedef long long pts_t;
 
class eTSFileSectionReader: public iDVBSectionReader, public sigc::trackable
{
  DECLARE_REF(eTSFileSectionReader);
  unsigned char sectionData[4096];
  unsigned int sectionSize;
  sigc::signal1<void, const uint8_t*> read;
 
public:
  eTSFileSectionReader(eMainloop *context);
  virtual ~eTSFileSectionReader();
  void data(unsigned char *packet, unsigned int size);
  RESULT setBufferSize(int size) { return 0; }
  RESULT start(const eDVBSectionFilterMask &mask);
  RESULT stop();
  RESULT connectRead(const sigc::slot1<void,const uint8_t*> &read, ePtr<eConnection> &conn);
};
 
class eDVBTSTools : public eDVBPMTParser
{
public:
  eDVBTSTools();
  ~eDVBTSTools();
 
  void setSource(ePtr<iTsSource> &source, const char *streaminfo_filename=NULL);
  int openFile(const char *filename, int nostreaminfo = 0);
 
  /* get (approximate) offset corresponding to PTS */
  int getOffset(off_t &offset, pts_t &pts, int marg=0);
  int getNextAccessPoint(pts_t &ts, const pts_t &start, int direction);
    /* this fixes up PTS to end up in a [0..len) range.
       discontinuities etc. are handled here.
      input:
        offset - approximate offset in file to resolve ambiguities
        pts - video-pts (i.e. current STC of video decoder)
      output:
        pts - zero-based PTS value
    */
  int fixupPTS(const off_t &offset, pts_t &pts);
  int calcLen(pts_t &len);
  int calcBitrate(); /* in bits/sec */
 
  enum {
    frametypeI = 1,
    frametypeP = 2,
    frametypeB = 4,
    frametypeAll = frametypeI | frametypeP | frametypeB
  };
  int findFrame(off_t &offset, size_t &len, int &direction, int frame_types = frametypeI);
  int findNextPicture(off_t &offset, size_t &len, int &distance, int frame_types = frametypeAll);
 
  /* Retrieve PMT. Returns 0 on success. */
  int findPMT(eDVBPMTParser::program &program);
 
protected:
  void closeSource();
 
    /* get first PTS *after* the given offset. */
    /* pts values are zero-based. */
  int getPTS(off_t &offset, pts_t &pts, int fixed=0);
 
  void calcBegin();
  void calcEnd();
  void calcBeginAndEnd();
 
  void takeSamples();
  int takeSample(off_t off, pts_t &p);
 
private:
  int m_pid;
  int m_packet_size;
 
  ePtr<iTsSource> m_source;
 
  int m_begin_valid, m_end_valid;
  pts_t m_pts_begin, m_pts_end;
  off_t m_offset_begin, m_offset_end;
  pts_t m_pts_length;
 
    /* for simple linear interpolation */
  std::map<pts_t, off_t> m_samples;
  int m_samples_taken;
 
  eMPEGStreamInformation m_streaminfo;
  off_t m_last_filelength;
  int m_futile;
  program m_program;
  bool m_pmtready;
 
  void PMTready(int error);
};
 
#endif