AFPToFSiTAlgorithm.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*
*
*   AFPToFSiTAlgorithm
*
*
*/
 
#include "Run3AFPMonitoring/AFPToFSiTAlgorithm.h"
#include "xAODForward/AFPSiHit.h"
#include "xAODForward/AFPStationID.h"
#include "xAODForward/AFPToFHit.h"
#include "xAODForward/AFPTrack.h"
 
#include <utility>
#include <vector>
 
AFPToFSiTAlgorithm::AFPToFSiTAlgorithm( const std::string& name, ISvcLocator* pSvcLocator )
  : AthMonitorAlgorithm( name, pSvcLocator ), m_afpSiHitContainerKey( "AFPSiHitContainer" ), m_afpToFHitContainerKey( "AFPToFHitContainer" ), m_afpTrackContainerKey( "AFPTrackContainer" ) {
    declareProperty( "AFPSiHitContainer", m_afpSiHitContainerKey );
    declareProperty( "AFPToFHitContainer", m_afpToFHitContainerKey );
    declareProperty( "AFPTrackContainer", m_afpTrackContainerKey );
    
}
 
AFPToFSiTAlgorithm::~AFPToFSiTAlgorithm() {}
 
StatusCode AFPToFSiTAlgorithm::initialize() {
 
    using namespace Monitored;
 
    m_GroupToFSiTCorr = buildToolMap<int>(m_tools, "AFPToFSiTTool", m_planes);
 
    // We must declare to the framework in initialize what SG objects we are going to use
    SG::ReadHandleKey<xAOD::AFPSiHitContainer> afpSiHitContainerKey( "AFPSiHits" );
    ATH_CHECK( m_afpSiHitContainerKey.initialize() );
    SG::ReadHandleKey<xAOD::AFPToFHitContainer> afpToFHitContainerKey( "AFPToFHits" );
    ATH_CHECK( m_afpToFHitContainerKey.initialize() );
    SG::ReadHandleKey<xAOD::AFPTrackContainer> afpTrackContainerKey( "AFPTracks" );
    ATH_CHECK( m_afpTrackContainerKey.initialize() );
 
    return AthMonitorAlgorithm::initialize();
}
 
