JRootDB.cc

Go to the documentation of this file.

#include <iostream>
#include <fstream>
#include <iomanip>
#include <memory>
 
#include "JDB/JSupport.hh"
#include "JDB/JDBincludes.hh"
#include "JDB/JDBDictionary.hh"
 
#include "JROOT/JRootStreamer.hh"
#include "JROOT/JROOTClassSelector.hh"
#include "JROOT/JASCIIFileStreamer.hh"
 
#include "JLang/JType.hh"
 
#include "JSupport/JFileRecorder.hh"
#include "JSupport/JMeta.hh"
 
#include "Jeep/JeepToolkit.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
namespace {
 
  using JLANG::JType;
  using JSUPPORT::JMeta;
 
  /**
   * File copy interace.
   */
  struct io_interface {
    virtual ~io_interface() {}
    virtual void copy(const std::string& input_file, const JMeta& meta, const std::string& output_file) const = 0;
  };
  
  /**
   * Multi-copy.
   */
  struct IO :
    std::map<std::string, std::shared_ptr<io_interface> > 
  {
    /**
     * File copy implementation.
     */
    template<class T>
    struct io : 
      public io_interface
    {
      virtual void copy(const std::string& input_file, const JMeta& meta, const std::string& output_file) const override
      {
        using namespace JPP;
 
        JASCIIFileReader<T> in(input_file.c_str(), JDBDictionary::getInstance());
        JFileRecorder<typename JTYPELIST<T, JMeta>::typelist> out(output_file.c_str());
 
        out.open();
 
        out.put(meta);
 
        for (T object; in >> object; ) {
          out.put(object);
        }
 
        out.close();      
      }
    };
  
    /**
     * Constructor.
     *
     * \param  type            data type
     */
    template<class T>
    IO(const JType<T>& type)
    {
      using namespace JPP;
 
      for_each(*this, type);
    }
 
    /**
     * Add data type.
     *
     * \param  type            data type
     */
    template<class T>
    void operator()(const JType<T>& type)
    {
      (*this)[JPP::getClassname(T::Class_Name())] = std::make_shared< io<T> >();
    }
  };
}
 
 
/**
 * \file
 *
 * Auxiliary program to convert ASCII data from data base into ROOT format.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  typedef JDBTypes_t typelist;
 
  string              inputFile;
  string              outputFile;
  JROOTClassSelector  selector;
  int                 debug;
 
  try {
 
    JParser<> zap("Auxiliary program to convert ASCII data from data base into ROOT format.");
    
    zap['f'] = make_field(inputFile,    "ASCII formatted input file."\
                          "\nFirst line should list data members of data structure given at option -C");
    zap['o'] = make_field(outputFile);
    zap['C'] = make_field(selector,     "name of data structure")     = getROOTClassSelection<typelist>();
    zap['d'] = make_field(debug)        = 1;
 
    zap['C'] = JPARSER::not_initialised();
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
  IO io(getType<typelist>());
 
  io[selector]->copy(inputFile, JMeta(argc,argv), outputFile);
}