SaveCmd.C

////////////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 2008 The Regents of the University of California
//
// This file is part of Qbox
//
// Qbox is distributed 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.
// See the file COPYING in the root directory of this distribution
// or <http://www.gnu.org/licenses/>.
//
////////////////////////////////////////////////////////////////////////////////
//
// SaveCmd.C:
//
////////////////////////////////////////////////////////////////////////////////
// $Id: SaveCmd.C,v 1.17 2008-09-08 16:26:36 fgygi Exp $


#include "SaveCmd.h"
#include "SampleWriter.h"
#include "isodate.h"
#include "release.h"

using namespace std;

////////////////////////////////////////////////////////////////////////////////
int SaveCmd::action(int argc, char **argv)
{
  string usage("  Use: save [-text] [-atomsonly]");
  usage += string(" [-serial] [-no_wfv] filename");
  if ( !(argc>=2 && argc<=4 ) )
  {
    if ( ui->onpe0() )
      cout << usage << endl;
    return 1;
  }

  // set default encoding and flags
  bool base64 = true;
  bool atomsonly = false;
  bool serial = false;
  bool save_wfv = true;
  char* filename = 0;

  // check for -text or -base64 or -atomsonly or -serial arguments
  for ( int i = 1; i < argc; i++ )
  {
    string arg(argv[i]);

    if ( arg=="-text" )
    {
      base64 = false;
    }
    else if ( arg=="-atomsonly" )
    {
      atomsonly = true;
    }
    else if ( arg=="-serial" )
    {
      serial = true;
    }
    else if ( arg=="-no_wfv" )
    {
      save_wfv = false;
    }
    else if ( arg[0] != '-' && i == argc-1 )
    {
      filename = argv[i];
    }
    else
    {
      if ( ui->onpe0() )
        cout << usage << endl;
      return 1;
    }
  }

  if ( filename == 0 )
  {
    if ( ui->onpe0() )
      cout << usage << endl;
    return 1;
  }
  SampleWriter swriter(s->ctxt_);
  string description = string(" Created ") + isodate() +
                       string(" by qbox-") + release() + string(" ");
  swriter.writeSample(*s, filename, description, base64, atomsonly, serial,
                      save_wfv);

  return 0;
}