PndRhoTupleQA.h

Go to the documentation of this file.

#ifndef PNDRHOTUPLEQA_H
#define PNDRHOTUPLEQA_H 1

#include "TLorentzVector.h"
#include "TVector3.h"
#include "TString.h"
#include "RhoCandList.h"
#include "RhoVtxPoca.h"

class RhoCandidate;
class RhoTuple;
class PndEventShape;
class PndAnalysis;
class RhoFitterBase;

class PndRhoTupleQA {
public:
        PndRhoTupleQA(PndAnalysis *ana=0, double pbarmom=15.15);
    ~PndRhoTupleQA();

        // *** QA methods
        // *** QA for candidates
        void qaCand(TString pre, RhoCandidate *cc, RhoTuple *n, bool skip=false);
        void qaP4(TString pre, TLorentzVector c, RhoTuple *n, bool skip=false);
        void qaP4Cms(TString pre, TLorentzVector c, RhoTuple *n, bool skip=false);
        void qaP4Cov(TString pre, RhoCandidate *c, RhoTuple *n, bool skip=false);
        void qaPos(TString pre, TVector3 p, RhoTuple *n, bool skip=false);
        void qaPull(TString pre, RhoCandidate *c, RhoTuple *n, bool skip=false);

        // *** QA for 2-prong
        void qaComp(TString pre, RhoCandidate *c, RhoTuple *n, bool covs=false, bool pulls=false);
        void qaKs0(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaPi0(TString pre, RhoCandidate *c, RhoTuple *n);

        // *** QA of event shape
        void qaEventShape(TString pre, PndEventShape *evsh, RhoTuple *n);
        void qaEventShapeShort(TString pre, PndEventShape *evsh, RhoTuple *n);

        // *** QA for parts of eventshape
        void qaESPidMult(TString pre, PndEventShape *evsh, double prob, double pmin, RhoTuple *n);
        void qaESMult(TString pre, PndEventShape *evsh, RhoTuple *n);
        void qaESSum(TString pre, PndEventShape *evsh, RhoTuple *n);
        void qaESMinMax(TString pre, PndEventShape *evsh, RhoTuple *n);
        void qaESEventVars(TString pre, PndEventShape *evsh, RhoTuple *n);

        // *** QA track, vtx, PID, decay
        void qaVtx(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaPoca(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaPRG(TString pre, RhoCandidate *c, RhoTuple *n);
        void qa2Body(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaDalitz(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaTrk(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaPid(TString pre, RhoCandidate *c, RhoTuple *n);

        // *** QA PndPidCandidate raw values
        void qaEmc(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaMvd(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaStt(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaGem(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaDrc(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaDsc(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaRich(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaTof(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaMuo(TString pre, RhoCandidate *c, RhoTuple *n);
        
        void qaRecoShort(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaRecoShortTree(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaRecoFull(TString pre, RhoCandidate *c, RhoTuple *n);
        void qaRecoFullTree(TString pre, RhoCandidate *c, RhoTuple *n);

        // *** QA for the full MC list as array.
        void qaMcList(TString pre, RhoCandList &l, RhoTuple *n, int max=10000);
        void qaMcList(RhoTuple *n, int max=10000);
        // *** QA for the MC truth of candidates and the difference of the relevant properties (pos, p4, etc.)
    void qaMcDiff(TString pre, RhoCandidate *c, RhoTuple *n, bool skip=false);
        void qaMc(TString pre, RhoCandidate *c, RhoTuple *n, bool skip=false);

        void qaFitter(TString pre, RhoFitterBase* fitter, RhoTuple *n, bool skip=false);

        // *** Setters and getters
        void SetIniP4(TLorentzVector _ini) {fIniP4 = _ini;}
        TLorentzVector GetIniP4() {return fIniP4;}

private:

    RhoCandidate *fDummyCand; //empty RhoCandidate to replace "mct" if there is no MCTruth partner (added by J.Puetz in June 2018);
        RhoVtxPoca  *fVtxPoca;
        PndAnalysis *fAnalysis;

        TLorentzVector fIniP4;    // initial 4-momentum
};

#endif