VP1TrackSummary.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
 
//                                                            //
//  Implementation of class VP1TrackSummary                   //
//                                                            //
//  Author: Thomas H. Kittelmann (Thomas.Kittelmann@cern.ch)  //
//  Initial version: November 2009                            //
//                                                            //
 
#include "VP1TrackSystems/VP1TrackSummary.h"
#include "VP1Utils/VP1DetInfo.h"
#include "AtlasDetDescr/AtlasDetectorID.h"
 
#include "TrkTrack/Track.h"
#include "TrkTrack/TrackStateOnSurface.h"
#include "TrkSurfaces/Surface.h"
#include "TrkMeasurementBase/MeasurementBase.h"
#include "TrkDetElementBase/TrkDetElementBase.h"
#include "TrkRIO_OnTrack/RIO_OnTrack.h"
#include "TrkCompetingRIOsOnTrack/CompetingRIOsOnTrack.h"
#include "TrkSegment/Segment.h"
 
#include <iostream>
//____________________________________________________________________
bool VP1TrackSummary::countHits( const Trk::Track* trk,
    unsigned& nPixelHits,
    unsigned& nSCTHits,
    unsigned& nTRTHits,
    unsigned& nhits_muon_phi, 
    unsigned& nhits_rpc, 
    unsigned& nhits_mdt, 
    unsigned& nhits_tgc, 
    unsigned& nhits_csc,
    unsigned& nhits_mm,
    unsigned& nhits_stgc
    )
{
    nPixelHits = nSCTHits = nTRTHits = nhits_muon_phi = nhits_rpc = nhits_mdt = nhits_tgc = nhits_csc = nhits_mm = nhits_stgc = 0;
    if (!trk)
        return false;
 
    const AtlasDetectorID * idhelper = VP1DetInfo::atlasIDHelper();
    if (!idhelper)
        return false;
 
    std::set<const Trk::TrkDetElementBase*> detelems;// Need several to handle #$$%ing cROTs. EJWM
    Trk::TrackStates::const_iterator tsos_iter = trk->trackStateOnSurfaces()->begin();
    Trk::TrackStates::const_iterator tsos_end = trk->trackStateOnSurfaces()->end();
    for (; tsos_iter != tsos_end; ++tsos_iter) {
        const Trk::MeasurementBase*       meas = *tsos_iter ? (*tsos_iter)->measurementOnTrack() : nullptr;
        VP1TrackSummary::addCounts(detelems, meas, idhelper, nPixelHits , nSCTHits , nTRTHits , nhits_muon_phi , nhits_rpc , nhits_mdt , nhits_tgc , nhits_csc, nhits_mm, nhits_stgc);
    }
    return true;
}
 
//____________________________________________________________________
bool VP1TrackSummary::countHits( const Trk::Segment* seg,
    unsigned& nPixelHits,
    unsigned& nSCTHits,
    unsigned& nTRTHits,
    unsigned& nhits_muon_phi, 
    unsigned& nhits_rpc, 
    unsigned& nhits_mdt, 
    unsigned& nhits_tgc, 
    unsigned& nhits_csc,
    unsigned& nhits_mm,
    unsigned& nhits_stgc
    )
{
    nPixelHits = nSCTHits = nTRTHits = nhits_muon_phi = nhits_rpc = nhits_mdt = nhits_tgc = nhits_csc = nhits_mm = nhits_stgc = 0;
    if (!seg)
        return false;
 
    const AtlasDetectorID * idhelper = VP1DetInfo::atlasIDHelper();
    if (!idhelper)
        return false;
 
    std::set<const Trk::TrkDetElementBase*> detelems;// Need several to handle #$$%ing cROTs. EJWM
    std::vector< const Trk::MeasurementBase * >::const_iterator tsos_iter = seg->containedMeasurements().begin(), tsos_end = seg->containedMeasurements().end();
    for (; tsos_iter != tsos_end; ++tsos_iter) {
        const Trk::MeasurementBase*       meas = *tsos_iter;
        addCounts(detelems, meas, idhelper, nPixelHits , nSCTHits , nTRTHits , nhits_muon_phi , nhits_rpc , nhits_mdt , nhits_tgc , nhits_csc, nhits_mm, nhits_stgc);
    }
    return true;
}
 
void VP1TrackSummary::addCounts( std::set<const Trk::TrkDetElementBase*>& detelems, const Trk::MeasurementBase* meas, 
    const AtlasDetectorID * idhelper,
    unsigned& nPixelHits,
    unsigned& nSCTHits,
    unsigned& nTRTHits,
    unsigned& nhits_muon_phi, 
    unsigned& nhits_rpc, 
    unsigned& nhits_mdt, 
    unsigned& nhits_tgc, 
    unsigned& nhits_csc,
    unsigned& nhits_mm,
    unsigned& nhits_stgc
    )
{
    const Trk::CompetingRIOsOnTrack*  crot = dynamic_cast<const Trk::CompetingRIOsOnTrack*>(meas);
 
    if (crot){
        unsigned int i = 0; 
        for (;i<crot->numberOfContainedROTs();++i){
            detelems.insert(crot->rioOnTrack(i).detectorElement());
        }
    }else if (meas) {
        const Trk::Surface * surf = &(meas->associatedSurface());
        detelems.insert( surf ? surf->associatedDetectorElement() : nullptr);
    }
 
    std::set<const Trk::TrkDetElementBase*>::const_iterator it = detelems.begin(), itEnd=detelems.end();
    for (;it!=itEnd;++it){
        const Trk::TrkDetElementBase* detelem=*it;
        if (!detelem) continue;
        Identifier id = detelem->identify();
        if (!id.is_valid())
            continue;
        if (idhelper->is_pixel(id)) ++nPixelHits;
        else if (idhelper->is_sct(id)) ++nSCTHits;
        else if (idhelper->is_trt(id)) ++nTRTHits;
        else if (idhelper->is_mdt(id)) ++nhits_mdt;
        else if (idhelper->is_rpc(id)) ++nhits_rpc;
        else if (idhelper->is_tgc(id)) ++nhits_tgc;
        else if (idhelper->is_csc(id)) ++nhits_csc;
        else if (idhelper->is_mm(id)) ++nhits_mm;
        else if (idhelper->is_stgc(id)) ++nhits_stgc;
    }// FIXME! Implement Muon phi hits
    nhits_muon_phi=0;
    detelems.clear();   
}