Matching between GEMChambermatch and GEMRechit only allowed when they're at the same etapartition.

Author: jo100sun

RecoMuon/MuonIdentification/plugins/MuonIdProducer.cc

@@ -166,6 +166,7 @@ MuonIdProducer::MuonIdProducer(const edm::ParameterSet& iConfig)
   //create mesh holder
   meshAlgo_ = std::make_unique<MuonMesh>(iConfig.getParameter<edm::ParameterSet>("arbitrationCleanerOptions"));
 
+  gemgeomToken_ = esConsumes<GEMGeometry, MuonGeometryRecord>();
   edm::InputTag rpcHitTag("rpcRecHits");
   rpcHitToken_ = consumes<RPCRecHitCollection>(rpcHitTag);
 
@@ -274,6 +275,7 @@ void MuonIdProducer::init(edm::Event& iEvent, const edm::EventSetup& iSetup) {
     iEvent.getByToken(glbQualToken_, glbQualHandle_);
   if (selectHighPurity_)
     iEvent.getByToken(pvToken_, pvHandle_);
+  gemgeom = &iSetup.getData(gemgeomToken_);
 }
 
 reco::Muon MuonIdProducer::makeMuon(edm::Event& iEvent,
@@ -296,6 +298,24 @@ reco::Muon MuonIdProducer::makeMuon(edm::Event& iEvent,
   return aMuon;
 }
 
+bool MuonIdProducer::checkBounds(const GeomDet* geomDet,
+                                       const GlobalPoint& global_position,
+                                       const float bordercut) {
+  const TrapezoidalPlaneBounds* bounds = dynamic_cast<const TrapezoidalPlaneBounds*>(&geomDet->surface().bounds());
+  LocalPoint localPoint = geomDet->surface().toLocal(global_position);
+  float wideWidth = bounds->width();
+  float narrowWidth = 2.f * bounds->widthAtHalfLength() - wideWidth;
+  float length = bounds->length();
+  float tangent = (wideWidth - narrowWidth) / (2.f * length);
+  float halfWidthAtY = tangent * localPoint.y() + 0.25f * (narrowWidth + wideWidth);
+  float distanceY = std::abs(localPoint.y()) - 0.5f * length;
+  float distanceX = std::abs(localPoint.x()) - halfWidthAtY;
+  if (distanceX < bordercut && distanceY < bordercut) {
+    return true;
+  }
+  return false;
+}
+
 reco::CaloMuon MuonIdProducer::makeCaloMuon(const reco::Muon& muon) {
   LogTrace("MuonIdentification") << "Creating a CaloMuon from a Muon";
 
@@ -1121,9 +1141,23 @@ void MuonIdProducer::fillMuonId(edm::Event& iEvent,
         gemHitMatch.x = gemRecHit.localPosition().x();
         gemHitMatch.mask = 0;
         gemHitMatch.bx = gemRecHit.BunchX();
-
+        
+        const GeomDet* geomDet = gemgeom->idToDetUnit(chamber.id);
+        const GlobalPoint& global_position = geomDet->toGlobal(lPos);
+        bool GEMmatched = false;
+        int ieta = 0;
+        if (const GEMChamber* gemChamber = dynamic_cast<const GEMChamber*>(geomDet)) {
+          for (const GEMEtaPartition* eta_partition : gemChamber->etaPartitions()) {
+            ieta = eta_partition->id().ieta();
+            if (ieta == gemRecHit.gemId().ieta()) {
+              if (checkBounds(eta_partition, global_position, 2)) {
+                GEMmatched = true;
+              }
+            }
+          }
+        }
         const double absDx = std::abs(gemRecHit.localPosition().x() - chamber.tState.localPosition().x());
-        if (absDx <= 5 or absDx * absDx <= 16 * localError.xx())
+        if ((absDx <= 5 or absDx * absDx <= 16 * localError.xx()) && GEMmatched)
           matchedChamber.gemHitMatches.push_back(gemHitMatch);
       }
 

RecoMuon/MuonIdentification/plugins/MuonIdProducer.h

@@ -61,6 +61,7 @@
 #include "RecoMuon/MuonIdentification/interface/MuonIdTruthInfo.h"
 #include "RecoMuon/MuonIdentification/interface/MuonArbitrationMethods.h"
 #include "DataFormats/Common/interface/ValueMap.h"
+#include "Geometry/GEMGeometry/interface/GEMGeometry.h"
 
 class MuonMesh;
 class MuonKinkFinder;
@@ -79,6 +80,8 @@ class MuonIdProducer : public edm::stream::EDProducer<> {
   static double sectorPhi(const DetId& id);
 
   static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
+  
+  bool checkBounds(const GeomDet* geomDet, const GlobalPoint& global_position, const float bordercut);
 
 private:
   void fillMuonId(edm::Event&,
@@ -281,5 +284,8 @@ class MuonIdProducer : public edm::stream::EDProducer<> {
 
   bool arbClean_;
   std::unique_ptr<MuonMesh> meshAlgo_;
+  edm::ESGetToken<GEMGeometry, MuonGeometryRecord> gemgeomToken_;
+  const GEMGeometry* gemgeom;
+  double GEM_edgecut_;
 };
 #endif