12 #ifndef TrigInDetAnalysisUtils_T_AnalysisConfig_H
13 #define TrigInDetAnalysisUtils_T_AnalysisConfig_H
20 #include "GaudiKernel/IToolSvc.h"
21 #include "GaudiKernel/ITHistSvc.h"
39 #ifdef XAODTRACKING_TRACKPARTICLE_H
58 template<
class Prov
ider>
70 const std::string& testChainName,
const std::string& testType,
const std::string& testKey,
71 const std::string& referenceChainName,
const std::string& referenceType,
const std::string& referenceKey,
72 const std::string& selectionChainName,
const std::string& selectionType,
const std::string& selectionKey,
101 std::vector<std::string> testChainNames; testChainNames.push_back(testChainName);
102 std::vector<std::string> referenceChainNames; referenceChainNames.push_back(referenceChainName);
103 std::vector<std::string> selectionChainNames; selectionChainNames.push_back(selectionChainName);
106 std::vector<std::string> testTypes; testTypes.push_back(testType);
107 std::vector<std::string> referenceTypes; referenceTypes.push_back(referenceType);
108 std::vector<std::string> selectionTypes; selectionTypes.push_back(selectionType);
112 std::vector<std::string> referenceKeys; referenceKeys.push_back(referenceKey);
113 std::vector<std::string> selectionKeys; selectionKeys.push_back(selectionKey);
116 std::vector<TrackFilter*> testFilters; testFilters.push_back(testFilter);
117 std::vector<TrackFilter*> referenceFilters; referenceFilters.push_back(referenceFilter);
118 std::vector<TrackFilter*> selectionFilters; selectionFilters.push_back(selectionFilter);
130 const std::string& testChainName,
const std::string& testType,
const std::string& testKey,
131 const std::string& referenceChainName,
const std::string& referenceType,
const std::string& referenceKey,
160 std::vector<std::string> testChainNames; testChainNames.push_back(testChainName);
161 std::vector<std::string> referenceChainNames; referenceChainNames.push_back(referenceChainName);
162 std::vector<std::string> selectionChainNames; selectionChainNames.push_back(
"NONE");
165 std::vector<std::string> testTypes; testTypes.push_back(testType);
166 std::vector<std::string> referenceTypes; referenceTypes.push_back(referenceType);
167 std::vector<std::string> selectionTypes; selectionTypes.push_back(
"");
171 std::vector<std::string> referenceKeys; referenceKeys.push_back(referenceKey);
172 std::vector<std::string> selectionKeys; selectionKeys.push_back(
"");
175 std::vector<TrackFilter*> testFilters; testFilters.push_back(testFilter);
176 std::vector<TrackFilter*> referenceFilters; referenceFilters.push_back(referenceFilter);
177 std::vector<TrackFilter*> selectionFilters;
226 if ( !
m_provider ) std::cerr <<
"ERROR T_AnalysisConfig::execute() called without initialising" << std::endl;
297 template<
class Collection>
298 std::pair< typename Collection::const_iterator, typename Collection::const_iterator >
300 const std::string&
key=
"" ) {
308 std::string key_collection =
key;
309 std::string key_tename =
"";
310 size_t pos = key_collection.find(
"/");
311 if (
pos!=std::string::npos ) {
313 key_tename =
key.substr( 0,
pos );
316 std::pair<
typename Collection::const_iterator,
317 typename Collection::const_iterator > itrpair;
322 const std::string* keyStr =
m_provider->evtStore()->keyToString(roi_link.
key(), checkCLID);
323 m_provider->msg(
MSG::DEBUG) <<
"Requesting range over (" <<
key <<
") associated to ROI from " << (keyStr ==
nullptr ?
"UNKNOWN" : *keyStr) <<
endmsg;
325 itrpair = (*m_tdt)->associateToEventView( handle, roi_link );
342 bool select( std::vector<TIDA::Vertex>& vertices,
348 for ( ; vtxitr!=vtx_end ; ++vtxitr ) {
349 if ( (*vtxitr)->vertexType()!=0 ) {
359 (*vtxitr)->nTrackParticles(),
361 (*vtxitr)->chiSquared(),
362 (*vtxitr)->numberDoF() ) );
371 bool select( std::vector<TIDA::Vertex>& vertices,
373 const std::string&
key=
"" ) {
380 if ( vtx_itrpair.first == vtx_itrpair.second ) {
385 m_provider->msg(
MSG::DEBUG) <<
"\txAOD::VertexContainer found with size " << (vtx_itrpair.second - vtx_itrpair.first)
388 return select( vertices, vtx_itrpair.first, vtx_itrpair.second );
394 bool select( std::vector<TIDA::Vertex>& vertices,
const std::string&
key=
"" ) {
400 if (
retrieve( xaodVtxCollection,
key ).isFailure()) {
401 m_provider->msg(MSG::WARNING) <<
"xAOD vertex container not found with key " <<
key <<
endmsg;
405 if ( xaodVtxCollection!=0 ) {
409 return select( vertices, xaodVtxCollection->
begin(), xaodVtxCollection->
end() );
415 for ( ; vtxitr != xaodVtxCollection->
end(); vtxitr++ ) {
416 if ( (*vtxitr)->nTrackParticles()>0 && (*vtxitr)->vertexType()!=0 ) {
424 (*vtxitr)->nTrackParticles(),
426 (*vtxitr)->chiSquared(),
427 (*vtxitr)->numberDoF() ) );
440 template<
class Collection>
444 const std::string&
key=
"" ) {
453 std::string key_collection =
key;
454 std::string key_tename =
"";
455 size_t pos = key_collection.find(
"/");
456 if (
pos!=std::string::npos ) {
458 key_tename =
key.substr( 0,
pos );
461 std::pair<
typename Collection::const_iterator,
462 typename Collection::const_iterator > itrpair;
467 const std::string* keyStr =
m_provider->evtStore()->keyToString(roi_link.
key(), checkCLID);
468 m_provider->msg(
MSG::DEBUG) <<
"Requesting range over (" <<
key <<
") associated to ROI from " << (keyStr ==
nullptr ?
"UNKNOWN" : *keyStr) <<
endmsg;
470 itrpair = (*m_tdt)->associateToEventView( handle, roi_link );
472 if ( itrpair.first != itrpair.second ) {
473 selector->selectTracks( itrpair.first, itrpair.second );
486 template<
class Collection>
491 std::string key_collection =
key;
492 std::string key_tename =
"";
493 size_t pos = key_collection.find(
"/");
494 if (
pos!=std::string::npos ) {
496 key_tename =
key.substr( 0,
pos );
499 std::vector< Trig::Feature<Collection> > trackcollections = citr->get<
Collection>( key_collection, TrigDefs::alsoDeactivateTEs, key_tename );
500 if ( !trackcollections.empty() ) {
502 for (
unsigned ifeat=0 ; ifeat<trackcollections.size() ; ifeat++ ) {
504 if ( !trackfeature.
empty() ) {
507 selector->selectTracks( trigtracks );
522 template<
class Collection>
527 std::vector< Trig::Feature<Collection> > trackcollections = citr->get<
Collection>(
key, TrigDefs::alsoDeactivateTEs );
528 if ( !trackcollections.empty() ) {
536 for (
unsigned ifeat=0 ; ifeat<trackcollections.size() ; ifeat++ ) {
544 std::vector< const Collection* > collectionVector;
551 for (
unsigned iv=collectionVector.size() ; iv-- ; ) {
555 if (
index!=iv )
continue;
558 selector->selectTracks( collectionVector[iv], truthmap );
572 template<
class Collection>
576 if (
m_provider->evtStore()->template contains<Collection>(
key ) ) {
581 collection = handle.
cptr();
582 return StatusCode::SUCCESS;
585 return StatusCode::FAILURE;
590 template<
class Collection>
607 template<
class Collection>
610 if (
key.empty() )
return false;
611 if ( !
m_provider->evtStore()->template contains<Collection>(
key ) )
return false;
615 if ( !(
sc.isSuccess() && collection ) )
return false;
620 if ( collection->size() == 0 ) {
625 selector->selectTracks( collection );
633 template<
class Collection>
636 std::vector<double>
v;
639 if (
m_provider->evtStore()->template contains<Collection>(
key ) ) {
641 if(
sc.isSuccess() && collection ) {
644 typename Collection::const_iterator trackitr = collection->begin();
645 typename Collection::const_iterator trackend = collection->end();
646 if ( trackitr!=trackend ) {
648 v[0] = (*trackitr)->vx();
649 v[1] = (*trackitr)->vy();
650 v[2] = (*trackitr)->vz();
662 template<
class Collection>
664 std::vector< Trig::Feature<Collection> > trackcollections = citr->get<
Collection>(
key, TrigDefs::alsoDeactivateTEs );
665 std::vector<double>
v;
666 if ( !trackcollections.empty() ) {
669 for (
unsigned ifeat=0 ; ifeat<trackcollections.size() ; ifeat++ ) {
672 if ( !trackfeature.
empty() ) {
677 typename Collection::const_iterator trackitr = trigtracks->begin();
678 typename Collection::const_iterator trackend = trigtracks->end();
679 if ( trackitr!=trackend ) {
681 v[0] = (*trackitr)->vx();
682 v[1] = (*trackitr)->vy();
683 v[2] = (*trackitr)->vz();
704 std::vector<TrackTrigObject>* elevec=0,
706 bool raw_track=
false,
708 # ifdef XAODTRACKING_TRACKPARTICLE_H
709 const std::string& containerName =
"Electrons"
711 const std::string& containerName =
"ElectronAODCollection"
719 # ifdef XAODTRACKING_TRACKPARTICLE_H
728 if( !
m_provider->evtStore()->template contains<Container>(containerName) ) {
729 m_provider->msg(MSG::WARNING) <<
"Error No Electron Container " << containerName <<
" !" <<
endmsg;
734 if(
sc.isFailure() || !container ) {
735 m_provider->msg(MSG::WARNING) <<
"Error retrieving container: " << containerName <<
" !" <<
endmsg;
741 Container::const_iterator elec = container->begin();
742 Container::const_iterator elec_end = container->end();
744 for( ; elec!=elec_end ; ++elec ){
755 bool good_electron =
false;
756 # ifdef XAODTRACKING_TRACKPARTICLE_H
764 long unsigned eleid = (
unsigned long)(&eleduff) ;
774 else trk_added = selectorRef.
selectTrack( (*elec)->trackParticle() );
776 if (trk_added) eleobj.
addChild( selectorRef.
tracks().back()->id() );
777 if (elevec) elevec->push_back( eleobj );
781 return selectorRef.
tracks().size();
791 # ifdef XAODTRACKING_TRACKPARTICLE_H
792 const std::string& containerName =
"Muons"
794 const std::string& containerName =
"StacoMuonCollection"
798 # ifdef XAODTRACKING_TRACKPARTICLE_H
810 if( !
m_provider->evtStore()->template contains<Container>(containerName) ) {
811 m_provider->msg(MSG::WARNING) <<
"Error No MuonCollection" << containerName <<
" !" <<
endmsg;
816 if(
sc.isFailure() || !container ) {
817 m_provider->msg(MSG::WARNING) <<
"Error retrieving " << containerName <<
" !" <<
endmsg;
821 auto muon = container->begin();
822 auto muon_end = container->end();
827 # ifdef XAODTRACKING_TRACKPARTICLE_H
838 return selectorRef.
tracks().size();
847 std::vector<TrackTrigObject>* tauvec=0,
849 int requireNtracks=0,
850 double EtCutOffline=0,
851 # ifdef XAODTRACKING_TRACKPARTICLE_H
852 const std::string& containerName =
"TauJets"
854 const std::string& containerName =
"TauRecContainer"
858 # ifdef XAODTRACKING_TRACKPARTICLE_H
872 if ( !
m_provider->evtStore()->template contains<Container>(containerName)) {
878 if (
sc != StatusCode::SUCCESS) {
879 m_provider->msg(MSG::WARNING) <<
" Offline tau retrieval not successful" <<
endmsg;
883 Container::const_iterator tau = container->begin();
884 Container::const_iterator tau_end = container->end();
887 for ( ; tau!=tau_end ; ++tau ) {
889 # ifdef XAODTRACKING_TRACKPARTICLE_H
892 # ifndef XAODTAU_VERSIONS_TAUJET_V3_H
893 int N = (*tau)->nTracks();
902 unsigned N = (*tau)->numTrack();
906 bool good_tau =
false;
907 # ifdef XAODTRACKING_TRACKPARTICLE_H
917 long unsigned tauid = (
unsigned long)(&duff) ;
925 bool trk_added =
false;
927 for (
unsigned i=
N ;
i-- ; ) {
928 # ifdef XAODTAU_TAUTRACK_H
930 std::vector< ElementLink<xAOD::TrackParticleContainer> > alink = (*tau)->track(
i)->trackLinks();
934 for (
size_t ilink=0 ; ilink<alink.size() ; ilink++ ) {
943 if ( trk_added ) tauobj.
addChild( selectorRef.
tracks().back()->id() );
945 if ( tauvec ) tauvec->push_back( tauobj );
949 return selectorRef.
tracks().size();
959 ToolHandle<Trig::TrigDecisionTool>*
m_tdt;
969 std::vector< std::vector<std::string> >
m_types;
970 std::vector< std::vector<std::string> >
m_keys;
1010 #endif // TrigInDetAnalysisUtils_T_AnalysisConfig_H