BesTuningIO Class Reference

#include <BesTuningIO.hh>

Public Member Functions
	BesTuningIO (std::vector< std::string >)
	~BesTuningIO ()
void	GetNextEvents (void)
void	GetMdcHits (void)
void	GetCgemHits (void)
void	GetTofHits (void)
void	GetEmcDigi (void)
void	GetMucHits (void)
void	GetRootEvent (int evtID)
void	GetMdcRootHits ()
void	GetCgemRootHits ()
void	GetTofRootHits ()
void	GetEmcRootDigi ()

Public Attributes
TFile *	f
TTree *	HitTree
TChain *	HitChain

Detailed Description

Definition at line 26 of file BesTuningIO.hh.

Constructor & Destructor Documentation

BesTuningIO()

BesTuningIO::BesTuningIO

(

std::vector< std::string >

name

)

Definition at line 41 of file BesTuningIO.cc.

        :m_tuningFile(name),m_evt(0)
{
    m_DigiMan = G4DigiManager::GetDMpointer();
    m_inputFileStream = new std::ifstream();
 
    if (ReadBoostRoot::GetFormatAR()){
        //tuning  input root file
        //TFile *f = new TFile(m_tuningFile);
        //HitTree = (TTree*)f->Get("HitTree");
        //m_TMcHitEvent = new TMcHitEvent();
    //TBranch *branch = HitTree->GetBranch("TMcHitEvent");
    // begin change from TTree to TChain
    HitChain = new TChain("HitTree");
    if (m_tuningFile.size()==0){
      std::cout << "there is no tuning file" << std::endl;
    }
    std::cout << "file number: " << m_tuningFile.size() << std::endl;
    for (int i = 0 ; i < m_tuningFile.size(); i++){
      
      //std::cout << "________________________________________________________________________________fileName : " << m_tuningFile[i] << std::endl;
      //HitChain->Add(&m_tuningFile[i]);
      HitChain->Add(m_tuningFile[i].c_str());
        }
        m_TMcHitEvent = new TMcHitEvent();
    TBranch *branch = HitChain->GetBranch("TMcHitEvent");
    
    // end   change from TTree to TChain
        branch->SetAddress(&m_TMcHitEvent);
    std::cout << "HitChain entries: " << HitChain->GetEntries() << std::endl;
    }
    else{// tuning input ascii file
        //m_inputFileStream->open(m_tuningFile.c_str());
        //if ((*m_inputFileStream).good()) {
        //    try {
        //        (*m_inputFileStream) >> m_version;
        //    } catch (AsciiDumpException& ) {
        //        std::cerr << "BesTuningIO::Got AsciiDumpException eror while reading VERSION block !!!" << std::endl;
        //    }
        //}else{
        //    std::cerr << "BesTuningIO::Open tuning input file error!!!" << std::endl;
        //}
    }
}

~BesTuningIO()

BesTuningIO::~BesTuningIO

(

)

Definition at line 86 of file BesTuningIO.cc.

                         {
    if (m_inputFileStream)  delete m_inputFileStream;
    if (m_evt) delete m_evt;
}

Member Function Documentation

GetCgemHits()

void BesTuningIO::GetCgemHits

(

void

)

Definition at line 163 of file BesTuningIO.cc.

