InDet::GNNTrackReaderTool Class Reference

InDet::GNNTrackReaderTool is a tool that reads track candidates of each event from a CSV file named as "track_runNumber_eventNumber.csv" or with a customized pre-fix. The directory, inputTracksDir, contains CSV files for all events. 1) If the inputTracksDir is not specified, the tool will read the CSV files from the current directory. 2) If the corresponding CSV file does not exist for a given event (runNumber, eventNumber), the tool will print an error message and return an empty list. More...

#include <GNNTrackReaderTool.h>

Inheritance diagram for InDet::GNNTrackReaderTool:

Collaboration diagram for InDet::GNNTrackReaderTool:

Public Member Functions
	GNNTrackReaderTool (const std::string &type, const std::string &name, const IInterface *parent)
virtual void	getTracks (uint32_t runNumber, uint32_t eventNumber, std::vector< std::vector< uint32_t > > &tracks) const override final
	Get track candidates from a CSV file named by runNumber and eventNumber. More...
virtual void	getTracks (uint32_t runNumber, uint32_t eventNumber, std::vector< std::vector< uint32_t >> &tracks, std::vector< std::vector< uint32_t >> &seeds) const override final
virtual MsgStream &	dump (MsgStream &out) const override
virtual std::ostream &	dump (std::ostream &out) const override

Protected Member Functions
	GNNTrackReaderTool ()=delete
	GNNTrackReaderTool (const GNNTrackReaderTool &)=delete
GNNTrackReaderTool &	operator= (const GNNTrackReaderTool &)=delete
MsgStream &	dumpevent (MsgStream &out) const

Protected Attributes
StringProperty	m_inputTracksDir {this, "inputTracksDir", "."}
StringProperty	m_csvPrefix {this, "csvPrefix", "track"}

Detailed Description

InDet::GNNTrackReaderTool is a tool that reads track candidates of each event from a CSV file named as "track_runNumber_eventNumber.csv" or with a customized pre-fix. The directory, inputTracksDir, contains CSV files for all events. 1) If the inputTracksDir is not specified, the tool will read the CSV files from the current directory. 2) If the corresponding CSV file does not exist for a given event (runNumber, eventNumber), the tool will print an error message and return an empty list.

Author

xiang.nosp@m.yang.nosp@m..ju@c.nosp@m.ern..nosp@m.ch

Definition at line 32 of file GNNTrackReaderTool.h.

Constructor & Destructor Documentation

GNNTrackReaderTool() [1/3]

InDet::GNNTrackReaderTool::GNNTrackReaderTool	(	const std::string &	type,
		const std::string &	name,
		const IInterface *	parent
	)

Definition at line 12 of file GNNTrackReaderTool.cxx.

                                                                         :
    base_class(type, name, parent)
{
    declareInterface<IGNNTrackReaderTool>(this);
}

GNNTrackReaderTool() [2/3]

InDet::GNNTrackReaderTool::GNNTrackReaderTool ( )

protecteddelete

GNNTrackReaderTool() [3/3]

InDet::GNNTrackReaderTool::GNNTrackReaderTool ( const GNNTrackReaderTool &  )

protecteddelete

Member Function Documentation

dump() [1/2]

MsgStream & InDet::GNNTrackReaderTool::dump ( MsgStream &  out ) const

overridevirtual

Definition at line 19 of file GNNTrackReaderTool.cxx.

{
  out<<std::endl;
  return dumpevent(out);
}

dump() [2/2]

std::ostream & InDet::GNNTrackReaderTool::dump ( std::ostream &  out ) const

overridevirtual

Definition at line 25 of file GNNTrackReaderTool.cxx.

{
  return out;
}

dumpevent()

MsgStream & InDet::GNNTrackReaderTool::dumpevent ( MsgStream &  out ) const

protected

Definition at line 30 of file GNNTrackReaderTool.cxx.

{
  out<<"|---------------------------------------------------------------------|"
       <<std::endl;
  out<<"| Number output tracks    | "<<std::setw(12)  
     <<"                              |"<<std::endl;
  out<<"|---------------------------------------------------------------------|"
     <<std::endl;
  return out;
}

getTracks() [1/2]

