DataWriterModule.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/core/Module.h>
#include <framework/datastore/StoreArray.h>
 
#include <TTree.h>
#include <TFile.h>
#include <string>
 
namespace Belle2 {
  class MCParticle;
  class KLMCluster;
  class ECLCluster;

  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