RootTreeWriter.h

// -*- mode: c++; mode: auto-fill; mode: flyspell-prog; -*-

#ifndef RootTreeWriter_h
#define RootTreeWriter_h 1

#include "RootProcessor.h"

// root includes

#include "TFile.h"
#include "TTree.h"

#include <fstream>

using namespace gloria;

namespace luiza {



   class RootTreeWriter : public RootProcessor {

  public:

    virtual RootTreeWriter*  newProcessor();


    RootTreeWriter();

    virtual void init() ;

    virtual void startAnalysis( GloriaDataContainer* data ) ;


    virtual void processData( GloriaDataContainer* data ) ;


    virtual void check( GloriaDataContainer* data ) ;


    virtual void concludeAnalysis( GloriaDataContainer* data ) ;


    virtual void end() ;


  protected:

     void InitializeRootTree(string treeName, GloriaFitsTable *source);

     void FillRootTree(GloriaFitsTable *source);

     void WriteRootTree();

    string _tableCollectionName ;

     string _rootFileName;

     bool _oneFile;

     bool _memoryResident;

     bool _addCounter;

     std::vector< std::string > _selectedColumns;

     std::vector< std::string > _intHeaderRecords;

     std::vector< std::string > _floatHeaderRecords;

     static TApplication* rootApl;

     TTree* _outputTree;

     string _outputFileName;

     TFile* _outputFile;

     int _nLoops ;

     int _nTables ;

     int _nWritten ;

     int _iRow;

     long *_intBranch;

     double *_floatBranch;

     bool _inLoop;

    void printEndMessage() const ;
  } ;

 } // end namespace luiza
#endif