/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/***************************************************************************
 *            mov_encoder_thread.cc
 *
 *  Tue May 17 16:00:01 CEST 2005
 *  Copyright  2005 Bent Bisballe
 *  deva@aasimon.org
 ****************************************************************************/

/*
 *    This file is part of MIaV.
 *
 *    MIaV is free software; you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation; either version 2 of the License, or
 *    (at your option) any later version.
 *
 *    MIaV is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License
 *    along with MIaV; if not, write to the Free Software
 *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA.
 */

/*
 * $Id$
 */

/*
 * $Log$
 * Revision 1.6  2005/05/19 14:10:22  deva
 *
 * Multithreading rulez?
 *
 * Revision 1.5  2005/05/19 10:55:49  deva
 * Test for block encoding of length strlen("IPIPP").
 *
 * Revision 1.4  2005/05/17 19:16:26  deva
 * Made new mpeg writer work, with proper file permissions.
 *
 * Revision 1.3  2005/05/17 15:13:15  deva
 * *** empty log message ***
 *
 * Revision 1.2  2005/05/17 15:12:51  deva
 * Fixed file rights (All read on files and directories, and all execute on directories).
 *
 * Revision 1.1  2005/05/17 14:30:56  deva
 * Added code, preparing threaded encoding.
 */

#include <config.h>
#include "mov_encoder_thread.h"
#include <errno.h>
#include "miav_config.h"

MovEncoderThread::MovEncoderThread(const char *filename)
{
  file = open(filename, 
              O_CREAT | O_WRONLY, //| O_LARGEFILE
              S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
  if(file == -1) {
    fprintf(stderr, "Could not open file for writing: %s\n", strerror(errno));
    return;
  }

  threads = config->readInt("encoding_threads");

  for(int cnt = 0; cnt < threads; cnt++) {
    encs.push_back(new MovEncoder());
    tids.push_back(new pthread_t);
    pthread_create (tids[cnt], NULL, thread_run, encs[cnt]);
  }

  current_encoder = 0;
  current_frame = 0;

  num_frames_in_block = config->readString("frame_sequence")->length();
  fprintf(stderr, "Frame sequence length [%d]\n", num_frames_in_block); fflush(stderr);

  inputqueue = new Queue<Frame>();
}

MovEncoderThread::~MovEncoderThread()
{
    
  fprintf(stderr, "Clear - encode last[%d]\n", current_encoder); fflush(stderr);
    
  // Push any hanging frames.
  encs[current_encoder]->encodeSequence(inputqueue);
    
  fprintf(stderr, "Clear - Readback\n"); fflush(stderr);
    
  /*
   * Readback mode
   */
  for(int cnt = 0; cnt <= current_encoder; cnt++) {
    
    //    fprintf(stderr, "pop[%d]-", cnt); fflush(stderr);
    
    Queue<Frame> *outputqueue = encs[cnt]->getResultSequence();
    Frame *f;
    while((f = outputqueue->pop())) {
      int i = write(file, f->data, f->size);
      if(i == -1) perror("Write failed");
      
      //      fprintf(stderr, "wrote[%d]-", i); fflush(stderr);
      
      delete f;
    }
  }
  
  fprintf(stderr, "Clear - join threads\n"); fflush(stderr);

  for(int cnt = 0; cnt < threads; cnt++) {
    encs[cnt]->running = false;
    encs[cnt]->encodeSequence(NULL);
    pthread_join(*tids[cnt], NULL);
    delete encs[cnt];
  }

  fprintf(stderr, "Clear - close file\n"); fflush(stderr);

  if(file != -1) close(file);

  fprintf(stderr, "Clear - done\n"); fflush(stderr);

}

void MovEncoderThread::encode(Frame* frame)
{
  if(file == -1) return;

  //  fprintf(stderr, "build[%d]-", current_encoder); fflush(stderr);
  inputqueue->bpush(frame);
  
  /*
   * Encode mode
   */
  // Switch frame
  current_frame++;
  if(current_frame >= num_frames_in_block) {

    //    fprintf(stderr, "push[%d]-", current_encoder); fflush(stderr);

    encs[current_encoder]->encodeSequence(inputqueue);
    inputqueue = new Queue<Frame>();

    // Switch encoder
    current_frame = 0;
    current_encoder++;
    if(current_encoder >= threads) {
      // switch mode
      /*
       * Readback mode
       */
      for(int cnt = 0; cnt < threads; cnt++) {

        //        fprintf(stderr, "pop[%d]-", cnt); fflush(stderr);

        Queue<Frame> *outputqueue = encs[cnt]->getResultSequence();
        Frame *f;
        while((f = outputqueue->pop())) {
          int i = write(file, f->data, f->size);
          if(i == -1) perror("Write failed");

          //          fprintf(stderr, "wrote[%d]-", i); fflush(stderr);

          delete f;
        }
      }
      current_encoder = 0;
    }
  }
}