{
  G4int cgemHitCollID = -1;
  cgemHitCollID = m_DigiMan->GetHitsCollectionID("BesCgemHitsCollection");
  if (cgemHitCollID>=0)
  {
    BesCgemHitsCollection* cgemDC = (BesCgemHitsCollection*)m_DigiMan->GetHitsCollection(cgemHitCollID);
    if (cgemDC)
    {
      G4int nHit = cgemDC->entries();
      if (nHit>0)
      {
        for (G4int i=0;i<nHit;i++)
        {
          delete (*cgemDC)[i];
        }
        cgemDC->GetVector()->clear();
      }
 
      std::vector<CgemHitType>::iterator iter;
      iter = (m_evt->cgemHit).hitCol.begin();
      // Loop over cgem hits
      for (; iter != (m_evt->cgemHit).hitCol.end(); iter++) 
      {
        BesCgemHit* newHit = new BesCgemHit();
 
        newHit->SetTrackID            ((*iter).m_ID_track   );
        newHit->SetLayerID            ((*iter).m_ID_layer   );
        newHit->SetPDGCode            ((*iter).m_pdg_code   );
        newHit->SetGlobalTime         ((*iter).m_global_time);
        newHit->SetTotalEnergyDeposit ((*iter).m_E_deposit  );
        newHit->SetStepLength         ((*iter).m_L_step     );
        newHit->SetPositionOfPrePoint (G4ThreeVector((*iter).m_XYZ_pre_x,(*iter).m_XYZ_pre_y,(*iter).m_XYZ_pre_z));
        newHit->SetPositionOfPostPoint(G4ThreeVector((*iter).m_XYZ_post_x,(*iter).m_XYZ_post_y,(*iter).m_XYZ_post_z));
        newHit->SetMomentumOfPrePoint (G4ThreeVector((*iter).m_P_pre_x,(*iter).m_P_pre_y,(*iter).m_P_pre_z));
        newHit->SetMomentumOfPostPoint(G4ThreeVector((*iter).m_P_post_x,(*iter).m_P_post_y,(*iter).m_P_post_z));
    /*
        newHit->SetPrePositionInCu    (G4ThreeVector((*iter).m_Cu_pre_x,(*iter).m_Cu_pre_y,(*iter).m_Cu_pre_z));
        newHit->SetPostPositionInCu   (G4ThreeVector((*iter).m_Cu_post_x,(*iter).m_Cu_post_y,(*iter).m_Cu_post_z));
        newHit->SetMomentumOfCuPre    (G4ThreeVector((*iter).m_P_Cu_pre_x,(*iter).m_P_Cu_pre_y,(*iter).m_P_Cu_pre_z));
        newHit->SetMomentumOfCuPost   (G4ThreeVector((*iter).m_P_Cu_post_x,(*iter).m_P_Cu_post_y,(*iter).m_P_Cu_post_z));
    */
//****************************************************************************************************
        cgemDC->insert(newHit);
      }
      //cgemDC->PrintAllHits();
    }
    else
    {
      std::cerr << "BesTuningIO::can't get cgemHitsCollection"<<std::endl;
    } 
  }
  else
  {
    std::cerr << "BesTuningIO::can't get cgemHitCollID"<<std::endl;
  }
}

GetCgemRootHits()

void BesTuningIO::GetCgemRootHits

(

)

Definition at line 343 of file BesTuningIO.cc.

