PndSimpleCombinerTask.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".              *
//****************************************************************************

// ************************************************************************
//
//  Analysis Task using PndSimpleCombiner
//
// ************************************************************************
//
// Parameters:
// - anadecay     : Decay specification, e.g. "phi -> K+ K-; D_s+ -> phi pi+" (automatic charged conjugation; particle used have to be defined beforehand)
//                  Keyword 'nocc' at end of decay definition suppresses automatic charged conjugation
//                  This string is handed over to PndSimpleCombiner
//
// - params       : configuration parameters, e.g. "fit4c:qamc". The string contains also parameters handled by PndSimpleCombiner; those handled by this task are:
//   - fit4c      : perform 4C fit on last resonance
//   - fitvtx     : perform vertex fit on all resonances when possible (at least two daughters)
//   - qamc       : stored MC information
//   - qaevtshape : store event shape information
//
// K.Goetzen 1/2015
//
// ************************************************************************

#ifndef PndSimpleCombinerTask_H
#define PndSimpleCombinerTask_H 1

#include <vector>
#include "FairTask.h"
#include "TLorentzVector.h"
#include "TString.h"

class TClonesArray;
class TH1F;
class TH2F;
class TDatabasePDG;

class PndAnalysis;
class RhoCandList;
class RhoCandidate;
class RhoTuple;
class PndSimpleCombiner;

typedef std::vector<TString> StringList;

class PndSimpleCombinerTask : public FairTask {

 public:
  // ** Default constructor
  PndSimpleCombinerTask(TString anadecay, TString anaparms, double p = 0, int run = 0, int mode = 0);

  // ** Destructor
  ~PndSimpleCombinerTask();

  void SetMultFactor(int fac) { fRunMult = fac; } // set run multiplicator

  // ** Virtual method Init
  virtual InitStatus Init();

  // ** Virtual method Exec
  virtual void Exec(Option_t *opt);

  virtual void Finish();

  void SetPidAlgo(TString algo) { fPidAlgo = algo; }
  void SetVerbose(int v = 1) { fVerbose = v; } // verbosity level, also passed to PndSimpleCombiner

 protected:
 private:
  StringList SplitString(TString s, TString delim);  // routine to split a string in pieces
  int CountChargedDaughters(RhoCandidate *c);        // information for vertex fitting
  void InitParms();                                  // parse parameter string
  void InitParticleDB();                             // initialize TDatabasePDG with input/pdg_table_evtgen.txt
  void UpdateParticleDB();                           // add new particles found in pgenerators/EvtGen/EvtGen/Private/evt.pdl

  int fVerbose;  // verbosity level
  int fEvtCount; // event counter
  int fRun;      // run number to be stored in ntuple (for unique event ID)
  int fMode;     // optional arbitrary mode number
  int fRunMult;  // factor for fRun to create unique event ID; default = 10000 (has be larger the num. of events per job!)

  TLorentzVector fIni;  // initial 4-vector; either computed from constructor parameter Mom, or taken from MC list
  TString fAnaDecay;    // decay string
  TString fAnaParms;    // parameter string; has to contain also parameters for PndSimpleCombiner
  int fNntp;            // number of ntuples to be created
  TString fPidAlgo;     // PID algo name
  bool fQaMC;           // flag to store MC list
  bool fQaEventShape;   // flag to store event shape variables
  int fQaRecoInfo;      // flag to store shotr (==1) or full (==2) reco info
  bool fQaEvShapeNtp;   // flag to store event shape variables in an extra ntuple
  bool fFit4C;          // flag to perform 4C fit
  bool fBest4C;         // flag to only store best 4C fitted candidate
  bool fFitVtx;         // flag to perform vtx fit
  double fFit4CChiCut;  // cut on chi2 from 4C fitter
  double fFitVtxChiCut; // cut on chi2 from vtx fitter

  // *** object instances needed
  PndAnalysis *fAnalysis;
  PndSimpleCombiner *fSimpleCombiner;

  std::vector<int> vmpdg;       // pdg code of the composites
  std::vector<RhoTuple *> vntp; // ntuples for the composites
  unsigned int fNodump;         // bit marker for ntuples not to be dumped (to save disc space); e.g. 101 -> don't dump ntp0 & ntp2
  RhoTuple *nmc;                // MC ntuple
  RhoTuple *nevt;               // MC ntuple

  TClonesArray *fOnlineFilterInfo;

  ClassDef(PndSimpleCombinerTask, 1);
};

#endif