ECLBackgroundModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Sam de Jong                                              *
 *               Alexandre Beaulieu                                       *
 *               Poyuan Chen                                              *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
//STL
#include <vector>
 
//Framework
#include <framework/core/HistoModule.h>
#include <framework/datastore/StoreArray.h>
 
 
class TH1F;
class TH2F;
 
namespace Belle2 {
  class ECLSimHit;
  class MCParticle;
  class ECLShower;
  class BeamBackHit;
  class ECLCrystalData;
 
#ifdef DOARICH
  class ARICHGeometryPar;
#endif
 
  class ECLBackgroundModule : public HistoModule {
 
  public:
 
    ECLBackgroundModule();
 
    virtual ~ECLBackgroundModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
    virtual void defineHisto() override;
 
  private:
 
    StoreArray<ECLSimHit> m_eclArray;
 
    StoreArray<MCParticle> m_mcParticles;
 
    StoreArray<BeamBackHit> m_BeamBackArray;
 
    StoreArray<ECLShower> m_eclShowerArray;
 
    int m_sampleTime;
 
    bool m_doARICH;
 
    std::vector<int> m_CryInt;
 
    int m_nEvent{0};
 
    TH1F* h_nECLSimHits{nullptr};
 
    TH1F* h_CrystalRadDoseTheta{nullptr};
    TH1F* h_CrystalRadDose{nullptr};
    TH1F* h_CrystalThetaID2{nullptr};
    TH1F* h_CrystalThetaID67{nullptr};
    TH2F* h_HitLocations{nullptr};
    TH1F* h_BarrelDose{nullptr};
 
    TH1F* hEdepPerRing{nullptr};
 
    TH1F* hNevtPerRing{nullptr};
 
 
 
    TH1F* h_DiodeRadDose{nullptr};
    TH1F* h_NeutronFlux{nullptr};
    TH1F* h_NeutronFluxThetaID2{nullptr};
    TH1F* h_NeutronFluxThetaID67{nullptr};
    TH1F* h_NeutronE{nullptr};
    TH1F* h_NeutronEThetaID0{nullptr};
    TH1F* h_PhotonE{nullptr};
 
 
    TH2F* h_ShowerVsTheta{nullptr};
    TH1F* h_Shower{nullptr};
 
    TH1F* h_ProdVert{nullptr};
    TH2F* h_ProdVertvsThetaId{nullptr};
 
    const double usInYr = 1e13;
    const double GeVtoJ = 1.6e-10;
 
    //from Alex Beaulieu's implementation
 
    ECLCrystalData* Crystal[8736] {0};
 
    int FillARICHBeamBack(BeamBackHit* aBBHit);
    int BuildECL();
    // Transform Histograms //
 
    int SetPosHistos(TH1F* h, TH2F* hFWD, TH2F* hBAR, TH2F* hBWD);
    TH2F* BuildPosHisto(TH1F* h, const char* sub);
    TH1F* BuildThetaIDWideHisto(TH1F* h_cry);
    TH1F* BuildARICHringIDHisto(TH1F* h_cell);
    int ARICHmod2row(int modID);
 
    //ECL channels
    static const int nECLCrystalTot = 8736;
    const int nECLCrystalBAR = 6624;
    const int nECLCrystalECF = 1152;
    const int nECLCrystalECB =  960;
    static const int nECLThetaID = 69;
 
    //ECL Pin Diodes
    const double DiodeArea = 2 * 2;
    const double DiodeThk = 0.1;
    const double SiRho = 2.33e-3;
    const double DiodeMass = DiodeArea * DiodeThk * SiRho;;
 
 
#ifdef DOARICH
    ARICHGeometryPar* m_arichgp {0};   
#endif
 
    //Below are the density for the BOARDS (not the HAPD themselves)
    //The values are Taken from Luka Šantelj's BeamBack_arich.cc
 
    const double HAPDarea      = 7.5 * 7.5;
    const double HAPDthickness = 0.2;
    const double HAPDmass      = 47.25e-3;
 
    //ARICH channels
    const int nHAPD = 420;
    const int nHAPDrings = 7;
    const int nHAPDperRing[7] = {42, 48, 54, 60, 66, 72, 78};
 
    // For shield studies
 
    TH1F* hEMDose{nullptr};
 
    TH1F* hEnergyPerCrystal{nullptr};
 
    TH1F* hDiodeFlux{nullptr};
 
    TH1F* hEgamma{nullptr};
    TH1F* hEneu{nullptr};
 
 
    TH1F* hARICHDoseBB{nullptr};
    TH1F* hHAPDFlux{nullptr};
 
    TH2F* hEnergyPerCrystalECF{nullptr};
    TH2F* hEnergyPerCrystalECB{nullptr};
    TH2F* hEnergyPerCrystalBAR{nullptr};
    TH1F* hEnergyPerCrystalWideTID{nullptr};
 
 
    TH2F* hEMDoseECF{nullptr};
    TH2F* hEMDoseECB{nullptr};
    TH2F* hEMDoseBAR{nullptr};
    TH1F* hEMDoseWideTID{nullptr};
 
    TH2F* hDiodeFluxECF{nullptr};
    TH2F* hDiodeFluxECB{nullptr};
    TH2F* hDiodeFluxBAR{nullptr};
    TH1F* hDiodeFluxWideTID{nullptr};
 
  };
 
} // end of Belle2 namespace