ShowerLib::FCALDistEnergyShowerLib Class Reference

Class for shower library shower lib. More...

#include <FCALDistEnergyShowerLib.h>

Inheritance diagram for ShowerLib::FCALDistEnergyShowerLib:

Collaboration diagram for ShowerLib::FCALDistEnergyShowerLib:

Public Member Functions
virtual	~FCALDistEnergyShowerLib ()
	default destructor
virtual std::vector< EnergySpot > *	getShower (const G4Track *track, ShowerLibStatistics *stats, int randomShift) const
	get shower for given G4 track
virtual double	getContainmentZ (const G4Track *track) const
	get average length of showers for the given energy
virtual double	getContainmentR (const G4Track *track) const
	get average lateral spread of the showers for the given energy
virtual bool	storeShower (HepMC::ConstGenParticlePtr genParticle, const Shower *shower)
	store shower in the library
virtual bool	writeToROOT (TFile *dest)
	write library to ROOT file
virtual ShowerLibStatistics *	createStatistics () const
virtual const std::string	getName () const
const std::string	statistics () const
	print library statistics
virtual const std::string	detector () const
	get detector tag
virtual int	particle_id () const
	get particle tag
virtual const std::string	release () const
	get Release tag
virtual void	release (const std::string &release)
	set Release tag
virtual const std::string	geometry () const
	get Geometry tag
virtual void	geometry (const std::string &geometry)
	set Geometry tag
virtual const std::string	geantVersion () const
	get geant version tag
virtual void	geantVersion (const std::string &version)
	set geant version tag
virtual const std::string	physicsList () const
	get geant 4 physics list name
virtual void	physicsList (const std::string &list)
	set geant 4 physics list name
virtual const std::string	comment () const
	get comment
virtual void	comment (const std::string &comment)
	set given string as comment
virtual void	addComment (const std::string &comment)
	add given string to comment

Static Public Member Functions
static IShowerLib *	readFromROOTFile (TFile *source)
	factory method. create a library from root file. returns NULL if file is invalid.
static IShowerLib *	createEmptyLib (const std::string &inputFile)
	factory method. create empty library with the given structure. returns NULL if file is invalid.

Protected Member Functions
virtual const std::string	printParameters () const
bool	readMeta (TTree *source)
	read metadata from the given TTree
bool	writeMeta (TTree *dest) const
	write metadata to the given TTree

Protected Attributes
std::string	m_detector
	name of the detector
int	m_particle
	ID of the generated particles.
std::string	m_release
	atlas software release
std::string	m_geometry
	atlas geometry version
std::string	m_geantVersion
	geant 4 version
std::string	m_physicsList
	used geant 4 physics list
std::string	m_comment
	comment
bool	m_filled
	is the library read from ROOT or from structure file

Private Types
typedef std::map< float, Shower >	distbin
typedef std::map< float, distbin >	library

Private Member Functions
	FCALDistEnergyShowerLib ()
bool	read (TTree *source)
	read library from given TTree
bool	write (TTree *dest) const
	write library to given TTree
bool	readStructure (std::vector< float > &structure)
float	distance (double x, double y) const

Private Attributes
library	m_libData
double	m_xrodcent
double	m_yrodcent
double	m_step

Detailed Description

Class for shower library shower lib.

   Store complete shower library (collection of energy bins)

Author

Wolfgang Ehrenfeld, University of Hamburg, Germany

Sasha Glazov, DESY Hamburg, Germany

Version

$Id: FCALDistEnergyShowerLib.h 769594 2016-08-23 13:48:34Z ssnyder $

Definition at line 36 of file FCALDistEnergyShowerLib.h.

Member Typedef Documentation

distbin

typedef std::map<float,Shower> ShowerLib::FCALDistEnergyShowerLib::distbin

private

Definition at line 93 of file FCALDistEnergyShowerLib.h.

library

typedef std::map<float,distbin> ShowerLib::FCALDistEnergyShowerLib::library

private

Definition at line 94 of file FCALDistEnergyShowerLib.h.

Constructor & Destructor Documentation

~FCALDistEnergyShowerLib()

virtual ShowerLib::FCALDistEnergyShowerLib::~FCALDistEnergyShowerLib ( )

inlinevirtual

default destructor

Definition at line 47 of file FCALDistEnergyShowerLib.h.

          {
              for ( auto& dist : m_libData) {  // dist bins
                  for (auto& ene : dist.second )  {  // energy map
                      for ( auto& spot : ene.second ) {
                          delete  spot;
                      }
                      ene.second.clear();
                  }
                  dist.second.clear();
              };
              m_libData.clear();
          }

FCALDistEnergyShowerLib()

ShowerLib::FCALDistEnergyShowerLib::FCALDistEnergyShowerLib ( )

inlineprivate

