TofDataSet Class Reference

#include <TofDataSet.h>

Public Member Functions
	TofDataSet ()
	~TofDataSet ()
RecordSet *	getBarrelData (unsigned int i) const
RecordSet *	getEndcapData (unsigned int i) const
RecordSet *	getEtfData (unsigned int i) const
void	setData (TTree *, unsigned int)
void	setBarrelDataFiles (std::vector< std::string > &)
void	setEndcapDataFiles (std::vector< std::string > &)
void	setEtfDataFiles (std::vector< std::string > &)
void	setBarrelData (RecBTofCalHitCol &)
void	setEndcapData (RecETofCalHitCol &)
void	setEtfData (RecBTofCalHitCol &)

Detailed Description

Definition at line 101 of file TofDataSet.h.

Constructor & Destructor Documentation

TofDataSet()

TofDataSet::TofDataSet

(

)

Definition at line 174 of file TofDataSet.cxx.

                       {
  for( unsigned int i=0; i<NBarrel; i++ ) {
    barrelData[i] = new RecordSet;
  }
  for( unsigned int i=0; i<NEndcap; i++ ) {
    endcapData[i] = new RecordSet;
  }
  for( unsigned int i=0; i<NEtf*NStrip; i++ ) {
    etfData[i] = new RecordSet;
  }
  return;
}

~TofDataSet()

TofDataSet::~TofDataSet

(

)

Definition at line 188 of file TofDataSet.cxx.

                        {
  for( unsigned int i=0; i<NBarrel; i++ ) {
    barrelData[i]->clear();
    delete barrelData[i];
  }
  for( unsigned int i=0; i<NEndcap; i++ ) {
    endcapData[i]->clear();
    delete endcapData[i];
  }
  for( unsigned int i=0; i<NEtf*NStrip; i++ ) {
    etfData[i]->clear();
    delete etfData[i];
  }
  return;
}

Member Function Documentation

getBarrelData()

RecordSet * TofDataSet::getBarrelData ( unsigned int i ) const

inline

Definition at line 107 of file TofDataSet.h.

{ return barrelData[i]; }

Referenced by TofCalibManager::doCalibration(), and TofCalibManager::fillRoot().

getEndcapData()

RecordSet * TofDataSet::getEndcapData ( unsigned int i ) const

inline

Definition at line 108 of file TofDataSet.h.

{ return endcapData[i]; }

Referenced by TofCalibManager::doCalibration(), and TofCalibManager::fillRoot().

getEtfData()

RecordSet * TofDataSet::getEtfData ( unsigned int i ) const

inline

Definition at line 109 of file TofDataSet.h.

{ return etfData[i];    }

Referenced by TofCalibManager::doCalibration(), and TofCalibManager::fillRoot().

setBarrelData()

void TofDataSet::setBarrelData

(

RecBTofCalHitCol &

bhitcol

)

Definition at line 274 of file TofDataSet.cxx.

                                                          {
  RecBTofCalHitCol::iterator iter = bhitcol.begin();
  for( ; iter!=bhitcol.end(); iter++ ) {
    int tofid = (*iter)->mod();
    if( tofid<0 || tofid>175 ) continue;
    if( fabs( (*iter)->dzHit() - 1.0 )>1.0e-6 ) continue;
 
    Record *r = new Record( (*iter) );
    if( r->cutBarrel() ) {
      barrelData[tofid]->push_back(r);
    }
    else {
      delete r;
    }
  }
  return;
}

Referenced by tofcalgsec::execute().

setBarrelDataFiles()

void TofDataSet::setBarrelDataFiles

(

std::vector< std::string > &

barrelFiles

)

Definition at line 205 of file TofDataSet.cxx.

                                                                       {
  TChain* data_barrel = new TChain("btrk");
  if( !data_barrel ) {
    std::cerr << " tofcalgsec Error Msg: creating a tree[barrel] fails in TofDataSet()"<<std::endl;
    throw "Error Msg: creating a tree fails in TofDataSet() ";
  }
  std::cout<<"begin reading barrel data file ... "<<std::endl;
  try{
    for( std::vector<std::string>::iterator it=barrelFiles.begin(); it!=barrelFiles.end(); it++ ) {
      std::cout << " Add file : " << (*it) << std::endl;
      data_barrel->Add( (*it).c_str() );
    }
  }
  catch(...){
    std::cerr << "tofcalgsec Error Msg : in TofDataSet::setDataFiles(std::vector<std::string>&) " << std::endl;
    return;
  }
  setData( data_barrel, 1 );
  delete data_barrel;
  return;
}

Referenced by tofcalgsec::finalize().

setData()

void TofDataSet::setData	(	TTree *	t,
		unsigned int	isBarrel )

