Merge pull request #27304 from Mourtz/L1Trigger

HGCal trigger support for Fireworks

Fireworks/Calo/plugins/BuildFile.xml

@@ -6,6 +6,7 @@
  <use name="DataFormats/HcalRecHit"/>
  <use name="DataFormats/JetReco"/>
  <use name="DataFormats/L1Trigger"/>
+ <use name="DataFormats/L1THGCal"/>
  <use name="DataFormats/METReco"/>
  <use name="DataFormats/PatCandidates"/>
  <use name="DataFormats/TauReco"/>

Fireworks/Calo/plugins/FWHGCalTriggerCellProxyBuilder.cc

@@ -0,0 +1,162 @@
+#include "Fireworks/Calo/plugins/FWL1THGCalProxyTemplate.cc"
+
+#include "Fireworks/Core/interface/FWEventItem.h"
+#include "Fireworks/Core/interface/FWGeometry.h"
+#include "Fireworks/Core/interface/BuilderUtils.h"
+
+#include "DataFormats/HcalDetId/interface/HcalDetId.h"
+#include "DataFormats/ForwardDetId/interface/HGCalTriggerDetId.h"
+#include "DataFormats/ForwardDetId/interface/HGCScintillatorDetId.h"
+#include "DataFormats/L1THGCal/interface/HGCalTriggerCell.h"
+// #include "L1Trigger/L1THGCal/plugins/geometries/HGCalTriggerGeometryV9Imp2.cc"
+
+#include "TEveBoxSet.h"
+
+class FWHGCalTriggerCellProxyBuilder : public FWL1THGCalProxyTemplate<l1t::HGCalTriggerCell> {
+public:
+  FWHGCalTriggerCellProxyBuilder(void) {}
+  ~FWHGCalTriggerCellProxyBuilder(void) override {}
+
+  REGISTER_PROXYBUILDER_METHODS();
+
+private:
+  FWHGCalTriggerCellProxyBuilder(const FWHGCalTriggerCellProxyBuilder &) = delete;                   // stop default
+  const FWHGCalTriggerCellProxyBuilder &operator=(const FWHGCalTriggerCellProxyBuilder &) = delete;  // stop default
+
+  void build(const l1t::HGCalTriggerCell &iData,
+             unsigned int iIndex,
+             TEveElement &oItemHolder,
+             const FWViewContext *) override;
+};
+
+void FWHGCalTriggerCellProxyBuilder::build(const l1t::HGCalTriggerCell &iData,
+                                           unsigned int iIndex,
+                                           TEveElement &oItemHolder,
+                                           const FWViewContext *) {
+  const long layer = item()->getConfig()->value<long>("Layer");
+  const double saturation_energy = item()->getConfig()->value<double>("EnergyCutOff");
+  const bool heatmap = item()->getConfig()->value<bool>("Heatmap");
+
+  const bool z_plus = item()->getConfig()->value<bool>("Z+");
+  const bool z_minus = item()->getConfig()->value<bool>("Z-");
+
+  bool h_hex(false);
+  TEveBoxSet *hex_boxset = new TEveBoxSet();
+  if (!heatmap)
+    hex_boxset->UseSingleColor();
+  hex_boxset->SetPickable(true);
+  hex_boxset->Reset(TEveBoxSet::kBT_Hex, true, 64);
+  hex_boxset->SetAntiFlick(true);
+
+  bool h_box(false);
+  TEveBoxSet *boxset = new TEveBoxSet();
+  if (!heatmap)
+    boxset->UseSingleColor();
+  boxset->SetPickable(true);
+  boxset->Reset(TEveBoxSet::kBT_FreeBox, true, 64);
+  boxset->SetAntiFlick(true);
+
+  const float energy = fmin(10 * iData.energy() / saturation_energy, 1.0);
+
+  std::unordered_set<unsigned> cells = getCellsFromTriggerCell(iData.detId());
+
+  for (std::unordered_set<unsigned>::const_iterator it = cells.begin(), itEnd = cells.end(); it != itEnd; ++it) {
+    const bool z = (*it >> 25) & 0x1;
+
+    // discard everything thats not at the side that we are intersted in
+    if (((z_plus & z_minus) != 1) && (((z_plus | z_minus) == 0) || !(z == z_minus || z == !z_plus)))
+      continue;
+
+    const float *corners = item()->getGeom()->getCorners(*it);
+    const float *parameters = item()->getGeom()->getParameters(*it);
+    const float *shapes = item()->getGeom()->getShapePars(*it);
+
+    if (corners == nullptr || parameters == nullptr || shapes == nullptr) {
+      continue;
+    }
+
+    const int total_points = parameters[0];
+    const bool isScintillator = (total_points == 4);
+    const uint8_t type = ((*it >> 28) & 0xF);
+
+    uint8_t ll = layer;
+    if (layer > 0) {
+      if (layer > 28) {
+        if (type == 8) {
+          continue;
+        }
+        ll -= 28;
+      } else {
+        if (type != 8) {
+          continue;
+        }
+      }
+
+      if (ll != ((*it >> (isScintillator ? 17 : 20)) & 0x1F))
+        continue;
+    }
+
+    // Scintillator
+    if (isScintillator) {
+      const int total_vertices = 3 * total_points;
+
+      std::vector<float> pnts(24);
+      for (int i = 0; i < total_points; ++i) {
+        pnts[i * 3 + 0] = corners[i * 3];
+        pnts[i * 3 + 1] = corners[i * 3 + 1];
+        pnts[i * 3 + 2] = corners[i * 3 + 2];
+
+        pnts[(i * 3 + 0) + total_vertices] = corners[i * 3];
+        pnts[(i * 3 + 1) + total_vertices] = corners[i * 3 + 1];
+        pnts[(i * 3 + 2) + total_vertices] = corners[i * 3 + 2] + shapes[3];
+      }
+      boxset->AddBox(&pnts[0]);
+      boxset->DigitColor(energy * 255, 0, 255 - energy * 255);
+
+      h_box = true;
+    }
+    // Silicon
+    else {
+      const int offset = 9;
+
+      float centerX = (corners[6] + corners[6 + offset]) / 2;
+      float centerY = (corners[7] + corners[7 + offset]) / 2;
+      float radius = fabs(corners[6] - corners[6 + offset]) / 2;
+      hex_boxset->AddHex(TEveVector(centerX, centerY, corners[2]), radius, 90.0, shapes[3]);
+      hex_boxset->DigitColor(energy * 255, 0, 255 - energy * 255);
+
+      h_hex = true;
+    }
+  }
+
+  if (h_hex) {
+    hex_boxset->RefitPlex();
+
+    hex_boxset->CSCTakeAnyParentAsMaster();
+    if (!heatmap) {
+      hex_boxset->CSCApplyMainColorToMatchingChildren();
+      hex_boxset->CSCApplyMainTransparencyToMatchingChildren();
+      hex_boxset->SetMainColor(item()->modelInfo(iIndex).displayProperties().color());
+      hex_boxset->SetMainTransparency(item()->defaultDisplayProperties().transparency());
+    }
+    oItemHolder.AddElement(hex_boxset);
+  }
+
+  if (h_box) {
+    boxset->RefitPlex();
+
+    boxset->CSCTakeAnyParentAsMaster();
+    if (!heatmap) {
+      boxset->CSCApplyMainColorToMatchingChildren();
+      boxset->CSCApplyMainTransparencyToMatchingChildren();
+      boxset->SetMainColor(item()->modelInfo(iIndex).displayProperties().color());
+      boxset->SetMainTransparency(item()->defaultDisplayProperties().transparency());
+    }
+    oItemHolder.AddElement(boxset);
+  }
+}
+
+REGISTER_FWPROXYBUILDER(FWHGCalTriggerCellProxyBuilder,
+                        l1t::HGCalTriggerCell,
+                        "HGCal Trigger Cell",
+                        FWViewType::kAll3DBits);