Definition at line 82 of file FCALDistEnergyShowerLib.h.

:m_xrodcent(0.0),m_yrodcent(0.0),m_step(0.0) {}

Member Function Documentation

addComment()

void ShowerLib::IShowerLib::addComment ( const std::string & comment )

inlinevirtualinherited

add given string to comment

Definition at line 141 of file IShowerLib.h.

{ m_comment += comment; }

comment() [1/2]

const std::string ShowerLib::IShowerLib::comment ( ) const

inlinevirtualinherited

get comment

Definition at line 139 of file IShowerLib.h.

{ return m_comment; }

comment() [2/2]

void ShowerLib::IShowerLib::comment ( const std::string & comment )

inlinevirtualinherited

set given string as comment

Definition at line 140 of file IShowerLib.h.

{ m_comment = comment; }

createEmptyLib()

IShowerLib * ShowerLib::FCALDistEnergyShowerLib::createEmptyLib ( const std::string & inputFile )

static

factory method. create empty library with the given structure. returns NULL if file is invalid.

Definition at line 79 of file FCALDistEnergyShowerLib.cxx.

  {
      /*
       * Eta Energy Structure format:
       *
       * VER PART DET DIST1 DIST2 ...
       * XRODCENT YRODCENT STEP
       * COMMENT
       *
       * where
       *
       * VER == 4
       * DIST(N+1) > DIST(N)
       * DIST(N) >= 0
       * First DIST is the beginning of the lib. If not zero, zero will be added
       * Last DIST is the end of the lib. If less then 4.5, 4.5 will be added
       */
      std::ifstream filestr(inputFile.c_str(),std::ios::in);
 
 
      if (!filestr.is_open()) {
          std::cout << "FCALDistEnergyShowerLib    " << inputFile << ": bad file!" << std::endl;
          return nullptr;
      }
 
      std::string instr;
      std::getline(filestr,instr);
      std::stringstream ss(instr);
 
      int ver;
 
      ss >> ver;
 
      if (ver != LIB_VERSION) {
          return nullptr;
      }
 
 
      int part;
      std::string det;
 
      ss >> part >> det;
 
      std::vector<float> etalist;
 
      float eta;
      float etaold;
 
      ss >> eta;
 
      if (ss.fail()) {
          std::cout << "file reading failed! (not enough data)" << std::endl;
          return nullptr;
      }
 
      if (eta != 0.) {
          if (eta < 0) {
              std::cout << "no negative dists allowed (" << eta << ")" << std::endl;
              return nullptr;
          }
          etalist.push_back(0.);
      }
 
      etalist.push_back(eta);
      etaold = eta;
 
      while (!ss.eof()) {
          ss >> eta;
          if ((ss.fail()) || (eta <= etaold)) {
              std::cout << "screwed dists! (" << eta << "<=" << etaold << ")" << std::endl;
              return nullptr;
          }
          if (eta < 4.5) {
              etalist.push_back(eta);
          } else if (eta > 4.5) {
              std::cout << "dist can't be bigger than 4.5 (" << eta << "). ignored" << std::endl;
          }
          etaold = eta;
      }
 
      //if (etalist.back() < 4.5) etalist.push_back(4.5);
 
      FCALDistEnergyShowerLib* newlib = new FCALDistEnergyShowerLib();
      if (!newlib->readStructure(etalist)) {
          std::cout << "this structure is not valid" << std::endl;
          delete newlib;
          return nullptr;
      }
 
      newlib->m_detector = std::move(det);
      newlib->m_particle = part;
 
      newlib->m_filled = false;
 
      std::getline(filestr,instr);
          std::stringstream ss1(instr);
      
      ss1 >> (newlib->m_xrodcent) >> (newlib->m_yrodcent) >> (newlib->m_step);
 
      std::getline(filestr,instr);
      newlib->m_comment = std::move(instr);
 
      return newlib;
  }

createStatistics()

ShowerLibStatistics * ShowerLib::FCALDistEnergyShowerLib::createStatistics ( ) const

virtual

Implements ShowerLib::IShowerLib.

Definition at line 632 of file FCALDistEnergyShowerLib.cxx.

  {
      std::map<int, std::string> names;
      std::map<int, int> sizes;
      for(library::const_iterator it = m_libData.begin(); it != m_libData.end(); ++it) {
        sizes[calcKey(it->first)]=it->second.size();
        float distlow = it->first;
        float disthigh;
        library::const_iterator it_copy = it;
        ++it_copy;
        if (it_copy == m_libData.end()) {
          disthigh = 4.5;
        } else {
          disthigh = it_copy->first;
        }
        std::stringstream ss;
        ss << std::showpos << std::fixed << std::setprecision(2);
        ss << "DIST: " << distlow << " .. " << disthigh;
        names[calcKey(it->first)] = ss.str();
      }
      return new ShowerLibStatistics(names, sizes);
  }

