LmvmFastSim.h

Go to the documentation of this file.

/* Copyright (C) 2023-2025 UGiessen, Giessen
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Cornelius Feier-Riesen [committer] */
 
#ifndef LMVM_FAST_SIM_H
#define LMVM_FAST_SIM_H
 
#include "LmvmHist.h"
#include "TObject.h"
 
#include <map>
#include <string>
#include <vector>
 
class TH1;
class TH2D;
class TH1D;
class TFile;
class CbmHistManager;
 
class LmvmFastSim : public TObject {
 
 public:
  LmvmFastSim() { ; }
  virtual ~LmvmFastSim() { ; }
 
  void DoFastSim(std::vector<std::string> files, const std::string& dataDir, const std::string& outFile,
                 int nofFastSimEv, const std::string& bgHistName);
 
 private:
  std::string fOutputDir   = "";
  std::string fOutFileName = "";
  std::string fPlutoMode   = "";  // Either "one" or "all". Means: is there only one Pluto per event implemented or all?
  int fNofSimEv            = 0;
  int fNofFastSimEv        = -1;
  int fNofPlutos           = -1;
  int fNofUrqmds           = -1;
  int fNofCats             = -1;
  std::vector<std::string> fPlutoNames;
  std::vector<std::string> fUrqmdNames;
  std::vector<std::string> fCatNames;  // own categories as e.g. analysis steps
  std::vector<std::string> fChargeCats;
  std::string fBgHistName = "";
 
  bool fDoChargeBased   = false;
  bool fDoParticleBased = false;
  bool fDoTrueEl        = false;
 
  // mix events
  bool fDoMixEvents = false;  // to save time, use mixing only if needed
  int fMixingDepth  = 5000;   // Mixing depth for Combinatorial BG Calculation from Fast Simulations
 
  std::vector<LmvmHist*> fH;
  LmvmHist fHMean;
 
  void CalcMinvAndFillHist(const std::vector<LmvmDataFastSim>& Plus, const std::vector<LmvmDataFastSim>& Minus,
                           const std::string& hName);
  void CalcMinvAndFillHist(const std::vector<LmvmDataFastSim>& Plus, const std::vector<LmvmDataFastSim>& Minus,
                           const std::string& hName, const std::string& opt);
 
  template<class T>
  void CreateMeanHist(const std::string& name);
 
  void CheckHistogramNames();
  void GetChargeCategoryVector();
  void GetIndividualCategories();
  void GetPlutoAndUrqmdNames(const std::vector<std::string>& files);
  void CreateHistos();
  bool ContainsString(const std::vector<std::string>& vName, const std::string& name);
 
  void CheckMemory(const std::string& text);
 
  void SaveHist();
 
  LmvmFastSim(const LmvmFastSim&);
  LmvmFastSim operator=(const LmvmFastSim&);
 
  ClassDef(LmvmFastSim, 1);
};
 
#endif