recplayer.c

   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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19 */
  20
  21 #include "recplayer.h"
  22
  23 #define _XOPEN_SOURCE 600
  24 #include <fcntl.h>
  25
  26 RecPlayer::RecPlayer(cRecording* rec)
  27 {
  28   log = Log::getInstance();
  29   file = NULL;
  30   fileOpen = 0;
  31   lastPosition = 0;
  32   recording = rec;
  33   for(int i = 1; i < 1000; i++) segments[i] = NULL;
  34
  35   // FIXME find out max file path / name lengths
  36 #if VDRVERSNUM < 10703
  37   indexFile = new cIndexFile(recording->FileName(), false);
  38 #else
  39   indexFile = new cIndexFile(recording->FileName(), false,  recording->IsPesRecording());
  40 #endif
  41   if (!indexFile) log->log("RecPlayer", Log::ERR, "Failed to create indexfile!");
  42
  43   scan();
  44 }
  45
  46 void RecPlayer::scan()
  47 {
  48   if (file) fclose(file);
  49   totalLength = 0;
  50   fileOpen = 0;
  51   totalFrames = 0;
  52
  53   int i = 1;
  54   while(segments[i++]) delete segments[i];
  55
  56   char fileName[2048];
  57 #if VDRVERSNUM < 10703
  58   for(i = 1; i < 255; i++)//maximum is 255 files instead of 1000, according to VDR HISTORY file...
  59   {
  60     snprintf(fileName, 2047, "%s/%03i.vdr", recording->FileName(), i);
  61 #else
  62   for(i = 1; i < 65535; i++)
  63   {
  64     if (recording->IsPesRecording())
  65       snprintf(fileName, 2047, "%s/%03i.vdr", recording->FileName(), i);
  66     else
  67       snprintf(fileName, 2047, "%s/%05i.ts", recording->FileName(), i);
  68 #endif
  69     log->log("RecPlayer", Log::DEBUG, "FILENAME: %s", fileName);
  70     file = fopen(fileName, "r");
  71     if (!file) break;
  72
  73     segments[i] = new Segment();
  74     segments[i]->start = totalLength;
  75     fseek(file, 0, SEEK_END);
  76     totalLength += ftell(file);
  77     totalFrames = indexFile->Last();
  78     log->log("RecPlayer", Log::DEBUG, "File %i found, totalLength now %llu, numFrames = %lu", i, totalLength, totalFrames);
  79     segments[i]->end = totalLength;
  80     fclose(file);
  81   }
  82
  83   file = NULL;
  84 }
  85
  86 RecPlayer::~RecPlayer()
  87 {
  88   log->log("RecPlayer", Log::DEBUG, "destructor");
  89   int i = 1;
  90   while(segments[i++]) delete segments[i];
  91   if (file) fclose(file);
  92 }
  93
  94 int RecPlayer::openFile(int index)
  95 {
  96   if (file) fclose(file);
  97
  98   char fileName[2048];
  99 #if VDRVERSNUM < 10703
 100   snprintf(fileName, 2047, "%s/%03i.vdr", recording->FileName(), index);
 101 #else
 102   if (recording->IsPesRecording())
 103     snprintf(fileName, 2047, "%s/%03i.vdr", recording->FileName(), index);
 104   else
 105     snprintf(fileName, 2047, "%s/%05i.ts", recording->FileName(), index);
 106 #endif
 107   log->log("RecPlayer", Log::DEBUG, "openFile called for index %i string:%s", index, fileName);
 108
 109   file = fopen(fileName, "r");
 110   if (!file)
 111   {
 112     log->log("RecPlayer", Log::DEBUG, "file failed to open");
 113     fileOpen = 0;
 114     return 0;
 115   }
 116   fileOpen = index;
 117   return 1;
 118 }
 119
 120 ULLONG RecPlayer::getLengthBytes()
 121 {
 122   return totalLength;
 123 }
 124
 125 ULONG RecPlayer::getLengthFrames()
 126 {
 127   return totalFrames;
 128 }
 129
 130 unsigned long RecPlayer::getBlock(unsigned char* buffer, ULLONG position, unsigned long amount)
 131 {
 132   if ((amount > totalLength) || (amount > 500000))
 133   {
 134     log->log("RecPlayer", Log::DEBUG, "Amount %lu requested and rejected", amount);
 135     return 0;
 136   }
 137
 138   if (position >= totalLength)
 139   {
 140     log->log("RecPlayer", Log::DEBUG, "Client asked for data starting past end of recording!");
 141     return 0;
 142   }
 143
 144   if ((position + amount) > totalLength)
 145   {
 146     log->log("RecPlayer", Log::DEBUG, "Client asked for some data past the end of recording, adjusting amount");
 147     amount = totalLength - position;
 148   }
 149
 150   // work out what block position is in
 151   int segmentNumber;
 152   for(segmentNumber = 1; segmentNumber < 1000; segmentNumber++)
 153   {
 154     if ((position >= segments[segmentNumber]->start) && (position < segments[segmentNumber]->end)) break;
 155     // position is in this block
 156   }
 157
 158   // we could be seeking around
 159   if (segmentNumber != fileOpen)
 160   {
 161     if (!openFile(segmentNumber)) return 0;
 162   }
 163
 164   ULLONG currentPosition = position;
 165   ULONG yetToGet = amount;
 166   ULONG got = 0;
 167   ULONG getFromThisSegment = 0;
 168   ULONG filePosition;
 169
 170   while(got < amount)
 171   {
 172     if (got)
 173     {
 174       // if(got) then we have already got some and we are back around
 175       // advance the file pointer to the next file
 176       if (!openFile(++segmentNumber)) return 0;
 177     }
 178
 179     // is the request completely in this block?
 180     if ((currentPosition + yetToGet) <= segments[segmentNumber]->end)
 181       getFromThisSegment = yetToGet;
 182     else
 183       getFromThisSegment = segments[segmentNumber]->end - currentPosition;
 184
 185     filePosition = currentPosition - segments[segmentNumber]->start;
 186     fseek(file, filePosition, SEEK_SET);
 187     if (fread(&buffer[got], getFromThisSegment, 1, file) != 1) return 0; // umm, big problem.
 188
 189     // Tell linux not to bother keeping the data in the FS cache
 190     posix_fadvise(file->_fileno, filePosition, getFromThisSegment, POSIX_FADV_DONTNEED);
 191
 192     got += getFromThisSegment;
 193     currentPosition += getFromThisSegment;
 194     yetToGet -= getFromThisSegment;
 195   }
 196
 197   lastPosition = position;
 198   return got;
 199 }
 200
 201 ULLONG RecPlayer::getLastPosition()
 202 {
 203   return lastPosition;
 204 }
 205
 206 cRecording* RecPlayer::getCurrentRecording()
 207 {
 208   return recording;
 209 }
 210
 211 ULLONG RecPlayer::positionFromFrameNumber(ULONG frameNumber)
 212 {
 213   if (!indexFile) return 0;
 214 #if VDRVERSNUM < 10703
 215   uchar retFileNumber;
 216   int retFileOffset;
 217   uchar retPicType;
 218 #else
 219   uint16_t retFileNumber;
 220   off_t retFileOffset;
 221   bool retPicType;
 222 #endif
 223   int retLength;
 224
 225
 226   if (!indexFile->Get((int)frameNumber, &retFileNumber, &retFileOffset, &retPicType, &retLength))
 227   {
 228     return 0;
 229   }
 230
 231 //  log->log("RecPlayer", Log::DEBUG, "FN: %u FO: %i", retFileNumber, retFileOffset);
 232   if (!segments[retFileNumber]) return 0;
 233   ULLONG position = segments[retFileNumber]->start + retFileOffset;
 234 //  log->log("RecPlayer", Log::DEBUG, "Pos: %llu", position);
 235
 236   return position;
 237 }
 238
 239 ULONG RecPlayer::frameNumberFromPosition(ULLONG position)
 240 {
 241   if (!indexFile) return 0;
 242
 243   if (position >= totalLength)
 244   {
 245     log->log("RecPlayer", Log::DEBUG, "Client asked for data starting past end of recording!");
 246     return 0;
 247   }
 248
 249   uchar segmentNumber;
 250   for(segmentNumber = 1; segmentNumber < 255; segmentNumber++)
 251   {
 252     if ((position >= segments[segmentNumber]->start) && (position < segments[segmentNumber]->end)) break;
 253     // position is in this block
 254   }
 255   ULONG askposition = position - segments[segmentNumber]->start;
 256   return indexFile->Get((int)segmentNumber, askposition);
 257
 258 }
 259
 260
 261 bool RecPlayer::getNextIFrame(ULONG frameNumber, ULONG direction, ULLONG* rfilePosition, ULONG* rframeNumber, ULONG* rframeLength)
 262 {
 263   // 0 = backwards
 264   // 1 = forwards
 265
 266   if (!indexFile) return false;
 267
 268 #if VDRVERSNUM < 10703
 269   uchar waste1;
 270   int waste2;
 271 #else
 272   uint16_t waste1;
 273   off_t waste2;
 274 #endif
 275
 276   int iframeLength;
 277   int indexReturnFrameNumber;
 278
 279   indexReturnFrameNumber = (ULONG)indexFile->GetNextIFrame(frameNumber, (direction==1 ? true : false), &waste1, &waste2, &iframeLength);
 280   log->log("RecPlayer", Log::DEBUG, "GNIF input framenumber:%lu, direction=%lu, output:framenumber=%i, framelength=%i", frameNumber, direction, indexReturnFrameNumber, iframeLength);
 281
 282   if (indexReturnFrameNumber == -1) return false;
 283
 284   *rfilePosition = positionFromFrameNumber(indexReturnFrameNumber);
 285   *rframeNumber = (ULONG)indexReturnFrameNumber;
 286   *rframeLength = (ULONG)iframeLength;
 287
 288   return true;
 289 }