detector()

const std::string ShowerLib::IShowerLib::detector ( ) const

inlinevirtualinherited

get detector tag

Definition at line 121 of file IShowerLib.h.

{ return m_detector; }

distance()

float ShowerLib::FCALDistEnergyShowerLib::distance ( double x, double y ) const

private

Definition at line 36 of file FCALDistEnergyShowerLib.cxx.

  {
      double stepy = m_step * sqrt(3.)/2.;
      double dy =  fmod(y-m_yrodcent,stepy);
      int ny = int( (y-m_yrodcent)/stepy );
      //additional half shift:
      double dx0 = (ny % 2 )*m_step/2.;
      double dx = fmod(x-m_xrodcent-dx0,m_step);
      double d1 = sqrt(dx*dx+dy*dy);
      double d2 = sqrt((m_step-dx)*(m_step-dx)+dy*dy);
      double d3 = sqrt((m_step/2.-dx)*(m_step/2.-dx)+(stepy-dy)*(stepy-dy));
      float dd = d1;
      if (dd > d2) dd = d2;
      if (dd > d3) dd = d3;
      return dd;
  }

geantVersion() [1/2]

const std::string ShowerLib::IShowerLib::geantVersion ( ) const

inlinevirtualinherited

get geant version tag

Definition at line 133 of file IShowerLib.h.

{ return m_geantVersion; }

geantVersion() [2/2]

void ShowerLib::IShowerLib::geantVersion ( const std::string & version )

inlinevirtualinherited

set geant version tag

Definition at line 134 of file IShowerLib.h.

{ m_geantVersion = version; }

geometry() [1/2]

const std::string ShowerLib::IShowerLib::geometry ( ) const

inlinevirtualinherited

get Geometry tag

Definition at line 130 of file IShowerLib.h.

{ return m_geometry; }

geometry() [2/2]

void ShowerLib::IShowerLib::geometry ( const std::string & geometry )

inlinevirtualinherited

set Geometry tag

Definition at line 131 of file IShowerLib.h.

{ m_geometry = geometry; }

getContainmentR()

double ShowerLib::FCALDistEnergyShowerLib::getContainmentR ( const G4Track * track ) const

virtual

get average lateral spread of the showers for the given energy

Implements ShowerLib::IShowerLib.

Definition at line 361 of file FCALDistEnergyShowerLib.cxx.

  {
      if (!m_filled) {
          std::cout << "Library is not created for production use" << std::endl;
          return 0;
      }
 
      //std::cout << "Starting getShower()" << std::endl;
 
      double x = track->GetPosition().getX();
      double y = track->GetPosition().getY();
 
      float dist = this->distance(x,y);
 
      G4int particleCode = track->GetDefinition()->GetPDGEncoding();
      if ( particleCode < 0 ) particleCode = -particleCode; // hack for positrons.
 
      if ( particleCode != m_particle ) {
          std::cout << "wrong particle: " << particleCode << "!=" << m_particle << std::endl;
          return 0;
      }
 
      library::const_iterator libit = m_libData.upper_bound(dist);
 
      if (libit == m_libData.begin()) {
          //this is really weird
          std::cout << "Something is wrong with dist: x=" << x << " y=" << y << " dist=" << dist << std::endl;
      } else {
          --libit;
      }
 
      //std::cout << "Found eta bin" << std::endl;
 
      if ((*libit).second.empty()) {
          std::cout << "The etabin corresponding to the eta is empty" << std::endl;
          return 0;
      }
      double trenergy = track->GetKineticEnergy();
      //std::cout << "Energy: " << trenergy << std::endl;
      distbin::const_iterator distit = (*libit).second.lower_bound(trenergy);
      if (distit == (*libit).second.end()) --distit; //if it points to the end, repoint it to the last shower in bin
      else if (distit != (*libit).second.begin()) { //find the closest energy. it's either found item or previous (if there is a previous)
          distbin::const_iterator distitch = distit;
          --distitch;
          if (((*distit).first - trenergy) > (trenergy - (*distitch).first)) { // the closest is the previous
              --distit;
          }
      }
      //std::cout << "Found closest energy:" << (*etait).first << std::endl;
      double rezR = (*distit).second.getRSize();
      distbin::const_iterator distiter = distit;
      int actualNumFS = 1;
      int spread = 2; //will calculate average Z for 5 showers max ( -2 .. 0 .. +2 )
      double upperEnergy = (*distit).first * 1.01; //we allow 1% off
      for (int i = 0; i < spread; i++) {
          ++distiter;
          if (distiter == (*libit).second.end()) {
              break;
          }
          if (upperEnergy < (*distiter).first) break; //energy diff too high
          //the shower is OK, including it to the average
          rezR += (*distiter).second.getRSize();
          actualNumFS++;
      }
      distiter = distit;
      if (distiter != (*libit).second.begin()) {
          double lowerEnergy = (*distit).first * 0.99; //we allow 1% off
          for (int i = 0; i < spread; i++) {
              --distiter;
              if (lowerEnergy > (*distiter).first) break; //energy diff too high
              //the shower is OK, including it to the average
              rezR += (*distiter).second.getRSize();
              actualNumFS++;
              if (distiter == (*libit).second.begin()) {
                  break;
              }
          }
      }
      return rezR/actualNumFS; //average Z size
  }

