ARICHNtupleModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Luka Santelj, Dino Tahirovic                             *
 *                                                                        *
 * The purpose of this module is to test the reconstruction of            *
 * the particles with ARICH.                                              *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef ARICHNTUPLEMODULE_H
#define ARICHNTUPLEMODULE_H
 
#include <framework/datastore/StoreArray.h>
#include <framework/core/Module.h>
 
#include <mdst/dataobjects/Track.h>
#include <mdst/dataobjects/MCParticle.h>
 
#include <arich/dataobjects/ARICHLikelihood.h>
#include <arich/dataobjects/ARICHAeroHit.h>
#include <arich/dataobjects/ARICHTrack.h>
#include <arich/dataobjects/ARICHInfo.h>
 
 
// ROOT
#include <string>
#include <TFile.h>
#include <TTree.h>
#include <arich/modules/arichNtuple/ARICHNtupleStruct.h>
 
namespace Belle2 {
  class ARICHNtupleModule : public Module {
 
  public:
 
    ARICHNtupleModule();
 
    virtual ~ARICHNtupleModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
  private:
 
    // required input
    StoreArray<ARICHTrack> m_arichTracks; 
    StoreArray<ARICHLikelihood> m_arichLikelihoods; 
    // optional input
    StoreArray<Track> m_tracks; 
    StoreArray<MCParticle> m_arichMCPs; 
    StoreArray<ARICHAeroHit> m_arichAeroHits; 
    StoreArray<ARICHInfo> m_arichInfo; 
    std::string m_outputFile; 
    TFile* m_file; 
    TTree* m_tree; 
    ARICH::ARICHTree m_arich; 
  };
 
} // Belle2 namespace
 
#endif