#include <MuonSeedPtExtractor.h>
Public Member Functions | |
MuonSeedPtExtractor (const edm::ParameterSet &) | |
Constructor with Parameter set and MuonServiceProxy. | |
virtual std::vector< double > | pT_extract (MuonTransientTrackingRecHit::ConstMuonRecHitPointer firstHit, MuonTransientTrackingRecHit::ConstMuonRecHitPointer secondHit) const |
void | setBeamSpot (const GlobalVector &gv) |
virtual | ~MuonSeedPtExtractor () |
Destructor. | |
Private Types | |
typedef std::map< std::string, std::vector< double > > | ParametersMap |
typedef std::map< std::string, std::vector< double > > | ScalesMap |
Private Member Functions | |
void | fillParametersForCombo (const std::string &name, const edm::ParameterSet &pset) |
void | fillScalesForCombo (const std::string &name, const edm::ParameterSet &pset) |
std::vector< double > | getPt (const std::vector< double > &vPara, double eta, double dPhi) const |
std::vector< double > | getPt (const std::vector< double > &vPara, double eta, double dPhi, const std::string &combination, const DTChamberId &outerDetId) const |
void | init (const edm::ParameterSet &par) |
int | stationCode (MuonTransientTrackingRecHit::ConstMuonRecHitPointer hit) const |
Private Attributes | |
bool | scaleDT_ |
GlobalVector | theBeamSpot |
ParametersMap | theParametersForCombo |
ScalesMap | theScalesForCombo |
Definition at line 15 of file MuonSeedPtExtractor.h.
typedef std::map<std::string, std::vector<double> > MuonSeedPtExtractor::ParametersMap [private] |
Definition at line 45 of file MuonSeedPtExtractor.h.
typedef std::map<std::string, std::vector<double> > MuonSeedPtExtractor::ScalesMap [private] |
Definition at line 46 of file MuonSeedPtExtractor.h.
MuonSeedPtExtractor::MuonSeedPtExtractor | ( | const edm::ParameterSet & | par | ) |
Constructor with Parameter set and MuonServiceProxy.
Definition at line 9 of file MuonSeedPtExtractor.cc.
References init().
: theBeamSpot(0.,0.,0.), scaleDT_( par.getParameter<bool>("scaleDT") ) { init(par); }
MuonSeedPtExtractor::~MuonSeedPtExtractor | ( | ) | [virtual] |
void MuonSeedPtExtractor::fillParametersForCombo | ( | const std::string & | name, |
const edm::ParameterSet & | pset | ||
) | [private] |
Definition at line 117 of file MuonSeedPtExtractor.cc.
References edm::ParameterSet::getParameter(), mergeVDriftHistosByStation::name, and theParametersForCombo.
Referenced by init().
{ theParametersForCombo[name] = pset.getParameter<std::vector<double> >(name); }
void MuonSeedPtExtractor::fillScalesForCombo | ( | const std::string & | name, |
const edm::ParameterSet & | pset | ||
) | [private] |
Definition at line 123 of file MuonSeedPtExtractor.cc.
References edm::ParameterSet::getParameter(), mergeVDriftHistosByStation::name, and theScalesForCombo.
Referenced by init().
{ theScalesForCombo[name] = pset.getParameter<std::vector<double> >(name); }
std::vector< double > MuonSeedPtExtractor::getPt | ( | const std::vector< double > & | vPara, |
double | eta, | ||
double | dPhi, | ||
const std::string & | combination, | ||
const DTChamberId & | outerDetId | ||
) | const [private] |
Definition at line 394 of file MuonSeedPtExtractor.cc.
References abs, h, theScalesForCombo, and DTChamberId::wheel().
{ double h = fabs(eta); double estPt = ( vPara[0] + vPara[1]*h + vPara[2]*h*h ) / dPhi; // changed by S.C. double estSPt = ( vPara[3] + vPara[4]*h + vPara[5]*h*h ) / dPhi; // scale the pt and spt , changed by S.C. int wheel = 0; if(combination[0] == 'D') { wheel = abs(outerDetId.wheel()); } std::ostringstream os; os << combination << "_" << wheel << "_scale"; ScalesMap::const_iterator scalesItr = theScalesForCombo.find(os.str()); if ( scalesItr != theScalesForCombo.end()) { double t1 = scalesItr->second[0]; double scaleFactor = 1./( 1. + t1/( estPt + 10. ) ) ; estSPt = estSPt * scaleFactor ; estPt = estPt * scaleFactor ; } // std::cout<<"estPt = "<<estPt<<std::endl; std::vector<double> paraPt ; paraPt.push_back( estPt ); paraPt.push_back( estSPt ) ; return paraPt ; }
std::vector< double > MuonSeedPtExtractor::getPt | ( | const std::vector< double > & | vPara, |
double | eta, | ||
double | dPhi | ||
) | const [private] |
Definition at line 381 of file MuonSeedPtExtractor.cc.
References h.
Referenced by pT_extract().
{ //std::cout<<" eta = "<<eta<<" dPhi = "<<dPhi<<" vPara[0] = "<<vPara[0]<<" vPara[1] = "<<vPara[1]<<" vPara[2] = "<<vPara[2]<<std::endl; double h = fabs(eta); double estPt = ( vPara[0] + vPara[1]*h + vPara[2]*h*h ) / dPhi; double estSPt = ( vPara[3] + vPara[4]*h + vPara[5]*h*h ) / dPhi; // std::cout<<"estPt = "<<estPt<<std::endl; std::vector<double> paraPt ; paraPt.push_back( estPt ); paraPt.push_back( estSPt ) ; return paraPt ; }
void MuonSeedPtExtractor::init | ( | const edm::ParameterSet & | par | ) | [private] |
Definition at line 17 of file MuonSeedPtExtractor.cc.
References fillParametersForCombo(), and fillScalesForCombo().
Referenced by MuonSeedPtExtractor().
{ // load pT seed parameters // DT combinations fillParametersForCombo("DT_12", par); fillParametersForCombo("DT_13", par); fillParametersForCombo("DT_14", par); fillParametersForCombo("DT_23", par); fillParametersForCombo("DT_24", par); fillParametersForCombo("DT_34", par); // CSC combinations fillParametersForCombo("CSC_01", par); fillParametersForCombo("CSC_12", par); fillParametersForCombo("CSC_02", par); fillParametersForCombo("CSC_13", par); fillParametersForCombo("CSC_03", par); fillParametersForCombo("CSC_14", par); fillParametersForCombo("CSC_23", par); fillParametersForCombo("CSC_24", par); fillParametersForCombo("CSC_34", par); // Overlap combinations fillParametersForCombo("OL_1213", par); fillParametersForCombo("OL_1222", par); fillParametersForCombo("OL_1232", par); fillParametersForCombo("OL_2213", par); fillParametersForCombo("OL_2222", par); // Single segments (CSC) fillParametersForCombo("SME_11", par); fillParametersForCombo("SME_12", par); fillParametersForCombo("SME_13", par); fillParametersForCombo("SME_21", par); fillParametersForCombo("SME_22", par); fillParametersForCombo("SME_31", par); fillParametersForCombo("SME_32", par); fillParametersForCombo("SME_41", par); fillParametersForCombo("SME_42", par); // Single segments (DT) fillParametersForCombo("SMB_10", par); fillParametersForCombo("SMB_11", par); fillParametersForCombo("SMB_12", par); fillParametersForCombo("SMB_20", par); fillParametersForCombo("SMB_21", par); fillParametersForCombo("SMB_22", par); fillParametersForCombo("SMB_30", par); fillParametersForCombo("SMB_31", par); fillParametersForCombo("SMB_32", par); fillScalesForCombo("CSC_01_1_scale", par); fillScalesForCombo("CSC_12_1_scale", par); fillScalesForCombo("CSC_12_2_scale", par); fillScalesForCombo("CSC_12_3_scale", par); fillScalesForCombo("CSC_13_2_scale", par); fillScalesForCombo("CSC_13_3_scale", par); fillScalesForCombo("CSC_14_3_scale", par); fillScalesForCombo("CSC_23_1_scale", par); fillScalesForCombo("CSC_23_2_scale", par); fillScalesForCombo("CSC_24_1_scale", par); fillScalesForCombo("CSC_34_1_scale", par); fillScalesForCombo("DT_12_1_scale", par); fillScalesForCombo("DT_12_2_scale", par); fillScalesForCombo("DT_13_1_scale", par); fillScalesForCombo("DT_13_2_scale", par); fillScalesForCombo("DT_14_1_scale", par); fillScalesForCombo("DT_14_2_scale", par); fillScalesForCombo("DT_23_1_scale", par); fillScalesForCombo("DT_23_2_scale", par); fillScalesForCombo("DT_24_1_scale", par); fillScalesForCombo("DT_24_2_scale", par); fillScalesForCombo("DT_34_1_scale", par); fillScalesForCombo("DT_34_2_scale", par); fillScalesForCombo("OL_1213_0_scale", par); fillScalesForCombo("OL_1222_0_scale", par); fillScalesForCombo("OL_1232_0_scale", par); fillScalesForCombo("OL_2213_0_scale", par); fillScalesForCombo("OL_2222_0_scale", par); fillScalesForCombo("SMB_10_0_scale", par); fillScalesForCombo("SMB_11_0_scale", par); fillScalesForCombo("SMB_12_0_scale", par); fillScalesForCombo("SMB_20_0_scale", par); fillScalesForCombo("SMB_21_0_scale", par); fillScalesForCombo("SMB_22_0_scale", par); fillScalesForCombo("SMB_30_0_scale", par); fillScalesForCombo("SMB_31_0_scale", par); fillScalesForCombo("SMB_32_0_scale", par); fillScalesForCombo("SME_11_0_scale", par); fillScalesForCombo("SME_12_0_scale", par); fillScalesForCombo("SME_13_0_scale", par); fillScalesForCombo("SME_21_0_scale", par); fillScalesForCombo("SME_22_0_scale", par); }
std::vector< double > MuonSeedPtExtractor::pT_extract | ( | MuonTransientTrackingRecHit::ConstMuonRecHitPointer | firstHit, |
MuonTransientTrackingRecHit::ConstMuonRecHitPointer | secondHit | ||
) | const [virtual] |
If firstHit and secondHit are the same, it calculates the pT from the phi bend of the single segment
Definition at line 129 of file MuonSeedPtExtractor.cc.
References abs, dPhi(), alignCSCRings::e, eta(), PV3DBase< T, PVType, FrameType >::eta(), Exception, getPt(), PV3DBase< T, PVType, FrameType >::perp(), PV3DBase< T, PVType, FrameType >::phi(), Pi, CSCDetId::ring(), relativeConstraints::ring, scaleDT_, mathSSE::sqrt(), DTChamberId::station(), CSCDetId::station(), stationCode(), AlCaHLTBitMon_QueryRunRegistry::string, theBeamSpot, theParametersForCombo, DTChamberId::wheel(), PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().
Referenced by MuonDTSeedFromRecHits::computePtWithoutVtx(), MuonDTSeedFromRecHits::computePtWithVtx(), MuonCSCSeedFromRecHits::createDefaultEndcapSeed(), SETSeedFinder::estimateMomentum(), MuonCSCSeedFromRecHits::makeSeed(), and MuonOverlapSeedFromRecHits::makeSeed().
{ GlobalPoint innerPoint = firstHit->globalPosition() - theBeamSpot; GlobalPoint outerPoint = secondHit->globalPosition() - theBeamSpot; MuonTransientTrackingRecHit::ConstMuonRecHitPointer innerHit = firstHit; MuonTransientTrackingRecHit::ConstMuonRecHitPointer outerHit = secondHit; // ways in which the hits could be in the wrong order if( (outerHit->isDT() && innerHit->isCSC()) || (outerHit->isDT() && innerHit->isDT() && (innerPoint.perp() > outerPoint.perp())) || (outerHit->isCSC() && innerHit->isCSC() && (fabs(innerPoint.z()) > fabs(outerPoint.z()))) ) { innerHit = secondHit; outerHit = firstHit; innerPoint = innerHit->globalPosition() - theBeamSpot; outerPoint = outerHit->globalPosition() - theBeamSpot; } double phiInner = innerPoint.phi(); double phiOuter = outerPoint.phi(); double etaInner = innerPoint.eta(); double etaOuter = outerPoint.eta(); //std::cout<<" inner pos = "<< innerPoint << " phi eta " << phiInner << " " << etaInner << std::endl; //std::cout<<" outer pos = "<< outerPoint << " phi eta " << phiOuter << " " << etaOuter << std::endl; //double thetaInner = firstHit->globalPosition().theta(); // if some of the segments is missing r-phi measurement then we should // use only the 4D phi estimate (also use 4D eta estimate only) // the direction is not so important (it will be corrected) /* bool firstOK = (4==allValidSets[iSet][firstMeasurement]->hit->dimension()); bool lastOK = (4==allValidSets[iSet][lastMeasurement]->hit->dimension()); if(!(firstOK * lastOK)){ if(!firstOK){ } if(!firstOK){ } } */ double dPhi = phiInner - phiOuter; if(dPhi < -TMath::Pi()){ dPhi += 2*TMath::Pi(); } else if(dPhi > TMath::Pi()){ dPhi -= 2*TMath::Pi(); } int sign = 1; if ( dPhi< 0.) { dPhi = -dPhi; sign = -1; } if (dPhi < 1.0e-6){ dPhi = 1.0e-6; } double eta = fabs(etaOuter);// what if it is 2D segment? use inner? std::vector <int> stationCoded(2); stationCoded[0] = stationCoded[1] = 999; DetId detId_inner = innerHit->hit()->geographicalId(); DetId detId_outer = outerHit->hit()->geographicalId(); stationCoded[0] = stationCode(innerHit); stationCoded[1] = stationCode(outerHit); std::ostringstream os0 ; std::ostringstream os1 ; os0 << abs(stationCoded[0]); os1 << abs(stationCoded[1]); //std::cout<<" st1 = "<<stationCoded[0]<<" st2 = "<<stationCoded[1]<<std::endl; //std::cout<<" detId_inner = "<<detId_inner.rawId()<<" detId_outer = "<< detId_outer.rawId()<<std::endl; std::string combination = "0"; std::string init_combination = combination; bool singleSegment = false; //if(detId_first == detId_second){ if( stationCoded[0] == stationCoded[1]){ // single segment - DT or CSC singleSegment = true; //eta = innerPoint.eta(); GlobalVector gv = innerHit->globalDirection(); double gvPerpPos = gv.x()*gv.x() + gv.y()*gv.y(); if (gvPerpPos < 1e-32) gvPerpPos = 1e-32; gvPerpPos=sqrt(gvPerpPos); // Psi is angle between the segment origin and segment direction // Use dot product between two vectors to get Psi in global x-y plane double cosDpsi = (gv.x()*innerPoint.x() + gv.y()*innerPoint.y()); if (cosDpsi!=0){ cosDpsi /= sqrt(innerPoint.x()*innerPoint.x() + innerPoint.y()*innerPoint.y()); cosDpsi /= gvPerpPos; cosDpsi = cosDpsi > 1 ? 1 : cosDpsi; cosDpsi = cosDpsi < -1 ? -1 : cosDpsi; } double axb = ( innerPoint.x()*gv.y() ) - ( innerPoint.y()*gv.x() ) ; sign = (axb < 0.) ? 1 : -1; double dpsi = fabs(acos(cosDpsi)) ; if ( dpsi > TMath::Pi()/2.) { dpsi = TMath::Pi() - dpsi; } if (fabs(dpsi) < 0.00005) { dpsi = 0.00005; } dPhi = dpsi; if(innerHit->isDT()) { DTChamberId dtCh(detId_inner); std::ostringstream os; os << dtCh.station() << abs(dtCh.wheel()); combination = "SMB_"+os.str(); } else if(innerHit->isCSC()) { CSCDetId cscId(detId_inner); std::ostringstream os; int ring = cscId.ring(); if(ring == 4) ring = 1; os << cscId.station() << ring; combination = "SME_"+os.str(); } else { throw cms::Exception("MuonSeedPtExtractor") << "Bad hit DetId"; } } else{ if(stationCoded[0]<0){ if(stationCoded[1]<0){ // DT-DT combination = "DT_" + os0.str() + os1.str(); } else{ // DT-CSC eta = fabs(etaInner); if(-1==stationCoded[0]){ switch (stationCoded[1]){ case 1: combination = "OL_1213"; break; case 2: combination = "OL_1222"; break; case 3: combination = "OL_1232"; break; default: // can not be break; } } else if (-2==stationCoded[0]){ if(1==stationCoded[1]){ combination = "OL_2213"; } else{ // can not be (not coded?) combination = "OL_2222";// in case } } else{ // can not be } } } else{ if(stationCoded[1]<0){ // CSC-DT // can not be } else{ // CSC-CSC combination = "CSC_" + os0.str() + os1.str(); if("CSC_04" == combination){ combination = "CSC_14"; } } } } std::vector<double> pTestimate(2);// //std::cout<<" combination = "<<combination<<std::endl; if(init_combination!=combination){ //std::cout<<" combination = "<<combination<<" eta = "<<eta<<" dPhi = "<<dPhi<<std::endl; ParametersMap::const_iterator parametersItr = theParametersForCombo.find(combination); if(parametersItr == theParametersForCombo.end()) { // edm::LogWarning("RecoMuon|MuonSeedGenerator|MuonSeedPtExtractor") << "Cannot find parameters for combo " << combination; edm::LogWarning("BadSegmentCombination") << "Cannot find parameters for combo " << combination; pTestimate[0] = pTestimate[1] = 100; // throw cms::Exception("MuonSeedPtEstimator") << "Cannot find parameters for combo " << combination; } else { if(scaleDT_ && outerHit->isDT() ) { pTestimate = getPt(parametersItr->second, eta, dPhi, combination, detId_outer); } else { pTestimate = getPt(parametersItr->second, eta, dPhi); } if(singleSegment){ pTestimate[0] = fabs(pTestimate[0]); pTestimate[1] = fabs(pTestimate[1]); } pTestimate[0] *= double(sign); } } else{ // often a MB3 - ME1/3 seed pTestimate[0] = pTestimate[1] = 100; // hmm } // std::cout<<" pTestimate[0] = "<<pTestimate[0]<<" pTestimate[1] = "<<pTestimate[1]<<std::endl; /* MuonRecHitContainer_clusters[cluster][iHit+1]->isDT()); if(specialCase){ DTChamberId dtCh(detId); DTChamberId dtCh_2(detId_2); specialCase = (dtCh.station() == dtCh_2.station()); } */ //return vPara; return pTestimate; }
void MuonSeedPtExtractor::setBeamSpot | ( | const GlobalVector & | gv | ) | [inline] |
Definition at line 30 of file MuonSeedPtExtractor.h.
References theBeamSpot.
Referenced by MuonSeedVFinder::setBeamSpot().
{theBeamSpot = gv;}
int MuonSeedPtExtractor::stationCode | ( | MuonTransientTrackingRecHit::ConstMuonRecHitPointer | hit | ) | const [private] |
Definition at line 359 of file MuonSeedPtExtractor.cc.
References query::result, CSCDetId::ring(), DTChamberId::station(), and CSCDetId::station().
Referenced by pT_extract().
{ DetId detId(hit->hit()->geographicalId()); int result = -999; if(hit->isDT() ){ DTChamberId dtCh(detId); //std::cout<<"first (DT) St/W/S = "<<dtCh.station()<<"/"<<dtCh.wheel()<<"/"<<dtCh.sector()<<"/"<<std::endl; result = -1 * dtCh.station(); } else if( hit->isCSC() ){ CSCDetId cscID(detId); //std::cout<<"first (CSC) E/S/R/C = "<<cscID.endcap()<<"/"<<cscID.station()<<"/"<<cscID.ring()<<"/"<<cscID.chamber()<<std::endl; result = cscID.station(); if(result == 1 && (1 == cscID.ring() || 4 == cscID.ring()) ) result = 0; } else if(hit->isRPC()){ } return result; }
bool MuonSeedPtExtractor::scaleDT_ [private] |
Definition at line 50 of file MuonSeedPtExtractor.h.
Referenced by pT_extract().
GlobalVector MuonSeedPtExtractor::theBeamSpot [private] |
Definition at line 49 of file MuonSeedPtExtractor.h.
Referenced by pT_extract(), and setBeamSpot().
Definition at line 47 of file MuonSeedPtExtractor.h.
Referenced by fillParametersForCombo(), and pT_extract().
Definition at line 48 of file MuonSeedPtExtractor.h.
Referenced by fillScalesForCombo(), and getPt().