JET1Alg.cxx

Go to the documentation of this file.

/*
 *   Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
 */
 
#include "JET1Alg.h"
#include "GlobalSimulation/topoc_pu_type.h"
#include "GlobalSimulation/main_output.h"
#include "xAODCore/AuxContainerBase.h"
 
namespace GlobalSim {
  
    StatusCode JET1Alg::initialize() {
 
        CHECK( m_inputTowersKey.initialize() );
        CHECK( m_outputKey.initialize() );
 
        return StatusCode::SUCCESS;
    }
 
    StatusCode JET1Alg::execute(const EventContext& ctx) const {
 
        ATH_MSG_DEBUG("Building WTAConeJets");
    // Read the GlobalCellTowers
    auto towers = SG::makeHandle(m_inputTowersKey, ctx);
    CHECK(towers.isValid());
    auto nTowers = towers->size();
    ATH_MSG_DEBUG("Reading " << nTowers << " cell towers from xAOD::BaseContainer");
 
    std::vector<WTATrigObj> input_towers;
    for(size_t i = 0; i < nTowers; i++){
        Object<topoc_pu_type> tower( *towers->at(i) );
        WTATrigObj this_tower(tower.ptt.bits().to_ulong(), tower.eta.bits().to_ulong(), tower.phi.bits().to_ulong(), 0, i);
        input_towers.push_back(this_tower);
        }
    
    // Do the usual
    std::unique_ptr<WTAConeMaker> MyWTAConeMaker = std::make_unique<WTACone2PassMaker>();
    WTAParameters MyWTAParameters = WTAParameters(
                              8, // const_et_cut = 2 GeV
                              20, // seed_et_cut = 5 GeV
                              4 // jet_dR = 4
                              );
 
    MyWTAConeMaker->m_WTAConeMakerParameter = MyWTAParameters; // Pass the WTAConeParameters
    WTAConeParallelHelper wta_parallel_helper;
    wta_parallel_helper.SetBlockN(4);
    wta_parallel_helper.CreateBlocks(input_towers);
    wta_parallel_helper.RunParallelWTA(MyWTAConeMaker);
    wta_parallel_helper.CheckJetInCore();
    std::vector<WTAJet> WTAJetList = wta_parallel_helper.GetAllJets(); // Bitwise Jets
 
    //Write out as an xAOD::BaseContainer of SG::AuxElements
    auto jets = SG::makeHandle(m_outputKey, ctx);
    CHECK( jets.record( std::make_unique<xAOD::BaseContainer>(), std::make_unique<xAOD::AuxContainerBase>()) );
    
    for(const auto& WTAJet: WTAJetList){
      jets->push_back( std::make_unique<SG::AuxElement>() );
      auto jet = Object<main_output_type>( *jets->back());
      
      jet.ptt = std::clamp(static_cast<int>(WTAJet.pt()), 0, (1 << jet.ptt.width) - 1); 
      jet.eta = std::clamp(static_cast<int>(WTAJet.eta()), 0, (1 << jet.eta.width) - 1);
      jet.phi = std::clamp(static_cast<int>(WTAJet.phi()), 0, (1 << jet.phi.width) - 1);
      //AM: The widths are not encoded anywhere here? This would be needed/nice
      WTA4JetERingInfo ering_info = WTAJet.GetERingInfo();
      jet.ring_zero_ptt = std::clamp(static_cast<int>(ering_info.ring0_Et), 0, (1 << jet.ring_zero_ptt.width) - 1);
      jet.ring_one_ptt = std::clamp(static_cast<int>(ering_info.ring1_Et), 0, (1 << jet.ring_one_ptt.width) - 1);
      jet.ring_two_ptt = std::clamp(static_cast<int>(ering_info.ring2_Et), 0, (1 << jet.ring_two_ptt.width) - 1);
      jet.ring_three_ptt = std::clamp(static_cast<int>(ering_info.ring3_Et), 0, (1 << jet.ring_three_ptt.width) - 1);
      jet.ring_four_ptt = std::clamp(static_cast<int>(ering_info.ring4_Et), 0, (1 << jet.ring_four_ptt.width) - 1);
      jet.total_tobs =std::clamp(static_cast<int>(ering_info.total_TobN), 0, (1 << jet.total_tobs.width) - 1);
      jet.ring_one_tobs = std::clamp(static_cast<int>(ering_info.ring1_TobN), 0, (1 << jet.ring_one_tobs.width) - 1);
      jet.ring_two_tobs = std::clamp(static_cast<int>(ering_info.ring2_TobN), 0, (1 << jet.ring_two_tobs.width) - 1);
      jet.ring_three_tobs = std::clamp(static_cast<int>(ering_info.ring3_TobN), 0, (1 << jet.ring_three_tobs.width) - 1);
      jet.ring_four_tobs = std::clamp(static_cast<int>(ering_info.ring4_TobN), 0, (1 << jet.ring_four_tobs.width) - 1);
      // The WTA engine does not provide the flags yet, so they are written as
      // false. The fields still have to be filled: without them the aux variable
      // does not exist and packing the word throws. Note et_overflow is genuinely
      // set in the reference data, so this is a placeholder, not a final value.
      jet.flag_et_overflow      = false;
      jet.flag_error            = false;
      jet.flag_next_tob_same_et = false;
      jet.flag_truncate         = false;
    }
 
        return StatusCode::SUCCESS;
    }
 
 
}