EcalGeometry.h

Go to the documentation of this file.

 
#ifndef DETDESCR_ECALGEOMETRY_H_
#define DETDESCR_ECALGEOMETRY_H_
 
#include <assert.h>
 
#include <algorithm>
#include <iomanip>
#include <iostream>
#include <map>
 
#include "DetDescr/EcalID.h"
#include "Framework/ConditionsObject.h"
#include "Framework/Configure/Parameters.h"
#include "Framework/Exception/Exception.h"
#include "Framework/Logger.h"
 
// ROOT
#include "TGeoPolygon.h"
#include "TGraph.h"
#include "TH2Poly.h"
#include "TList.h"
 
namespace ecal {
class EcalGeometryProvider;
}
 
namespace ldmx {
 
class EcalGeometry : public framework::ConditionsObject {
 public:
  static constexpr const char* CONDITIONS_OBJECT_NAME{"EcalGeometry"};
 
  virtual ~EcalGeometry() = default;
 
  EcalID getID(double x, double y, double z, bool fallible = false) const;
 
  EcalID getID(double x, double y, int layer_id, bool fallible = false) const;
 
  EcalID getID(double x, double y, int layer_id, int module_id,
               bool fallible = false) const;
 
  std::tuple<double, double, double> getPosition(EcalID id) const;
 
  std::pair<double, double> getPositionInModule(int cell_id) const;
 
  double getZPosition(int layer) const {
    return std::get<2>(layer_pos_xy_.at(layer));
  }
 
  double getEcalFrontZ() const { return ecal_front_z_; }
 
  int getNumLayers() const { return layer_pos_xy_.size(); }
 
  int getNumModulesPerLayer() const { return 7; }
 
  int getNumCellsPerModule() const {
    return cell_id_in_module_.GetNumberOfBins();
  }
 
  std::vector<EcalID> getNN(EcalID id) const {
    auto list = nn_map_.at(EcalID(0, id.module(), id.cell()));
    for (auto& flat : list)
      flat = EcalID(id.layer(), flat.module(), flat.cell());
    return list;
  }
 
  bool isNN(EcalID centroid, EcalID probe) const {
    for (auto& id : getNN(centroid)) {
      if (id == probe) return true;
    }
    return false;
  }
 
  std::vector<EcalID> getNNN(EcalID id) const {
    auto list = nnn_map_.at(EcalID(0, id.module(), id.cell()));
    for (auto& flat : list)
      flat = EcalID(id.layer(), flat.module(), flat.cell());
    return list;
  }
 
  bool isNNN(EcalID centroid, EcalID probe) const {
    for (auto& id : getNNN(centroid)) {
      if (id == probe) return true;
    }
    return false;
  }
 
  double getModuleMinR() const { return module_r_min_; }
 
  double getModuleMaxR() const { return module_r_max_; }
 
  double getCellMinR() const { return cell_r_min_; }
 
  double getCellMaxR() const { return cell_r_max_; }
 
  TH2Poly* getCellPolyMap() const {
    for (auto const& [cell_id, cell_center] : cell_pos_in_module_) {
      cell_id_in_module_.Fill(cell_center.first, cell_center.second, cell_id);
    }
    return &cell_id_in_module_;
  }
 
  static EcalGeometry* debugMake(const framework::config::Parameters& p) {
    return new EcalGeometry(p);
  }
 
 private:
  EcalGeometry(const framework::config::Parameters& ps);
  friend class ecal::EcalGeometryProvider;
 
  void buildLayerMap();
 
  void buildModuleMap();
 
  void buildCellMap();
 
  void buildCellModuleMap();
 
  void buildNeighborMaps();
 
  double distanceToEdge(EcalID id) const;
 
  bool isEdgeCell(EcalID cellModuleID) const {
    return (distanceToEdge(cellModuleID) < cell_r_max_);
  }
 
  bool isInside(double normX, double normY) const;
 
 private:
  double gap_;
 
  double cell_r_min_{0};
 
  double module_r_min_{0};
 
  double cell_r_max_{0};
 
  double module_r_max_{0};
 
  bool corners_side_up_;
 
  double layer_shift_x_;
 
  double layer_shift_y_;
 
  bool layer_shift_odd_;
 
  bool layer_shift_odd_bilayer_;
 
  double si_thickness_;
 
  double n_cell_r_height_{0};
 
  double ecal_front_z_{0};
 
  std::vector<double> layer_z_positions_;
 
 private:
  std::map<int, std::tuple<double, double, double>> layer_pos_xy_;
 
  std::map<int, std::pair<double, double>> module_pos_xy_;
 
  std::map<int, std::pair<double, double>> cell_pos_in_module_;
 
  std::map<EcalID, std::pair<double, double>> cell_pos_in_layer_;
 
  std::map<EcalID, std::tuple<double, double, double>> cell_global_pos_;
 
  std::map<EcalID, std::vector<EcalID>> nn_map_;
 
  std::map<EcalID, std::vector<EcalID>> nnn_map_;
 
  mutable TH2Poly cell_id_in_module_;
 
  enableLogging("EcalGeometry")
};
 
}  // namespace ldmx
 
#endif