JPrintTree.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
#include <iomanip>

#include "TError.h"
#include "TROOT.h"
#include "TFile.h"

#include "km3net-dataformat/offline/Head.hh"
#include "km3net-dataformat/offline/Evt.hh"

#include "JDAQ/JDAQEventIO.hh"
#include "JDAQ/JDAQTimesliceIO.hh"
#include "JDAQ/JDAQSummarysliceIO.hh"

#include "antares-dataformat/TimeSlice.hh"
#include "antares-dataformat/PhysicsEvent.hh"

#include "JROOT/JTreeParameters.hh"
#include "JROOT/JTreeReader.hh"
#include "JSupport/JSupport.hh"

#include "JLang/JSinglePointer.hh"
#include "JLang/JBool.hh"
#include "JLang/JType.hh"
#include "JLang/JTypeList.hh"
#include "JLang/JNullType.hh"

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


namespace {

  using namespace JPP;

  
  /**
   * Auxiliary class to build list of valid TreeReader pointers.
   */
  class JTreeList :
    public std::vector< JSinglePointer<JAbstractTreeReader> >
  {
  public:
    /**
     * Constructor.
     *
     * \param  typelist   type list
     */
    template<class T>
    JTreeList(JType<T> typelist) 
    {
      for_each(*this, typelist);
    }


    /**
     * Addition of class.
     *
     * \param  type       type
     */
    template<class T>
    void operator()(JType<T> type)
    {
      add(JType<T>(), JBool<JTreeParametersAvailable<T>::result>());
    }

  private:
    /**
     * Addition of TTree compatible class.
     */
    template<class T>
    void add(JType<T>, JBool<true>)
    {
      this->push_back(new JTreeReader<T>());
    }
  
  
    /**
     * No addition of TTree incompatible class.
     */
    template<class T>
    void add(JType<T>, JBool<false>)
    {}
  };
}


/**
 * \file
 *
 * Auxiliary program to print ROOT TTree information.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;

  vector<string> inputFile;
  int            debug;

  try { 

    JParser<> zap("Auxiliary program to print ROOT TTree information.");
    
    zap['f'] = make_field(inputFile);
    zap['d'] = make_field(debug)           = 1;

    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }

  cout.tie(&cerr);

  gErrorIgnoreLevel = kFatal;

  JType<JAppend<JAllTypes_t, JAntaresTypes_t>::typelist> typelist;

  JTreeList inspector(typelist);

  int number_of_errors = 0;

  for (vector<string>::const_iterator file_name = inputFile.begin(); file_name != inputFile.end(); ++file_name) {

    TFile* fp = TFile::Open(file_name->c_str());

    cout << *file_name << flush;

    if (fp != NULL) {

      cout << endl;

      for (JTreeList::iterator i = inspector.begin(); i != inspector.end(); ++i) {

        if ((*i)->load(fp)) {
          cout << setw(24) << left  <<  (**i)->GetName()                 << ' ' 
               << setw(32) << left  <<  (*i)->getDictionary()->GetName() << ' ' 
               << setw(10) << right <<  (**i)->GetEntries()              << ' '
               << setw( 6) << right << ((**i)->GetTotBytes() >> 20)      << " [MB]" << endl;
        }
      }

    } else {

      ++number_of_errors;

      cout << " not opened." << endl;
    }
  }

  return (number_of_errors == 0 ? 0 : 1);
}