.. _program_listing_file_Root_HLTJetGetter.cxx:

Program Listing for File HLTJetGetter.cxx
=========================================

|exhale_lsh| :ref:`Return to documentation for file <file_Root_HLTJetGetter.cxx>` (``Root/HLTJetGetter.cxx``)

.. |exhale_lsh| unicode:: U+021B0 .. UPWARDS ARROW WITH TIP LEFTWARDS

.. code-block:: cpp

   /******************************************
    *
    * This class gets HLT jets from the TDT and can be expanded to get other features
    *
    * Merlin Davies (merlin.davies@cern.ch)
    * Caterina Doglioni (caterina.doglioni@cern.ch)
    * John Alison (john.alison@cern.ch)
    *
    *
    ******************************************/
   
   // c++ include(s):
   #include <iostream>
   #include <vector>
   
   // EL include(s):
   #include <EventLoop/Job.h>
   #include <EventLoop/StatusCode.h>
   #include <EventLoop/Worker.h>
   
   // EDM include(s):
   #include "xAODEventInfo/EventInfo.h"
   #include "xAODJet/JetContainer.h"
   #include "xAODJet/JetAuxContainer.h"
   #include "xAODJet/Jet.h"
   
   // package include(s):
   #include "xAODAnaHelpers/HelperFunctions.h"
   #include "xAODAnaHelpers/HLTJetGetter.h"
   #include "TrigConfxAOD/xAODConfigTool.h"
   #include "TrigDecisionTool/TrigDecisionTool.h"
   
   // this is needed to distribute the algorithm to the workers
   ClassImp(HLTJetGetter)
   
   HLTJetGetter :: HLTJetGetter () :
   Algorithm("HLTJetGetter")
   {
   }
   
   
   EL::StatusCode HLTJetGetter :: setupJob (EL::Job& job)
   {
       ANA_MSG_INFO( "Calling setupJob");
       job.useXAOD ();
       xAOD::Init( "HLTJetGetter" ).ignore(); // call before opening first file
       return EL::StatusCode::SUCCESS;
   }
   
   
   
   EL::StatusCode HLTJetGetter :: histInitialize ()
   {
       ANA_CHECK( xAH::Algorithm::algInitialize());
       return EL::StatusCode::SUCCESS;
   }
   
   
   
   EL::StatusCode HLTJetGetter :: fileExecute ()
   {
       return EL::StatusCode::SUCCESS;
   }
   
   
   
   EL::StatusCode HLTJetGetter :: changeInput (bool /*firstFile*/)
   {
       return EL::StatusCode::SUCCESS;
   }
   
   EL::StatusCode HLTJetGetter :: initialize ()
   {
   
   
       ANA_MSG_INFO( "Initializing HLTJetGetter Interface... ");
   
       m_event = wk()->xaodEvent();
       m_store = wk()->xaodStore();
   
       //
       // Grab the TrigDecTool from the ToolStore
       //
   
       /*
       if ( asg::ToolStore::contains<Trig::TrigDecisionTool>( "TrigDecisionTool" ) ) {
           m_trigDecTool = asg::ToolStore::get<Trig::TrigDecisionTool>("TrigDecisionTool");
       } else {
           Info ("Initialize()", "the Trigger Decision Tool is not yet initialized...[%s]. Doing so now.", m_name.c_str());
           m_ownTDTAndTCT = true;
   
           m_trigConfTool = new TrigConf::xAODConfigTool( "xAODConfigTool" );
           ANA_CHECK( m_trigConfTool->initialize());
           ToolHandle< TrigConf::ITrigConfigTool > configHandle( m_trigConfTool );
   
           m_trigDecTool = new Trig::TrigDecisionTool( "TrigDecisionTool" );
           ANA_CHECK( m_trigDecTool->setProperty( "ConfigTool", configHandle ));
           ANA_CHECK( m_trigDecTool->setProperty( "TrigDecisionKey", "xTrigDecision" ));
           ANA_CHECK( m_trigDecTool->setProperty( "OutputLevel", MSG::ERROR));
           ANA_CHECK( m_trigDecTool->initialize());
           ANA_MSG_INFO( "Successfully configured Trig::TrigDecisionTool!");
       }
       */
   
       if(!m_trigDecTool_handle.isUserConfigured()){
         ANA_MSG_FATAL("A configured " << m_trigDecTool_handle.typeAndName() << " must have been previously created! Are you creating one in xAH::BasicEventSelection?" );
         return EL::StatusCode::FAILURE;
       }
   
       // If there is no InputContainer we must stop
       if ( m_inContainerName.empty() ) {
           ANA_MSG_ERROR( "InputContainer is empty!");
           return EL::StatusCode::FAILURE;
       }
   
       return EL::StatusCode::SUCCESS;
   }
   
   
   EL::StatusCode HLTJetGetter :: execute ()
   {
       ANA_MSG_DEBUG( "Getting HLT jets... ");
   
       //
       // Create the new container and its auxiliary store.
       //
       xAOD::JetContainer*     hltJets    = new xAOD::JetContainer();
       xAOD::JetAuxContainer*  hltJetsAux = new xAOD::JetAuxContainer();
       hltJets->setStore( hltJetsAux ); //< Connect the two
   
       //Retrieving jets via trigger decision tool:
       const Trig::ChainGroup * chainGroup = m_trigDecTool_handle->getChainGroup(m_triggerList); //Trigger list:
       
       std::vector<std::string> trigger_list = chainGroup->getListOfTriggers();
   
       auto chainFeatures = chainGroup->features(); //Gets features associated to chain defined above
   
       auto JetFeatureContainers = chainFeatures.containerFeature<xAOD::JetContainer>(m_inContainerName.c_str());
   
       ANA_CHECK( m_store->record( hltJets,    m_outContainerName));
       ANA_CHECK( m_store->record( hltJetsAux, m_outContainerName+"Aux."));
   
       for( auto fContainer : JetFeatureContainers ) {
           for( auto trigJet : *fContainer.cptr() ) {
               xAOD::Jet *Jet = new xAOD::Jet();
               *Jet = *trigJet;
               hltJets->push_back( Jet );
           }//end trigJet loop
       }//end feature container loop
       
       if(msgLvl(MSG::VERBOSE)) m_store->print();
   
       return EL::StatusCode::SUCCESS;
   }
   
   
   
   EL::StatusCode HLTJetGetter :: postExecute ()
   {
       ANA_MSG_DEBUG( "Calling postExecute");
       return EL::StatusCode::SUCCESS;
   }
   
   
   
   EL::StatusCode HLTJetGetter :: finalize ()
   {
       ANA_MSG_INFO( "Deleting tool instances...");
   
       // this is necessary because in most cases the pointer will be set to null
       // after deletion in BasicEventSelection, but it will not propagate here
       if ( m_ownTDTAndTCT ) {
         //if ( m_trigDecTool_handle )  { delete m_trigDecTool_handle; m_trigDecTool_handle = nullptr;  }
         if ( m_trigDecTool_handle.isInitialized() )  m_trigDecTool_handle->finalize();
         if ( m_trigConfTool ) {  delete m_trigConfTool; m_trigConfTool = nullptr; }
       }
   
       return EL::StatusCode::SUCCESS;
   }
   
   
   
   EL::StatusCode HLTJetGetter :: histFinalize ()
   {
       ANA_MSG_INFO( "Calling histFinalize");
       ANA_CHECK( xAH::Algorithm::algFinalize());
       return EL::StatusCode::SUCCESS;
   }