{
 
  G4int THCID = -1;
  THCID = m_DigiMan->GetHitsCollectionID("BesCgemHitsCollection");
  if (THCID>=0) 
  {
    BesCgemHitsCollection* cgemDC =(BesCgemHitsCollection*)(m_DigiMan->GetHitsCollection(THCID));
    if(cgemDC)
    {
      int nHit = cgemDC->entries();
      if(nHit > 0)
      {
        for(int i = 0; i < nHit; i++)
        {
          delete (*cgemDC)[i];
        }
          cgemDC->GetVector()->clear();
      }
    }
 
   int nHits = m_TMcHitEvent->getMcHitCgemCol()->GetEntries();
   //std::cout << "nHits: " << nHits << std::endl;
   for(int i = 0; i < nHits; i++)
   {
     m_TMcHitCgem = m_TMcHitEvent->getMcHitCgem(i);
       
     BesCgemHit* cgemHit = new BesCgemHit();
     cgemHit->SetTrackID             (m_TMcHitCgem->GetTrackID            ());
     cgemHit->SetLayerID             (m_TMcHitCgem->GetLayerID            ());
     cgemHit->SetPDGCode             (m_TMcHitCgem->GetPDGCode            ());
     cgemHit->SetParentID            (m_TMcHitCgem->GetParentID           ());
     cgemHit->SetGlobalTime          (m_TMcHitCgem->GetGlobalTime         ());
     cgemHit->SetTotalEnergyDeposit  (m_TMcHitCgem->GetTotalEnergyDeposit ());
     cgemHit->SetStepLength          (m_TMcHitCgem->GetStepLength         ());
     
     TVector3 XYZ_pre  = m_TMcHitCgem->GetPositionOfPrePoint ();
     TVector3 XYZ_post = m_TMcHitCgem->GetPositionOfPostPoint();
     TVector3 P_pre    = m_TMcHitCgem->GetMomentumOfPrePoint ();
     TVector3 P_post   = m_TMcHitCgem->GetMomentumOfPostPoint();
 
     G4ThreeVector tmp_XYZ_pre  = G4ThreeVector(XYZ_pre.x()  , XYZ_pre.y()  , XYZ_pre.z() );
     G4ThreeVector tmp_XYZ_post = G4ThreeVector(XYZ_post.x() , XYZ_post.y() , XYZ_post.z());
     G4ThreeVector tmp_P_pre    = G4ThreeVector(P_pre.x()    , P_pre.y()    , P_pre.z()   );
     G4ThreeVector tmp_P_post   = G4ThreeVector(P_post.x()   , P_post.y()   , P_post.z()  );
 
     cgemHit->SetPositionOfPrePoint  ( tmp_XYZ_pre  );
     cgemHit->SetPositionOfPostPoint ( tmp_XYZ_post );
     cgemHit->SetMomentumOfPrePoint  ( tmp_P_pre    );
     cgemHit->SetMomentumOfPostPoint ( tmp_P_post   );
 
     /*
// ******************************************************************
     TVector3 Cu_pre  = m_TMcHitCgem->GetPrePositionInCu ();
     TVector3 Cu_post = m_TMcHitCgem->GetPostPositionInCu();
     TVector3 P_Cu_pre    = m_TMcHitCgem->GetMomentumOfCuPre ();
     TVector3 P_Cu_post   = m_TMcHitCgem->GetMomentumOfCuPost();
 
     G4ThreeVector tmp_Cu_pre  = G4ThreeVector(Cu_pre.x()  , Cu_pre.y()  , Cu_pre.z()  );
     G4ThreeVector tmp_Cu_post = G4ThreeVector(Cu_post.x() , Cu_post.y() , Cu_post.z() );
     G4ThreeVector tmp_Cu_P_pre    = G4ThreeVector(P_Cu_pre.x()    ,P_Cu_pre.y()    , P_Cu_pre.z()    );
     G4ThreeVector tmp_Cu_P_post   = G4ThreeVector(P_Cu_post.x()   ,P_Cu_post.y()   , P_Cu_post.z()   );
 
     cgemHit->SetPrePositionInCu  ( tmp_Cu_pre  );
     cgemHit->SetPostPositionInCu ( tmp_Cu_post );
     cgemHit->SetMomentumOfCuPre  ( tmp_Cu_P_pre    );
     cgemHit->SetMomentumOfCuPost ( tmp_Cu_P_post  );
// ******************************************************************************************     
     */
     //cgemHit->Print(); 
 
     cgemDC->insert(cgemHit);
     delete  m_TMcHitCgem;
   }
  }
}

Referenced by GetRootEvent().

GetEmcDigi()

void BesTuningIO::GetEmcDigi

(

void

)

Definition at line 221 of file BesTuningIO.cc.

                            {
  G4int THCID = -1;
  THCID = m_DigiMan->GetDigiCollectionID("BesEmcDigitsCollection");
  if (THCID>=0) {
    BesEmcDigitsCollection* emcDC = new BesEmcDigitsCollection("BesEmcDigitizer","BesEmcDigitsCollection");
   m_DigiMan->SetDigiCollection(THCID,emcDC);
  }
}

Referenced by GetNextEvents().

GetEmcRootDigi()

void BesTuningIO::GetEmcRootDigi

(

)

