SimplifiedGeometry.h

Go to the documentation of this file.

#ifndef FASTSIM_SIMPLIFIEDGEOMETRY_H
#define FASTSIM_SIMPLIFIEDGEOMETRY_H
 
#include "DataFormats/Math/interface/LorentzVector.h"
 
#include <memory>
#include <vector>
 
// Author: L. Vanelderen, S. Kurz
// Date: 29 May 2017
 
class DetLayer;
class MagneticField;
class GeometricSearchTracker;
class TH1F;
 
namespace edm {
  class ParameterSet;
}
 
namespace fastsim {
  class InteractionModel;
  class SimplifiedGeometryFactory;
 
 
  class SimplifiedGeometry {
  public:
 
    SimplifiedGeometry(double geomProperty);
 
    SimplifiedGeometry(SimplifiedGeometry&&) = default;
    SimplifiedGeometry(SimplifiedGeometry const&) = delete;
 
    SimplifiedGeometry& operator=(SimplifiedGeometry&&) = default;
    SimplifiedGeometry& operator=(SimplifiedGeometry const&) = delete;
 
    virtual ~SimplifiedGeometry();
 
    // HACK
 
    enum CaloType { NONE, TRACKERBOUNDARY, PRESHOWER1, PRESHOWER2, ECAL, HCAL, VFCAL };
 
    void setCaloType(CaloType type) { caloType_ = type; }
 
    CaloType getCaloType() const { return caloType_; }
 
    // END
 
 
    void setIndex(int index) { index_ = index; }
 
 
    int index() const { return index_; }
 
 
    const double getGeomProperty() const { return geomProperty_; }
 
 
    virtual const double getThickness(const math::XYZTLorentzVector& position) const = 0;
 
 
    virtual const double getThickness(const math::XYZTLorentzVector& position,
                                      const math::XYZTLorentzVector& momentum) const = 0;
 
 
    const double getNuclearInteractionThicknessFactor() const { return nuclearInteractionThicknessFactor_; }
 
 
    const DetLayer* getDetLayer() const { return detLayer_; }
 
 
    virtual const double getMagneticFieldZ(const math::XYZTLorentzVector& position) const = 0;
 
 
    virtual bool isForward() const = 0;
 
 
    const std::vector<InteractionModel*>& getInteractionModels() const { return interactionModels_; }
 
    friend std::ostream& operator<<(std::ostream& os, const SimplifiedGeometry& layer);
 
    friend class fastsim::SimplifiedGeometryFactory;
 
  protected:
    double geomProperty_;  
    int index_;  
    const DetLayer* detLayer_;                 
    std::unique_ptr<TH1F> magneticFieldHist_;  
    std::unique_ptr<TH1F> thicknessHist_;      
    double nuclearInteractionThicknessFactor_;  
    std::vector<InteractionModel*>
        interactionModels_;  
    CaloType caloType_;      
  };
 
  std::ostream& operator<<(std::ostream& os, const SimplifiedGeometry& layer);
 
}  // namespace fastsim
 
#endif