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();

        // 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);

}

#endif