Atlfast::TrackMaker Class Reference

#include <TrackMaker.h>

Collaboration diagram for Atlfast::TrackMaker:

[legend]List of all members.

Public Member Functions
	TrackMaker (const std::string &name, ISvcLocator *pSvcLocator)
virtual	~TrackMaker ()
virtual StatusCode	initialize ()
virtual StatusCode	execute ()
virtual StatusCode	finalize ()
Private Member Functions
virtual Track *	create (const HepMC::GenParticle *)
virtual bool	isAcceptable (Track *)
TrackTrajectory	extractTrajectory (const HepMC::GenParticle *)
Private Attributes
double	m_vlMax
double	m_vtMax
double	m_mcPtMin
double	m_mcEtaMax
bool	m_doSmearing
double	m_bField
string	m_muonParamFile
string	m_pionParamFile
string	m_electronParamFile
string	m_bremParamFile
TesIO *	m_tesIO
HepMC_helper::IMCselector *	m_ncutter
std::string	m_mcLocation
TrackSmearer *	m_smearer
ChargeService *	m_chargeService
std::string	m_outputLocation

---

Detailed Description

A class that makes tracks

TrackMaker sequence

Definition at line 80 of file TrackMaker.h.

---

Constructor & Destructor Documentation

Atlfast::TrackMaker::TrackMaker	(	const std::string &	name,
		ISvcLocator *	pSvcLocator
	)

Definition at line 45 of file TrackMaker.cxx.

                                                                           : 
    Algorithm( name, pSvcLocator ), 
    m_tesIO(0),
    m_ncutter(0),
    m_smearer(0),
    m_chargeService(0)
  {
    
    // Setting the parameter defaults - these can be changed by user via
    // jobOptions
    m_mcPtMin           = 0.5*GeV;
    m_mcEtaMax          = 2.5;
    m_vlMax             = 40.0*cm;
    m_vtMax             = 5.05*cm;
    m_doSmearing        = true;
    m_bField            = 2.0;
    m_muonParamFile     = "atlfastDatafiles/Atlfast_MuonResParam_CSC.dat";
    m_pionParamFile     = "atlfastDatafiles/Atlfast_PionResParam_DC1_NewUnits.dat";
    m_electronParamFile = "atlfastDatafiles/Atlfast_ElectronResParam_CSC.dat";
    m_bremParamFile     = "atlfastDatafiles/Atlfast_ElectronBremParam_CSC.dat";
    
    // Default paths for entities in the TES
    m_outputLocation    = "/Event/AtlfastTracks";
    
    // Declare the paramemters to Athena so that
    // they can be over-written via the job options file
    declareProperty( "McPtMinimum",           m_mcPtMin ) ;
    declareProperty( "McEtaMaximum",          m_mcEtaMax ) ;
    declareProperty( "vlMaximum",             m_vlMax ) ;
    declareProperty( "vtMaximum",             m_vtMax ) ;
    declareProperty( "DoSmearing",            m_doSmearing ); 
    declareProperty( "Bfield",                m_bField );
    declareProperty( "OutputLocation",        m_outputLocation ) ;
    declareProperty( "MuonParamFile",         m_muonParamFile );
    declareProperty( "PionParamFile",         m_pionParamFile );
    declareProperty( "ElectronParamFile",     m_electronParamFile );
    declareProperty( "ElectronBremParamFile", m_bremParamFile );
  }

Atlfast::TrackMaker::~TrackMaker

(

)

[virtual]

Definition at line 87 of file TrackMaker.cxx.

  {

    if (m_ncutter) delete m_ncutter;
    if (m_tesIO) delete m_tesIO;
    if (m_smearer) delete m_smearer;
    if (m_chargeService) delete m_chargeService;
  } 

---

Member Function Documentation

StatusCode Atlfast::TrackMaker::initialize

(

)

[virtual]

