PndRhoTupleQA.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

#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 = nullptr, 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 qaCompP4(TString pre, RhoCandidate *c, RhoTuple *n);
  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