IValidationNtupleTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/

// IValidationNtupleTool.h
//   Header file for interface of ValidationNtupleTools
// (c) ATLAS Detector software
// Sebastian.Fleischmann@cern.ch
 
#ifndef TRK_IVALIDATIONNTUPLETOOL_H
#define TRK_IVALIDATIONNTUPLETOOL_H
 
#include "GaudiKernel/IAlgTool.h"
#include "TrkEventPrimitives/FitQualityOnSurface.h"
#include "TrkParameters/TrackParameters.h"
//#include "GeneratorObjects/HepMcParticleLink.h"
#include <vector>
 
class TrackTruth;
class TTree;
class HepMcParticleLink;
 
namespace Trk {
static const InterfaceID IID_IValidationNtupleTool("IValidationNtupleTool",1,0);
 
class TrackParticleBase;
class MeasurementBase;     
class Track;
class TrackStateOnSurface;
 

 
class IValidationNtupleTool : virtual public IAlgTool {
public:
    static const InterfaceID& interfaceID();

    virtual StatusCode addNtupleItems (
        TTree* tree ) = 0;

    virtual StatusCode writeTrackData (
        const Trk::Track&,
        const int iterationIndex,
        const unsigned int fitStatCode = 0 ) = 0;

    virtual StatusCode writeTrackParticleData (
        const Trk::TrackParticleBase& ) = 0;
    
    virtual StatusCode fillTrackData (
        const Trk::Track&,
        const int iterationIndex,
        const unsigned int fitStatCode = 0 ) = 0;

    virtual StatusCode fillTrackParticleData (
        const Trk::TrackParticleBase&) = 0;
        
    virtual StatusCode fillTrackParameter (
        const Trk::TrackParameters*,
        const int iterationIndex ) = 0;
    
    virtual StatusCode fillMeasurementData (
        const Trk::MeasurementBase*,
        const Trk::TrackParameters* ) = 0;
        
    virtual StatusCode fillOutlierData (
        const Trk::MeasurementBase*,
        const Trk::TrackParameters*,
        const Trk::FitQualityOnSurface* ) = 0;

    virtual StatusCode fillHoleData (
        const Trk::TrackStateOnSurface&) = 0;
 
    virtual StatusCode fillTrackTruthData ( const TrackParameters*&,
                                            const TrackTruth&, 
                                            const int truthIndex = -1 ) = 0;
    
    virtual StatusCode writeRecord( TTree* tree ) = 0;
};
 
inline const InterfaceID& Trk::IValidationNtupleTool::interfaceID() {
    return IID_IValidationNtupleTool;
}
 
} // end of namespace
 
#endif // TRK_IVALIDATIONNTUPLETOOL_H