getContainmentZ()

double ShowerLib::FCALDistEnergyShowerLib::getContainmentZ ( const G4Track * track ) const

virtual

get average length of showers for the given energy

Implements ShowerLib::IShowerLib.

Definition at line 280 of file FCALDistEnergyShowerLib.cxx.

  {
      if (!m_filled) {
          std::cout << "Library is not created for production use" << std::endl;
          return 0;
      }
 
      //std::cout << "Starting getShower()" << std::endl;
 
      double x = track->GetPosition().getX();
      double y = track->GetPosition().getY();
 
      float dist = this->distance(x,y);
 
      G4int particleCode = track->GetDefinition()->GetPDGEncoding();
      if ( particleCode < 0 ) particleCode = -particleCode; // hack for positrons.
 
      if ( particleCode != m_particle ) {
          std::cout << "wrong particle: " << particleCode << "!=" << m_particle << std::endl;
          return 0;
      }
 
      library::const_iterator libit = m_libData.upper_bound(dist);
 
      if (libit == m_libData.begin()) {
          //this is really weird
          std::cout << "Something is wrong with dist: x=" << x << " y=" << y << " dist=" << dist << std::endl;
      } else {
          --libit;
      }
 
      //std::cout << "Found eta bin" << std::endl;
 
      if ((*libit).second.empty()) {
          std::cout << "The etabin corresponding to the eta is empty" << std::endl;
          return 0;
      }
      double trenergy = track->GetKineticEnergy();
      //std::cout << "Energy: " << trenergy << std::endl;
      distbin::const_iterator distit = (*libit).second.lower_bound(trenergy);
      if (distit == (*libit).second.end()) --distit; //if it points to the end, repoint it to the last shower in bin
      else if (distit != (*libit).second.begin()) { //find the closest energy. it's either found item or previous (if there is a previous)
          distbin::const_iterator distitch = distit;
          --distitch;
          if (((*distit).first - trenergy) > (trenergy - (*distitch).first)) { // the closest is the previous
              --distit;
          }
      }
      //std::cout << "Found closest energy:" << (*etait).first << std::endl;
      double rezZ = (*distit).second.getZSize();
      distbin::const_iterator distiter = distit;
      int actualNumFS = 1;
      int spread = 2; //will calculate average Z for 5 showers max ( -2 .. 0 .. +2 )
      double upperEnergy = (*distit).first * 1.01; //we allow 1% off
      for (int i = 0; i < spread; i++) {
          ++distiter;
          if (distiter == (*libit).second.end()) {
              break;
          }
          if (upperEnergy < (*distiter).first) break; //energy diff too high
          //the shower is OK, including it to the average
          rezZ += (*distiter).second.getZSize();
          actualNumFS++;
      }
      distiter = distit;
      if (distiter != (*libit).second.begin()) {
          double lowerEnergy = (*distit).first * 0.99; //we allow 1% off
          for (int i = 0; i < spread; i++) {
              --distiter;
              if (lowerEnergy > (*distiter).first) break; //energy diff too high
              //the shower is OK, including it to the average
              rezZ += (*distiter).second.getZSize();
              actualNumFS++;
              if (distiter == (*libit).second.begin()) {
                  break;
              }
          }
      }
      return rezZ/actualNumFS; //average Z size
  }

getName()

virtual const std::string ShowerLib::FCALDistEnergyShowerLib::getName ( ) const

inlinevirtual

Reimplemented from ShowerLib::IShowerLib.

Definition at line 74 of file FCALDistEnergyShowerLib.h.

{ return "FCALDist Energy ShowerLib"; }

getShower()

std::vector< EnergySpot > * ShowerLib::FCALDistEnergyShowerLib::getShower ( const G4Track * track, ShowerLibStatistics * stats, int randomShift ) const

virtual

get shower for given G4 track

Implements ShowerLib::IShowerLib.

