BesRootIO Class Reference

#include <BesRootIO.hh>

Public Member Functions
	BesRootIO (G4int, G4int, G4int, G4int, G4int, G4int, G4int, G4int, G4int, G4int, G4String)
	~BesRootIO ()
void	SetMdcDigiFlag (G4int flag)
void	SetCgemDigiFlag (G4int flag)
void	SaveEmcDigiRootEvent ()
void	SaveHitRootEvent (G4int, const G4Event *)
void	SaveMdcHitRoot ()
void	SaveCgemHitRoot ()
void	SaveTofHitRoot ()

Public Attributes
TFile *	f
TTree *	HitTree

Detailed Description

Definition at line 18 of file BesRootIO.hh.

Constructor & Destructor Documentation

BesRootIO()

BesRootIO::BesRootIO	(	G4int	mdcTruFlag,
		G4int	mdcDigiFlag,
		G4int	cgemTruFlag,
		G4int	cgemDigiFlag,
		G4int	tofTruFlag,
		G4int	tofDigiFlag,
		G4int	emcTruFlag,
		G4int	emcDigiFlag,
		G4int	mucTruFlag,
		G4int	mucDigiFlag,
		G4String	name
	)

Definition at line 41 of file BesRootIO.cc.

        :m_mdcTruFlag(mdcTruFlag),m_mdcDigiFlag(mdcDigiFlag),
        m_cgemTruFlag(cgemTruFlag),m_cgemDigiFlag(cgemDigiFlag),
        m_tofTruFlag(tofTruFlag),m_tofDigiFlag(tofDigiFlag),
        m_emcTruFlag(emcTruFlag),m_emcDigiFlag(emcDigiFlag),
        m_mucTruFlag(mucTruFlag),m_mucDigiFlag(mucDigiFlag),m_rootFile(name)
{
    m_DigiMan = G4DigiManager::GetDMpointer();
 
    f = new TFile(m_rootFile,"RECREATE");
    //create a MdcHit TTree
    m_TMcHitEvent = new TMcHitEvent();
    HitTree = new TTree("HitTree","staff data from HitCollection");
 
    HitTree->Branch("TMcHitEvent", "TMcHitEvent", &m_TMcHitEvent, 32000, 99);
}

~BesRootIO()

BesRootIO::~BesRootIO

(

)

Definition at line 63 of file BesRootIO.cc.

{
    f->Write();
}

Member Function Documentation

SaveCgemHitRoot()

void BesRootIO::SaveCgemHitRoot

(

)

Definition at line 165 of file BesRootIO.cc.

{   
  //std::cout << "SaveCgemHitRoot" << std::endl;
    G4int THCID = -1;
    THCID = m_DigiMan->GetHitsCollectionID("BesCgemHitsCollection");
    if (THCID>=0)
    {
      BesCgemHitsCollection *cgemDC=(BesCgemHitsCollection*)(m_DigiMan->GetHitsCollection(THCID));
      G4int nHits = cgemDC->entries();
        //std::cout << "nHits: " << nHits << std::endl;
      if (nHits>0)
      {
        BesCgemHit* cgemHit;
        for (int i=0;i<nHits;i++)
        {
          cgemHit=(*cgemDC)[i];
          TMcHitCgem* tCgemHit = new TMcHitCgem();
 
          tCgemHit->SetTrackID             (cgemHit->GetTrackID            ());
          tCgemHit->SetLayerID             (cgemHit->GetLayerID            ());
          tCgemHit->SetPDGCode             (cgemHit->GetPDGCode            ());
          tCgemHit->SetGlobalTime          (cgemHit->GetGlobalTime         ());
          tCgemHit->SetTotalEnergyDeposit  (cgemHit->GetTotalEnergyDeposit ());
          tCgemHit->SetStepLength          (cgemHit->GetStepLength         ());
          
          G4ThreeVector XYZ_pre  = cgemHit->GetPositionOfPrePoint ();
          G4ThreeVector XYZ_post = cgemHit->GetPositionOfPostPoint();
          G4ThreeVector P_pre    = cgemHit->GetMomentumOfPrePoint ();
          G4ThreeVector P_post   = cgemHit->GetMomentumOfPostPoint();
 
          TVector3 tmp_XYZ_pre  = TVector3(XYZ_pre.x()  , XYZ_pre.y()  , XYZ_pre.z()  );
          TVector3 tmp_XYZ_post = TVector3(XYZ_post.x() , XYZ_post.y() , XYZ_post.z() );
          TVector3 tmp_P_pre    = TVector3(P_pre.x()    , P_pre.y()    , P_pre.z()    );
          TVector3 tmp_P_post   = TVector3(P_post.x()   , P_post.y()   , P_post.z()   );
 
          tCgemHit->SetPositionOfPrePoint  ( tmp_XYZ_pre  );
          tCgemHit->SetPositionOfPostPoint ( tmp_XYZ_post );
          tCgemHit->SetMomentumOfPrePoint  ( tmp_P_pre    );
          tCgemHit->SetMomentumOfPostPoint ( tmp_P_post   );
      /*
// ******************************************************************
          G4ThreeVector Cu_pre  = cgemHit->GetPrePositionInCu ();
          G4ThreeVector Cu_post = cgemHit->GetPostPositionInCu();
          G4ThreeVector P_Cu_pre    = cgemHit->GetMomentumOfCuPre ();
          G4ThreeVector P_Cu_post   = cgemHit->GetMomentumOfCuPost();
          
          TVector3 tmp_Cu_pre  = TVector3(Cu_pre.x()  , Cu_pre.y()  , Cu_pre.z()  );
          TVector3 tmp_Cu_post = TVector3(Cu_post.x() , Cu_post.y() , Cu_post.z() );
          TVector3 tmp_P_Cu_pre    = TVector3(P_Cu_pre.x()    ,P_Cu_pre.y()    , P_Cu_pre.z()    );
          TVector3 tmp_P_Cu_post   = TVector3(P_Cu_post.x()   ,P_Cu_post.y()   , P_Cu_post.z()   );
       
          tCgemHit->SetPrePositionInCu  ( tmp_Cu_pre  );
          tCgemHit->SetPostPositionInCu ( tmp_Cu_post );
          tCgemHit->SetMomentumOfCuPre  ( tmp_P_Cu_pre    );
          tCgemHit->SetMomentumOfCuPost ( tmp_P_Cu_post  );
// ******************************************************************************************
          */
              //cgemHit->Print();
          m_TMcHitEvent->addMcHitCgem(tCgemHit);
 
        } /* End of 'for (int i=0;i<nHits;i++)' */
      } /* End of 'if (nHits>0)' */
    } /* End of 'if (THCID>=0)' */
} /* End of 'void BesRootIO::SaveCgemHitRoot()' */

