MarlinDDKalTest.h

Go to the documentation of this file.

#ifndef MarlinDDKalTest_h
#define MarlinDDKalTest_h

#include "MarlinTrk/IMarlinTrkSystem.h"

#ifdef MARLINTRK_DIAGNOSTICS_ON
#include "MarlinTrk/DiagnosticsController.h"
#endif

#include "DDKalTest/DDKalDetector.h"

//LCIO:
#include "lcio.h"
#include "UTIL/BitField64.h" 
#include "UTIL/LCTOOLS.h"
#include <LCRTRelations.h>

#include "streamlog/streamlog.h"

#include "TObjArray.h"
#include "TVector3.h"

#include <cmath>
#include <vector>


class TKalDetCradle ;
class TVKalDetector ;
class DDVMeasLayer ;
class THelicalTrack ;

class DDCylinderMeasLayer;

namespace EVENT{
  class TrackerHit ;
}

namespace MarlinTrk{
  
  class MarlinDDKalTest : public MarlinTrk::IMarlinTrkSystem {
    
  public:
    
    friend class MarlinDDKalTestTrack;
    
    // define some configuration constants
    static const bool FitBackward   = kIterBackward ;
    static const bool FitForward    = kIterForward ;
    static const bool OrderOutgoing  = true ;
    static const bool OrderIncoming  = false ;
    
    MarlinDDKalTest() ;
    MarlinDDKalTest(const MarlinDDKalTest&) = delete;
    MarlinDDKalTest const& operator=(const MarlinDDKalTest&) = delete;

    ~MarlinDDKalTest() ;
    
    virtual void setOption(unsigned CFGOption, bool val) ;

    void init() ; 
    
    virtual std::string name() { return "DDKalTest" ; }

    MarlinTrk::IMarlinTrack* createTrack()  ;
    
    
  protected:
    
    void includeMultipleScattering( bool on )  ;
    
    void includeEnergyLoss( bool on )  ;
    
    void storeActiveMeasurementModuleIDs(TVKalDetector* detector);  
    
    void getSensitiveMeasurementModules( int detElementID, std::vector< const DDVMeasLayer *>& measmodules) const; 
    
    void getSensitiveMeasurementModulesForLayer( int layerID, std::vector<const DDVMeasLayer *>& measmodules) const;
    
    //  void init(bool MSOn, bool EnergyLossOn) ;
    bool is_initialised=false;
    
    //** find the measurment layer for a given hit 
    const DDVMeasLayer* findMeasLayer( EVENT::TrackerHit * trkhit) const ; 
    //** find the measurment layer for a given det element ID and point in space 
    const DDVMeasLayer* findMeasLayer( int detElementID, const TVector3& point) const ;
    
    // get the last layer crossed by the helix when extrapolating from the present position to the pca to point
    const DDVMeasLayer* getLastMeasLayer(THelicalTrack const& helix, TVector3 const& point) const ;
    
    const DDCylinderMeasLayer* getIPLayer() const { return _ipLayer; }
    

    // members:

    const DDCylinderMeasLayer* _ipLayer=nullptr;
    
    TKalDetCradle* _det=nullptr;         // the detector cradle
    
    std::multimap< int,const DDVMeasLayer *> _active_measurement_modules{};
    
    std::multimap< int,const DDVMeasLayer *> _active_measurement_modules_by_layer{};

    std::vector< DDKalDetector* > _detectors{};
    
#ifdef MARLINTRK_DIAGNOSTICS_ON

  private:    
    MarlinTrk::DiagnosticsController _diagnostics;

  public:    

    virtual void * getDiagnositicsPointer() { return &_diagnostics ; }
            
#endif

    
  } ;
  
} // end of namespace MarlinTrk

#endif