Merge CV, ver=5.1; ops/methods from HV, and interface from CV where possible

[goodguy/history.git] / cinelerra-5.1 / libzmpeg3 / mpeg3toc.C

diff --git a/cinelerra-5.1/libzmpeg3/mpeg3toc.C b/cinelerra-5.1/libzmpeg3/mpeg3toc.C
new file mode 100644 (file)
index 0000000..4364fcc
--- /dev/null
+++ b/cinelerra-5.1/libzmpeg3/mpeg3toc.C
@@ -0,0 +1,133 @@
+// New version.
+// Very basic table of contents utility since most of the time it's going to be
+// built inside a graphical program.
+#include "libzmpeg3.h"
+#include <stdint.h>
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <signal.h>
+
+int done;
+
+void sigint(int n)
+{
+  done = 1;
+}
+
+void thumbnail(int track)
+{
+#if 0
+  int64_t framenum; uint8_t *tdat; int mw, mh;
+  mpeg3_get_thumbnail(track, &framenum, &tdat, &mw, &mh);
+  char fn[256];  FILE *fp;
+  sprintf(fn,"/tmp/dat/s%df%05d.pbm",track,framenum);
+  if( !(fp=fopen(fn,"w")) ) return;
+  fprintf(fp,"P5\n%d %d\n255\n",w,h);
+  fwrite(tdat,mw,mh,fp);
+  fclose(fp);
+#endif
+}
+
+int main(int argc, char *argv[])
+{
+  int i;
+  char *src = 0, *dst = 0;
+  int program = 0;
+  int verbose = 0;
+
+  if(argc < 3) {
+    fprintf(stderr, "Table of contents generator version %d.%d.%d\n"
+      "Create a table of contents for a DVD or mpeg stream.\n"
+      "Usage: mpeg3toc <path> <output>\n"
+      "\n"
+      "-v Print tracking information\n"
+      "-p <n> program number (default 0)\n"
+      "\n"
+      "The path should be absolute unless you plan\n"
+      "to always run your movie editor from the same directory\n"
+      "as the filename.  For renderfarms the filesystem prefix\n"
+      "should be / and the movie directory mounted under the same\n"
+      "directory on each node.\n\n"
+      "Example: mpeg3toc -v /cdrom/video_ts/vts_01_0.ifo titanic.toc\n",
+      mpeg3_major(),
+      mpeg3_minor(),
+      mpeg3_release());
+    exit(1);
+  }
+
+  for( i=1; i<argc; ++i ) {
+    if( !strcmp(argv[i], "-v") ) {
+      verbose = 1;
+    }
+    else if( !strcmp(argv[i], "-p") ) {
+      if( ++i < argc ) program = atoi(argv[i]);
+    }
+    else if(argv[i][0] == '-') {
+      fprintf(stderr, "Unrecognized command %s\n", argv[i]);
+      exit(1);
+    }
+    else if(!src) {
+      src = argv[i];
+    }
+    else if(!dst) {
+      dst = argv[i];
+    }
+    else {
+      fprintf(stderr, "Ignoring argument \"%s\"\n", argv[i]);
+    }
+  }
+
+  if(!src) {
+    fprintf(stderr, "source path not supplied.\n");
+    exit(1);
+  }
+
+  if(!dst) {
+    fprintf(stderr, "destination path not supplied.\n");
+    exit(1);
+  }
+
+  int64_t total_bytes;
+  zmpeg3_t *file = mpeg3_start_toc(src, dst, program, &total_bytes);
+  if(!file) exit(1);
+  struct timeval new_time;
+  struct timeval prev_time;
+  struct timeval start_time;
+  gettimeofday(&prev_time, 0);
+  gettimeofday(&start_time, 0);
+  done = 0;
+  signal(SIGINT, sigint);
+  //mpeg3_set_thumbnail_callback(file, thumbnail);
+  mpeg3_set_cpus(file, 3);
+  
+  while(!done) {
+    int64_t bytes_processed = 0;
+    mpeg3_do_toc(file, &bytes_processed);
+
+    gettimeofday(&new_time, 0);
+    if( verbose && new_time.tv_sec - prev_time.tv_sec > 1 ) {
+      int64_t elapsed_seconds = new_time.tv_sec - start_time.tv_sec;
+      int64_t total_seconds = elapsed_seconds * total_bytes / bytes_processed;
+      int64_t eta = total_seconds - elapsed_seconds;
+      fprintf(stderr, _LD"%% ETA: "_LD"m"_LD"s        \r", 
+        bytes_processed * 100 / total_bytes,
+        eta / 60, eta % 60);
+      fflush(stdout);
+      prev_time = new_time;
+    }
+
+    if(bytes_processed >= total_bytes) break;
+  }
+
+  mpeg3_stop_toc(file);
+  gettimeofday(&new_time, 0);
+  int64_t elapsed = new_time.tv_sec - start_time.tv_sec;
+  if(verbose) {
+    fprintf(stderr, _LD"m"_LD"s elapsed           \n", 
+      elapsed / 60, elapsed % 60);
+  }
+
+  return 0;
+}
+