Definition at line 106 of file TofDataSet.cxx.

                                                          {
  if( t && t->GetEntries()>0 ) {
    rootRecord item;
    t->SetBranchAddress( "run",     &item.run     );
    t->SetBranchAddress( "event",   &item.event   );
    t->SetBranchAddress( "tofid",   &item.tofid   );
    if( NULL != t->FindBranch( "strip" ) ) {
      t->SetBranchAddress( "strip",   &item.strip   );
    }
    t->SetBranchAddress( "qleft",   &item.qleft   );
    t->SetBranchAddress( "qright",  &item.qright  );
    t->SetBranchAddress( "tleft",   &item.tleft   );
    t->SetBranchAddress( "tright",  &item.tright  );
    t->SetBranchAddress( "zrhit",   &item.zrhit   );
    if( NULL != t->FindBranch( "dt" ) ) {
      t->SetBranchAddress( "dt",      &item.dt      );
    }
    t->SetBranchAddress( "texp",    &item.texp    );
    t->SetBranchAddress( "path",    &item.path    );
    t->SetBranchAddress( "phi",     &item.phi     );
    t->SetBranchAddress( "theta",   &item.theta        );
    t->SetBranchAddress( "p",       &item.p       );
    t->SetBranchAddress( "hitcase", &item.hitcase );
 
    for( unsigned int i=0; i<t->GetEntries(); i++ ) {
      t->GetEntry(i);
      if( isBarrel==1 && item.hitcase>=0 && item.hitcase<=2 ) {
    Record *r = new Record( item );
    if( r->cutBarrel() ) {
      unsigned int tofID = item.tofid;
      barrelData[tofID]->push_back(r);
    }
    else {
      delete r;
    }
      }
      else if( isBarrel==0 && ( item.hitcase==3 || item.hitcase==4 ) ) {
    Record *r = new Record( item );
    if( r->cutEndcap() ) {
      unsigned int tofID = item.tofid;
      endcapData[tofID]->push_back(r);
    }
    else {
      delete r;
    }
      }
      else if( isBarrel==2 && item.hitcase>=5 && item.hitcase<=6 ) {
    Record *r = new Record( item );
    if( r->cutEtf() ) {
      unsigned int tofID = item.tofid;
      unsigned int strip = item.strip;
      unsigned int id = tofID*12 + strip;
      etfData[id]->push_back(r);
    }
    else {
      delete r;
    }
      }
    }
  }
  else {
    std::cerr << "Error: a invalid tree or a blank tree, When converting a tree to TofDataSet,exit" << std::endl;
  }
 
  return;
}

Referenced by setBarrelDataFiles(), setEndcapDataFiles(), and setEtfDataFiles().

setEndcapData()

void TofDataSet::setEndcapData

(

RecETofCalHitCol &

ehitcol

)

Definition at line 293 of file TofDataSet.cxx.

                                                          {
  RecETofCalHitCol::iterator iter = ehitcol.begin();
  for( ; iter!=ehitcol.end(); iter++ ) {
    int tofid = (*iter)->mod();
    if( tofid<0 || tofid>95 ) continue;
 
    Record *r = new Record( (*iter) );
    if( r->cutEndcap() ) {
      endcapData[tofid]->push_back(r);
    }
    else {
      delete r;
    }
  }
  return;
}

Referenced by tofcalgsec::execute().

setEndcapDataFiles()

void TofDataSet::setEndcapDataFiles

(

std::vector< std::string > &

endcapFiles

)

Definition at line 228 of file TofDataSet.cxx.

                                                                       {
  TChain* data_endcap = new TChain("etrk");
  if( !data_endcap ) {
    std::cerr << " tofcalgsec Error Msg: creating a tree[endcap] fails in TofDataSet()"<<std::endl;
    throw "Error Msg: creating a tree fails in TofDataSet() ";
  }
  std::cout<<"begin reading endcap data file ... "<<std::endl;
  try{
    for( std::vector<std::string>::iterator it=endcapFiles.begin(); it!=endcapFiles.end(); it++ ) {
      std::cout << " Add file : " << (*it) << std::endl;
      data_endcap->Add( (*it).c_str() );
    }
  }
  catch(...){
    std::cerr << "tofcalgsec Error Msg : in TofDataSet::setDataFiles(std::vector<std::string>&) " << std::endl;
    return;
  }
  setData( data_endcap, 0 );
  delete data_endcap;
  return;
}

Referenced by tofcalgsec::finalize().

setEtfData()

void TofDataSet::setEtfData

(

RecBTofCalHitCol &

bhitcol

)

Definition at line 311 of file TofDataSet.cxx.

                                                       {
  RecBTofCalHitCol::iterator iter = bhitcol.begin();
  for( ; iter!=bhitcol.end(); iter++ ) {
    int tofid = (*iter)->mod();
    int strip = int( (*iter)->sinTheta() );
    if( tofid<0 || tofid>71 ) continue;
    if( strip<0 || strip>11 ) continue;
    if( fabs( (*iter)->dzHit() )>1.0e-6 ) continue;
    unsigned int id = tofid*12 + strip;
    Record *r = new Record( (*iter) );
    if( r->cutEtf() ) {
      etfData[id]->push_back(r);
    }
    else {
      delete r;
    }
  }
  return;
}

Referenced by tofcalgsec::execute().

setEtfDataFiles()

void TofDataSet::setEtfDataFiles

(

std::vector< std::string > &

etfFiles

)

Definition at line 251 of file TofDataSet.cxx.

                                                                 {
  TChain* data_etf = new TChain("etf");
  if( !data_etf ) {
    std::cerr << " tofcalgsec Error Msg: creating a tree[etf] fails in TofDataSet()"<<std::endl;
    throw "Error Msg: creating a tree fails in TofDataSet() ";
  }
  std::cout<<"begin reading etf data file ... "<<std::endl;
  try{
    for( std::vector<std::string>::iterator it=etfFiles.begin(); it!=etfFiles.end(); it++ ) {
      std::cout << " Add file : " << (*it) << std::endl;
      data_etf->Add( (*it).c_str() );
    }
  }
  catch(...){
    std::cerr << "tofcalgsec Error Msg : in TofDataSet::setDataFiles(std::vector<std::string>&) " << std::endl;
    return;
  }
  setData( data_etf, 2 );
  delete data_etf;
  return;
}

Referenced by tofcalgsec::finalize().

---

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

• TofDataSet.h
• TofDataSet.cxx