Add ME0 Trigger Object (Stub) and Producer

Author: woohyeonHeo

L1Trigger/L1TGEM/interface/ME0StubAlgoChamber.h

@@ -10,7 +10,6 @@
 
 namespace l1t {
   namespace me0 {
-    // map<int, vector<vector<uint64_t>>> cross_partition_cancellation(vector<vector<uint64_t>> segments);
     std::vector<std::vector<ME0StubPrimitive>> cross_partition_cancellation(
         std::vector<std::vector<ME0StubPrimitive>>& segments, int cross_part_seg_width);
     std::vector<ME0StubPrimitive> process_chamber(const std::vector<std::vector<UInt192>>& chamber_data,

L1Trigger/L1TGEM/interface/ME0StubPrimitive.h

@@ -1,5 +1,5 @@
-#ifndef DataFormats_ME0StubPrimitive_H
-#define DataFormats_ME0StubPrimitive_H
+#ifndef L1Trigger_L1TGEM_ME0StubPrimitive_H
+#define L1Trigger_L1TGEM_ME0StubPrimitive_H
 
 #include <vector>
 #include <cstdint>

L1Trigger/L1TGEM/plugins/ME0StubBuilder.h

@@ -21,28 +21,15 @@
 #include "FWCore/ParameterSet/interface/ParameterSet.h"
 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
 
-// using namespace l1t::me0;
-
 class ME0StubAlgorithmBase;
 
 class ME0StubBuilder {
 public:
-  /** Configure the algorithm via ctor.
-     * Receives ParameterSet percolated down from EDProducer
-     * which owns this Builder.
-     */
   explicit ME0StubBuilder(const edm::ParameterSet&);
-  /// Destructor
   ~ME0StubBuilder();
 
-  /** Find stubs in each ensemble of 6 GEM layers, build ME0Stub's ,
-     *  and fill into output collection.
-     */
   void build(const GEMPadDigiCollection* paddigis, ME0StubCollection& oc);
 
-  /** Cache pointer to geometry _for current event_
-     */
-
   static void fillDescription(edm::ParameterSetDescription& descriptions);
 
 private:

L1Trigger/L1TGEM/plugins/ME0StubProducer.cc

@@ -20,35 +20,29 @@
 #include "DataFormats/GEMDigi/interface/ME0StubCollection.h"
 #include "DataFormats/GEMDigi/interface/ME0Stub.h"
 
-// #include "RecoLocalMuon/GEMSegment/plugins/GEMSegmentBuilder.h"
 #include "L1Trigger/L1TGEM/plugins/ME0StubBuilder.h"
 #include "Geometry/Records/interface/MuonGeometryRecord.h"
 
-// using namespace l1t::me0;
-
 class ME0StubProducer : public edm::stream::EDProducer<> {
 public:
-  /// Constructor
   explicit ME0StubProducer(const edm::ParameterSet&);
-  /// Destructor
   ~ME0StubProducer() override {}
-  /// Produce the ME0Stub collection
+
   void produce(edm::Event&, const edm::EventSetup&) override;
 
   static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
 
 private:
-  int iev;  // events through
+  int iev;
   edm::EDGetTokenT<GEMPadDigiCollection> theGEMPadDigiToken;
   std::unique_ptr<ME0StubBuilder> segmentBuilder_;
   edm::ESGetToken<GEMGeometry, MuonGeometryRecord> gemGeomToken_;
 };
 
 ME0StubProducer::ME0StubProducer(const edm::ParameterSet& ps) : iev(0) {
   theGEMPadDigiToken = consumes<GEMPadDigiCollection>(ps.getParameter<edm::InputTag>("InputCollection"));
-  segmentBuilder_ = std::make_unique<ME0StubBuilder>(ps);  // pass on the Parameter Set
+  segmentBuilder_ = std::make_unique<ME0StubBuilder>(ps);
   gemGeomToken_ = esConsumes<GEMGeometry, MuonGeometryRecord>();
-  // register what this produces
   produces<ME0StubCollection>();
 }
 
@@ -62,17 +56,13 @@ void ME0StubProducer::fillDescriptions(edm::ConfigurationDescriptions& descripti
 void ME0StubProducer::produce(edm::Event& ev, const edm::EventSetup& setup) {
   LogDebug("ME0StubProducer") << "start producing segments for " << ++iev << "th event with GEM data";
 
-  // get the collection of GEMDigi
   edm::Handle<GEMPadDigiCollection> gemPadDigis;
   ev.getByToken(theGEMPadDigiToken, gemPadDigis);
 
-  // create empty collection of Segments
   auto oc = std::make_unique<ME0StubCollection>();
 
-  // fill the collection
-  segmentBuilder_->build(gemPadDigis.product(), *oc);  //@@ FILL oc
+  segmentBuilder_->build(gemPadDigis.product(), *oc);
 
-  // put collection in event
   ev.put(std::move(oc));
 }
 

L1Trigger/L1TGEM/src/ME0StubAlgoChamber.cc

@@ -120,7 +120,6 @@ std::vector<ME0StubPrimitive> l1t::me0::process_chamber(
 
   // pick the best N outputs from each partition
   for (int i = 0; i < static_cast<int>(segments.size()); ++i) {
-    // segments[i] = segment_sorter(segments[i], config.num_outputs);
     segment_sorter(segments[i], config.num_outputs);
   }
 
@@ -133,13 +132,11 @@ std::vector<ME0StubPrimitive> l1t::me0::process_chamber(
   }
   joined_segments.push_back(segments[14]);
   for (int i = 0; i < static_cast<int>(joined_segments.size()); ++i) {
-    // joined_segments[i] = segment_sorter(joined_segments[i], config.num_outputs);
     segment_sorter(joined_segments[i], config.num_outputs);
   }
 
   // concatenate together all of the segments, sort them, and pick the best N outputs
   std::vector<ME0StubPrimitive> concatenated = concatVector(joined_segments);
-  // concatenated = segment_sorter(concatenated, config.num_outputs);
   segment_sorter(concatenated, config.num_outputs);
 
   return concatenated;

L1Trigger/L1TGEM/src/ME0StubAlgoPartition.cc

@@ -56,8 +56,6 @@ std::vector<ME0StubPrimitive> l1t::me0::cancel_edges(
     */
 
   std::vector<ME0StubPrimitive> canceled_segements = segments;
-  // std::vector<ME0StubPrimitive> canceled_segements;
-  // std::copy(segments.begin(), segments.end(), std::back_inserter(canceled_segements));
   std::vector<int> comps;
 
   bool ghost;

L1Trigger/L1TGEM/src/ME0StubAlgoPatUnitMux.cc

@@ -8,11 +8,9 @@ uint64_t l1t::me0::parse_data(const UInt192& data, int strip, int max_span) {
   if (strip < max_span / 2 + 1) {
     data_shifted = data << (max_span / 2 - strip);
     parsed_data = (data_shifted & UInt192(0xffffffffffffffff >> (64 - max_span))).to_ullong();
-    // parsed_data = (data_shifted & UInt192(pow(2,max_span)-1)).to_ullong();
   } else {
     data_shifted = data >> (strip - max_span / 2);
     parsed_data = (data_shifted & UInt192(0xffffffffffffffff >> (64 - max_span))).to_ullong();
-    // parsed_data = (data_shifted & UInt192(pow(2,max_span)-1)).to_ullong();
   }
   return parsed_data;
 }

L1Trigger/L1TGEM/src/ME0StubPrimitive.cc

@@ -35,23 +35,23 @@ void ME0StubPrimitive::update_quality() {
   }
 }
 void ME0StubPrimitive::fit(int max_span) {
-  if (id!=0) {
+  if (id != 0) {
     std::vector<double> tmp;
-    for (double cent : centroid) {
-      tmp.push_back(cent-(max_span/2+1));
+    tmp.reserve(centroid.size()); for (double cent : centroid) {
+      tmp.push_back(cent - (max_span / 2 + 1));
     }
     std::vector<double> x;
     std::vector<double> centroids;
-    for (uint32_t i=0; i < tmp.size(); ++i) {
-      if (tmp[i] != -1*(max_span/2+1)) {
-        x.push_back(i-2.5);
+    for (uint32_t i = 0; i < tmp.size(); ++i) {
+      if (tmp[i] != -1 * (max_span / 2 + 1)) {
+        x.push_back(i - 2.5);
         centroids.push_back(tmp[i]);
       }
     }
     std::vector<double> fit = llse_fit(x, centroids);
     bend_ang = fit[0];
     substrip = fit[1];
-    mse      = fit[2];
+    mse = fit[2];
   }
 }
 std::vector<double> ME0StubPrimitive::llse_fit(const std::vector<double>& x, const std::vector<double>& y) {