Definition at line 184 of file FCALDistEnergyShowerLib.cxx.

  {
      if (!m_filled) {
          std::cout << "Library is not created for production use" << std::endl;
          return nullptr;
      }
 
      //std::cout << "Starting getShower()" << std::endl;
 
      double x = track->GetPosition().getX();
      double y = track->GetPosition().getY();
 
      float dist = this->distance(x,y);
 
      G4int particleCode = track->GetDefinition()->GetPDGEncoding();
      if ( particleCode < 0 ) particleCode = -particleCode; // hack for positrons.
 
      if ( particleCode != m_particle ) {
          std::cout << "wrong particle: " << particleCode << "!=" << m_particle << std::endl;
          return nullptr;
      }
 
      //std::cout << "Extracted particle parameters: " << eta << std::endl;
 
      library::const_iterator libit = m_libData.upper_bound(dist);
 
      if (libit == m_libData.begin()) {
          //this is really weird
          std::cout << "Something is wrong with dist: x=" << x << " y=" << y << " dist=" << dist << std::endl;
      } else {
          --libit;
      }
 
      //std::cout << "Found eta bin" << std::endl;
 
      if ((*libit).second.empty()) {
          std::cout << "The etabin corresponding to the eta is empty" << std::endl;
          return nullptr;
      }
      double trenergy = track->GetKineticEnergy();
      //std::cout << "Energy: " << trenergy << std::endl;
      distbin::const_iterator distit = (*libit).second.lower_bound(trenergy);
      if (distit == (*libit).second.end()) --distit; //if it points to the end, repoint it to the last shower in bin
      else if (distit != (*libit).second.begin()) { //find the closest energy. it's either found item or previous (if there is a previous)
          distbin::const_iterator distitch = distit;
          --distitch;
          if (((*distit).first - trenergy) > (trenergy - (*distitch).first)) { // the closest is the previous
              --distit;
          }
      }
      //std::cout << "Found closest energy:" << (*distit).first << std::endl;
      if (randomShift > 0) {
          double upperEnergy = (*distit).first * 1.01; //we allow 1% off
          for (int i = 0; i < randomShift; i++) {
              ++distit;
              if (distit == (*libit).second.end()) {
                  --distit;
                  break;
              }
              if ((*distit).first > upperEnergy) break; //energy diff too high
          }
      }
      if ((randomShift < 0)&&(distit != (*libit).second.begin())) {
          double lowerEnergy = (*distit).first * 0.99; //we allow 1% off
          for (int i = 0; i > randomShift; i--) {
              --distit;
              if (distit == (*libit).second.begin()) {
                  //distit++;
                  break;
              }
              if (lowerEnergy > (*distit).first) break; //energy diff too high
          }
      }
      //std::cout << "Found out energy" << std::endl;
      //std::cout << "Shower with num hits:" << (*etait).second.size() << std::endl;
      std::vector<EnergySpot>* outshower = new std::vector<EnergySpot>();//((*etait).second);
      Shower::const_iterator iter;
      //std::cout << "Created out shower" << std::endl;
 
      float energyScale = track->GetKineticEnergy() / (*distit).first;
      //std::cout << "Scale: " << energyScale << std::endl;
 
      for (iter = (*distit).second.begin() /*outshower->begin()*/; iter != (*distit).second.end() /*outshower->end()*/; ++iter) {
          EnergySpot tmp( (*iter)->GetPosition(), (*iter)->GetEnergy(), (*iter)->GetTime() );
          tmp.SetEnergy(tmp.GetEnergy() * energyScale);
          outshower->push_back(tmp);
          //(*iter).SetEnergy((*iter).GetEnergy() * energyScale);
      }
      //std::cout << "Scaled" << std::endl;
      if (stats != nullptr) {
          stats->recordShowerLibUse(calcKey((*libit).first));
      }
      //std::cout << "Done" << std::endl;
      return outshower;
  }

particle_id()

int ShowerLib::IShowerLib::particle_id ( ) const

inlinevirtualinherited

get particle tag

Definition at line 124 of file IShowerLib.h.

{ return m_particle; }

physicsList() [1/2]

const std::string ShowerLib::IShowerLib::physicsList ( ) const

inlinevirtualinherited

get geant 4 physics list name

Definition at line 136 of file IShowerLib.h.

{ return m_physicsList; }

physicsList() [2/2]

void ShowerLib::IShowerLib::physicsList ( const std::string & list )

inlinevirtualinherited

set geant 4 physics list name

Definition at line 137 of file IShowerLib.h.

{ m_physicsList = list; }

printParameters()

const std::string ShowerLib::FCALDistEnergyShowerLib::printParameters ( ) const

protectedvirtual

Implements ShowerLib::IShowerLib.

Definition at line 613 of file FCALDistEnergyShowerLib.cxx.

  {
      std::stringstream ss;
      ss << std::fixed << std::setprecision(3);
      ss << "Distance calculator parameters: xrod_cent=" << m_xrodcent << " yrod_cent=" << m_yrodcent << " step=" << m_step;
      return ss.str();
  }

read()

bool ShowerLib::FCALDistEnergyShowerLib::read ( TTree * source )

private

read library from given TTree

