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