LayeringExtensionImpl.h

Go to the documentation of this file.

/*
 * LayeringExtensionImpl.h
 *
 * Simple implementation of the LayeringExtension interface.
 *
 *  Created on: Mar 6, 2014
 *      Author: Christian Grefe, CERN
 */

#ifndef DDRec_LAYERINGEXTENSIONIMPL_H_
#define DDRec_LAYERINGEXTENSIONIMPL_H_

// DDRec
#include "DDRec/Extensions/LayeringExtension.h"

// DD4hep
#include "DD4hep/Detector.h"

// C++
#include <map>
#include <vector>

class TGeoManager;

namespace DD4hep {
namespace DDRec {

class LayeringExtensionImpl: public LayeringExtension {
public:
  typedef Geometry::Position Position;
  
        LayeringExtensionImpl();

  LayeringExtensionImpl(const LayeringExtensionImpl& /*e*/, const Geometry::DetElement& /*d*/) {};

         virtual ~LayeringExtensionImpl();

         virtual int numberOfLayers() const;

         virtual int numberOfSensors(int layerIndex) const;

         virtual Geometry::DetElement layer(int layerIndex) const;

        virtual const std::vector<Geometry::DetElement>& sensors(int layerIndex) const;

        virtual const std::vector<Geometry::DetElement>& absorbers(int layerIndex) const;

        virtual double totalThickness() const;

        virtual double totalRadiationLength() const;

        virtual double totalInteractionLength() const;

        virtual double thickness(int layerIndex) const;

        virtual double radiationLength(int layerIndex) const;

        virtual double interactionLength(int layerIndex) const;

        virtual double absorberThickness(int layerIndex) const;

        virtual double absorberRadiationLength(int layerIndex) const;

        virtual double absorberInteractionLength(int layerIndex) const;

        virtual double sensorThickness(int layerIndex) const;

        virtual double sensorRadiationLength(int layerIndex) const;

        virtual double sensorInteractionLength(int layerIndex) const;

        void setLayer(int layerIndex, Geometry::DetElement layer, const Position& normal);

protected:
        struct LayerAttributes {
                LayerAttributes();
                Geometry::DetElement layer;
                Position normal;
                std::vector<Geometry::DetElement> sensors;
                std::vector<Geometry::DetElement> absorbers;
                double thickness;
                double radiationLength;
                double interactionLength;
                double absorberThickness;
                double absorberRadiationLength;
                double absorberInteractionLength;
                double sensorThickness;
                double sensorRadiationLength;
                double sensorInteractionLength;
                bool isCalculated;
                void calculate();
        private:
                double addElement(const Geometry::DetElement& det);
                TGeoManager* _tgeoManager;
        };

        mutable std::map<int, LayerAttributes> _layerMap; 

        void checkMap(int layerIndex) const;
};

} /* namespace DDRec */
} /* namespace DD4hep */

#endif /* DDReconstruction_LAYERINGEXTENSIONIMPL_H_ */