/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/***************************************************************************
 *            mov_encoder_writer.cc
 *
 *  Sun May 22 12:51:36 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.2  2005/05/26 12:48:36  deva
 * *** empty log message ***
 *
 * Revision 1.1  2005/05/22 15:49:22  deva
 * Added multithreaded encoding support.
 *
 */
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>

#include <pthread.h>
#include <semaphore.h>

#include <errno.h>

#include <config.h>
#include "mov_encoder_writer.h"

MovEncoderWriter::MovEncoderWriter(const char* fname, FramePriorityQueue *q, sem_t *s, pthread_mutex_t *m)
{
  // Append suffix
  char f[256];
  sprintf(f, "%s.mpg", fname);

  filename = string(fname);
  
  file = open(f, 
              O_CREAT | O_WRONLY | O_SYNC, //| 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;
  }

  sem = s;
  queue = q;
  frame_number = 0;
  mutex = m;

  running = true;

  sequence_number = 0;
}

MovEncoderWriter::~MovEncoderWriter()
{
  if(file != -1) close(file);
}

//#define TEST_BIG_FILE 10*1024*1024
void MovEncoderWriter::run()
{
#ifdef TEST_BIG_FILE
  char big_buf[TEST_BIG_FILE];
#endif/* TEST_BIG_FILE*/
  //  fprintf(stderr, "\t\t\t\t\t\t\t\t\tFile Writer ready!\n"); fflush(stderr);

  Frame *frame;

  if(file == -1) return;

  while(running) {
    sem_wait(sem);

    if(queue->size() == 0) continue;

    pthread_mutex_lock(mutex);
    frame = queue->top();
    if(frame->number == frame_number) queue->pop();
    pthread_mutex_unlock(mutex);

    /*
    if(!frame) {
      fprintf(stderr, "\t\t\t\t\t\t\t\t\tNULL frame detected");
      continue;
    }
    */

    //    fprintf(stderr, "\t\t\t\t\t\t\t\t\tChecking frame [%d] against expected [%d]\n", 
    //        frame->number, frame_number); fflush(stderr);
    while(frame->number == frame_number) {

      int ret;
#ifdef TEST_BIG_FILE
      ret = write(file, big_buf, TEST_BIG_FILE);
#else /* TEST_BIG_FILE */
      ret = write(file, frame->data, frame->size);
#endif /* TEST_BIG_FILE */

      if(ret == -1) {
        fprintf(stderr, "Error in filewriter\n"); fflush(stderr);
        switch(errno) {
        case EAGAIN: // Non-blocking I/O has been selected using O_NONBLOCK and the write would block.
        case EBADF:  // fd is not a valid file descriptor or is not open for writing.
        case EFAULT: // buf is outside your accessible address space.
        case EINTR:  // The call was interrupted by a signal before any data was written.
        case EINVAL: // fd is attached to an object which is unsuitable for writing.
        case EIO:    // A low-level I/O error occurred while modifying the inode.
        case ENOSPC: // The device containing the file referred to by fd has no room for the data.
        case EPIPE:  // fd is connected to a pipe or socket whose reading end is closed.  When this hap-
                     // pens the writing process will also receive a SIGPIPE signal.  (Thus,  the  write
                     // return  value  is  seen only if the program catches, blocks or ignores this sig-
                     // nal.)
          fprintf(stderr, "Could not write to file: %s\n", strerror(errno)); fflush(stderr);
          return;

        case EFBIG:  // An attempt was made to write a file that exceeds the implementation-defined max-
                     // imum file size or the process' file size limit, or to write at a  position  past
                     // than the maximum allowed offset.
          fprintf(stderr, "Attempting to write to a new file\n"); fflush(stderr);
          // Close current file
          if(file != -1) close(file);
          {
            sequence_number++;
            // Create new filename
            char f[256];
            sprintf(f, "%s-%d.mpg", filename.c_str(), sequence_number);
            
            // Open new file
            file = open(f, 
                        O_CREAT | O_WRONLY | O_SYNC, //| 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)); fflush(stderr);
              return;
            }
          }
          // Do write again.
#ifdef TEST_BIG_FILE
          ret = write(file, big_buf, TEST_BIG_FILE);
#else /* TEST_BIG_FILE */
          ret = write(file, frame->data, frame->size);
#endif /* TEST_BIG_FILE */
          if(ret == -1) {
            fprintf(stderr, "Could not write to file: %s\n", strerror(errno)); fflush(stderr);
            return;
          }
          fprintf(stderr, "Success\n"); fflush(stderr);
          break;
        }
      }

      delete frame;
      
      //      fprintf(stderr, "\t\t\t\t\t\t\t\t\tWrite frame [%d]\n", frame->number); fflush(stderr);

      frame_number++;

      pthread_mutex_lock(mutex);
      frame = queue->top();
      if(frame->number == frame_number) queue->pop();
      pthread_mutex_unlock(mutex);
    
    }
  }
}