Definition at line 490 of file FCALDistEnergyShowerLib.cxx.

  {
      /*
       * Dist Energy library format:
       * |       x      |       y      |       z      |    e     |  time  |  - name of branch in TTree
       * ------------------------------------------------------------------
       * |  xrod cent   |  yrod cent   |step (roddist)|   not    |  not   |  - library header
       * | (parameter)  | (parameter)  | (parameter)  |   used   |  used  |
       * ------------------------------------------------------------------
       * |num of showers| min dist for |     not      |   not    |  not   |  - dist bin header
       * | in dist bin  | cur dist bin |     used     |   used   |  used  |
       * ------------------------------------------------------------------
       * | num of hits  |shower r-size |shower z-size |  truth   |  not   |  - shower header
       * |  in shower   |for cont.check|for cont.check|  energy  |  used  |
       * ------------------------------------------------------------------
       * |x-coord of hit|y-coord of hit|z-coord of hit|dep.energy|hit time|  - hit
       */
      int nentr = source->GetEntriesFast();
      if (nentr < 3) return false;
      Float_t x,y,z,e,time;
      source->SetBranchAddress("x",&x);
      source->SetBranchAddress("y",&y);
      source->SetBranchAddress("z",&z);
      source->SetBranchAddress("e",&e);
      source->SetBranchAddress("time",&time);
      int entr = 0;
 
          source->GetEntry(entr++);
      m_xrodcent = x;
      m_yrodcent = y;
      m_step = z;
 
      do {
          //read eta bin header
          source->GetEntry(entr++); //x - nshowers, y - min dist in the current dist bin
          int nsh = (int)(x+0.1); // +0.1 just in case - c++ has low round
          float curDist = y;
          distbin * curbin = &(m_libData[curDist]); //creating a new dist bin
          for(int i = 0; i < nsh; i++) {
              //read shower header
              source->GetEntry(entr++); //x - nhits, y - r size, z - z size, e - gen energy
              int nhits = (int)(x+0.1);
              float curEnergy = e;
              Shower * shower = &((*curbin)[curEnergy]);
              shower->setRSize(y);
              shower->setZSize(z);
              for(int j = 0; j < nhits; j++) {
                  source->GetEntry(entr++); //variables mean what the name suggests
                  shower->push_back(new ShowerEnergySpot(G4ThreeVector(x,y,z),e,time));
              }
          }
      } while (entr < nentr);
 
      if (entr != nentr) {
          return false;
      }
 
      m_filled = true;
      return true;
  }

readFromROOTFile()

IShowerLib * ShowerLib::FCALDistEnergyShowerLib::readFromROOTFile ( TFile * source )

static

factory method. create a library from root file. returns NULL if file is invalid.

Definition at line 53 of file FCALDistEnergyShowerLib.cxx.

  {
      TParameter<int>* ver;
      ver = (TParameter<int>*)source->Get("version");
 
      if ((ver == nullptr) || (ver->GetVal() != LIB_VERSION)) return nullptr;
 
      TTree* TTreeMeta = (TTree*)source->Get("meta");
      TTree* TTreeLib  = (TTree*)source->Get("library");
 
      if ((TTreeMeta == nullptr) || (TTreeLib == nullptr)) return nullptr;
 
      std::cout << "FCALDistEnergyShowerLib header found." << std::endl;
 
      FCALDistEnergyShowerLib* newlib = new FCALDistEnergyShowerLib();
 
      if (!(newlib->readMeta(TTreeMeta)) || !(newlib->read(TTreeLib))) {
          delete newlib;
          std::cout << "FCALDistEnergyShowerLib read unsuccessful." << std::endl;
          return nullptr;
      }
 
      return newlib;
 
  }

readMeta()

bool ShowerLib::IShowerLib::readMeta ( TTree * source )

protectedinherited

read metadata from the given TTree

Definition at line 16 of file IShowerLib.cxx.

                                         {
      char detector[40];
      char release[40];
      char geometry[40];
      char geant[40];
      char physics[40];
      char comment[100];
      source->SetBranchAddress("detector",&detector);
      source->SetBranchAddress("particle",&m_particle);
      source->SetBranchAddress("release",&release);
      source->SetBranchAddress("geometry",&geometry);
      source->SetBranchAddress("geantVersion",&geant);
      source->SetBranchAddress("physicsList",&physics);
      source->SetBranchAddress("comment",&comment);
      source->GetEntry(0);
      m_detector = detector;
      m_release = release;
      m_geometry = geometry;
      m_geantVersion = geant;
      m_physicsList = physics;
      m_comment = comment;
      return true;
  }

readStructure()

bool ShowerLib::FCALDistEnergyShowerLib::readStructure ( std::vector< float > & structure )

private