void InDet::GNNTrackReaderTool::getTracks ( uint32_t  runNumber, uint32_t  eventNumber, std::vector< std::vector< uint32_t > > &  tracks  ) const

finaloverridevirtual

Get track candidates from a CSV file named by runNumber and eventNumber.

Parameters

runNumber	run number of the event.
eventNumber	event number of the event.
tracks	a list of track candidates in terms of spacepoint indices as read from the CSV file.

Definition at line 41 of file GNNTrackReaderTool.cxx.

{
  std::string fileName = m_inputTracksDir +  "/" + m_csvPrefix + "_" \
                         + std::to_string(runNumber) + "_" + std::to_string(eventNumber) + ".csv";
 
  trackCandidates.clear();
  std::ifstream csvFile(fileName);
 
  if (!csvFile.is_open()) {
    ATH_MSG_ERROR("Cannot open file " << fileName);
    return;
  } else {
    ATH_MSG_INFO("File " << fileName << " is opened.");
  }
 
  std::string line;
  while(std::getline(csvFile, line)){
      std::stringstream lineStream(line);
      std::string cell;
      std::vector<uint32_t> trackCandidate;
      // allow both "," and " " as delimiter
      char delimiter = ',';
      if (line.find(delimiter) == std::string::npos) {
        delimiter = ' ';
      }
      while (std::getline(lineStream, cell, delimiter)) {
          uint32_t cellId = 0;
          try {
              cellId = std::stoi(cell);
          } catch (const std::invalid_argument& ia) {
              std::cout << "Invalid argument: " << ia.what() << " for cell " << cell << std::endl;
              continue;
          }
 
          if (std::find(trackCandidate.begin(), trackCandidate.end(), cellId) == trackCandidate.end()) {
              trackCandidate.push_back(cellId);
          }
      }
      trackCandidates.push_back(std::move(trackCandidate));
  }
}

getTracks() [2/2]

void InDet::GNNTrackReaderTool::getTracks ( uint32_t  runNumber, uint32_t  eventNumber, std::vector< std::vector< uint32_t >> &  tracks, std::vector< std::vector< uint32_t >> &  seeds  ) const

finaloverridevirtual

Definition at line 87 of file GNNTrackReaderTool.cxx.

                                                 {
  std::string fileName = m_inputTracksDir + "/" + m_csvPrefix + "_" +
                         std::to_string(runNumber) + "_" +
                         std::to_string(eventNumber) + ".csv";
 
  trackCandidates.clear();
  std::ifstream csvFile(fileName);
 
  if (!csvFile.is_open()) {
    ATH_MSG_ERROR("Cannot open file " << fileName);
    return;
  } else {
    ATH_MSG_INFO("File " << fileName << " is opened.");
  }
 
  std::string line;
  while (std::getline(csvFile, line)) {
    std::istringstream lineStream(line);
    std::string CLString, SPString;
    char delimiter = ',';
    if (line.find(delimiter) == std::string::npos) {
      delimiter = ' ';
    }
 
    if (std::getline(lineStream, CLString, '|') &&
        std::getline(lineStream, SPString)) {
      std::istringstream CLStream(CLString), SPStream(SPString);
      std::vector<uint32_t> cls, sps;
      std::string number;
 
      while (std::getline(CLStream, number, delimiter)) {
        cls.push_back(std::stoi(number));
      }
 
      while (std::getline(SPStream, number, delimiter)) {
        sps.push_back(std::stoi(number));
      }
 
      trackCandidates.push_back(cls);
      seeds.push_back(sps);
    }
  }
  csvFile.close();
 
  ATH_MSG_DEBUG("Length of track list " << trackCandidates.size());
}

operator=()

GNNTrackReaderTool& InDet::GNNTrackReaderTool::operator= ( const GNNTrackReaderTool &  )

protecteddelete

Member Data Documentation

m_csvPrefix

StringProperty InDet::GNNTrackReaderTool::m_csvPrefix {this, "csvPrefix", "track"}

protected

Definition at line 67 of file GNNTrackReaderTool.h.

m_inputTracksDir

StringProperty InDet::GNNTrackReaderTool::m_inputTracksDir {this, "inputTracksDir", "."}

protected

Definition at line 66 of file GNNTrackReaderTool.h.

---

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

• GNNTrackReaderTool.h
• GNNTrackReaderTool.cxx