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 99 of file TofDataSet.h.

Constructor & Destructor Documentation

TofDataSet()

TofDataSet::TofDataSet

(

)

Definition at line 172 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 179 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 105 of file TofDataSet.h.

{ return barrelData[i]; }

getEndcapData()

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

inline

Definition at line 106 of file TofDataSet.h.

{ return endcapData[i]; }

getEtfData()

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

inline

Definition at line 107 of file TofDataSet.h.

{ return etfData[i]; }

setBarrelData()

void TofDataSet::setBarrelData

(

RecBTofCalHitCol &

bhitcol

)

Definition at line 285 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 198 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 104 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 300 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 227 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 314 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 256 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