DataWriterModule.h

/***************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Jo-Frederik Krohn                                        *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 *************************************************************************/
#ifndef DataWriterModule_H
#define DataWriterModule_H
 
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
 
#include <mdst/dataobjects/MCParticle.h>
#include <mdst/dataobjects/KLMCluster.h>
#include <mdst/dataobjects/ECLCluster.h>
 
#include <TTree.h>
#include <TFile.h>
#include <string>
 
namespace Belle2 {
  class DataWriterModule : public Module {
 
  public:
 
    DataWriterModule();
 
    virtual ~DataWriterModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
  protected:
 
  private:
 
    std::string m_outPath = "KlIdBKGClassifierTrainingTuples.root";
 
    // KLM cluster variables for pure KLM and KLM+ECL Klongs
    Float_t    m_KLMnCluster;
    Float_t    m_KLMnLayer;
    Float_t    m_KLMnInnermostLayer;
    Float_t m_KLMglobalZ;
    Float_t m_KLMtime;
    Float_t m_KLMavInterClusterDist;
    Float_t m_KLMhitDepth;
    Float_t m_KLMenergy;
    Float_t m_KLMinvM;
    Float_t m_KLMTruth;
    Float_t m_KLMnextCluster;
    Float_t m_KLMTrackSepDist;
    Float_t m_KLMTrackSepAngle;
 
    Float_t m_KLMInitialTrackSepAngle;
    Float_t m_KLMTrackRotationAngle;
    Float_t m_KLMTrackClusterSepAngle;
    Float_t m_KLMAngleToMC;
 
 
    // variables of closest ECL cluster with respect to KLM cluster
    Float_t m_KLMECLDist;
    Float_t m_KLMECLE;
    Float_t m_KLMECLdeltaL;   // new
    Float_t m_KLMECLminTrackDist; //new
    Float_t m_KLMECLE9oE25;
    Float_t m_KLMECLTiming;
    Float_t m_KLMECLTerror;
    Float_t m_KLMECLEerror;
    Float_t m_KLMtrackToECL;
    Float_t m_KLMKLid;
    Float_t m_KLMMCMom;
    Float_t m_KLMMCPhi;
    Float_t m_KLMMCTheta;
    Float_t m_KLMMom;
    Float_t m_KLMPhi;
    Float_t m_KLMTheta;
 
    Float_t m_KLMMCStatus  ;
    Float_t m_KLMMCLifetime;
    Float_t m_KLMMCPDG     ;
    Float_t m_KLMMCPrimaryPDG     ;
    Float_t m_KLMECLHypo;
    Float_t m_KLMECLZMVA;
    Float_t m_KLMECLZ40;
    Float_t m_KLMECLZ51;
    Float_t m_KLMECLUncertaintyPhi;
    Float_t m_KLMECLUncertaintyTheta;
    Float_t m_KLMMCWeight;
    Float_t m_KLMtrackFlag;
    Float_t m_KLMeclFlag;
 
 
    // ECL cluster variables for pure ECL Klongs
    Float_t m_ECLE;
    Float_t m_ECLE9oE25;
    Float_t m_ECLTiming;
    Float_t m_ECLR;
    Float_t m_ECLEerror;
    Float_t m_ECLminTrkDistance; // new
    Float_t m_ECLdeltaL; // new
 
 
    Float_t m_ECLZ51;
    Float_t m_ECLZ40;
    Float_t m_ECLE1oE9;
    Float_t m_ECL2ndMom;
    Float_t m_ECLnumChrystals;
    Float_t m_ECLLAT;
    Float_t m_ECLZMVA;
    Float_t m_ECLKLid;
    Float_t m_ECLMCStatus  ;
    Float_t m_ECLMCLifetime;
    Float_t m_ECLMCPDG     ;
    Float_t m_ECLMCPrimaryPDG     ;
    Float_t m_ECLDeltaTime        ;
    Float_t m_ECLUncertaintyEnergy;
    Float_t m_ECLUncertaintyTheta ;
    Float_t m_ECLUncertaintyPhi   ;
    Float_t m_ECLMCMom;
    Float_t m_ECLMCPhi;
    Float_t m_ECLMCTheta;
    Float_t m_ECLMom;
    Float_t m_ECLPhi;
    Float_t m_ECLTheta;
    Float_t m_ECLZ;
    Float_t m_ECLTruth;
    Float_t m_isBeamBKG;
    Float_t m_ECLMCWeight;
 
    Float_t m_isSignal;
 
 
    StoreArray<MCParticle> m_mcParticles;
    StoreArray<KLMCluster> m_klmClusters;
    StoreArray<ECLCluster> m_eclClusters;
 
    TFile* m_f = nullptr; //
    TTree* m_treeKLM = nullptr;
    TTree* m_treeECLhadron = nullptr;
    TTree* m_treeECLgamma = nullptr;
 
    bool m_useKLM;
 
    bool m_useECL;
 
 
  }; // end class
} // end namespace Belle2
 
#endif