JRootDB.cc

Go to the documentation of this file.

#include <iostream>
#include <fstream>
#include <sstream>
#include <iomanip>
#include <memory>

#include "JDB/JSupport.hh"
#include "JDB/JDBincludes.hh"
#include "JDB/JDBDictionary.hh"

#include "JROOT/JRootStreamer.hh"
#include "JROOT/JRootDictionary.hh"
#include "JROOT/JROOTClassSelector.hh"
#include "JLang/JNullStream.hh"

#include "JSupport/JFileRecorder.hh"

#include "Jeep/JeepToolkit.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"

namespace {

  /**
   * File copy interace.
   */
  struct io_interface {
    virtual void copy(const std::string& input_file, const std::string& output_file) const = 0;
  };
  
  /**
   * File copy implementation.
   */
  template<class T>
  struct io : 
    public io_interface
  {
    virtual void copy(const std::string& input_file, const std::string& output_file) const override
    {
      using namespace std;
      using namespace JPP;

      T object;

      JRootReadableClass cls(object);

      JRootReader reader(null, JEquationParameters(), JDBDictionary::getInstance());

      ifstream in(input_file.c_str());

      // reader header line

      vector<JRootReadableClass> columns;

      string buffer;

      getline(in, buffer);

      for (istringstream is(buffer); is >> buffer; ) {
        columns.push_back(cls.find(buffer.c_str()));
      }

      JFileRecorder<T> out(output_file.c_str());

      out.open();

      while (getline(in, buffer)) {

        size_t i = 0;
        size_t n = 0;

        for (istringstream is(buffer); is >> buffer && i != columns.size(); ++i) {

          if (columns[i].is_valid()) {

            JRedirectString redirect(reader, buffer);

            if (reader.getObject(columns[i])) {
              n += 1;
            }
          }
        }

        if (n != 0) {
          out.put(object);
        }
      }

      in .close();
      out.close();      
    }
  };
  
  /**
   * Multi-copy.
   */
  struct IO :
    std::map<std::string, std::shared_ptr<io_interface> > 
  {
    /**
     * Add data type.
     *
     * \param  type            data type
     */
    template<class T>
    void operator()(const JLANG::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;

  for_each(io, JType<typelist>());

  io[selector]->copy(inputFile, outputFile);
}