Definition at line 621 of file FCALDistEnergyShowerLib.cxx.

  {
      std::vector<float>::const_iterator iter;
 
      for (iter = structure.begin(); iter != structure.end(); ++iter) {
          m_libData[(*iter)];
      }
 
      return true;
  }

release() [1/2]

const std::string ShowerLib::IShowerLib::release ( ) const

inlinevirtualinherited

get Release tag

Definition at line 127 of file IShowerLib.h.

{ return m_release; }

release() [2/2]

void ShowerLib::IShowerLib::release ( const std::string & release )

inlinevirtualinherited

set Release tag

Definition at line 128 of file IShowerLib.h.

{ m_release = release; }

statistics()

const std::string ShowerLib::IShowerLib::statistics ( ) const

inherited

print library statistics

Definition at line 71 of file IShowerLib.cxx.

                                               {
 
      if (m_filled) {
      std::stringstream out;
 
      out << "======================================================" << std::endl;
      out << "             Statistics of the shower library         " << std::endl;
      out << "======================================================" << std::endl;
      out << "Library Type: "<< getName() <<" PARTICLE: " << m_particle << " DETECTOR: "<< m_detector << std::endl;
      out << m_release << " " << m_geometry << " " << m_geantVersion << " " << m_physicsList << std::endl;
      if (printParameters().length() > 0)
          out << printParameters() << std::endl;
      out << m_comment << std::endl;
      return out.str();
      } else { //statistics for the genlib, shorter version
      std::stringstream out;
      out << getName() <<" PARTICLE: " << m_particle << " DETECTOR: "<< m_detector << ", ";
      out << m_release << " " << m_geometry << " " << m_geantVersion << " " << m_physicsList << std::endl;
      if (printParameters().length() > 0)
          out << printParameters() << std::endl;
      out << m_comment << std::endl;
      return out.str();
      }
  }

storeShower()

bool ShowerLib::FCALDistEnergyShowerLib::storeShower ( HepMC::ConstGenParticlePtr genParticle, const Shower * shower )

virtual

store shower in the library

Implements ShowerLib::IShowerLib.

Definition at line 442 of file FCALDistEnergyShowerLib.cxx.

  {
      if (m_filled) {
          std::cout << "ERROR: filled" << std::endl;
          return false;
      }
 
      double x = genParticle->production_vertex()->position().x();
      double y = genParticle->production_vertex()->position().y();
 
      float dist = this->distance(x,y);
 
      if ( genParticle->pdg_id() != m_particle ) {
          std::cout << "ERROR: wrong pdgcode: " << m_particle << " != " << genParticle->pdg_id() << std::endl;
          return false;
      }
 
      library::iterator libit = m_libData.upper_bound(dist);
      if (libit == m_libData.begin()) {
          //this is really weird
          std::cout << "Something is wrong with dist: x=" << x << " y=" << y << " dist=" << dist << std::endl;
      } else {
          --libit;
      }
      (*libit).second.insert(distbin::value_type(genParticle->momentum().e(),(*shower)));
      return true;
  }

write()

bool ShowerLib::FCALDistEnergyShowerLib::write ( TTree * dest ) const

private

write library to given TTree

Definition at line 551 of file FCALDistEnergyShowerLib.cxx.

  {
      /*
       * Dist Energy library format:
       * |       x      |       y      |       z      |    e     |  time  |  - name of branch in TTree
       * ------------------------------------------------------------------
       * |  xrod cent   |  yrod cent   |step (roddist)|   not    |  not   |  - library header
       * | (parameter)  | (parameter)  | (parameter)  |   used   |  used  |
       * ------------------------------------------------------------------
       * |num of showers| min dist for |     not      |   not    |  not   |  - dist bin header
       * | in dist bin  | cur dist bin |     used     |   used   |  used  |
       * ------------------------------------------------------------------
       * | num of hits  |shower r-size |shower z-size |  truth   |  not   |  - shower header
       * |  in shower   |for cont.check|for cont.check|  energy  |  used  |
       * ------------------------------------------------------------------
       * |x-coord of hit|y-coord of hit|z-coord of hit|dep.energy|hit time|  - hit
       */
      Float_t x,y,z,e,time;
      dest->Branch("x",&x);
      dest->Branch("y",&y);
      dest->Branch("z",&z);
      dest->Branch("e",&e);
      dest->Branch("time",&time);
 
      x = m_xrodcent;
      y = m_yrodcent;
      z = m_step;
      e = 0;
      time = 0;
      dest->Fill();
      
      library::const_iterator libit;
      for (libit = m_libData.begin(); libit != m_libData.end(); ++libit) {
          x = (*libit).second.size();
          y = (*libit).first;
          z = 0;
          e = 0;
          time = 0;
          dest->Fill(); //eta bin header
          distbin::const_iterator distit;
          for (distit = (*libit).second.begin(); distit != (*libit).second.end(); ++distit) {
              x = (*distit).second.size();
              y = (*distit).second.getRSize();
              z = (*distit).second.getZSize();
              e = (*distit).first;
              time = 0;
              dest->Fill(); //shower header
              Shower::const_iterator iter;
              for (iter = (*distit).second.begin(); iter != (*distit).second.end(); ++iter) {
                  x = (*iter)->GetPosition().x();
                  y = (*iter)->GetPosition().y();
                  z = (*iter)->GetPosition().z();
                  e = (*iter)->GetEnergy();
                  time = (*iter)->GetTime();
                  dest->Fill();
              }
          }
      }
      //dest->Write();
      return true;
  }