Definition at line 246 of file BesTuningIO.cc.

                                {
  G4int THCID = -1;
  THCID = m_DigiMan->GetDigiCollectionID("BesEmcDigitsCollection");
  //cout << "THCID " << THCID << endl; 
  if (THCID>=0) {
    BesEmcDigitsCollection* emcDC = new BesEmcDigitsCollection("BesEmcDigitizer","BesEmcDigitsCollection");
 
  //std::cout << "GetEmcRootDigi "  << emcDC << std::endl;
    if(emcDC){
      int nHit = emcDC->entries();
      //std::cout << "nHit: " << nHit << std::endl;
      if(nHit > 0){
        for(int i = 0; i < nHit; i++){
          delete (*emcDC)[i];
        }
    emcDC->GetVector()->clear();
      }
    }
    
   int nHits = m_TMcHitEvent->getMcDigiEmcCol()->GetEntries();
   //std::cout << "nHits: " << nHits << std::endl;
   for(int i = 0; i < nHits; i++){
     m_TMcDigiEmc = m_TMcHitEvent->getMcDigiEmc(i);
       
     BesEmcDigi* emcDigi = new BesEmcDigi();
     
     emcDigi->SetPartId(m_TMcDigiEmc->GetPartId());    
     emcDigi->SetThetaNb(m_TMcDigiEmc->GetThetaNb());
     emcDigi->SetPhiNb(m_TMcDigiEmc->GetPhiNb());   
     emcDigi->SetEnergy(m_TMcDigiEmc->GetEnergy());
     emcDigi->SetTime(m_TMcDigiEmc->GetTime());
     emcDigi->SetTrackIndex(m_TMcDigiEmc->GetTrackIndex());
     //emcDigi->Print();
     //std::cout << "SetEnergy" <<  emcDigi->GetEnergy() << std::endl;
 
     emcDC->insert(emcDigi);
     delete m_TMcDigiEmc;
 
   }
 
   //std::cout << "insert" << std::endl;
   m_DigiMan->SetDigiCollection(THCID,emcDC);
 
 
  }
 
}

Referenced by GetRootEvent().

GetMdcHits()

void BesTuningIO::GetMdcHits

(

void

)

Definition at line 120 of file BesTuningIO.cc.

                            {
    G4int mdcHitCollID = -1;
    mdcHitCollID = m_DigiMan->GetHitsCollectionID("BesMdcHitsCollection");
    if (mdcHitCollID>=0){
        BesMdcHitsCollection* mdcDC = (BesMdcHitsCollection*)m_DigiMan->GetHitsCollection(mdcHitCollID);
        if (mdcDC){
            G4int nHit = mdcDC->entries();
            if (nHit>0){
                for (G4int i=0;i<nHit;i++)
                {
                    delete (*mdcDC)[i];
                }
                mdcDC->GetVector()->clear();
            }
 
            std::vector<MdcHitType>::iterator iter;
            iter = (m_evt->mdcHit).hitCol.begin();
            // Loop over mdc hits
            for (; iter != (m_evt->mdcHit).hitCol.end(); iter++) {
                BesMdcHit* newHit = new BesMdcHit();
                newHit->SetTrackID ((*iter).trackIndex);
                newHit->SetLayerNo ((*iter).layerNo);
                newHit->SetCellNo ((*iter).cellNo);
                newHit->SetEdep ((*iter).energyDeposit);
                newHit->SetPos (G4ThreeVector((*iter).posX,(*iter).posY,(*iter).posZ));
                newHit->SetDriftD ((*iter).driftDistance);
                newHit->SetTheta((*iter).theta);
                newHit->SetPosFlag((*iter).posFlag);
                newHit->SetEnterAngle((*iter).enterAngle);
                newHit->SetDriftT (0.);
                newHit->SetGlobalT((*iter).globalT);
                mdcDC->insert(newHit);
 
            }
            //mdcDC->PrintAllHits();
        }else{
            std::cerr << "BesTuningIO::can't get mdcHitsCollection"<<std::endl;
        }
    }else{
        std::cerr << "BesTuningIO::can't get mdcHitCollID"<<std::endl;
    }
}

