CMS 3D CMS Logo

List of all members | Public Member Functions | Private Member Functions | Private Attributes
CandOneToOneDeltaRMatcher Class Reference
Inheritance diagram for CandOneToOneDeltaRMatcher:
edm::EDProducer edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

 CandOneToOneDeltaRMatcher (const edm::ParameterSet &)
 
 ~CandOneToOneDeltaRMatcher ()
 
- Public Member Functions inherited from edm::EDProducer
 EDProducer ()
 
ModuleDescription const & moduleDescription () const
 
virtual ~EDProducer ()
 
- Public Member Functions inherited from edm::ProducerBase
void callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
 
std::vector< edm::ProductResolverIndex > const & indiciesForPutProducts (BranchType iBranchType) const
 
 ProducerBase ()
 
void registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
 
std::function< void(BranchDescription const &)> registrationCallback () const
 used by the fwk to register list of products More...
 
void resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)
 
virtual ~ProducerBase () noexcept(false)
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfoconsumesInfo () const
 
 EDConsumerBase ()
 
 EDConsumerBase (EDConsumerBase const &)=delete
 
 EDConsumerBase (EDConsumerBase &&)=default
 
ProductResolverIndexAndSkipBit indexFrom (EDGetToken, BranchType, TypeID const &) const
 
void itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
void itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
std::vector< ProductResolverIndexAndSkipBit > const & itemsToGetFrom (BranchType iType) const
 
void labelsForToken (EDGetToken iToken, Labels &oLabels) const
 
void modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
 
EDConsumerBase const & operator= (EDConsumerBase const &)=delete
 
EDConsumerBaseoperator= (EDConsumerBase &&)=default
 
bool registeredToConsume (ProductResolverIndex, bool, BranchType) const
 
bool registeredToConsumeMany (TypeID const &, BranchType) const
 
void updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
 
virtual ~EDConsumerBase () noexcept(false)
 

Private Member Functions

std::vector< int > AlgoBruteForce (int, int)
 
std::vector< int > AlgoSwitchMethod (int, int)
 
double lenght (const std::vector< int > &)
 
void produce (edm::Event &, const edm::EventSetup &) override
 

Private Attributes

std::string algoMethod_
 
std::vector< std::vector< float > > AllDist
 
edm::EDGetTokenT< reco::CandidateViewmatchedToken_
 
edm::EDGetTokenT< reco::CandidateViewsourceToken_
 

Additional Inherited Members

- Public Types inherited from edm::EDProducer
typedef EDProducer ModuleType
 
- Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList TypeLabelList
 
- Public Types inherited from edm::EDConsumerBase
typedef ProductLabels Labels
 
- Static Public Member Functions inherited from edm::EDProducer
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &descriptions)
 
- Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > consumes (edm::InputTag const &tag)
 
