pipelist.c

来自「Motion JPEG编解码器源代码」· C语言 代码 · 共 208 行

/*
 *  pipelist.[ch] - provides two functions to read / write pipe
 *                  list files, the "recipes" for lavpipe
 *
 *  Copyright (C) 2001, pHilipp Zabel <[e-mail]>
 *  Copyright (C) 2001, Matthew Marjanovic <[e-mail]>
 *
 *  This program 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.
 *
 *  This program 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 this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "mjpeg_logging.h"

#include "pipelist.h"

extern int verbose;

int read_pipe_list (char *name, PipeList *pl)
{
   FILE *fd;
   char  line[1024];
   int   i, j, n;

   fd = fopen (name, "r");

   fgets (line, 1024, fd);
   if (strcmp (line, "LAV Pipe List\n") == 0) {

      /* 1. video norm */

      fgets (line, 1024, fd);
      if(line[0]!='N' && line[0]!='n' && line[0]!='P' && line[0]!='p')
      {
         mjpeg_error("Pipe list second line is not NTSC/PAL");
         exit (1);
      }
	  mjpeg_debug("Pipe list norm is %s",line[0]=='N'||line[0]=='n'?"NTSC":"PAL");
      if(line[0]=='N'||line[0]=='n')
         pl->video_norm = 'n';
      else
         pl->video_norm = 'p';

      /* 2. input streams */

      fgets (line, 1024, fd);
      if (sscanf (line, "%d", &(pl->source_count)) != 1) {
         mjpeg_error( "pipelist: # of input streams expected, \"%s\" found", line);
         return -1;
      }
	  
	  mjpeg_info( "Pipe list contains %d input streams", pl->source_count);

      pl->source_cmd = (char **) malloc (pl->source_count * sizeof (char *));

      for (i=0; i<pl->source_count; i++) {
         fgets(line,1024,fd);
         n = strlen(line);
         if(line[n-1]!='\n') {
            mjpeg_error("Input cmdline in pipe list too long");
            exit(1);
         }
         line[n-1] = 0; /* Get rid of \n at end */
         pl->source_cmd[i] = (char *) malloc (n);
         strncpy (pl->source_cmd[i], line, n);
      }
      
      /* 3. sequences */
      
      pl->frame_count = 0;
      pl->segment_count = 0;
      pl->segments = (PipeSegment **) malloc (32 * sizeof (PipeSegment *));
      while (fgets (line, 1024, fd)) {

         PipeSegment *seq = (PipeSegment *) malloc (sizeof (PipeSegment));
         
         /* 3.1. frames in sequence */

         if (sscanf (line, "%d", &(seq->frame_count)) != 1) {
            mjpeg_error( "pipelist: # of frames in sequence expected, \"%s\" found", line);
            return -1;
         }
         if (seq->frame_count < 1) {
			 mjpeg_error_exit1( "Pipe list contains sequence of length < 1 frame");
         }

		 mjpeg_debug( "Pipe list sequence %d contains %d frames", 
					  pl->segment_count, seq->frame_count);
         pl->frame_count += seq->frame_count;
         
         /* 3.2. input streams */
     
         n = !fgets (line, 1024, fd);
         if (sscanf (line, "%d", &(seq->input_count)) != 1) {
            mjpeg_error( "pipelist: # of streams in sequence expected, \"%s\" found", line);
            return -1;
         }
         seq->input_index = (int *) malloc (seq->input_count * sizeof (int));
         seq->input_offset = (unsigned long *) malloc (seq->input_count * sizeof (unsigned long));
         for (i=0; i<seq->input_count; i++) {
            if (!fgets (line, 1024, fd)) n++;
            j = sscanf (line, "%d %lud", &(seq->input_index[i]), &(seq->input_offset[i]));
            if (j == 1) {
               /* if no offset is given, assume ofs = 0 */
               seq->input_offset[i] = 0;
               j++;
            }
            if (j != 2) {
               mjpeg_error( "pipelist: input stream index & offset expected, \"%s\" found", line);
               return -1;
            }
            if (seq->input_index[i] >= pl->source_count) {
               mjpeg_error( "Sequence requests input stream that is not contained in pipe list");
               exit (1);
            }
         }

         /* 3.3. output cmd */

         fgets (line, 1024, fd);
         if (n > 0) {
            mjpeg_error( "Error in pipe list: Unexpected end");
            mjpeg_error( "\"%s\"", line);
            exit (1);
         }
         n = strlen(line);
         if(line[n-1]!='\n') {
            mjpeg_error("Output cmdline in pipe list too long");
            exit(1);
         }
         line[n-1] = 0; /* Get rid of \n at end */
         seq->output_cmd = (char *) malloc (n);
         strncpy (seq->output_cmd, line, n);
         
         pl->segments[pl->segment_count++] = seq;
         if ((pl->segment_count % 32) == 0)
         pl->segments = (PipeSegment **) realloc (pl->segments, 
                   (pl->segment_count + 32) * sizeof (PipeSegment *));
            pl->segments = (PipeSegment **) realloc (pl->segments, sizeof (pl->segments) + 32 * sizeof (PipeSegment *));
      }
      return 0;
   }
   /* errno = EBADMSG; */
   return -1;
}

int write_pipe_list (char *name, PipeList *pl)
{
   FILE *fd;
   int   i, n;

   fd = fopen (name, "w");

   fprintf (fd, "LAV Pipe List\n");

   /* 1. video norm */

   fprintf (fd, "%s\n", (pl->video_norm == 'n') ? "NTSC" : "PAL");

   /* 2. input streams */

   fprintf (fd, "%d\n", pl->source_count);

   for (i=0; i<pl->source_count; i++)
      fprintf (fd, "%s\n", pl->source_cmd[i]);

   /* 3. sequences */

   for (i=0; i<pl->segment_count; i++) {
      
      PipeSegment *seq = pl->segments[i];

      /* 3.1. frames in sequence */
      
      fprintf (fd, "%d\n", seq->frame_count);

      /* 3.2. input streams */
      
      fprintf (fd, "%d\n", seq->input_count);
      for (n=0; n<seq->input_count; n++)
         fprintf (fd, "%d %lud\n", seq->input_index[n], seq->input_offset[n]);

      /* 3.3. output cmd */
      
      fprintf (fd, "%s\n", seq->output_cmd);
   }
   return 0;

}