TrackMaker.h

Go to the documentation of this file.

// ================================================
// TrackMaker class description
// ================================================
//
// THIS TEXT TO BE REPLACED BY ATLAS STANDARD FORMAT
//
//
// This version....
//
//
// Namespace Atlfast::
//
// class: TrackMaker
//
// Description: 
//
//  Default Algorithm which knows how to simulate Tracks from MC truth 
//
//
//
// ................................................................
//

#ifndef ATLFAST_TRACKMAKER_H
#define ATLFAST_TRACKMAKER_H

// STL
#include <vector>
#include <string>

// Gaudi 
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/IAlgorithm.h"
#include "GaudiKernel/Algorithm.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/DataObject.h"

// Other
#include "CLHEP/Vector/LorentzVector.h"
#include "CLHEP/Vector/ThreeVector.h"
#include "HepMC/GenEvent.h"
#include "HepMC/GenParticle.h"
#include "HepMC/ParticleDataTable.h"


// Atlfast
#include "AtlfastEvent/IKinematic.h"
//#include "AtlfastEvent/TrackCollection.h"
#include "AtlfastEvent/CollectionDefs.h"
#include "AtlfastAlgs/TrackSmearer.h"
#include "AtlfastUtils/TesIO.h"
#include "AtlfastUtils/HeaderPrinter.h"
#include "AtlfastEvent/MCparticleCollection.h"


//***************************************************************
// TrackMaker class declaration
//
//****************************************************************
namespace HepMC_helper
{
  class IMCselector;
}

namespace Atlfast 
{
  using std::string;

  class ChargeService;

  class Track;
 
  class TrackMaker : public Algorithm{
    
  public:
    
    //-------------------------
    // Constructors/Destructors
    //-------------------------
    
    TrackMaker( const std::string& name, ISvcLocator* pSvcLocator ) ; 
    virtual ~TrackMaker();
    
    
    //------------------------------------------------------
    // Methods used by Gaudi to run the algorithm
    //------------------------------------------------------
    
    virtual StatusCode initialize() ;
    virtual StatusCode execute() ;
    virtual StatusCode finalize() ;
    
  private:
    
    // -------------------------------
    // Parameters of this algorithm
    // (descriptions above)
    //--------------------------------
    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;
    
    
    //--------------------------------------------
    // Selector for HepMC generator particles
    //---------------------------------------------
    
    
    TesIO* m_tesIO;
    HepMC_helper::IMCselector*  m_ncutter ;
    //holds the Stotegate MC location
    std::string m_mcLocation;
    
    //--------------------------------------------
    // Smearer helper class which knows all about how to do smearing
    // for Tracks
    // There may need to be more of these added in order
    // to treat different species differently.
    // Alternately the single smearer object could be made to
    // dispatch on particle type. Not thought through yet.
    //---------------------------------------------
    
    
    TrackSmearer*  m_smearer;
    ChargeService* m_chargeService;    
    
    //--------------------------------------------------
    // Paths in the Transient Event store to get/put things
    // These are currently set in member variables so that
    // they can be overwritten by the job options service
    //--------------------------------------------------
 
    // TES output location 
    std::string m_outputLocation;
    
    //-------------------------------
    // Private methods
    //-------------------------------
    
    // Create a Track 
    virtual Track* create( const HepMC::GenParticle* );

    // Test if Track passes kinematic requirements
    //  virtual bool isAcceptable( Track* );
    virtual bool isAcceptable(Track*);
    TrackTrajectory extractTrajectory( const HepMC::GenParticle* ) ;
  };

}  //end of namespace bracket

#endif

---