ClusterNtuple.h

Go to the documentation of this file.

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

#ifndef CLUSTERNTUPLE_H
#define CLUSTERNTUPLE_H
 
#include "MuonLinearSegmentMakerUtilities/Cluster.h"
#include <vector>
#include <algorithm>
 
class TTree;
 
namespace ClusterSeg {
 
  enum FillType {chi2,chi2T,residual,residualT};
 
  struct ClusterNtuple {
 
    void initForWrite(TTree& tree);
    void initForRead(TTree& tree);
 
    void fill(const double& fillVal, FillType theFillType );
    void fill( const Cluster& cluster );
    void fill( const std::vector<Cluster*>& clusters );
 
    bool read( std::vector<Cluster*>& clusters);
    static void clean( std::vector<Cluster*>& clusters);   
 
    void reset();
    void init();
 
    static const int CLUSTERSIZE = 50000;
 
    double residualT[CLUSTERSIZE] = {};
    double residual[CLUSTERSIZE] = {};
    int nMDT = 0;
    int nMDTT = 0;
    double chi2T[CLUSTERSIZE] = {};
    double chi2[CLUSTERSIZE] = {};
    int ntracksT = 0;
    int ntracks = 0;
    int nclusters = 0;
    double x[CLUSTERSIZE] = {};
    double y[CLUSTERSIZE] = {};
    double z[CLUSTERSIZE] = {};
    int tIndex[CLUSTERSIZE] = {};
    int pIndex[CLUSTERSIZE] = {};
    bool isPhi[CLUSTERSIZE] = {};
    bool isMatch[CLUSTERSIZE] = {};
    int barcode[CLUSTERSIZE] = {};
  };
 
}
 
 
#endif