ARICHNtupleModule.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#pragma once
 
#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 event() 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