EDGetToken consumes (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken consumes (TypeToGet const &id, edm::InputTag const &tag)
 
ConsumesCollector consumesCollector ()
 Use a ConsumesCollector to gather consumes information from helper functions. More...
 
template<typename ProductType , BranchType B = InEvent>
void consumesMany ()
 
void consumesMany (const TypeToGet &id)
 
template<BranchType B>
void consumesMany (const TypeToGet &id)
 
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > mayConsume (edm::InputTag const &tag)
 
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 

Detailed Description

Definition at line 18 of file CandOneToOneDeltaRMatcher.cc.

Constructor & Destructor Documentation

CandOneToOneDeltaRMatcher::CandOneToOneDeltaRMatcher ( const edm::ParameterSet cfg)

Definition at line 60 of file CandOneToOneDeltaRMatcher.cc.

60  :
61  sourceToken_( consumes<CandidateView>( cfg.getParameter<InputTag>( "src" ) ) ),
62  matchedToken_( consumes<CandidateView>( cfg.getParameter<InputTag>( "matched" ) ) ),
63  algoMethod_( cfg.getParameter<string>( "algoMethod" ) ) {
64  produces<CandViewMatchMap>("src2mtc");
65  produces<CandViewMatchMap>("mtc2src");
66 }
T getParameter(std::string const &) const
edm::EDGetTokenT< reco::CandidateView > sourceToken_
edm::EDGetTokenT< reco::CandidateView > matchedToken_
CandOneToOneDeltaRMatcher::~CandOneToOneDeltaRMatcher ( )

Definition at line 68 of file CandOneToOneDeltaRMatcher.cc.

68  {
69 }

Member Function Documentation

vector< int > CandOneToOneDeltaRMatcher::AlgoBruteForce ( int  nMin,
int  nMax 
)
private

Definition at line 220 of file CandOneToOneDeltaRMatcher.cc.

References lenght(), and stdcomb::next_combination().

Referenced by produce().

220  {
221 
222  vector<int> ca;
223  vector<int> cb;
224  vector<int> bestCB;
225  float totalDeltaR=0;
226  float BestTotalDeltaR=1000;
227 
228  for(int i1=0; i1<nMax; i1++) ca.push_back(i1);
229  for(int i1=0; i1<nMin; i1++) cb.push_back(i1);
230 
231  do
232  {
233  //do your processing on the new combination here
234  for(int cnt=0;cnt<TMath::Factorial(nMin); cnt++)
235  {
236  totalDeltaR = lenght(cb);
237  if ( totalDeltaR < BestTotalDeltaR ) {
238  BestTotalDeltaR = totalDeltaR;
239  bestCB=cb;
240  }
241  next_permutation( cb.begin() , cb.end() );
242  }
243  }
244  while(next_combination( ca.begin() , ca.end() , cb.begin() , cb.end() ));
245 
246  return bestCB;
247 }
double lenght(const std::vector< int > &)
bool next_combination(BidIt n_begin, BidIt n_end, BidIt r_begin, BidIt r_end)
Definition: combination.h:22
vector< int > CandOneToOneDeltaRMatcher::AlgoSwitchMethod ( int  nMin,
int  nMax 
)
private

Definition at line 260 of file CandOneToOneDeltaRMatcher.cc.

References AllDist, and DEFINE_FWK_MODULE.

Referenced by produce().

260  {
261 
262  vector<int> bestCB;
263  for(int i1=0; i1<nMin; i1++) {
264  int minInd=0;
265  for(int i2=1; i2<nMax; i2++) if( AllDist[i1][i2] < AllDist[i1][minInd] ) minInd = i2;
266  bestCB.push_back(minInd);
267  }
268 
269  bool inside = true;
270  while( inside ) {
271  inside = false;
272  for(int i1=0;i1<nMin;i1++){
273  for(int i2=i1+1;i2<nMin;i2++){
274  if ( bestCB[i1] == bestCB[i2] ) {
275  inside = true;
276  if ( AllDist[i1][(bestCB[i1])] <= AllDist[i2][(bestCB[i2])]) {
277  AllDist[i2][(bestCB[i2])]= 1000;
278  int minInd=0;
279  for(int i3=1; i3<nMax; i3++) if( AllDist[i2][i3] < AllDist[i2][minInd] ) minInd = i3;
280  bestCB[i2]= minInd;
281  } else {
282  AllDist[i1][(bestCB[i1])]= 1000;
283  int minInd=0;
284  for(int i3=1; i3<nMax; i3++) if( AllDist[i1][i3] < AllDist[i1][minInd] ) minInd = i3;
285  bestCB[i1]= minInd;
286  }
287  } // End if
288  }
289  }
290  } // End while
291 
292  return bestCB;
293 
294 }
std::vector< std::vector< float > > AllDist
double CandOneToOneDeltaRMatcher::lenght ( const std::vector< int > &  best)
private

Definition at line 201 of file CandOneToOneDeltaRMatcher.cc.

References AllDist.

Referenced by AlgoBruteForce().

201  {
202  double myLenght=0;
203  int row=0;
204  for(vector<int>::const_iterator it=best.begin(); it!=best.end(); it++ ) {
205  myLenght+=AllDist[row][*it];
206  row++;
207  }
208  return myLenght;
209 }
std::vector< std::vector< float > > AllDist
void CandOneToOneDeltaRMatcher::produce ( edm::Event evt,
const edm::EventSetup es 
)
overrideprivate

Definition at line 71 of file CandOneToOneDeltaRMatcher.cc.

References AlgoBruteForce(), algoMethod_, AlgoSwitchMethod(), AllDist, edm::View< T >::begin(), EnergyCorrector::c, HiRegitMuonDetachedTripletStep_cff::DeltaR, edm::View< T >::end(), Exception, edm::Event::getByToken(), createfilelist::int, electronMatch_cfi::matched, matchedToken_, hpstanc_transforms::max, min(), eostools::move(), edm::Event::put(), edm::View< T >::refAt(), edm::View< T >::size(), source, and sourceToken_.

71  {
72 
75  evt.getByToken( sourceToken_, source ) ;
76  evt.getByToken( matchedToken_, matched ) ;
77 
78  edm::LogVerbatim("CandOneToOneDeltaRMatcher") << "======== Source Collection =======";
79  for( CandidateView::const_iterator c = source->begin(); c != source->end(); ++c ) {
80  edm::LogVerbatim("CandOneToOneDeltaRMatcher") << " pt source " << c->pt() << " " << c->eta() << " " << c->phi() << endl;
81  }
82  edm::LogVerbatim("CandOneToOneDeltaRMatcher") << "======== Matched Collection =======";
83  for( CandidateView::const_iterator c = matched->begin(); c != matched->end(); ++c ) {
84  edm::LogVerbatim("CandOneToOneDeltaRMatcher") << " pt source " << c->pt() << " " << c->eta() << " " << c->phi() << endl;
85  }
86 
87  const int nSrc = source->size();
88  const int nMtc = matched->size();
89 
90  const int nMin = min( source->size() , matched->size() );
91  const int nMax = max( source->size() , matched->size() );
92  if( nMin < 1 ) return;
93 
94  if( nSrc <= nMtc ) {
95  for(CandidateView::const_iterator iSr = source->begin();
96  iSr != source->end();
97  iSr++) {
98  vector <float> tempAllDist;
99  for(CandidateView::const_iterator iMt = matched->begin();
100  iMt != matched->end();
101  iMt++) {
102  tempAllDist.push_back(DeltaR( iSr->p4() , iMt->p4() ) );
103  }
104  AllDist.push_back(tempAllDist);
105  tempAllDist.clear();
106  }
107  } else {
108  for(CandidateView::const_iterator iMt = matched->begin();
109  iMt != matched->end();
110  iMt++) {
111  vector <float> tempAllDist;
112  for(CandidateView::const_iterator iSr = source->begin();
113  iSr != source->end();
114  iSr++) {
115  tempAllDist.push_back(DeltaR( iSr->p4() , iMt->p4() ) );
116  }
117  AllDist.push_back(tempAllDist);
118  tempAllDist.clear();
119  }
120  }
121 
122  /*
123  edm::LogVerbatim("CandOneToOneDeltaRMatcher") << "======== The DeltaR Matrix =======";
124  for(int m0=0; m0<nMin; m0++) {
125  // for(int m1=0; m1<nMax; m1++) {
126  edm::LogVerbatim("CandOneToOneDeltaRMatcher") << setprecision(2) << fixed << (m1 AllDist[m0][m1] ;
127  //}
128  edm::LogVerbatim("CandOneToOneDeltaRMatcher") << "\n";
129  }
130  */
131 
132  // Loop size if Brute Force
133  int nLoopToDo = (int) ( TMath::Factorial(nMax) / TMath::Factorial(nMax - nMin) );
134  edm::LogVerbatim("CandOneToOneDeltaRMatcher") << "nLoop:" << nLoopToDo << endl;
135  edm::LogVerbatim("CandOneToOneDeltaRMatcher") << "Choosen Algo is:" << algoMethod_ ;
136  vector<int> bestCB;
137 
138  // Algo is Brute Force
139  if( algoMethod_ == "BruteForce") {
140 
141  bestCB = AlgoBruteForce(nMin,nMax);
142 
143  // Algo is Switch Method
144  } else if( algoMethod_ == "SwitchMode" ) {
145 
146  bestCB = AlgoSwitchMethod(nMin,nMax);
147 
148  // Algo is Brute Force if nLoop < 10000
149  } else if( algoMethod_ == "MixMode" ) {
150 
151  if( nLoopToDo < 10000 ) {
152  bestCB = AlgoBruteForce(nMin,nMax);
153  } else {
154  bestCB = AlgoSwitchMethod(nMin,nMax);
155  }
156 
157  } else {
158  throw cms::Exception("OneToOne Constructor") << "wrong matching method in ParameterSet";
159  }
160 
161  for(int i1=0; i1<nMin; i1++) edm::LogVerbatim("CandOneToOneDeltaRMatcher") << "min: " << i1 << " " << bestCB[i1] << " " << AllDist[i1][bestCB[i1]];
162 
163 /*
164  auto matchMapSrMt = std::make_unique<CandViewMatchMap>(CandViewMatchMap::ref_type( CandidateRefProd( source ),
165  CandidateRefProd( matched ) ) );
166  auto matchMapMtSr = std::make_unique<CandViewMatchMap>(CandViewMatchMap::ref_type( CandidateRefProd( matched ),
167  CandidateRefProd( source ) ) );
168 */
169 
170  auto matchMapSrMt = std::make_unique<CandViewMatchMap>();
171  auto matchMapMtSr = std::make_unique<CandViewMatchMap>();
172 
173  for( int c = 0; c != nMin; c ++ ) {
174  if( source->size() <= matched->size() ) {
175  matchMapSrMt->insert( source ->refAt(c ), matched->refAt(bestCB[c] ) );
176  matchMapMtSr->insert( matched->refAt(bestCB[c] ), source ->refAt(c ) );
177  } else {
178  matchMapSrMt->insert( source ->refAt(bestCB[c] ), matched->refAt(c ) );
179  matchMapMtSr->insert( matched->refAt(c ), source ->refAt(bestCB[c] ) );
180  }
181  }
182 
183 /*
184  for( int c = 0; c != nMin; c ++ ) {
185  if( source->size() <= matched->size() ) {
186  matchMapSrMt->insert( CandidateRef( source, c ), CandidateRef( matched, bestCB[c] ) );
187  matchMapMtSr->insert( CandidateRef( matched, bestCB[c] ), CandidateRef( source, c ) );
188  } else {
189  matchMapSrMt->insert( CandidateRef( source, bestCB[c] ), CandidateRef( matched, c ) );
190  matchMapMtSr->insert( CandidateRef( matched, c ), CandidateRef( source, bestCB[c] ) );
191  }
192  }
193 */
194  evt.put(std::move(matchMapSrMt), "src2mtc" );
195  evt.put(std::move(matchMapMtSr), "mtc2src" );
196 
197  AllDist.clear();
198 }
edm::EDGetTokenT< reco::CandidateView > sourceToken_
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:122
std::vector< std::vector< float > > AllDist
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:460
size_type size() const
std::vector< int > AlgoSwitchMethod(int, int)
std::vector< int > AlgoBruteForce(int, int)
RefToBase< value_type > refAt(size_type i) const
const_iterator begin() const
T min(T a, T b)
Definition: MathUtil.h:58
boost::indirect_iterator< typename seq_t::const_iterator > const_iterator
Definition: View.h:86
const_iterator end() const
static std::string const source
Definition: EdmProvDump.cc:43
def move(src, dest)
Definition: eostools.py:510
edm::EDGetTokenT< reco::CandidateView > matchedToken_

Member Data Documentation

std::string CandOneToOneDeltaRMatcher::algoMethod_
private

Definition at line 31 of file CandOneToOneDeltaRMatcher.cc.

Referenced by produce().

std::vector< std::vector<float> > CandOneToOneDeltaRMatcher::AllDist
private

Definition at line 30 of file CandOneToOneDeltaRMatcher.cc.

Referenced by AlgoSwitchMethod(), lenght(), and produce().

edm::EDGetTokenT<reco::CandidateView> CandOneToOneDeltaRMatcher::matchedToken_
private

Definition at line 29 of file CandOneToOneDeltaRMatcher.cc.

Referenced by produce().

edm::EDGetTokenT<reco::CandidateView> CandOneToOneDeltaRMatcher::sourceToken_
private

Definition at line 28 of file CandOneToOneDeltaRMatcher.cc.

Referenced by produce().