bool TopMuonPairSelector::processEvent(cafe::Event &event)
{
//get pointer to statistics collector
StatPointer stat ;
event.get("StatPointer", stat) ;
Collection<TMBEMCluster> em_from(event.getCollection<TMBEMCluster>(_electronBranch.c_str()));
Collection<TMBMuon> mu_from(event.getCollection<TMBMuon>(_muonBranch.c_str()));
Collection<TMBMuon> all_muons( event.getMuons() );
if (em_from.size() == 0 && mu_from.size() == 0 && debug ) {
out() << "PAIR SELECTOR WARNING! Electron/Muon branch "
<< _electronBranch << "/" << _muonBranch << " is empty or is not existing! "
<< endl ;
return true ;
}
if (em_from.size() >=2 && debug) {
out() << "PAIR SELECTOR ERROR! 2 electrons found in branch \""
<< _electronBranch << "\". Considering only leading electron outside of a jet!"
<< endl ;
// return false ;
}
l_mom.SetPxPyPzE( 0 , 0 , 0 , 0 );
lepton_charge = 0;
jet_array = event.getCollection<TMBJet>(_jet_branch.c_str());
if( em_from.size() == 0 )
{
for( int i = 0 ; i < mu_from.size() ; i++ )
{
bool is_in_jet = false;
if( dR_cut > 0 )
{
for( int j = 0 ; j < jet_array.size() ; j++ )
{
if( do_dr_mujet_lead_muon && mu_from[i].DeltaR( jet_array[j] ) < dR_cut )
is_in_jet = true;
}
}
if( use_only_leading_lepton && i > 0 )
is_in_jet = true;
if( mu_from[i].Pt() > l_mom.Pt() && mu_from[i].charge() != 0 && !is_in_jet )
{
l_mom = mu_from[i];
l_track = const_cast<TMBTrack*>(mu_from[i].GetChargedTrack());
lepton_charge = mu_from[i].charge();
}
}
}
else
{
for( int i = 0 ; i < em_from.size() ; i++ )
{
if( em_from[i].Pt() > l_mom.Pt() && em_from[i].charge() != 0 )
{
bool common_track = false;
for( int j = 0 ; j < all_muons.size() ; j++ )
{
if( remove_common_track && ( all_muons[j].GetChargedTrack() && all_muons[j].isLoose() == 1 )
&& ( em_from[i].getPtrChp()->DeltaR( *all_muons[j].GetChargedTrack() ) < 1e-4 ) )
common_track = true;
}
if( use_only_leading_lepton && i > 0 )
common_track = true;
if( !common_track )
{
l_mom = em_from[i];
l_track = const_cast<TMBTrack*>(em_from[i].getPtrChp());
lepton_charge = em_from[i].charge();
}
}
}
}
if (_selectSameSign)
{
stat.EventSelected("Same charge selection") ;
}
else
{
stat.EventSelected("Opposite charge selection") ;
}
SelectUserObjects<TMBMuon>::processEvent(event);
return true;
}
