HGCalTriggerGeometryBase.h

Go to the documentation of this file.

#ifndef __L1Trigger_L1THGCal_HGCalTriggerGeometryBase_h__
#define __L1Trigger_L1THGCal_HGCalTriggerGeometryBase_h__

#include <iostream>
#include <unordered_set>
#include <unordered_map>

#include "DataFormats/GeometryVector/interface/GlobalPoint.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ESHandle.h"

#include "Geometry/HGCalGeometry/interface/HGCalGeometry.h"
#include "Geometry/CaloTopology/interface/HGCalTopology.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/HcalTowerAlgo/interface/HcalGeometry.h"

// Pure virtual trigger geometry class
// Provides the interface to access trigger cell and module mappings
class HGCalTriggerGeometryBase {
public:
  typedef std::unordered_map<unsigned, unsigned> geom_map;
  typedef std::unordered_set<unsigned> geom_set;
  typedef std::set<unsigned> geom_ordered_set;

  HGCalTriggerGeometryBase(const edm::ParameterSet& conf);
  virtual ~HGCalTriggerGeometryBase() {}

  const std::string& name() const { return name_; }

  bool isV9Geometry() const { return !calo_geometry_.isValid(); }
  const edm::ESHandle<CaloGeometry>& caloGeometry() const { return calo_geometry_; }
  const HGCalGeometry* eeGeometry() const {
    return (hgc_ee_geometry_.isValid()
                ? hgc_ee_geometry_.product()
                : (static_cast<const HGCalGeometry*>(calo_geometry_->getSubdetectorGeometry(DetId::Forward, HGCEE))));
  }
  const HGCalGeometry* fhGeometry() const {
    return (hgc_hsi_geometry_.isValid()
                ? hgc_hsi_geometry_.product()
                : (static_cast<const HGCalGeometry*>(calo_geometry_->getSubdetectorGeometry(DetId::Forward, HGCHEF))));
  }
  const HcalGeometry* bhGeometry() const {
    if (hgc_hsc_geometry_.isValid()) {
      throw cms::Exception("HGCalTriggerGeometry") << "bhGeometry cannot be used with the V9 geometry";
    }
    return (static_cast<const HcalGeometry*>(calo_geometry_->getSubdetectorGeometry(DetId::Hcal, HcalEndcap)));
  }
  const HGCalGeometry* hsiGeometry() const { return fhGeometry(); }
  const HGCalGeometry* hscGeometry() const {
    if (!hgc_hsc_geometry_.isValid()) {
      throw cms::Exception("HGCalTriggerGeometry") << "hscGeometry cannot be used with the V7 and V8 geometries";
    }
    return hgc_hsc_geometry_.product();
  }
  const HGCalTopology& eeTopology() const { return eeGeometry()->topology(); }
  const HGCalTopology& fhTopology() const { return fhGeometry()->topology(); }
  const HcalTopology& bhTopology() const { return bhGeometry()->topology(); }
  const HGCalTopology& hsiTopology() const { return hsiGeometry()->topology(); }
  const HGCalTopology& hscTopology() const { return hscGeometry()->topology(); }

  // non-const access to the geometry class
  virtual void initialize(const edm::ESHandle<CaloGeometry>&) = 0;
  virtual void initialize(const edm::ESHandle<HGCalGeometry>&,
                          const edm::ESHandle<HGCalGeometry>&,
                          const edm::ESHandle<HGCalGeometry>&) = 0;
  virtual void reset();

  // const access to the geometry class
  virtual unsigned getTriggerCellFromCell(const unsigned cell_det_id) const = 0;
  virtual unsigned getModuleFromCell(const unsigned cell_det_id) const = 0;
  virtual unsigned getModuleFromTriggerCell(const unsigned trigger_cell_det_id) const = 0;

  virtual geom_set getCellsFromTriggerCell(const unsigned cell_det_id) const = 0;
  virtual geom_set getCellsFromModule(const unsigned cell_det_id) const = 0;
  virtual geom_set getTriggerCellsFromModule(const unsigned trigger_cell_det_id) const = 0;

  virtual geom_ordered_set getOrderedCellsFromModule(const unsigned cell_det_id) const = 0;
  virtual geom_ordered_set getOrderedTriggerCellsFromModule(const unsigned trigger_cell_det_id) const = 0;

  virtual geom_set getNeighborsFromTriggerCell(const unsigned trigger_cell_det_id) const = 0;

  virtual GlobalPoint getTriggerCellPosition(const unsigned trigger_cell_det_id) const = 0;
  virtual GlobalPoint getModulePosition(const unsigned module_det_id) const = 0;

  virtual bool validTriggerCell(const unsigned trigger_cell_id) const = 0;
  virtual bool disconnectedModule(const unsigned module_id) const = 0;
  virtual unsigned triggerLayer(const unsigned id) const = 0;

protected:
  void setCaloGeometry(const edm::ESHandle<CaloGeometry>& geom) { calo_geometry_ = geom; }
  void setEEGeometry(const edm::ESHandle<HGCalGeometry>& geom) { hgc_ee_geometry_ = geom; }
  void setHSiGeometry(const edm::ESHandle<HGCalGeometry>& geom) { hgc_hsi_geometry_ = geom; }
  void setHScGeometry(const edm::ESHandle<HGCalGeometry>& geom) { hgc_hsc_geometry_ = geom; }

private:
  const std::string name_;

  edm::ESHandle<CaloGeometry> calo_geometry_;
  edm::ESHandle<HGCalGeometry> hgc_ee_geometry_;
  edm::ESHandle<HGCalGeometry> hgc_hsi_geometry_;
  edm::ESHandle<HGCalGeometry> hgc_hsc_geometry_;
};

#include "FWCore/PluginManager/interface/PluginFactory.h"
typedef edmplugin::PluginFactory<HGCalTriggerGeometryBase*(const edm::ParameterSet&)> HGCalTriggerGeometryFactory;

#endif