core/Transcoder.cpp

   1 /******************************************************************************
   2 * FOBS C++ wrapper code
   3 * Copyright (c) 2004 Omnividea Multimedia S.L
   4 * Coded by Jose San Pedro Wandelmer
   5 *
   6 *    This file is part of FOBS.
   7 *
   8 *    FOBS is free software; you can redistribute it and/or modify
   9 *    it under the terms of the GNU Lesser General Public License as
  10 *    published by the Free Software Foundation; either version 2.1
  11 *    of the License, or (at your option) any later version.
  12 *
  13 *    FOBS is distributed in the hope that it will be useful,
  14 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 *    GNU Lesser General Public License for more details.
  17 *
  18 *    You should have received a copy of the GNU Lesser General Public
  19 *    License along with FOBS; if not, write to the Free Software
  20 *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  21 ******************************************************************************/
  22
  23
  24 extern "C"
  25 {
  26 #include "libavformat/avformat.h"
  27 #include "libavcodec/avcodec.h"
  28 #include "libswscale/swscale.h"
  29 }
  30
  31 #include "Transcoder.h"
  32 #include "PacketBuffer.h"
  33 #include <iostream>
  34 #include <cstring> // strdup
  35 #include <cstdlib> // free
  36
  37
  38 using namespace std;
  39 using namespace omnividea::fobs;
  40
  41 static Decoder iDecoder("");
  42
  43 Transcoder::Transcoder(char *fIn, char *fOut)
  44 {
  45    completion  = 0.0;
  46    this->fIn = strdup(fIn);
  47    this->fOut = strdup(fOut);
  48    vWidth = -1; //from Input
  49    vHeight = -1; //from Input
  50    vBitRate = 0.0; //from Input
  51    vFrameRate = 0.0; //from Input
  52
  53    aSamplesPerSec = -1; //from Input
  54    aChannels = -1; //from Input
  55    aBitRate = 0.0; //from Input
  56
  57    vCodec = NULL; //from Input
  58    aCodec = NULL; //from Input
  59    mFormat = NULL; //from Input
  60
  61    customVideoSettings = false;
  62    customAudioSettings = false;
  63
  64    abortSignal = false;
  65
  66    e = new Encoder(fOut);
  67    d = new Decoder(fIn);
  68
  69 }
  70 Transcoder::~Transcoder()
  71 {
  72   free(fIn);
  73   free(fOut);
  74   if(vCodec) free(vCodec);
  75   if(aCodec) free(aCodec);
  76   if(mFormat) free(mFormat);
  77
  78   if(e)e->close();
  79   if(d)d->close();
  80   delete e;
  81   delete d;
  82 }
  83
  84 ReturnCode Transcoder::chooseVideoCodec(int width, int height, double bitRate, double frameRate, char *fcc)
  85 {
  86   vWidth = width;
  87   vHeight = height;
  88   vBitRate = bitRate;
  89   vFrameRate = frameRate;
  90   if(vCodec) free(vCodec);
  91   if(fcc) vCodec = strdup(fcc);
  92   else vCodec = NULL;
  93   customVideoSettings = true;
  94   return OkCode;
  95 }
  96 ReturnCode Transcoder::chooseAudioCodec(int samplesPerSec, int channels, double bitRate, char *fcc)
  97 {
  98   aSamplesPerSec = samplesPerSec;
  99   aChannels = channels;
 100   aBitRate = bitRate;
 101   if(aCodec) free(aCodec);
 102   if(fcc) aCodec = strdup(fcc);
 103   else aCodec = NULL;
 104   customAudioSettings = true;
 105   return OkCode;
 106 }
 107
 108 ReturnCode Transcoder::chooseFormat(char *fcc)
 109 {
 110   if(mFormat) free(mFormat);
 111   if(fcc)mFormat = strdup(fcc);
 112   else mFormat = NULL;
 113   e->chooseFormat(fcc);
 114   return OkCode;
 115 }
 116
 117 ReturnCode Transcoder::transcode()
 118 {
 119   ReturnCode error = OkCode;
 120   AVPacket *pkt;
 121   int vCounter = 0;
 122   int aCounter = 0;
 123   d->setAudioResampleFlag(true);
 124   error = d->open();
 125
 126   if(isOk(error))
 127   {
 128       if(customVideoSettings)
 129       {
 130         error = e->setVideoParameters(vWidth, vHeight, vBitRate, vFrameRate, vCodec);
 131       }
 132       else
 133       {
 134         error = e->setVideoParameters(d);
 135       }
 136   }
 137   if(isOk(error))
 138   {
 139       if(customAudioSettings)
 140       {
 141         error = e->setAudioParameters(aSamplesPerSec, aChannels, aBitRate, aCodec);
 142       }
 143       else
 144       {
 145         error = e->setAudioParameters(d);
 146       }
 147   }
 148   TimeStamp duration;
 149   if(isOk(error))
 150   {
 151     error = e->open();
 152     duration = d->getDurationMilliseconds();
 153   }
 154
 155   if(isError(error))
 156   {
 157     d->close();
 158     e->close();
 159     return error;
 160   }
 161
 162   while(isOk(error) && abortSignal == false)
 163   {
 164     AVFrame avframe; //FIXME/XXX remove this
 165     AVPacket opkt;
 166     bool packetFrame;
 167     av_init_packet(&opkt);
 168
 169     avcodec_get_frame_defaults(&avframe);
 170
 171     packetFrame = false;
 172     error = d->readNextFrame();
 173     if(isOk(error))
 174     {
 175       while(d->videoBuffer->count() > 0)
 176       {
 177         error = d->placeAtNextFrame(true);
 178         if(customVideoSettings)
 179         {
 180             error = d->decodeFrame();
 181             if(error == NoFrameError)
 182             {
 183                 error = OkCode;
 184                 continue;
 185             }
 186             if(isOk(error))
 187             {
 188                 byte *rgb = d->getRGB();
 189                 if(rgb == NULL) error = GenericError;
 190                 if(isOk(error))
 191                 {
 192                     log(Packets, "Adding video frame: %u\n", ++vCounter);
 193                     e->addFrame((char *)rgb, d->getWidth(), d->getHeight());
 194                 }
 195             }
 196         }
 197         else
 198         {
 199             pkt = d->videoBuffer->extractNext();
 200             e->videoStream->codec->coded_frame = &avframe;
 201             opkt.stream_index= e->videoStreamIndex;
 202             e->videoStream->codec->frame_number++;
 203             packetFrame = true;
 204
 205             AVRational iTimebase = d->inputFile->streams[d->videoStreamIndex]->time_base;
 206             AVRational oTimebase = e->videoStream->time_base;
 207
 208             //opkt.pts = (int64_t)(pkt->dts*iTimebase.num/(double)iTimebase.den * (double)oTimebase.den/oTimebase.num);
 209             //opkt.dts = (int64_t)(pkt->dts*iTimebase.num/(double)iTimebase.den * (double)oTimebase.den/oTimebase.num);
 210             if(pkt->pts != AV_NOPTS_VALUE) {
 211                 opkt.pts= av_rescale_q(av_rescale_q(pkt->pts, iTimebase, AV_TIME_BASE_Q) - e->start_time, AV_TIME_BASE_Q,  oTimebase);
 212                 //opkt.pts = av_rescale_q(pkt->pts, iTimebase, oTimebase);
 213             } else {
 214                 opkt.pts = AV_NOPTS_VALUE;
 215             }
 216             {
 217                 int64_t dts;
 218                 if (pkt->dts == AV_NOPTS_VALUE)
 219                     dts = AV_NOPTS_VALUE;//ist->next_pts;
 220                 else
 221                     dts= av_rescale_q(pkt->dts, iTimebase, AV_TIME_BASE_Q);
 222                 opkt.dts= av_rescale_q(dts - e->start_time, AV_TIME_BASE_Q,  oTimebase);
 223
 224                 //opkt.dts = av_rescale_q(pkt->dts, iTimebase, oTimebase);
 225             }
 226
 227             //cout << "Video Pkt " << pkt->dts << " - Converted into= " << opkt.dts << endl;
 228         }
 229
 230       }
 231 //      else
 232       while(d->audioBuffer->count() > 0)
 233       {
 234         error = d->placeAtNextFrame(false);
 235         if(!isOk(error)) {
 236             log(Packets, "Error while placing at next audio frame %d\n", error);
 237             continue;
 238         }
 239         if(customAudioSettings)
 240         {
 241             error = d->decodeAudioFrame();
 242             if(isOk(error) && d->getAudioSamplesSize()>0)
 243             {
 244                 log(Packets, "Adding audio frame: %u\n", ++aCounter);
 245                 if(d->getAudioSamplesSize() <= 0)
 246                 {
 247                     log(Packets, "Audio frame was empty!\n");
 248                 }
 249                 else {
 250                    e->addAudioFrame(d->getAudioSamples(), d->getAudioSamplesSize(), d->getAudioChannelNumber(), d->getAudioSampleRate());
 251                 }
 252             }
 253         }
 254         else
 255         {
 256             //error = d->placeAtNextFrame(false);
 257             pkt = d->audioBuffer->extractNext();
 258             e->audioStream->codec->coded_frame = &avframe;
 259             opkt.stream_index= e->audioStreamIndex;
 260             e->audioStream->codec->frame_number++;
 261             packetFrame = true;
 262
 263             AVRational iTimebase = d->inputFile->streams[d->audioStreamIndex]->time_base;
 264             AVRational oTimebase = e->audioStream->time_base;
 265             //opkt.pts = (int64_t)(pkt->dts*iTimebase.num/(double)iTimebase.den * (double)oTimebase.den/oTimebase.num);
 266             //opkt.dts = (int64_t)(pkt->dts*iTimebase.num/(double)iTimebase.den * (double)oTimebase.den/oTimebase.num);
 267             if(pkt->pts != AV_NOPTS_VALUE) {
 268                 opkt.pts= av_rescale_q(av_rescale_q(pkt->pts, iTimebase, AV_TIME_BASE_Q) - e->start_time, AV_TIME_BASE_Q,  oTimebase);
 269                 //opkt.pts = av_rescale_q(pkt->pts, iTimebase, oTimebase);
 270             } else {
 271                 opkt.pts = AV_NOPTS_VALUE;
 272             }
 273             {
 274                 int64_t dts;
 275                 if (pkt->dts == AV_NOPTS_VALUE)
 276                     dts = AV_NOPTS_VALUE;//ist->next_pts;
 277                 else
 278                     dts= av_rescale_q(pkt->dts, iTimebase, AV_TIME_BASE_Q);
 279                 opkt.dts= av_rescale_q(dts - e->start_time, AV_TIME_BASE_Q,  oTimebase);
 280
 281                 //opkt.dts = av_rescale_q(pkt->dts, iTimebase, oTimebase);
 282             }
 283         }
 284         //cout << "Audio Pkt " << pkt->dts << endl;
 285       }
 286     }
 287
 288     if(packetFrame)
 289     {
 290         avframe.key_frame = pkt->flags & PKT_FLAG_KEY;
 291         //opkt.data=  pkt->data;
 292         //opkt.size=  pkt->size;
 293
 294         //opkt.pts= pkt->pts; //FIXME ist->pts?
 295         //opkt.dts= pkt->dts;
 296
 297         opkt.flags= pkt->flags;
 298
 299
 300         if(av_parser_change(d->inputFile->streams[pkt->stream_index]->parser,
 301                             opkt.stream_index==e->videoStreamIndex?e->videoStream->codec:e->audioStream->codec,
 302                             &opkt.data, &opkt.size, pkt->data, pkt->size, pkt->flags & PKT_FLAG_KEY))
 303         {
 304             opkt.destruct= av_destruct_packet;
 305         }
 306         e->addPacket(&opkt);
 307
 308         av_free_packet(&opkt);
 309         av_free_packet(pkt);
 310     }
 311
 312     double p = d->getNextFrameTime()*100000.0/duration;
 313     if(p > 100.0) p = 100.0;
 314     completion = p;
 315
 316   }
 317   return OkCode;
 318 }
 319
 320 double Transcoder::getCompletion() const
 321 {
 322   return completion;
 323 }
 324
 325 void Transcoder::abort()
 326 {
 327   abortSignal = true;
 328 }