ssphys/SSPhys/InfoCommand.cpp

   1 // InfoCommand.cpp: implementation of the CInfoCommand class.
   2 //
   3 //////////////////////////////////////////////////////////////////////
   4
   5 #include "stdafx.h"
   6 #include "InfoCommand.h"
   7 #include "Formatter.h"
   8 #include <SSPhysLib\SSFiles.h>
   9 #include <SSPhysLib\SSObject.h>
  10
  11 //////////////////////////////////////////////////////////////////////
  12 // Construction/Destruction
  13 //////////////////////////////////////////////////////////////////////
  14
  15 CInfoCommand::CInfoCommand()
  16   : CCommand ("info"),
  17     m_bOnlyRecords (false),
  18     m_bAllRecords (false),
  19     m_bDisplayAtOffset (false),
  20     m_Offset (0)
  21 {
  22
  23 }
  24
  25 COptionInfoList CInfoCommand::GetOptionsInfo () const
  26 {
  27   COptionInfoList options = CCommand::GetOptionsInfo();
  28   options.push_back (COptionInfo ('a', 'a', "all", "display information about all records", COptionInfo::noArgument));
  29   options.push_back (COptionInfo ('r', 'r', "records", "display basic information about the records", COptionInfo::noArgument));
  30   options.push_back (COptionInfo ('o', 'o', "offset", "display record at specific offset", COptionInfo::requiredArgument));
  31   return options;
  32 }
  33
  34 bool CInfoCommand::SetOption (const COption& option)
  35 {
  36   switch (option.id)
  37   {
  38   case 'a':
  39     m_bAllRecords = true;
  40     break;
  41   case 'r':
  42     m_bOnlyRecords = true;
  43     break;
  44   case 'o':
  45    m_bDisplayAtOffset = true;
  46    m_Offset = atoi ((const char*) option.value);
  47     break;
  48   default:
  49     return false;
  50   }
  51   return true;
  52 }
  53
  54 bool CInfoCommand::SetArguments (CArguments& args)
  55 {
  56   if (args.empty ())
  57     throw SSException ("no argument");
  58
  59   m_PhysFile = args.front ();
  60   args.pop ();
  61   return true;
  62 }
  63
  64 void CInfoCommand::Execute ()
  65 {
  66   std::auto_ptr <SSRecordFile> pFile (SSRecordFile::MakeFile (m_PhysFile));
  67   g_pFormatter->SetFileName (m_PhysFile);
  68 //  std::auto_ptr<ISSObjectVisitor> pFormatter (CPhysFormatterFactory::MakeFormatter ());
  69   if (pFile.get ())
  70   {
  71     if (m_bAllRecords)
  72     {
  73       SSRecordPtr pRecord = pFile->GetFirstRecord();
  74       while (pRecord)
  75       {
  76         if (m_bOnlyRecords)
  77         {
  78           pRecord->Dump (std::cout);
  79         }
  80         else
  81         {
  82           std::auto_ptr<SSObject> pObject (SSObject::MakeObject(pRecord));
  83           g_pFormatter->Format (*pObject);
  84           std::cout << std::endl;
  85         }
  86         pRecord = pFile->FindNextRecord(pRecord);
  87       }
  88     }
  89     else if (m_bDisplayAtOffset)
  90     {
  91       SSRecordPtr pRecord = pFile->GetRecord(m_Offset);
  92       std::auto_ptr<SSObject> pObject (SSObject::MakeObject(pRecord));
  93       g_pFormatter->Format (*pObject);
  94     }
  95     else
  96       pFile->Dump (std::cout);
  97   }
  98 }