#include <TofDataSet.h>

Public Member Functions
	TofDataSet ()

	~TofDataSet ()

RecordSet *	getBarrelData (unsigned int i) const

RecordSet *	getEndcapData (unsigned int i) const

void	setData (TTree *, bool)

void	setBarrelDataFiles (std::vector< std::string > &)

void	setEndcapDataFiles (std::vector< std::string > &)

void	setBarrelData (RecBTofCalHitCol &)

void	setEndcapData (RecETofCalHitCol &)

Detailed Description

Definition at line 92 of file TofDataSet.h.

Constructor & Destructor Documentation

◆ TofDataSet()

TofDataSet::TofDataSet ( )

Definition at line 149 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;
  }
  return;
}

◆ ~TofDataSet()

TofDataSet::~TofDataSet ( )

Definition at line 160 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];
  }
  return;
}

Member Function Documentation

◆ getBarrelData()

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

inline

Definition at line 98 of file TofDataSet.h.

98{ return barrelData[i]; }

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

◆ getEndcapData()

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

inline

Definition at line 99 of file TofDataSet.h.

99{ return endcapData[i]; }

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

◆ setBarrelData()

void TofDataSet::setBarrelData ( RecBTofCalHitCol & bhitcol )

Definition at line 219 of file TofDataSet.cxx.

                                                          {
  RecBTofCalHitCol::iterator iter = bhitcol.begin();
  for( ; iter!=bhitcol.end(); iter++ ) {
    int tofid = (*iter)->mod();
    if( tofid<0 || tofid>175 ) 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 173 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, true );
  delete data_barrel;
  return;
}

Referenced by tofcalgsec::finalize().

◆ setData()

void TofDataSet::setData	(	TTree *	t,
		bool	isBarrel
	)

Definition at line 99 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   );
    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   );
    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 && 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 && ( 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 {
    std::cerr << "Error: a invalid tree or a blank tree, When converting a tree to TofDataSet,exit" << std::endl;
  }
 
  return;
}

Referenced by setBarrelDataFiles(), and setEndcapDataFiles().

◆ setEndcapData()

void TofDataSet::setEndcapData ( RecETofCalHitCol & ehitcol )

Definition at line 237 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;
}