[ViewVC] Diff of: cvsroot/UserCode/MitPhysics/Utils/src/ElectronTools.cc

Comparing UserCode/MitPhysics/Utils/src/ElectronTools.cc (file contents):
Revision 1.49 by mingyang, Mon Oct 8 17:22:59 2012 UTC vs.
Revision 1.52 by pharris, Sat Jan 12 11:49:50 2013 UTC

+//--------------------------------------------------------------------------------------------------
-+
+Bool_t ElectronTools::PassConversionFilterPFAOD(const Electron *ele,
-+
+                                           const DecayParticleCol *conversions,
-+
+                                           const BaseVertex *vtx,
-+
+                                           UInt_t nWrongHitsMax,
-+
+                                           Double_t probMin,
-+
+                                           Double_t lxyMin,
-+
+                                           Bool_t matchCkf,
-+
+                                           Bool_t requireArbitratedMerged,
-+
+                                           Double_t trkptMin)
-+
+{
-+
-+
+  Bool_t isGoodConversion = kFALSE;
-+
-+
+  for (UInt_t ifc=0; ifc<conversions->GetEntries(); ifc++) {
-+
+    Bool_t ConversionMatchFound = kFALSE;
-+
+    for (UInt_t d=0; d<conversions->At(ifc)->NDaughters(); d++) {
-+
+      const ChargedParticle *pParticle = 0;
-+
+      pParticle = dynamic_cast<const ChargedParticle*>(conversions->At(ifc)->Daughter(d));
-+
+      if(pParticle == 0) continue;
-+
+      const Track* trk = 0;
-+
+      trk = pParticle->Trk();
-+
+      if(trk == 0) continue;
-+
+      if (ele->GsfTrk() == trk || (matchCkf && ele->TrackerTrk()==trk) ) {
-+
+        ConversionMatchFound = kTRUE;
-+
+        break;
-+
+      }
-+
+    }
-+
-+
+    // if match between the e-track and one of the conversion legs
-+
+    if (ConversionMatchFound == kTRUE){
-+
+      isGoodConversion =  (conversions->At(ifc)->Prob() > probMin) &&
-+
+        (!requireArbitratedMerged || conversions->At(ifc)->Quality().Quality(ConversionQuality::arbitratedMerged)) &&
-+
+        (conversions->At(ifc)->LxyCorrected(vtx) > lxyMin);
-+
-+
+      if (isGoodConversion == kTRUE) {
-+
+        for (UInt_t d=0; d<conversions->At(ifc)->NDaughters(); d++) {
-+
+          const ChargedParticle *pParticle = 0;
-+
+          pParticle = dynamic_cast<const ChargedParticle*>(conversions->At(ifc)->Daughter(d));
-+
+          if(pParticle == 0) continue;
-+
+          const Track* trk = 0;
-+
+          trk = pParticle->Trk();
-+
+          if(trk == 0) continue;
-+
+          if (trk) {
-+
+            if (trk->Pt()<trkptMin) isGoodConversion = kFALSE;
-+
+            const StableData *sd = dynamic_cast<const StableData*>
-+
+              (conversions->At(ifc)->DaughterDat(d));
-+
+            if (sd->NWrongHits() > nWrongHitsMax)
-+
+              isGoodConversion = kFALSE;
-+
+          } else {
-+
+            isGoodConversion = kFALSE;
-+
+          }
-+
+        }
-+
+      }
-+
+    }
-+
-+
+    if (isGoodConversion == kTRUE) break;
-+
-+
+  } // loop over all conversions
-+
-+
+  return !isGoodConversion;
-+
+}
-+
+//--------------------------------------------------------------------------------------------------
+Bool_t ElectronTools::PassD0Cut(const Electron *ele, const VertexCol *vertices, Double_t fD0Cut, Int_t nVertex)
+  Bool_t d0cut = kFALSE;
+  return true;
-+
+std::pair<Double_t,Double_t> ElectronTools::ComputeEPCombination( const Electron * ele, const float regression_energy,
-+
+                                                  const float regression_energy_error) {
-+
-+
+  enum Classification { GSF_ELECTRON_UNKNOWN=-1,
-+
+                        GSF_ELECTRON_GOLDEN=0,
-+
+                        GSF_ELECTRON_BIGBREM=1,
-+
+                        GSF_ELECTRON_BADTRACK=2,
-+
+                        GSF_ELECTRON_SHOWERING=3,
-+
+                        GSF_ELECTRON_GAP=4 } ;
-+
-+
+  float newEnergyError_ = regression_energy_error;
-+
+  float scEnergy = regression_energy;
-+
-+
+  int elClass = ele->Classification();
-+
-+
+  float trackMomentum  = ele->PIn() ;
-+
+  float errorTrackMomentum_ = 999. ;
-+
-+
+  // the electron's track momentum error was not available in bambu versions less than 029
-+
+  if(ele->TrackMomentumError() > 0)
-+
+    errorTrackMomentum_ = ele->TrackMomentumError();
-+
+  else if ( ele->GsfTrk()->PtErr() > 0)
-+
+    errorTrackMomentum_ = ele->GsfTrk()->PtErr()*cosh(ele->GsfTrk()->Eta());
-+
+  else
-+
+    assert(0);
-+
-+
+  float finalMomentum = ele->E(); // initial
-+
+  float finalMomentumError = 999.;
-+
-+
+  // first check for large errors
-+
-+
+  if (errorTrackMomentum_/trackMomentum > 0.5 && regression_energy_error/regression_energy <= 0.5) {
-+
+    finalMomentum = regression_energy;    finalMomentumError = regression_energy_error;
-+
+  }
-+
+  else if (errorTrackMomentum_/trackMomentum <= 0.5 && regression_energy_error/regression_energy > 0.5){
-+
+    finalMomentum = trackMomentum;  finalMomentumError = errorTrackMomentum_;
-+
+  }
-+
+  else if (errorTrackMomentum_/trackMomentum > 0.5 && regression_energy_error/regression_energy > 0.5){
-+
+    if (errorTrackMomentum_/trackMomentum < regression_energy_error/regression_energy) {
-+
+      finalMomentum = trackMomentum; finalMomentumError = errorTrackMomentum_;
-+
+    }
-+
+    else{
-+
+      finalMomentum = regression_energy; finalMomentumError = regression_energy_error;
-+
+    }
-+
+  }
-+
+  // then apply the combination algorithm
-+
+  else {
-+
+     // calculate E/p and corresponding error
-+
+    float eOverP = regression_energy / trackMomentum;
-+
+    float errorEOverP = sqrt(
-+
+                             (regression_energy_error/trackMomentum)*(regression_energy_error/trackMomentum) +
-+
+                             (regression_energy*errorTrackMomentum_/trackMomentum/trackMomentum)*
-+
+                             (regression_energy*errorTrackMomentum_/trackMomentum/trackMomentum));
-+
-+
-+
+    bool eleIsNotInCombination = false ;
-+
+    if ( (eOverP  > 1 + 2.5*errorEOverP) || (eOverP  < 1 - 2.5*errorEOverP) || (eOverP < 0.8) || (eOverP > 1.3) )
-+
+      { eleIsNotInCombination = true ; }
-+
+     if (eleIsNotInCombination)
-+
+       {
-+
+         if (eOverP > 1)
-+
+           { finalMomentum = regression_energy ; finalMomentumError = regression_energy_error ; }
-+
+         else
-+
+           {
-+
+             if (elClass == GSF_ELECTRON_GOLDEN)
-+
+               { finalMomentum = regression_energy; finalMomentumError = regression_energy_error; }
-+
+             if (elClass == GSF_ELECTRON_BIGBREM)
-+
+               {
-+
+                 if (regression_energy<36)
-+
+                   { finalMomentum = trackMomentum ; finalMomentumError = errorTrackMomentum_ ; }
-+
+                 else
-+
+                   { finalMomentum = regression_energy ; finalMomentumError = regression_energy_error ; }
-+
+               }
-+
+             if (elClass == GSF_ELECTRON_BADTRACK)
-+
+               { finalMomentum = regression_energy; finalMomentumError = regression_energy_error ; }
-+
+             if (elClass == GSF_ELECTRON_SHOWERING)
-+
+               {
-+
+                 if (regression_energy<30)
-+
+                   { finalMomentum = trackMomentum ; finalMomentumError = errorTrackMomentum_; }
-+
+                 else
-+
+                   { finalMomentum = regression_energy; finalMomentumError = regression_energy_error;}
-+
+               }
-+
+             if (elClass == GSF_ELECTRON_GAP)
-+
+               {
-+
+                 if (regression_energy<60)
-+
+                   { finalMomentum = trackMomentum ; finalMomentumError = errorTrackMomentum_ ; }
-+
+                 else
-+
+                   { finalMomentum = regression_energy; finalMomentumError = regression_energy_error ; }
-+
+               }
-+
+           }
-+
+       }
-+
+     else
-+
+       {
-+
+         // combination
-+
+         finalMomentum = (regression_energy/regression_energy_error/regression_energy_error + trackMomentum/errorTrackMomentum_/errorTrackMomentum_) /
-+
+           (1/regression_energy_error/regression_energy_error + 1/errorTrackMomentum_/errorTrackMomentum_);
-+
+         float finalMomentumVariance = 1 / (1/regression_energy_error/regression_energy_error + 1/errorTrackMomentum_/errorTrackMomentum_);
-+
+         finalMomentumError = sqrt(finalMomentumVariance);
-+
+       }
-+
+  }
-+
-+
+  std::pair<Double_t,Double_t> value;
-+
+  value.first = finalMomentum;
-+
+  value.second = finalMomentumError;
-+
+  return value;
-+
-+
+}

Diff Legend

-–
+Removed lines
-+
+Added lines
-<
+Changed lines
->
+Changed lines

Comparing UserCode/MitPhysics/Utils/src/ElectronTools.cc (file contents): Revision 1.49 by mingyang, Mon Oct 8 17:22:59 2012 UTC vs. Revision 1.52 by pharris, Sat Jan 12 11:49:50 2013 UTC

Diff Legend

Comparing UserCode/MitPhysics/Utils/src/ElectronTools.cc (file contents):
Revision 1.49 by mingyang, Mon Oct 8 17:22:59 2012 UTC vs.
Revision 1.52 by pharris, Sat Jan 12 11:49:50 2013 UTC