X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;ds=sidebyside;f=src%2Fklvwalk.cpp;h=671506047f95e0d9b37a74222b822c20168a9172;hb=f6761c9d335a482925d7ad9b1b72543684afcb77;hp=3704f10f8e42b479fad033052af91d22125d8cbd;hpb=8095eaa320551b6795d0368c0ad0c227a3167caa;p=asdcplib.git diff --git a/src/klvwalk.cpp b/src/klvwalk.cpp index 3704f10..6715060 100755 --- a/src/klvwalk.cpp +++ b/src/klvwalk.cpp @@ -1,10 +1,37 @@ -// -// klvwalk.cpp -// - -#include -#include -#include +/* +Copyright (c) 2005-2009, John Hurst +All rights reserved. + +Redistribution and use in source and binary forms, with or without +modification, are permitted provided that the following conditions +are met: +1. Redistributions of source code must retain the above copyright + notice, this list of conditions and the following disclaimer. +2. Redistributions in binary form must reproduce the above copyright + notice, this list of conditions and the following disclaimer in the + documentation and/or other materials provided with the distribution. +3. The name of the author may not be used to endorse or promote products + derived from this software without specific prior written permission. + +THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR +IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES +OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. +IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, +INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT +NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, +DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY +THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF +THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +*/ +/*! \file klvwalk.cpp + \version $Id$ + \brief KLV+MXF test +*/ + +#include "AS_DCP.h" +#include "MXF.h" +#include #include #include #include @@ -13,135 +40,241 @@ #include using namespace ASDCP; +using Kumu::DefaultLogSink; //------------------------------------------------------------------------------------------ // +// command line option parser class + +static const char* PROGRAM_NAME = "klvwalk"; // program name for messages +typedef std::list FileList_t; + +// Increment the iterator, test for an additional non-option command line argument. +// Causes the caller to return if there are no remaining arguments or if the next +// argument begins with '-'. +#define TEST_EXTRA_ARG(i,c) if ( ++i >= argc || argv[(i)][0] == '-' ) \ + { \ + fprintf(stderr, "Argument not found for option -%c.\n", (c)); \ + return; \ + } + +// +void +banner(FILE* stream = stdout) +{ + fprintf(stream, "\n\ +%s (asdcplib %s)\n\n\ +Copyright (c) 2005-2009 John Hurst\n\ +%s is part of the asdcplib DCP tools package.\n\ +asdcplib may be copied only under the terms of the license found at\n\ +the top of every file in the asdcplib distribution kit.\n\n\ +Specify the -h (help) option for further information about %s\n\n", + PROGRAM_NAME, ASDCP::Version(), PROGRAM_NAME, PROGRAM_NAME); +} -// There is no header file thet defines this function. -// You just have to know it's there... -void set_debug_mode(bool info_mode, bool debug_mode); +// +void +usage(FILE* stream = stdout) +{ + fprintf(stream, "\ +USAGE: %s [-r] [-v] [ ...]\n\ +\n\ + %s [-h|-help] [-V]\n\ +\n\ + -h | -help - Show help\n\ + -r - When KLV data is an MXF OPAtom file, display OPAtom headers\n\ + -v - Verbose. Prints informative messages to stderr\n\ + -V - Show version information\n\ +\n\ + NOTES: o There is no option grouping, all options must be distinct arguments.\n\ + o All option arguments must be separated from the option by whitespace.\n\ +\n", PROGRAM_NAME, PROGRAM_NAME); +} +// +// + class CommandOptions + { + CommandOptions(); + + public: + bool error_flag; // true if the given options are in error or not complete + bool version_flag; // true if the version display option was selected + bool help_flag; // true if the help display option was selected + bool verbose_flag; // true if the informative messages option was selected + bool read_mxf_flag; // true if the -r option was selected + FileList_t inFileList; // File to operate on + + CommandOptions(int argc, const char** argv) : + error_flag(true), version_flag(false), help_flag(false), verbose_flag(false), read_mxf_flag(false) + { + for ( int i = 1; i < argc; i++ ) + { + + if ( (strcmp( argv[i], "-help") == 0) ) + { + help_flag = true; + continue; + } + + if ( argv[i][0] == '-' && isalpha(argv[i][1]) && argv[i][2] == 0 ) + { + switch ( argv[i][1] ) + { + case 'h': help_flag = true; break; + case 'r': read_mxf_flag = true; break; + case 'V': version_flag = true; break; + case 'v': verbose_flag = true; break; + + default: + fprintf(stderr, "Unrecognized option: %s\n", argv[i]); + return; + } + } + else + { + if ( argv[i][0] != '-' ) + inFileList.push_back(argv[i]); + + else + { + fprintf(stderr, "Unrecognized option: %s\n", argv[i]); + return; + } + } + } + + if ( help_flag || version_flag ) + return; + + if ( inFileList.empty() ) + { + fputs("Input filename(s) required.\n", stderr); + return; + } + + error_flag = false; + } + }; + + +//--------------------------------------------------------------------------------------------------- +// int -main(int argc, char** argv) +main(int argc, const char** argv) { - Result_t result = RESULT_OK; - bool read_mxf = false; - bool rewrite_mxf = false; - int arg_i = 1; - set_debug_mode(true, true); + CommandOptions Options(argc, argv); - if ( strcmp(argv[1], "-r") == 0 ) - { - read_mxf = true; - arg_i++; - } - else if ( strcmp(argv[1], "-w") == 0 ) + if ( Options.version_flag ) + banner(); + + if ( Options.help_flag ) + usage(); + + if ( Options.version_flag || Options.help_flag ) + return 0; + + if ( Options.error_flag ) { - rewrite_mxf = true; - arg_i++; - assert(argc - arg_i == 2); + fprintf(stderr, "There was a problem. Type %s -h for help.\n", PROGRAM_NAME); + return 3; } - fprintf(stderr, "Opening file %s\n", argv[arg_i]); + FileList_t::iterator fi; + Result_t result = RESULT_OK; - if ( read_mxf ) + for ( fi = Options.inFileList.begin(); ASDCP_SUCCESS(result) && fi != Options.inFileList.end(); fi++ ) { - ASDCP::FileReader Reader; - ASDCP::MXF::OPAtomHeader Header; - - result = Reader.OpenRead(argv[arg_i]); + if (Options.verbose_flag) + fprintf(stderr, "Opening file %s\n", ((*fi).c_str())); + + if ( Options.read_mxf_flag ) // dump MXF + { + Kumu::FileReader Reader; + const Dictionary* Dict = &DefaultCompositeDict(); + ASDCP::MXF::OPAtomHeader Header(Dict); + + result = Reader.OpenRead((*fi).c_str()); + + if ( ASDCP_SUCCESS(result) ) + result = Header.InitFromFile(Reader); + + if ( ASDCP_SUCCESS(result) ) + Header.Dump(stdout); + + if ( ASDCP_SUCCESS(result) && Header.m_RIP.PairArray.size() > 2 ) + { + MXF::Array::const_iterator pi = Header.m_RIP.PairArray.begin(); - if ( ASDCP_SUCCESS(result) ) - result = Header.InitFromFile(Reader); + for ( pi++; pi != Header.m_RIP.PairArray.end() && ASDCP_SUCCESS(result); pi++ ) + { + result = Reader.Seek((*pi).ByteOffset); - // if ( ASDCP_SUCCESS(result) ) - Header.Dump(); + if ( ASDCP_SUCCESS(result) ) + { + MXF::Partition TmpPart(Dict); + result = TmpPart.InitFromFile(Reader); - if ( ASDCP_SUCCESS(result) ) - { - ASDCP::MXF::OPAtomIndexFooter Index; - result = Reader.Seek(Header.FooterPartition); + if ( ASDCP_SUCCESS(result) && TmpPart.BodySID > 0 ) + TmpPart.Dump(stdout); + } + } + } if ( ASDCP_SUCCESS(result) ) { - Index.m_Lookup = &Header.m_Primer; - result = Index.InitFromFile(Reader); + ASDCP::MXF::OPAtomIndexFooter Index(Dict); + result = Reader.Seek(Header.FooterPartition); + + if ( ASDCP_SUCCESS(result) ) + { + Index.m_Lookup = &Header.m_Primer; + result = Index.InitFromFile(Reader); + } + + if ( ASDCP_SUCCESS(result) ) + Index.Dump(stdout); } if ( ASDCP_SUCCESS(result) ) - Index.Dump(); + Header.m_RIP.Dump(stdout); } - } - else if ( rewrite_mxf ) - { - ASDCP::FileReader Reader; - ASDCP::FileWriter Writer; - ASDCP::MXF::OPAtomHeader Header; - ASDCP::MXF::OPAtomIndexFooter Index; - - result = Reader.OpenRead(argv[arg_i++]); - - if ( ASDCP_SUCCESS(result) ) - result = Header.InitFromFile(Reader); - - if ( ASDCP_SUCCESS(result) ) - result = Reader.Seek(Header.FooterPartition); - - if ( ASDCP_SUCCESS(result) ) - result = Index.InitFromFile(Reader); - - Header.m_Primer.ClearTagList(); - - if ( ASDCP_SUCCESS(result) ) - result = Writer.OpenWrite(argv[arg_i]); - - if ( ASDCP_SUCCESS(result) ) - result = Header.WriteToFile(Writer); - -// if ( ASDCP_SUCCESS(result) ) -// result = Index.WriteToFile(Writer); - - // essence packets - - // index - - // RIP - } - else // dump klv - { - ASDCP::FileReader Reader; - KLVFilePacket KP; - - result = Reader.OpenRead(argv[arg_i]); - - if ( ASDCP_SUCCESS(result) ) - result = KP.InitFromFile(Reader); - - while ( ASDCP_SUCCESS(result) ) + else // dump klv { - KP.Dump(stderr, true); - result = KP.InitFromFile(Reader); + Kumu::FileReader Reader; + KLVFilePacket KP; + + result = Reader.OpenRead((*fi).c_str()); + + if ( ASDCP_SUCCESS(result) ) + result = KP.InitFromFile(Reader); + + while ( ASDCP_SUCCESS(result) ) + { + KP.Dump(stdout, DefaultCompositeDict(), true); + result = KP.InitFromFile(Reader); + } + + if( result == RESULT_ENDOFFILE ) + result = RESULT_OK; } - - if( result == RESULT_ENDOFFILE ) - result = RESULT_OK; } - if ( result != RESULT_OK ) + if ( ASDCP_FAILURE(result) ) { fputs("Program stopped on error.\n", stderr); - + if ( result != RESULT_FAIL ) { - fputs(GetResultString(result), stderr); + fputs(result, stderr); fputc('\n', stderr); } - + return 1; } - + return 0; } @@ -149,13 +282,3 @@ main(int argc, char** argv) // // end klvwalk.cpp // - - - - - - - - - -