writeMeta()

bool ShowerLib::IShowerLib::writeMeta ( TTree * dest ) const

protectedinherited

write metadata to the given TTree

Definition at line 40 of file IShowerLib.cxx.

                                                {
      char detector[40];
      char release[40];
      char geometry[40];
      char geant[40];
      char physics[40];
      char comment[100];
      int  particle(m_particle);
      m_detector.copy(detector,39,0);
      detector[m_detector.length()] = '\0';
      m_release.copy(release,39,0);
      release[m_release.length()] = '\0';
      m_geometry.copy(geometry,39,0);
      geometry[m_geometry.length()] = '\0';
      m_geantVersion.copy(geant,39,0);
      geant[m_geantVersion.length()] = '\0';
      m_physicsList.copy(physics,39,0);
      physics[m_physicsList.length()] = '\0';
      m_comment.copy(comment,99,0);
      comment[m_comment.length()] = '\0';
      source->Branch("detector",&detector,"detector/C");
      source->Branch("particle",&particle,"particle/I");
      source->Branch("release",&release,"release/C");
      source->Branch("geometry",&geometry,"geometry/C");
      source->Branch("geantVersion",&geant,"geantVersion/C");
      source->Branch("physicsList",&physics,"physicsList/C");
      source->Branch("comment",&comment,"comment/C");
      source->Fill();
      return true;
  }

writeToROOT()

bool ShowerLib::FCALDistEnergyShowerLib::writeToROOT ( TFile * dest )

virtual

write library to ROOT file

Implements ShowerLib::IShowerLib.

Definition at line 470 of file FCALDistEnergyShowerLib.cxx.

  {
      if (m_libData.empty()) return false;
      TParameter<int> ver("version",LIB_VERSION);
 
      dest->WriteObject(&ver,"version");
 
      TTree TTreeMeta;
      TTree TTreeLib;
 
      write(&TTreeLib);
      writeMeta(&TTreeMeta);
 
      dest->WriteObject(&TTreeLib,"library");
      dest->WriteObject(&TTreeMeta,"meta");
 
      return true;
  }

Member Data Documentation

m_comment

std::string ShowerLib::IShowerLib::m_comment

protectedinherited

comment

Definition at line 112 of file IShowerLib.h.

m_detector

std::string ShowerLib::IShowerLib::m_detector

protectedinherited

name of the detector

Definition at line 106 of file IShowerLib.h.

m_filled

bool ShowerLib::IShowerLib::m_filled

protectedinherited

is the library read from ROOT or from structure file

Definition at line 114 of file IShowerLib.h.

m_geantVersion

std::string ShowerLib::IShowerLib::m_geantVersion

protectedinherited

geant 4 version

Definition at line 110 of file IShowerLib.h.

m_geometry

std::string ShowerLib::IShowerLib::m_geometry

protectedinherited

atlas geometry version

Definition at line 109 of file IShowerLib.h.

m_libData

library ShowerLib::FCALDistEnergyShowerLib::m_libData

private

Definition at line 96 of file FCALDistEnergyShowerLib.h.

m_particle

int ShowerLib::IShowerLib::m_particle

protectedinherited

ID of the generated particles.

Definition at line 107 of file IShowerLib.h.

m_physicsList

std::string ShowerLib::IShowerLib::m_physicsList

protectedinherited

used geant 4 physics list

Definition at line 111 of file IShowerLib.h.

m_release

std::string ShowerLib::IShowerLib::m_release

protectedinherited

atlas software release

Definition at line 108 of file IShowerLib.h.

m_step

double ShowerLib::FCALDistEnergyShowerLib::m_step

private

Definition at line 101 of file FCALDistEnergyShowerLib.h.

m_xrodcent

double ShowerLib::FCALDistEnergyShowerLib::m_xrodcent

private

Definition at line 99 of file FCALDistEnergyShowerLib.h.

m_yrodcent

double ShowerLib::FCALDistEnergyShowerLib::m_yrodcent

private

Definition at line 100 of file FCALDistEnergyShowerLib.h.

---

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

• FCALDistEnergyShowerLib.h
• FCALDistEnergyShowerLib.cxx