Fireworks/Calo/plugins/FWHGCalTriggerClusterProxyBuilder.cc

@@ -0,0 +1,168 @@
+#include "Fireworks/Calo/plugins/FWL1THGCalProxyTemplate.cc"
+
+#include "Fireworks/Core/interface/FWEventItem.h"
+#include "Fireworks/Core/interface/FWGeometry.h"
+#include "Fireworks/Core/interface/BuilderUtils.h"
+
+#include "DataFormats/L1THGCal/interface/HGCalMulticluster.h"
+#include "DataFormats/L1THGCal/interface/HGCalClusterT.h"
+
+#include "DataFormats/HcalDetId/interface/HcalDetId.h"
+#include "DataFormats/ForwardDetId/interface/HGCalTriggerDetId.h"
+#include "DataFormats/ForwardDetId/interface/HGCScintillatorDetId.h"
+
+#include "TEveBoxSet.h"
+#include "TColor.h"
+#include "TROOT.h"
+
+class FWHGCalTriggerClusterProxyBuilder : public FWL1THGCalProxyTemplate<l1t::HGCalMulticluster> {
+public:
+  FWHGCalTriggerClusterProxyBuilder(void) {}
+  ~FWHGCalTriggerClusterProxyBuilder(void) override {}
+
+  REGISTER_PROXYBUILDER_METHODS();
+
+private:
+  FWHGCalTriggerClusterProxyBuilder(const FWHGCalTriggerClusterProxyBuilder &) = delete;  // stop default
+  const FWHGCalTriggerClusterProxyBuilder &operator=(const FWHGCalTriggerClusterProxyBuilder &) =
+      delete;  // stop default
+
+  void build(const l1t::HGCalMulticluster &iData,
+             unsigned int iIndex,
+             TEveElement &oItemHolder,
+             const FWViewContext *) override;
+};
+
+void FWHGCalTriggerClusterProxyBuilder::build(const l1t::HGCalMulticluster &iData,
+                                              unsigned int iIndex,
+                                              TEveElement &oItemHolder,
+                                              const FWViewContext *) {
+  const long layer = item()->getConfig()->value<long>("Layer");
+  const double saturation_energy = item()->getConfig()->value<double>("EnergyCutOff");
+  const bool heatmap = item()->getConfig()->value<bool>("Heatmap");
+
+  const bool z_plus = item()->getConfig()->value<bool>("Z+");
+  const bool z_minus = item()->getConfig()->value<bool>("Z-");
+
+  bool h_hex(false);
+  TEveBoxSet *hex_boxset = new TEveBoxSet();
+  if (!heatmap)
+    hex_boxset->UseSingleColor();
+  hex_boxset->SetPickable(true);
+  hex_boxset->Reset(TEveBoxSet::kBT_Hex, true, 64);
+  hex_boxset->SetAntiFlick(true);
+
+  bool h_box(false);
+  TEveBoxSet *boxset = new TEveBoxSet();
+  if (!heatmap)
+    boxset->UseSingleColor();
+  boxset->SetPickable(true);
+  boxset->Reset(TEveBoxSet::kBT_FreeBox, true, 64);
+  boxset->SetAntiFlick(true);
+
+  const float energy = fmin(iData.energy() / (saturation_energy * 100), 1.0);
+
+  for (const auto &t_detid : iData.constituents()) {
+    std::unordered_set<unsigned> cells = getCellsFromTriggerCell(t_detid.first);
+
+    for (auto &it : cells) {
+      const bool z = (it >> 25) & 0x1;
+
+      // discard everything thats not at the side that we are intersted in
+      if (((z_plus & z_minus) != 1) && (((z_plus | z_minus) == 0) || !(z == z_minus || z == !z_plus)))
+        continue;
+
+      const float *corners = item()->getGeom()->getCorners(it);
+      const float *parameters = item()->getGeom()->getParameters(it);
+      const float *shapes = item()->getGeom()->getShapePars(it);
+
+      if (corners == nullptr || parameters == nullptr || shapes == nullptr) {
+        continue;
+      }
+
+      const int total_points = parameters[0];
+      const bool isScintillator = (total_points == 4);
+      const uint8_t type = ((it >> 28) & 0xF);
+
+      uint8_t ll = layer;
+      if (layer > 0) {
+        if (layer > 28) {
+          if (type == 8) {
+            continue;
+          }
+          ll -= 28;
+        } else {
+          if (type != 8) {
+            continue;
+          }
+        }
+
+        if (ll != ((it >> (isScintillator ? 17 : 20)) & 0x1F))
+          continue;
+      }
+
+      // Scintillator
+      if (isScintillator) {
+        const int total_vertices = 3 * total_points;
+
+        std::vector<float> pnts(24);
+        for (int i = 0; i < total_points; ++i) {
+          pnts[i * 3 + 0] = corners[i * 3];
+          pnts[i * 3 + 1] = corners[i * 3 + 1];
+          pnts[i * 3 + 2] = corners[i * 3 + 2];
+
+          pnts[(i * 3 + 0) + total_vertices] = corners[i * 3];
+          pnts[(i * 3 + 1) + total_vertices] = corners[i * 3 + 1];
+          pnts[(i * 3 + 2) + total_vertices] = corners[i * 3 + 2] + shapes[3];
+        }
+        boxset->AddBox(&pnts[0]);
+        boxset->DigitColor(energy * 255, 0, 255 - energy * 255);
+
+        h_box = true;
+      }
+      // Silicon
+      else {
+        const int offset = 9;
+
+        float centerX = (corners[6] + corners[6 + offset]) / 2;
+        float centerY = (corners[7] + corners[7 + offset]) / 2;
+        float radius = fabs(corners[6] - corners[6 + offset]) / 2;
+        hex_boxset->AddHex(TEveVector(centerX, centerY, corners[2]), radius, 90.0, shapes[3]);
+        hex_boxset->DigitColor(energy * 255, 0, 255 - energy * 255);
+
+        h_hex = true;
+      }
+    }
+  }
+
+  if (h_hex) {
+    hex_boxset->RefitPlex();
+
+    hex_boxset->CSCTakeAnyParentAsMaster();
+    if (!heatmap) {
+      hex_boxset->CSCApplyMainColorToMatchingChildren();
+      hex_boxset->CSCApplyMainTransparencyToMatchingChildren();
+      hex_boxset->SetMainColor(item()->modelInfo(iIndex).displayProperties().color());
+      hex_boxset->SetMainTransparency(item()->defaultDisplayProperties().transparency());
+    }
+    oItemHolder.AddElement(hex_boxset);
+  }
+
+  if (h_box) {
+    boxset->RefitPlex();
+
+    boxset->CSCTakeAnyParentAsMaster();
+    if (!heatmap) {
+      boxset->CSCApplyMainColorToMatchingChildren();
+      boxset->CSCApplyMainTransparencyToMatchingChildren();
+      boxset->SetMainColor(item()->modelInfo(iIndex).displayProperties().color());
+      boxset->SetMainTransparency(item()->defaultDisplayProperties().transparency());
+    }
+    oItemHolder.AddElement(boxset);
+  }
+}
+
+REGISTER_FWPROXYBUILDER(FWHGCalTriggerClusterProxyBuilder,
+                        l1t::HGCalMulticluster,
+                        "HGCal Trigger Cluster",
+                        FWViewType::kAll3DBits);