Referenced by GetNextEvents().

GetMdcRootHits()

void BesTuningIO::GetMdcRootHits

(

)

Definition at line 294 of file BesTuningIO.cc.

                                {
 
  G4int THCID = -1;
  THCID = m_DigiMan->GetHitsCollectionID("BesMdcHitsCollection");
  if (THCID>=0) {
    BesMdcHitsCollection* mdcDC = (BesMdcHitsCollection*) (m_DigiMan->GetHitsCollection(THCID));
    if(mdcDC){
      int nHit = mdcDC->entries();
      if(nHit > 0){
        for(int i = 0; i < nHit; i++){
          delete (*mdcDC)[i];
        }
    mdcDC->GetVector()->clear();
      }
    }
    
   int nHits = m_TMcHitEvent->getMcHitMdcCol()->GetEntries();
   //std::cout << "nHits: " << nHits << std::endl;
   for(int i = 0; i < nHits; i++){
     m_TMcHitMdc = m_TMcHitEvent->getMcHitMdc(i);
       
     BesMdcHit* mdcHit = new BesMdcHit();
     
     mdcHit->SetTrackID(m_TMcHitMdc->GetTrackID());       
     mdcHit->SetLayerNo(m_TMcHitMdc->GetLayerNo()); 
     mdcHit->SetCellNo(m_TMcHitMdc->GetCellNo());
     mdcHit->SetEdep(m_TMcHitMdc->GetEdep()); 
     mdcHit->SetDriftD(m_TMcHitMdc->GetDriftD());
     mdcHit->SetDriftT(m_TMcHitMdc->GetDriftT());
     mdcHit->SetGlobalT(m_TMcHitMdc->GetGlobalT());
     mdcHit->SetTheta(m_TMcHitMdc->GetTheta());
     mdcHit->SetEnterAngle(m_TMcHitMdc->GetEnterAngle());
     mdcHit->SetPosFlag(m_TMcHitMdc->GetPosFlag());
 
     TVector3 tTemp = m_TMcHitMdc->GetPos();
     G4ThreeVector gTemp = G4ThreeVector(tTemp.X(), tTemp.Y(), tTemp.Z());
     mdcHit->SetPos(gTemp);
     //mdcHit->Print(); 
 
     mdcDC->insert(mdcHit);
     delete  m_TMcHitMdc;
 
   }
 
 
  }
 
}

Referenced by GetRootEvent().

GetMucHits()

void BesTuningIO::GetMucHits ( void )

inline

Definition at line 40 of file BesTuningIO.hh.

{};

Referenced by GetNextEvents(), and GetRootEvent().

GetNextEvents()

void BesTuningIO::GetNextEvents

(

void

)

Definition at line 91 of file BesTuningIO.cc.

                               {
    if (m_evt) delete m_evt;
 
    m_evt = new HitEVENT;
    try {
        (*m_inputFileStream) >> *m_evt;
    } catch (AsciiWrongTag& ex) {
        std::cerr << "wrong tag, got " << ex.got()
        << " expected: " << ex.expected()
        << std::endl;
        delete m_evt;
        m_evt=0;
        return;
    } catch (AsciiDumpException&) {
        std::cerr<<"BesTuningIO: Reach file end!"<<std::endl;
        delete m_evt;
        m_evt=0;
        return;
    }
 
    if (ReadBoostRoot::GetMdc())GetMdcHits();
 
    if (ReadBoostRoot::GetTof())GetTofHits();
 
    if (ReadBoostRoot::GetEmc())GetEmcDigi();
 
    if (ReadBoostRoot::GetMuc())GetMucHits();
}

Referenced by BesEventAction::EndOfEventAction().

GetRootEvent()

void BesTuningIO::GetRootEvent

(

int

evtID

)

