BesTuningIO Class Reference

#include <BesTuningIO.hh>

Public Member Functions
	BesTuningIO (std::vector< std::string >)
	~BesTuningIO ()
void	GetNextEvents (void)
void	GetMdcHits (void)
void	GetTofHits (void)
void	GetEmcDigi (void)
void	GetMucHits (void)
void	GetRootEvent (int evtID)
void	GetMdcRootHits ()
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 39 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 84 of file BesTuningIO.cc.

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

Member Function Documentation

GetEmcDigi()

void BesTuningIO::GetEmcDigi

(

void

)

Definition at line 161 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 185 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 118 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 233 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 39 of file BesTuningIO.hh.

{};

Referenced by GetNextEvents(), and GetRootEvent().

GetNextEvents()

void BesTuningIO::GetNextEvents

(

void

)

Definition at line 89 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 173 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::GetTof())GetTofRootHits();
    if (ReadBoostRoot::GetEmc())GetEmcRootDigi();
    if (ReadBoostRoot::GetMuc())GetMucHits();
}

Referenced by BesEventAction::EndOfEventAction().

GetTofHits()

void BesTuningIO::GetTofHits ( void  )

inline

Definition at line 37 of file BesTuningIO.hh.

{};

Referenced by GetNextEvents().

GetTofRootHits()

void BesTuningIO::GetTofRootHits

(

)

Definition at line 282 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 42 of file BesTuningIO.hh.

HitChain

TChain* BesTuningIO::HitChain

Definition at line 44 of file BesTuningIO.hh.

Referenced by BesTuningIO(), and GetRootEvent().

HitTree

TTree* BesTuningIO::HitTree

Definition at line 43 of file BesTuningIO.hh.

---

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

• BesTuningIO.hh
• BesTuningIO.cc