StatusCode AFPToFSiTAlgorithm::fillHistograms( const EventContext& ctx ) const {
    SG::ReadHandle<xAOD::AFPSiHitContainer> afpSiHitContainer( m_afpSiHitContainerKey, ctx );
    if ( !afpSiHitContainer.isValid() ) {
        ATH_MSG_WARNING( "evtStore() does not contain hits collection with name " << m_afpSiHitContainerKey );
        return StatusCode::SUCCESS;
    }
    ATH_CHECK( afpSiHitContainer.initialize() );
 
    SG::ReadHandle<xAOD::AFPTrackContainer> afpTrackContainer( m_afpTrackContainerKey, ctx );
    if ( !afpTrackContainer.isValid() ) {
        ATH_MSG_WARNING( "evtStore() does not contain hits collection with name " << m_afpTrackContainerKey );
        return StatusCode::SUCCESS;
    }
    ATH_CHECK( afpTrackContainer.initialize() );
 
    SG::ReadHandle<xAOD::AFPToFHitContainer> afpToFHitContainer( m_afpToFHitContainerKey, ctx );
    if ( !afpToFHitContainer.isValid() ) {
        ATH_MSG_WARNING( "evtStore() does not contain hits collection with name " << m_afpToFHitContainerKey );
        return StatusCode::SUCCESS;
    }
    ATH_CHECK( afpToFHitContainer.initialize() );
 
    Monitored::Scalar<float> lqbar[ 2 ] = { Monitored::Scalar<float>( "lqBar_A", 0.0 ),
                                            Monitored::Scalar<float>( "lqBar_C", 0.0 ) };
 
    Monitored::Scalar<float> lqbar_tight[ 2 ] = { Monitored::Scalar<float>( "lqBar_tight_A", 0.0 ),
                                                  Monitored::Scalar<float>( "lqBar_tight_C", 0.0 ) };
    Monitored::Scalar<float> fsp0_rows_tight[ 2 ] = { Monitored::Scalar<float>( "fsp0_rows_tight_A", 0.0 ),
                                                      Monitored::Scalar<float>( "fsp0_rows_tight_C", 0.0 ) };
    Monitored::Scalar<float> fsp0_columns_tight[ 2 ] = { Monitored::Scalar<float>( "fsp0_columns_tight_A", 0.0 ),
                                                         Monitored::Scalar<float>( "fsp0_columns_tight_C", 0.0 ) };
 
    Monitored::Scalar<float> trainHits[ 2 ] = { Monitored::Scalar<float>( "trainHits_A", 0.0 ),
                                                Monitored::Scalar<float>( "trainHits_C", 0.0 ) };
 
    Monitored::Scalar<float> tofHits[ 2 ] = { Monitored::Scalar<float>( "tofHits_A", 0.0 ),
                                              Monitored::Scalar<float>( "tofHits_C", 0.0 ) };
    Monitored::Scalar<float> fsp0Hits[ 2 ] = { Monitored::Scalar<float>( "fsp0Hits_A", 0.0 ),
                                               Monitored::Scalar<float>( "fsp0Hits_C", 0.0 ) };
 
    Monitored::Scalar<float> fs_rows_full[ 2 ] = { Monitored::Scalar<float>( "fs_rows_full_A", 0.0 ),
                                                Monitored::Scalar<float>( "fs_rows_full_C", 0.0 ) };
    Monitored::Scalar<float> fs_rows[ 2 ] = { Monitored::Scalar<float>( "fs_rows_A", 0.0 ),
                                                Monitored::Scalar<float>( "fs_rows_C", 0.0 ) };
    Monitored::Scalar<float> fs_columns[ 2 ] = { Monitored::Scalar<float>( "fs_columns_A", 0.0 ),
                                                   Monitored::Scalar<float>( "fs_columns_C", 0.0 ) };
 
 
    std::vector<std::pair<int, int>> fsp[ 2 ][ 2 ] = {};
    std::vector<std::pair<int, int>> fsp_full[ 4 ][ 2 ] = {};
 
    for ( const xAOD::AFPSiHit* hitsItr : *afpSiHitContainer ) {
        if ( hitsItr->stationID() != 0 && hitsItr->stationID() != 3 ) continue;
        int side = hitsItr->stationID() / 2;
        if ( hitsItr->pixelLayerID() >= 0 && hitsItr->pixelLayerID() < 4 )
        {
            int plane_full = hitsItr->pixelLayerID();
            fsp_full[ plane_full ][ side ].emplace_back( hitsItr->pixelRowIDChip(), hitsItr->pixelColIDChip() );
        }           
 
        if ( hitsItr->pixelLayerID() != 0 && hitsItr->pixelLayerID() != 2 ) continue;
        int plane = hitsItr->pixelLayerID() / 2;
 
        fsp[ plane ][ side ].emplace_back( hitsItr->pixelRowIDChip(), hitsItr->pixelColIDChip() );
    }
 
    int numToFHitsTrain[ 4 ][ 2 ] = {};
    bool ToFHasHit = false;
 
    for ( const xAOD::AFPToFHit* hitsItr : *afpToFHitContainer ) {
        int side = hitsItr->stationID() / 2;
        float bar = hitsItr->trainID() + hitsItr->barInTrainID() / 4.0;
        for ( auto& coord : fsp[ 0 ][ side ] ) {
            bool match = false;
            if ( fsp[ 0 ][ side ].size() == 2 && fsp[ 1 ][ side ].size() == 2 ) {
                for ( auto& coord2 : fsp[ 1 ][ side ] ) {
                    match |= std::abs( coord2.first - coord.first ) < 3 && std::abs( coord2.second - coord.second ) < 2;
                }
            }
            if ( match ) {
                lqbar_tight[ side ] = bar;
                fsp0_rows_tight[ side ] = coord.first;
                fsp0_columns_tight[ side ] = coord.second;
 
                fill( "AFPToFSiTTool", lqbar_tight[ side ], fsp0_rows_tight[ side ] );
                fill( "AFPToFSiTTool", lqbar_tight[ side ], fsp0_columns_tight[ side ] );
 
                lqbar_tight[ side ] = -0.5;
 
                fill( "AFPToFSiTTool", lqbar_tight[ side ], fsp0_rows_tight[ side ] );
                fill( "AFPToFSiTTool", lqbar_tight[ side ], fsp0_columns_tight[ side ] );
            }
        }
        //additional TofSitCorr hists 
        for (int j = 0; j < 4; j++)
            for ( auto& coord : fsp_full[ j ][ side ] )
            {
                lqbar[ side ] = bar;
                fs_rows[ side ] = coord.first;
                fs_columns[ side ] = coord.second;
                fs_rows_full[ side ] = coord.first;
                fill(m_tools[m_GroupToFSiTCorr.at(m_planes.at(j))], lqbar[ side ], fs_rows[ side ]);
                fill(m_tools[m_GroupToFSiTCorr.at(m_planes.at(j))], lqbar[ side ], fs_columns[ side ]);
                fill( "AFPToFSiTTool", lqbar[ side ], fs_rows_full[ side ] );
            }
        
        for (int j = 0; j < 4; j++)
        {
            if ( fsp_full[ j ][ side ].empty() )
            {
                lqbar[ side ] = bar;
                fs_columns[ side ] = -5;
                for ( int k = -10; k < -1; ++k ) 
                {
                    fs_rows[ side ] = k;
                    fs_rows_full[ side ] = k;
                    fill(m_tools[m_GroupToFSiTCorr.at(m_planes.at(j))], lqbar[ side ], fs_rows[ side ]);
                }
                fill(m_tools[m_GroupToFSiTCorr.at(m_planes.at(j))], lqbar[ side ], fs_columns[ side ]);
            }
        } 
 
        ++numToFHitsTrain[ hitsItr->trainID() ][ side ];
        ToFHasHit = true;
 
        ++tofHits[ side ];
    }
 
    for ( auto side : { 0, 1 } ) {
        bool wasMatch = false;
        if ( fsp[ 0 ][ side ].size() == 2 && fsp[ 1 ][ side ].size() == 2 ) {
            for ( auto& coord : fsp[ 0 ][ side ] ) {
                bool match = false;
                for ( auto& coord2 : fsp[ 1 ][ side ] ) {
                    match |= std::abs( coord2.first - coord.first ) < 3 && std::abs( coord2.second - coord.second ) < 2;
                }
                if ( match ) {
                    for ( int itrain = 0; itrain < 4; ++itrain ) {
                        trainHits[ side ] = itrain + numToFHitsTrain[ itrain ][ side ] / 5.0;
                        fsp0_rows_tight[ side ] = coord.first;
                        fsp0_columns_tight[ side ] = coord.second;
 
                        fill( "AFPToFSiTTool", trainHits[ side ], fsp0_rows_tight[ side ] );
                        fill( "AFPToFSiTTool", trainHits[ side ], fsp0_columns_tight[ side ] );
                    }
                    if ( numToFHitsTrain[ 0 ][ side ] > 4 || numToFHitsTrain[ 1 ][ side ] > 4
                         || numToFHitsTrain[ 2 ][ side ] > 4 || numToFHitsTrain[ 3 ][ side ] > 4 ) {
                        trainHits[ side ] = -1;
                        fsp0_rows_tight[ side ] = coord.first;
                        fsp0_columns_tight[ side ] = coord.second;
 
                        fill( "AFPToFSiTTool", trainHits[ side ], fsp0_rows_tight[ side ] );
                        fill( "AFPToFSiTTool", trainHits[ side ], fsp0_columns_tight[ side ] );
                    }
                }
                wasMatch |= match;
            }
        }
 
        if ( !ToFHasHit ) {
            for ( auto& coord : fsp[ 0 ][ side ] ) {
                if ( wasMatch ) {
                    lqbar_tight[ side ] = -1;
                    fsp0_rows_tight[ side ] = coord.first;
                    fsp0_columns_tight[ side ] = coord.second;
 
                    fill( "AFPToFSiTTool", lqbar_tight[ side ], fsp0_rows_tight[ side ] );
                    fill( "AFPToFSiTTool", lqbar_tight[ side ], fsp0_columns_tight[ side ] );
                }
            }
            for (int j = 0; j < 4; j++)
                for ( auto& coord : fsp_full[ j ][ side ] )
                {
                    lqbar[ side ] = -1;
                    fs_rows[ side ] = coord.first;
                    fs_columns[ side ] = coord.second;
                    fill(m_tools[m_GroupToFSiTCorr.at(m_planes.at(j))], lqbar[ side ], fs_rows[ side ]);
                    fill(m_tools[m_GroupToFSiTCorr.at(m_planes.at(j))], lqbar[ side ], fs_columns[ side ]);
                }
        }
 
        fsp0Hits[ side ] = fsp[ 0 ][ side ].size();
        fill( "AFPToFSiTTool", tofHits[ side ], fsp0Hits[ side ] );
    }
 
    return fillHistograms_eff(*afpTrackContainer, *afpToFHitContainer);
}
 
