From 10ec2f09566b1fe57d950aef9a26b7d3e91995e4 Mon Sep 17 00:00:00 2001
From: Rintaro Kuroiwa <rkuroiwa@google.com>
Date: Wed, 29 Jan 2014 13:40:13 -0800
Subject: [PATCH] MPD generator driver program

Change-Id: I9f34efbb322c3315583ec0bddf7d48e5a4e22f92
---
 app/mpd_generator.cc      | 86 +++++++++++++++++++++++++++++++++++++++
 app/mpd_generator_flags.h | 14 +++++++
 mpd/mpd.gyp               |  1 +
 packager.gyp              | 13 ++++++
 4 files changed, 114 insertions(+)
 create mode 100644 app/mpd_generator.cc
 create mode 100644 app/mpd_generator_flags.h

diff --git a/app/mpd_generator.cc b/app/mpd_generator.cc
new file mode 100644
index 0000000000..a633306139
--- /dev/null
+++ b/app/mpd_generator.cc
@@ -0,0 +1,86 @@
+#include "app/mpd_generator_flags.h"
+#include "base/logging.h"
+#include "base/strings/string_split.h"
+#include "base/strings/stringprintf.h"
+#include "mpd/util/mpd_writer.h"
+
+namespace {
+const char kUsage[] =
+    "MPD generation driver program.\n"
+    "This program accepts MediaInfo files in human readable text "
+    "format and outputs an MPD.\n"
+    "The main use case for this is to output MPD for VOD.\n"
+    "Limitations:\n"
+    " Each MediaInfo can only have one of VideoInfo, AudioInfo, or TextInfo.\n"
+    " There will be at most 3 AdaptationSets in the MPD, i.e. 1 video, 1 "
+    "audio, and 1 text.\n"
+    "Sample Usage:\n"
+    "%s --input=\"video1.txt,video2.txt,audio1.txt\" "
+    "--output=\"video_audio.mpd\"";
+
+enum ExitStatus {
+  kSuccess = 0,
+  kEmptyInputError,
+  kEmptyOutputError,
+  kFailedToWriteMpdToFileError
+};
+
+ExitStatus CheckRequiredFlags() {
+  if (FLAGS_input.empty()) {
+    LOG(ERROR) << "--input is required.";
+    return kEmptyInputError;
+  }
+
+  if (FLAGS_output.empty()) {
+    LOG(ERROR) << "--output is required.";
+    return kEmptyOutputError;
+  }
+
+  return kSuccess;
+}
+
+ExitStatus RunMpdGenerator() {
+  DCHECK_EQ(CheckRequiredFlags(), kSuccess);
+  std::vector<std::string> base_urls;
+  std::vector<std::string> input_files;
+  typedef std::vector<std::string>::const_iterator Iterator;
+
+  base::SplitString(FLAGS_input, ',', &input_files);
+
+
+  if (!FLAGS_base_urls.empty()) {
+    base::SplitString(FLAGS_base_urls, ',', &base_urls);
+  }
+
+  dash_packager::MpdWriter mpd_writer;
+  for (Iterator it = base_urls.begin(); it != base_urls.end(); ++it)
+    mpd_writer.AddBaseUrl(*it);
+
+  for (Iterator it = input_files.begin(); it != input_files.end(); ++it) {
+    if (!mpd_writer.AddFile(it->c_str())) {
+      LOG(WARNING) << "MpdWriter failed to read " << *it << ", skipping.";
+    }
+  }
+
+  if (!mpd_writer.WriteMpdToFile(FLAGS_output.c_str())) {
+    LOG(ERROR) << "Failed to write MPD to " << FLAGS_output;
+    return kFailedToWriteMpdToFileError;
+  }
+
+  return kSuccess;
+}
+
+}  // namespace
+
+int main(int argc, char** argv) {
+  google::SetUsageMessage(base::StringPrintf(kUsage, argv[0]));
+  google::ParseCommandLineFlags(&argc, &argv, true);
+
+  ExitStatus status = CheckRequiredFlags();
+  if (status != kSuccess) {
+    google::ShowUsageWithFlags(argv[0]);
+    return status;
+  }
+
+  return RunMpdGenerator();
+}
diff --git a/app/mpd_generator_flags.h b/app/mpd_generator_flags.h
new file mode 100644
index 0000000000..329462e1be
--- /dev/null
+++ b/app/mpd_generator_flags.h
@@ -0,0 +1,14 @@
+// Copyright (c) 2014 Google Inc. All rights reserved.
+
+#ifndef APP_MPD_GENERATOR_FLAGS_H_
+#define APP_MPD_GENERATOR_FLAGS_H_
+
+#include <gflags/gflags.h>
+
+DEFINE_string(input, "", "Comma separated list of MediaInfo input files.");
+DEFINE_string(output, "", "MPD output file name.");
+DEFINE_string(base_urls,
+              "",
+              "Comma separated BaseURLs for the MPD. The values will be added "
+              "as <BaseURL> element(s) immediately under the <MPD> element.");
+#endif  // APP_MPD_GENERATOR_FLAGS_H_
diff --git a/mpd/mpd.gyp b/mpd/mpd.gyp
index 41a80d1f81..f65156bc85 100644
--- a/mpd/mpd.gyp
+++ b/mpd/mpd.gyp
@@ -75,6 +75,7 @@
         'util/mpd_writer.h',
       ],
       'dependencies': [
+        '../packager.gyp:file',
         'mpd_builder',
       ],
     },
diff --git a/packager.gyp b/packager.gyp
index 9956781ad4..b9d0cc1146 100644
--- a/packager.gyp
+++ b/packager.gyp
@@ -302,5 +302,18 @@
         'third_party/gflags/gflags.gyp:gflags',
       ],
     },
+    {
+      'target_name': 'mpd_generator',
+      'type': 'executable',
+      'sources': [
+        'app/mpd_generator.cc',
+        'app/mpd_generator_flags.h',
+      ],
+      'dependencies': [
+        'base/base.gyp:base',
+        'mpd/mpd.gyp:mpd_util',
+        'third_party/gflags/gflags.gyp:gflags',
+      ],
+    },
   ],
 }