Definition at line 101 of file TrackMaker.cxx.

  {
    MsgStream log( messageService(), name() ) ;
    
    //get the Global Event Data from singleton pattern
    GlobalEventData* ged         = GlobalEventData::Instance();
    int lumi                     = ged->lumi();
    int randSeed                 = ged->randSeed() ;
    m_mcLocation                 = ged->mcLocation();  
    std::vector<int> invisibles  = ged->invisibles();

    log << MSG::INFO << "Got invisibles " << invisibles << endreq ;

    m_tesIO = new TesIO(m_mcLocation, ged->justHardScatter());
    
    //=======================================================
    // Set up NCutter to select the required HepMC::GenParticles
    HepMC_helper::IMCselector* charged;
    try
    {
      charged = new HepMC_helper::IsCharged() ;
    }
    catch(std::string errMsg)
    {
      log << MSG::ERROR << "Error making IsCharged " << errMsg<< endreq ;
    }
    catch(...)
    {
      log << MSG::ERROR << "Unknown Error making IsCharged " << endreq ;
    }
    HepMC_helper::IMCselector* cuts            = new HepMC_helper::MCCuts( m_mcPtMin, m_mcEtaMax ) ;
    HepMC_helper::IMCselector* finalState      = new HepMC_helper::IsFinalState();
    HepMC_helper::IMCselector* fromHardScatter = new HepMC_helper::IsFromHardScatter();
    HepMC_helper::IMCselector* invisible       = new HepMC_helper::RejectType(invisibles);


    vector<HepMC_helper::IMCselector*> selectors;
    selectors.push_back(fromHardScatter);
    selectors.push_back(finalState);
    selectors.push_back(charged);
    selectors.push_back(cuts);
    selectors.push_back(invisible);
    
    m_ncutter = new HepMC_helper::NCutter(selectors);
    
    delete cuts;
    delete charged;
    delete finalState;
    delete fromHardScatter;
    delete invisible;


    try
    {
      m_chargeService = new ChargeService() ;
    }
    catch(std::string errMsg)
    {
      log << MSG::ERROR << "Error making ChargeService " << errMsg<< endreq ;
    }
    catch(...)
    {
      log << MSG::ERROR << "Unknown Error making ChargeService " << endreq ;
    }
    m_smearer = new TrackSmearer( randSeed, log, m_muonParamFile, 
                                                 m_pionParamFile,
                                                 m_electronParamFile,
                                                 m_bremParamFile );
    log << MSG::DEBUG << "got lumi " << endreq;

    HeaderPrinter hp("Atlfast Track Maker:", log);
    hp.add("Luminosity              ",     lumi);        
    hp.add("Minimum four-vector Pt  ",     m_mcPtMin);
    hp.add("Maximum four-vector Eta ",     m_mcEtaMax);
    hp.add("Maximum Transverse Vertex ",   m_vtMax);
    hp.add("Maximum Longitudinal Vertex ", m_vlMax);
    hp.add("Do Smearing             ",     m_doSmearing);
    hp.add("B Field                 ",     m_bField);
    hp.add("Muon Parameters         ",     m_muonParamFile);
    hp.add("Pion Parameters         ",     m_pionParamFile);    
    hp.add("Electron Parameters     ",     m_electronParamFile);    
    hp.add("Electron Brem Parameters",     m_bremParamFile);    
    hp.add("Random Number Seed      ",     randSeed);
    hp.add("MC location             ",     m_mcLocation);
    hp.add("Output Location         ",     m_outputLocation);    
    hp.print();
    log << MSG::INFO << "Initialised successfully " << endreq;    
    return StatusCode::SUCCESS;
  }

StatusCode Atlfast::TrackMaker::execute

(

)

[virtual]

Definition at line 221 of file TrackMaker.cxx.

  {

    //................................
    // make a message logging stream
    
    MsgStream log( messageService(), name() ) ;
    log << MSG::DEBUG << "Execute() " << endreq;
    

    //.........................................................
    // We now need to access the HepMC event record(s) in the event store
    // and iterate over them to extract HepMC::GenParticles.
    // This method will only select the particles which are required
    // and add them to a local store (mcParticles)
    
    MCparticleCollection  mcParticles ;
    TesIoStat stat; 
    std::string message;
    
    stat = m_tesIO->getMC( mcParticles, m_ncutter );
    message = stat?  "Found MCparticles in TES":"No MC particles found in TES";
    log<<MSG::DEBUG << message <<" "<<mcParticles.size()<<endreq;
    // ......................
    // Make a container object which will store pointers to all isolated 
    // Tracks which are successfully created.  Since it is going to go 
    // into the event store, it has to be a special Athena container. 
    // This is typedefined in an include file
    
    TrackCollection* tracks = new TrackCollection ;
    

    //.........................................................
    // From each mc Track create a reconstructed Track. 
    // If all requirements are satisfied add to the collection
    
    Track* candidate ;
    MCparticleCollectionCIter src ;
    
    for( src = mcParticles.begin() ; src != mcParticles.end() ; ++src ) 
    { 
      log << MSG::DEBUG << *src << endreq;
      candidate = this->create( *src ); 
      log << MSG::DEBUG << "Created: " << candidate << endreq;
      
      if( this->isAcceptable(candidate) ) 
      {    
        log << MSG::DEBUG << "Passed cuts" << endreq;
        tracks->push_back(candidate) ; 
        log << MSG::DEBUG << "pushed back track" <<endreq;
      }
      else 
      {
        log << MSG::DEBUG << "Failed cuts" << endreq;
        delete candidate ;
      }
    }
    
    //......................................
    // Register any Tracks which have been successfilly created in the 
    // transient event store. If there are none then we delete the empty list.

    stat = m_tesIO->store(tracks, m_outputLocation);
    message = stat  ?  "Stored tracks in TES"  :  "Error storing tracks in TES";
    log << MSG::DEBUG << message << " " << tracks->size() << endreq;
    
    return stat; 

  }