StatusCode AFPToFSiTAlgorithm::fillHistograms_eff(
        const xAOD::AFPTrackContainer& afpTrackContainer,
        const xAOD::AFPToFHitContainer& afpToFHitContainer) const {
    // Initialize monitored variables for histogram filling
    Monitored::Scalar<bool> tof_eff_OFF_passed[2] = {
            Monitored::Scalar<bool>( "tof_eff_OFF_passed_A", false ),
            Monitored::Scalar<bool>( "tof_eff_OFF_passed_C", false )
        };
    Monitored::Scalar<uint8_t> tof_eff_OFF_trains[2] = {
            Monitored::Scalar<uint8_t>( "tof_eff_OFF_trains_A", 0 ),
            Monitored::Scalar<uint8_t>( "tof_eff_OFF_trains_C", 0 )
        };
    Monitored::Scalar<uint8_t> tof_eff_OFF_bars[2] = {
            Monitored::Scalar<uint8_t>( "tof_eff_OFF_bars_A", 0 ),
            Monitored::Scalar<uint8_t>( "tof_eff_OFF_bars_C", 0 )
        };
 
    bool bar_hit[2][5] = {};
    bool channel_present[2][16] = {};
    bool multihit[2] = {};
    uint8_t track_train[2] = {};
    std::size_t track_count[2] = {};
 
    // Load the necessary information
    for (const xAOD::AFPTrack* tracksItr : afpTrackContainer) { 
        const auto side = tracksItr->stationID() == 3;
 
        // Ignore tracks that are not from FAR stations
        if (tracksItr->stationID() != 0 && tracksItr->stationID() != 3) 
            continue;
 
        const auto xLocal = tracksItr->xLocal();
        for (uint8_t train = 0; train < 4; ++train) {
            bool matching = xLocal < m_tofTrainsCoordinates[side][train] - m_tofTrainGapSize
                         && xLocal > m_tofTrainsCoordinates[side][train + 1];
            if (matching)
                track_train[side] = train;
        }
        ++track_count[side];
    }
 
    for (const xAOD::AFPToFHit* hitsItr : afpToFHitContainer) {
        const auto side = hitsItr->stationID() == 3;
        const auto train = hitsItr->trainID();
        const auto bar = hitsItr->barInTrainID();
        const auto channel = 4 * train + bar;
 
        // Ignore hits with an impossible origin
        if (hitsItr->stationID() != 0 && hitsItr->stationID() != 3)
            continue;
        if (channel >= 16) continue;
 
        if (train == track_train[side]) 
            bar_hit[side][bar] = bar_hit[side][4] = true;
        if (channel_present[side][channel])
            multihit[side] = true;
        channel_present[side][channel] = true;
    }
 
    for (uint8_t side : {0, 1}) {
        // Cut on only 1 SiT track in the monitored station
        if (track_count[side] != 1) continue;
        // Cut on maximum of 1 hit in each ToF channel
        if (multihit[side]) continue;
 
        tof_eff_OFF_trains[side] = track_train[side];
        for (uint8_t bar = 0; bar <= 4; ++bar) { // 4 = train total
            tof_eff_OFF_bars[side] = bar;
            tof_eff_OFF_passed[side] = bar_hit[side][bar];
            fill( "AFPToFSiTTool", tof_eff_OFF_passed[side], tof_eff_OFF_bars[side], tof_eff_OFF_trains[side] );
        }
    }
 
    return StatusCode::SUCCESS;
}