UZH

Physik-Institut

CMS Wiki Pages

User Tools

Site Tools

Trace: • mc • mtcatkfec • mtcafull • pyroot_tfile • todo • pyroot_ttree • cgalloni
users:clange

Differences

This shows you the differences between two versions of the page.

Link to this comparison view

Both sides previous revisionPrevious revision
users:clange [2013/09/20 21:10] – clangeusers:clange [2014/05/07 18:25] (current) – clange
Line 20: Line 20:
  
   * [[http://www.physik.unizh.ch/seminars/tpseminar/|Seminar on Particle and Astrophysics (Autumn Semester 2013)]]   * [[http://www.physik.unizh.ch/seminars/tpseminar/|Seminar on Particle and Astrophysics (Autumn Semester 2013)]]
-  * + 
 + 
 +==== Code ==== 
 + 
 +Sorting a vector based on another vector: http://stackoverflow.com/questions/236172/how-do-i-sort-a-stdvector-by-the-values-of-a-different-stdvector 
 + 
 + struct tlvordering { 
 + bool operator ()(std::pair<size_t, std::vector<TLorentzVector>::const_iterator> const& a, std::pair<size_t, std::vector<TLorentzVector>::const_iterator> const& b) { 
 + return (*(a.second)).Pt() > (b.second->Pt()); 
 + } 
 + }; 
 +   
 +   
 + template <typename T> 
 + std::vector<T> sort_from_tlvref( 
 + std::vector<T> const& in, 
 + std::vector<std::pair<size_t, std::vector<TLorentzVector>::const_iterator> > const& reference 
 + ) { 
 + std::vector<T> ret(in.size()); 
 +   
 + size_t const size = in.size(); 
 + for (size_t i = 0; i < size; ++i) 
 + ret[i] = in[reference[i].first]; 
 +   
 + return ret; 
 + } 
 +   
 +   
 + void sortJetsByPt(std::vector<unsigned int>& selJets, // index vector 
 + std::vector<TLorentzVector>& selJetsP4, 
 + std::vector<int>& selJetsSVMassBins, 
 + std::vector<double>& selJetsSVMassSum, 
 + std::vector< std::vector<bool> >& selJetsBTags, 
 + std::vector<JetFlavor::Code>& selJetsFlav) { 
 +   
 + // make a pair of index order and TLorentzVector 
 + std::vector<std::pair<size_t, std::vector<TLorentzVector>::const_iterator> > order(selJetsP4.size()); 
 + size_t n = 0; 
 + for (std::vector<TLorentzVector>::const_iterator it = selJetsP4.begin(); it != selJetsP4.end(); ++it, ++n) { 
 + order[n] = std::make_pair(n, it); 
 + } 
 + // sort order pair by jet Pt 
 + sort(order.begin(), order.end(), tlvordering()); 
 + // user order pair as reference for sorting all vectors 
 + selJets = sort_from_tlvref(selJets, order); 
 + selJetsP4 = sort_from_tlvref(selJetsP4, order); 
 + selJetsSVMassBins = sort_from_tlvref(selJetsSVMassBins, order); 
 + selJetsSVMassSum = sort_from_tlvref(selJetsSVMassSum, order); 
 + selJetsBTags = sort_from_tlvref(selJetsBTags, order); 
 + selJetsFlav = sort_from_tlvref(selJetsFlav, order); 
 +  
 + // loop over jets and change order 
 + //   for (size_t iJ = 0; iJ < selJets.size()-1; ++iJ) { 
 + //      
 + //     if (selJetsP4[iJ].Pt() < selJetsP4[iJ+1].Pt()) { 
 + //       jetOrder[iJ] = iJ+1 
 + //     } 
 + //   } 
 +   
 + }
users/clange.1379704233.txt.gz · Last modified: by clange

Page Tools

Except where otherwise noted, content on this wiki is licensed under the following license: CC Attribution-Share Alike 4.0 International
CC Attribution-Share Alike 4.0 International Donate Powered by PHP Valid HTML5 Valid CSS Driven by DokuWiki