Definition at line 233 of file BesTuningIO.cc.

                                       {
    //std::cout << "evtID: " << evtID << std::endl;
    //HitTree->GetEntry(evtID);
    HitChain->GetEntry(evtID);
    //std::cout << "HitChain" << std::endl;
    if (ReadBoostRoot::GetMdc())GetMdcRootHits();
    if (ReadBoostRoot::GetCgem())GetCgemRootHits();
    if (ReadBoostRoot::GetTof())GetTofRootHits();
    if (ReadBoostRoot::GetEmc())GetEmcRootDigi();
    if (ReadBoostRoot::GetMuc())GetMucHits();
}

Referenced by BesEventAction::EndOfEventAction().

GetTofHits()

void BesTuningIO::GetTofHits ( void )

inline

Definition at line 38 of file BesTuningIO.hh.

{};

Referenced by GetNextEvents().

GetTofRootHits()

void BesTuningIO::GetTofRootHits

(

)

Definition at line 420 of file BesTuningIO.cc.

                                {
 
  //retrieve G4Svc
  ISvcLocator* svcLocator = Gaudi::svcLocator();
  IG4Svc* tmpSvc;
  StatusCode sc = svcLocator->service("G4Svc", tmpSvc);
  G4Svc* m_G4Svc = dynamic_cast<G4Svc *>(tmpSvc);
 
  const double m_beamTime = m_TMcHitEvent->getBeamTime();
  m_G4Svc->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));
    if(tofDC){
      int nHit = tofDC->entries();
      if(nHit > 0){
        for(int i = 0; i < nHit; i++){
          delete (*tofDC)[i];
        }
    tofDC->GetVector()->clear();
      }
    }
    
   int nHits = m_TMcHitEvent->getMcHitTofCol()->GetEntries();
   //std::cout << "nHits: " << nHits << std::endl;
   for(int i = 0; i < nHits; i++){
     m_TMcHitTof = m_TMcHitEvent->getMcHitTof(i);
       
     BesTofHit* tofHit = new BesTofHit();
 
     tofHit->SetTrackIndex(m_TMcHitTof->GetTrackIndex());
     tofHit->SetG4Index(m_TMcHitTof->GetG4Index());
     tofHit->SetPartId(m_TMcHitTof->GetPartId());
     tofHit->SetScinNb(m_TMcHitTof->GetScinNb());
     tofHit->SetEdep(m_TMcHitTof->GetEdep());
     tofHit->SetStepL(m_TMcHitTof->GetStepL());
     tofHit->SetTrackL(m_TMcHitTof->GetTrackL());
     tofHit->SetTime(m_TMcHitTof->GetTime());
     tofHit->SetDeltaT(m_TMcHitTof->GetDeltaT());
     tofHit->SetCharge(m_TMcHitTof->GetCharge());
     
     TVector3 tTemp = m_TMcHitTof->GetPos();
     G4ThreeVector gTemp(tTemp.X(), tTemp.Y(), tTemp.Z());
     tofHit->SetPos(gTemp);
 
     tTemp = m_TMcHitTof->GetPDirection();
     gTemp = G4ThreeVector(tTemp.X(), tTemp.Y(), tTemp.Z());
     tofHit->SetPDirection(gTemp);
 
     tTemp = m_TMcHitTof->GetMomentum();
     gTemp = G4ThreeVector(tTemp.X(), tTemp.Y(), tTemp.Z());
     tofHit->SetMomentum(gTemp);
     
     //tofHit->Print(); 
 
     tofDC->insert(tofHit);
     delete  m_TMcHitTof;
 
   }
 
   //std::cout << "tofDC: " <<  tofDC->entries() << std::endl;
 
  }
  
}

Referenced by GetRootEvent().

Member Data Documentation

f

TFile* BesTuningIO::f

Definition at line 43 of file BesTuningIO.hh.

HitChain

TChain* BesTuningIO::HitChain

Definition at line 45 of file BesTuningIO.hh.

Referenced by BesTuningIO(), and GetRootEvent().

HitTree

TTree* BesTuningIO::HitTree

Definition at line 44 of file BesTuningIO.hh.

---

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

• BesTuningIO.hh
• BesTuningIO.cc