StatusCode Atlfast::TrackMaker::finalize

(

)

[virtual]

Definition at line 195 of file TrackMaker.cxx.

  {
    
    MsgStream log( messageService(), name() ) ;
    
    log << MSG::INFO << "Finalised successfully " << endreq ;
    
    return StatusCode::SUCCESS ;
  }

Track * Atlfast::TrackMaker::create

(

const HepMC::GenParticle *

)

[private, virtual]

Definition at line 305 of file TrackMaker.cxx.

  {
    
    // Construct the Trajectory parameters corresponding to the source particle
    TrackTrajectory originalTrajectory = this->extractTrajectory( src ) ;

    if ( m_doSmearing ) 
    {
      // Ask our Smearer make a smeared set of Trajectory parameters
      // If this needs to be different for different particles then
      // it can be done here. Best  method of dispatching to be decided later

      Track myTrack( originalTrajectory, src );
      Track smearedTrack = m_smearer->smear( myTrack );
      
      // Return a new track
      Track* newTrack = new Track( smearedTrack );
      return newTrack;
    } 
    else 
    {
      // Just return the track with unsmeared four-momentum
      return new Track( originalTrajectory, src ) ;
    }
  }

bool Atlfast::TrackMaker::isAcceptable

(

Track *

)

[private, virtual]

Definition at line 342 of file TrackMaker.cxx.

  {
    HepPoint3D vertex = candidate->trajectory().startPoint();
    int pdg = abs( candidate->pdg_id() );
    if ( abs( vertex.z() ) > m_vlMax ) return false;
 
    // test track for vl and vt
    if ( pdg == 11 || pdg == 13 ) 
    {
      if ( vertex.perp() > m_vtMax ) return false;
    }
    return true ;
  }

TrackTrajectory Atlfast::TrackMaker::extractTrajectory

(

const HepMC::GenParticle *

)

[private]

Definition at line 358 of file TrackMaker.cxx.

  {
    // Look up vertex
    Hep3Vector vertex = (particle->production_vertex())->position().getV();
    // Get 4-momentum and explicitly convert to 3-momentum
    Hep3Vector threeMomentum = particle->momentum().getV() ;
    // Obtain charge of particle
    double charge=m_chargeService->operator()(particle);
    
    return TrackTrajectory( threeMomentum, vertex, charge, m_bField ) ;
    
  }

---

Member Data Documentation

double Atlfast::TrackMaker::m_vlMax [private]

Definition at line 106 of file TrackMaker.h.

double Atlfast::TrackMaker::m_vtMax [private]

Definition at line 107 of file TrackMaker.h.

double Atlfast::TrackMaker::m_mcPtMin [private]

Definition at line 108 of file TrackMaker.h.

double Atlfast::TrackMaker::m_mcEtaMax [private]

Definition at line 109 of file TrackMaker.h.

bool Atlfast::TrackMaker::m_doSmearing [private]

Definition at line 110 of file TrackMaker.h.

double Atlfast::TrackMaker::m_bField [private]

Definition at line 111 of file TrackMaker.h.

string Atlfast::TrackMaker::m_muonParamFile [private]

Definition at line 112 of file TrackMaker.h.

string Atlfast::TrackMaker::m_pionParamFile [private]

Definition at line 113 of file TrackMaker.h.

string Atlfast::TrackMaker::m_electronParamFile [private]

Definition at line 114 of file TrackMaker.h.

string Atlfast::TrackMaker::m_bremParamFile [private]

Definition at line 115 of file TrackMaker.h.

TesIO* Atlfast::TrackMaker::m_tesIO [private]

Definition at line 123 of file TrackMaker.h.

HepMC_helper::IMCselector* Atlfast::TrackMaker::m_ncutter [private]

Definition at line 124 of file TrackMaker.h.

std::string Atlfast::TrackMaker::m_mcLocation [private]

Definition at line 126 of file TrackMaker.h.

TrackSmearer* Atlfast::TrackMaker::m_smearer [private]

Definition at line 138 of file TrackMaker.h.

ChargeService* Atlfast::TrackMaker::m_chargeService [private]

Definition at line 139 of file TrackMaker.h.

std::string Atlfast::TrackMaker::m_outputLocation [private]

Definition at line 148 of file TrackMaker.h.

---

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

• /afs/cern.ch/user/s/sdean/project_builds/AtlasOffline-rel_1/Simulation/Atlfast/AtlfastAlgs/AtlfastAlgs/TrackMaker.h
• /afs/cern.ch/user/s/sdean/project_builds/AtlasOffline-rel_1/Simulation/Atlfast/AtlfastAlgs/src/TrackMaker.cxx

---