CaloSubdetectorGeometry.h

Go to the documentation of this file.

#ifndef GEOMETRY_CALOGEOMETRY_CALOSUBDETECTORGEOMETRY_H
#define GEOMETRY_CALOGEOMETRY_CALOSUBDETECTORGEOMETRY_H 1

#include <memory>
#include <vector>
#include <set>
#if !defined(__CINT__) && !defined(__MAKECINT__) && !defined(__REFLEX__)
#include <atomic>
#endif
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "Geometry/CaloGeometry/interface/CaloCellGeometry.h"
#include "DataFormats/Math/interface/deltaR.h"

class CaloSubdetectorGeometry {
public:
  typedef std::vector<std::shared_ptr<const CaloCellGeometry> > CellSet;
  typedef CaloCellGeometry::CCGFloat CCGFloat;

  typedef std::set<DetId> DetIdSet;

  typedef CaloCellGeometry::ParMgr ParMgr;
  typedef CaloCellGeometry::ParVec ParVec;
  typedef CaloCellGeometry::ParVecVec ParVecVec;

  typedef std::vector<CCGFloat> TrVec;
  typedef std::vector<unsigned int> IVec;
  typedef std::vector<CCGFloat> DimVec;

  CaloSubdetectorGeometry();

  virtual ~CaloSubdetectorGeometry();

  CaloSubdetectorGeometry(const CaloSubdetectorGeometry&) = delete;
  CaloSubdetectorGeometry& operator=(const CaloSubdetectorGeometry&) = delete;

  virtual void newCell(const GlobalPoint& f1,
                       const GlobalPoint& f2,
                       const GlobalPoint& f3,
                       const CCGFloat* parm,
                       const DetId& detId) = 0;

  virtual bool present(const DetId& id) const;

  virtual std::shared_ptr<const CaloCellGeometry> getGeometry(const DetId& id) const;

  virtual const std::vector<DetId>& getValidDetIds(DetId::Detector det = DetId::Detector(0), int subdet = 0) const;

  // Get closest cell, etc...
  virtual DetId getClosestCell(const GlobalPoint& r) const;

  virtual DetIdSet getCells(const GlobalPoint& r, double dR) const;
  virtual CellSet getCellSet(const GlobalPoint& r, double dR) const;

  CCGFloat deltaPhi(const DetId& detId) const;

  CCGFloat deltaEta(const DetId& detId) const;

  void allocateCorners(CaloCellGeometry::CornersVec::size_type n);

  CaloCellGeometry::CornersMgr* cornersMgr() { return m_cmgr; }

  void allocatePar(ParVec::size_type n, unsigned int m);

  ParMgr* parMgr() { return m_parMgr; }
  const ParMgr* parMgrConst() const { return m_parMgr; }

  ParVecVec& parVecVec() { return m_parVecVec; }
  const ParVecVec& parVecVec() const { return m_parVecVec; }

  virtual unsigned int numberOfShapes() const { return 1; }
  virtual unsigned int numberOfParametersPerShape() const { return 1; }
  virtual unsigned int numberOfTransformParms() const { return 6; }

  virtual void fillDefaultNamedParameters() const { return; }

  virtual void getSummary(TrVec& trVector, IVec& iVector, DimVec& dimVector, IVec& dinsVector) const;

  virtual void initializeParms() { return; }

protected:
  virtual unsigned int indexFor(const DetId& id) const;
  virtual unsigned int sizeForDenseIndex(const DetId& id) const;

  virtual const CaloCellGeometry* getGeometryRawPtr(uint32_t index) const = 0;
  virtual std::shared_ptr<const CaloCellGeometry> cellGeomPtr(uint32_t index) const;

  ParVecVec m_parVecVec;

  static CCGFloat deltaR(const GlobalPoint& p1, const GlobalPoint& p2) { return reco::deltaR(p1, p2); }

  void addValidID(const DetId& id);

  std::vector<DetId> m_validIds;

private:
  ParMgr* m_parMgr;

  CaloCellGeometry::CornersMgr* m_cmgr;

  bool m_sortedIds;

#if !defined(__CINT__) && !defined(__MAKECINT__) && !defined(__REFLEX__)
  mutable std::atomic<std::vector<CCGFloat>*> m_deltaPhi;
  mutable std::atomic<std::vector<CCGFloat>*> m_deltaEta;
#else
  mutable std::vector<CCGFloat>* m_deltaPhi;
  mutable std::vector<CCGFloat>* m_deltaEta;
#endif
};

#endif