Referenced by SaveHitRootEvent().

SaveEmcDigiRootEvent()

void BesRootIO::SaveEmcDigiRootEvent

(

)

Definition at line 92 of file BesRootIO.cc.

                                    {
  //std::cout << "SaveEmcDigiRootEvent" << std::endl;
  G4int emcDigiCollID = -1;
  emcDigiCollID = m_DigiMan->GetDigiCollectionID("BesEmcDigitsCollection");
  if(emcDigiCollID>=0)
  {
    BesEmcDigitsCollection* emcDC = (BesEmcDigitsCollection*)m_DigiMan->GetDigiCollection(emcDigiCollID);
    G4int nHits = emcDC->entries();
    //std::cout << "nHits: " << nHits << std::endl;
 
    if(nHits>0){
      BesEmcDigi* emcDigi;
      for (int i = 0; i < nHits; i++) {
        emcDigi=(*emcDC)[i];
        TMcDigiEmc* tEmcDigi = new TMcDigiEmc();
 
        tEmcDigi->SetPartId(emcDigi->GetPartId());    
        tEmcDigi->SetThetaNb(emcDigi->GetThetaNb());
        tEmcDigi->SetPhiNb(emcDigi->GetPhiNb());   
        tEmcDigi->SetEnergy(emcDigi->GetEnergy());
        tEmcDigi->SetTime(emcDigi->GetTime());
        tEmcDigi->SetTrackIndex(emcDigi->GetTrackIndex());
    //emcDigi->Print();
        //std::cout << "SetEnergy" <<  emcDigi->GetEnergy() << std::endl;
 
        m_TMcHitEvent->addMcDigiEmc(tEmcDigi);
       //std::cout << "addMcDigiEmc" << std::endl;
      }
    }
  }
}

Referenced by SaveHitRootEvent().

SaveHitRootEvent()

void BesRootIO::SaveHitRootEvent	(	G4int	runId,
		const G4Event *	evt
	)

Definition at line 68 of file BesRootIO.cc.

{
    //std::cout << "Save Hit Root Event" << std::endl;
    if (m_mdcDigiFlag)
    {
        SaveMdcHitRoot();
    }
    if (m_cgemDigiFlag)
    {
        SaveCgemHitRoot();
    }
    if (m_tofDigiFlag)
    {
        SaveTofHitRoot();
    }
    if (m_emcDigiFlag){
        SaveEmcDigiRootEvent();
    }
 
    HitTree->Fill();
    
    m_TMcHitEvent->Clear();
}

Referenced by BesEventAction::EndOfEventAction().

SaveMdcHitRoot()

void BesRootIO::SaveMdcHitRoot

(

)

Definition at line 124 of file BesRootIO.cc.

