vdr.h

   1 /*
   2     Copyright 2004-2005 Chris Tallon
   3
   4     This file is part of VOMP.
   5
   6     VOMP is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     VOMP is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with VOMP; if not, write to the Free Software
  18     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19 */
  20
  21 #ifndef VDR_H
  22 #define VDR_H
  23
  24 #include <stdio.h>
  25 #include <time.h>
  26 #ifndef WIN32
  27   #include <pthread.h>
  28 #else
  29   //Find threading replacements
  30 #endif
  31 #include <vector>
  32 #include <algorithm>
  33
  34 #include "defines.h"
  35 #include "log.h"
  36 #include "dsock.h"
  37 #include "tcp.h"
  38 #include "channel.h"
  39 #include "event.h"
  40 #include "rectimer.h"
  41 #include "recinfo.h"
  42 #include "mark.h"
  43 #include "wol.h"
  44 #include "media.h"
  45
  46 using namespace std;
  47
  48 typedef vector<Event*> EventList;
  49 typedef vector<Channel*> ChannelList;
  50 typedef vector<RecTimer*> RecTimerList;
  51
  52 struct VDRServer
  53 {
  54   char* ip;
  55   char* name;
  56 };
  57
  58 struct RecTimerSorter     // : public binary_function<double, double, bool>
  59 {
  60   bool operator() (const RecTimer* a, const RecTimer* b)
  61   {
  62     return a->startTime < b->startTime;
  63   }
  64 };
  65
  66 struct ServerSorter
  67 {
  68   bool operator() (const VDRServer& a, const VDRServer& b)
  69   {
  70     if (strcmp(b.name, a.name) > 0) return true;
  71     return false;
  72   }
  73 };
  74
  75 class RecMan;
  76
  77 class VDR
  78 {
  79
  80   public:
  81     VDR();
  82     ~VDR();
  83     static VDR* getInstance();
  84
  85     int init(int port);
  86     int shutdown();
  87
  88     void findServers(vector<VDRServer>& servers);
  89     void cancelFindingServer();
  90     void setServerIP(char*);
  91     int connect();
  92     void disconnect();
  93     bool isConnected() { return connected; }
  94     ULONG getChannelNumberWidth() { return channelNumberWidth; }
  95
  96     void setReceiveWindow(size_t size);
  97
  98     // protocol functions
  99     // for the following, if result == false then the connection has died
 100     //  doLogin
 101     //  getRecordingList
 102     //  getChannelsList
 103     //  getChannelSchedule
 104     //  getRecTimersList
 105     // isConnected can be called after the following to determine if still ok
 106     //  deleteRecording
 107     //  streamRecording
 108     //  positionFromFrameNumber
 109     //  streamChannel
 110     //  getBlock
 111     //  stopStreaming
 112     //  configLoad
 113     //  configSave
 114     //  setEventTimer
 115
 116     int doLogin();
 117
 118     bool       getRecordingsList(RecMan* recman);
 119     RecInfo*   getRecInfo(char* fileName);
 120     int        deleteRecording(char* fileName);
 121     char*      moveRecording(char* fileName, char* newPath);
 122     ULLONG     streamRecording(char* fileName, ULONG* lengthFrames);
 123     ULLONG     positionFromFrameNumber(ULONG frameNumber);
 124     ULONG      frameNumberFromPosition(ULLONG position);
 125     bool       getNextIFrame(ULONG frameNumber, ULONG direction, ULLONG* rfilePosition, ULONG* rframeNumber, ULONG* rframeLength);
 126                // Direction: 0=backwards, 1=forwards
 127     MarkList*  getMarks(char* fileName);
 128
 129     ChannelList* getChannelsList(ULONG type);
 130     int          streamChannel(ULONG number);
 131     void         getChannelPids(Channel* channel);
 132
 133     UCHAR*     getBlock(ULLONG position, UINT maxAmount, UINT* amountReceived);
 134     //get image blocks separate - we can do this in parallel
 135     UCHAR*     getImageBlock(ULONG position, UINT maxAmount, UINT* amountReceived);
 136     int        stopStreaming();
 137     EventList* getChannelSchedule(ULONG number);
 138     EventList* getChannelSchedule(ULONG number, time_t start, ULONG duration);
 139     int        configSave(char* section, char* key, const char* value);
 140     char*      configLoad(char* section, char* key);
 141     ULONG      setEventTimer(char* timerString);
 142
 143     RecTimerList* getRecTimersList();
 144     /**
 145       * ge a list of media entries
 146       * if parent==NULL this is the configured base list
 147       */
 148     MediaList* getMediaList(const char* parent=NULL,int mediaType=MEDIA_TYPE_ALL);
 149     /**
 150       * start loading a JPEG image
 151       * return size, 0 if not found
 152       */
 153     ULONG loadImage(const char * filename, ULONG xsize=0,ULONG ysize=0);
 154
 155     // end
 156
 157     const static ULONG VIDEO = 1;
 158     const static ULONG RADIO = 2;
 159
 160
 161
 162
 163     // obselete
 164     ULLONG     rescanRecording(ULONG* lengthFrames);                    // FIXME obselete
 165
 166
 167
 168   private:
 169     UCHAR*     getBlock(ULLONG position, UINT maxAmount, UINT* amountReceived, ULONG cmd);
 170     static VDR* instance;
 171     Log* logger;
 172     int initted;
 173     int findingServer;
 174     TCP* tcp;
 175     int port;
 176     char serverIP[16];
 177     bool connected;
 178
 179     ULONG maxChannelNumber;
 180     ULONG channelNumberWidth;
 181
 182 #ifndef WIN32
 183     pthread_mutex_t mutex;
 184 #else
 185     HANDLE mutex;
 186 #endif
 187
 188     UCHAR* packet;
 189     ULONG packetLength;
 190     ULONG packetPos;
 191
 192     const static ULONG VDR_LOGIN               = 1;
 193     const static ULONG VDR_GETRECORDINGLIST    = 2;
 194     const static ULONG VDR_DELETERECORDING     = 3;
 195     const static ULONG VDR_GETCHANNELLIST      = 5;
 196     const static ULONG VDR_STREAMCHANNEL       = 6;
 197     const static ULONG VDR_GETBLOCK            = 7;
 198     const static ULONG VDR_STOPSTREAMING       = 8;
 199     const static ULONG VDR_STREAMRECORDING     = 9;
 200     const static ULONG VDR_GETCHANNELSCHEDULE  = 10;
 201     const static ULONG VDR_CONFIGSAVE          = 11;
 202     const static ULONG VDR_CONFIGLOAD          = 12;
 203     const static ULONG VDR_RESCANRECORDING     = 13;  // FIXME obselete
 204     const static ULONG VDR_GETTIMERS           = 14;
 205     const static ULONG VDR_SETTIMER            = 15;
 206     const static ULONG VDR_POSFROMFRAME        = 16;
 207     const static ULONG VDR_FRAMEFROMPOS        = 17;
 208     const static ULONG VDR_MOVERECORDING       = 18;
 209     const static ULONG VDR_GETNEXTIFRAME       = 19;
 210     const static ULONG VDR_GETRECINFO          = 20;
 211     const static ULONG VDR_GETMARKS            = 21;
 212     const static ULONG VDR_GETCHANNELPIDS      = 22;
 213     const static ULONG VDR_GETMEDIALIST        = 30;
 214     const static ULONG VDR_GETIMAGE            = 31;
 215     const static ULONG VDR_GETIMAGEBLOCK       = 32;
 216     int  getPacket();
 217     void freePacket();
 218     int  serverError();
 219     char*  extractString();
 220     UCHAR  extractUCHAR();
 221     ULONG  extractULONG();
 222     ULLONG extractULLONG();
 223     long   extractLONG();
 224 };
 225
 226 #endif
 227
 228 /*
 229
 230 index.vdr file format for video:
 231
 232 For every video frame:
 233 {
 234   File offset    4 bytes
 235   Picture type   1 byte
 236   File number    1 byte
 237   Zero           2 bytes
 238 }
 239
 240 Picture types:
 241
 242 #define NO_PICTURE 0
 243 #define I_FRAME    1
 244 #define P_FRAME    2
 245 #define B_FRAME    3
 246
 247 */