Fireworks/Calo/plugins/FWItemTriggerCellAccessors.cc

@@ -0,0 +1,31 @@
+// -*- C++ -*-
+//
+// Package:     Core
+// Class  :     FWItemCSCSegmentAccessor
+//
+// Implementation:
+//     An example of how to write a plugin based FWItemAccessorBase derived class.
+//
+// Original Author:  Giulio Eulisse
+//         Created:  Thu Feb 18 15:19:44 EDT 2008
+//
+
+// system include files
+#include <cassert>
+
+#include "TClass.h"
+
+// user include files
+#include "DataFormats/L1Trigger/interface/BXVector.h"
+#include "DataFormats/L1THGCal/interface/HGCalTriggerCell.h"
+#include "DataFormats/L1THGCal/interface/HGCalMulticluster.h"
+
+#include "Fireworks/Core/interface/FWItemRandomAccessor.h"
+
+REGISTER_TEMPLATE_FWITEMACCESSOR(BXVectorAccessor<BXVector<l1t::HGCalTriggerCell>>,
+                                 BXVector<l1t::HGCalTriggerCell>,
+                                 "HGCalTriggerCellCollectionAccessor");
+
+REGISTER_TEMPLATE_FWITEMACCESSOR(BXVectorAccessor<BXVector<l1t::HGCalMulticluster>>,
+                                 BXVector<l1t::HGCalMulticluster>,
+                                 "HGCalTriggerClusterCollectionAccessor");