{   
  //std::cout << "SaveMdcHitRoot" << std::endl;
    G4int THCID = -1;
    THCID = m_DigiMan->GetHitsCollectionID("BesMdcHitsCollection");
    if (THCID>=0)
    {
        BesMdcHitsCollection* mdcDC = (BesMdcHitsCollection*) (m_DigiMan->GetHitsCollection(THCID));
        G4int nHits = mdcDC->entries();
    //std::cout << "nHits: " << nHits << std::endl;
        if (nHits>0)
        {
            BesMdcHit* mdcHit;
            for (int i=0;i<nHits;i++)
            {
                mdcHit=(*mdcDC)[i];
                TMcHitMdc* tMdcHit = new TMcHitMdc();
                tMdcHit->SetTrackID(mdcHit->GetTrackID());       
                tMdcHit->SetLayerNo(mdcHit->GetLayerNo()); 
                tMdcHit->SetCellNo(mdcHit->GetCellNo());
                tMdcHit->SetEdep(mdcHit->GetEdep()); 
                tMdcHit->SetDriftD(mdcHit->GetDriftD());
                tMdcHit->SetDriftT(mdcHit->GetDriftT());
                tMdcHit->SetGlobalT(mdcHit->GetGlobalT());
                tMdcHit->SetTheta(mdcHit->GetTheta());
                tMdcHit->SetEnterAngle(mdcHit->GetEnterAngle());
                tMdcHit->SetPosFlag(mdcHit->GetPosFlag());
 
                G4ThreeVector gTemp = mdcHit->GetPos();
                TVector3 tTemp = TVector3(gTemp.x(), gTemp.y(), gTemp.z());
                tMdcHit->SetPos(tTemp);
 
        //mdcHit->Print();
                m_TMcHitEvent->addMcHitMdc(tMdcHit);
            }
        }
    }
 
 
}

Referenced by SaveHitRootEvent().

SaveTofHitRoot()

void BesRootIO::SaveTofHitRoot

(

)

Definition at line 230 of file BesRootIO.cc.

{
   //retrieve G4Svc
    ISvcLocator* svcLocator = Gaudi::svcLocator();
    IG4Svc* tmpSvc;
    StatusCode sc = svcLocator->service("G4Svc", tmpSvc);
    G4Svc*  m_G4Svc = dynamic_cast<G4Svc *>(tmpSvc);
    double m_beamTime = m_G4Svc->GetBeamTime() * ns;
    m_TMcHitEvent->setBeamTime(m_beamTime);
    //std::cout << "beamtime: " << m_beamTime << std::endl;
 
    G4int THCID = -1;
    THCID = m_DigiMan->GetHitsCollectionID("BesTofHitsCollection");
    if (THCID>=0)
    {
        BesTofHitsCollection* tofDC = (BesTofHitsCollection*) (m_DigiMan->GetHitsCollection(THCID));
        G4int nHits = tofDC->entries();
    //std::cout << "nHits: " << nHits << std::endl;
        if (nHits>0)
        {
            BesTofHit* tofHit;
            for (int i=0;i<nHits;i++)
            {
                tofHit=(*tofDC)[i];
                TMcHitTof* tTofHit = new TMcHitTof();
                tTofHit->SetTrackIndex(tofHit->GetTrackIndex());
                tTofHit->SetG4Index(tofHit->GetG4Index());
                tTofHit->SetPartId(tofHit->GetPartId());
                tTofHit->SetScinNb(tofHit->GetScinNb());
                tTofHit->SetEdep(tofHit->GetEdep());
                tTofHit->SetStepL(tofHit->GetStepL());
                tTofHit->SetTrackL(tofHit->GetTrackL());
                tTofHit->SetTime(tofHit->GetTime());
                tTofHit->SetDeltaT(tofHit->GetDeltaT());
                tTofHit->SetCharge(tofHit->GetCharge());
 
                G4ThreeVector gTemp = tofHit->GetPos();
                TVector3 tTemp(gTemp.x(), gTemp.y(), gTemp.z());
                tTofHit->SetPos(tTemp);
 
                gTemp = tofHit->GetPDirection();
                tTemp = TVector3(gTemp.x(), gTemp.y(), gTemp.z());
                tTofHit->SetPDirection(tTemp);
 
                gTemp = tofHit->GetMomentum();
                tTemp = TVector3(gTemp.x(), gTemp.y(), gTemp.z());
                tTofHit->SetMomentum(tTemp);
 
        //tofHit->Print();
                m_TMcHitEvent->addMcHitTof(tTofHit);
            }
        }
    }
 
}

Referenced by SaveHitRootEvent().

SetCgemDigiFlag()

void BesRootIO::SetCgemDigiFlag ( G4int  flag )

inline

Definition at line 31 of file BesRootIO.hh.

{m_cgemDigiFlag=flag;}

SetMdcDigiFlag()

void BesRootIO::SetMdcDigiFlag ( G4int  flag )

inline

Definition at line 30 of file BesRootIO.hh.

{m_mdcDigiFlag=flag;}

Member Data Documentation

f

TFile* BesRootIO::f

Definition at line 24 of file BesRootIO.hh.

Referenced by BesRootIO(), and ~BesRootIO().

HitTree

TTree* BesRootIO::HitTree

Definition at line 25 of file BesRootIO.hh.

Referenced by BesRootIO(), and SaveHitRootEvent().

---

The documentation for this class was generated from the following files:

• BesRootIO.hh
• BesRootIO.cc