CbmStsAnaBase.h

Go to the documentation of this file.

/* Copyright (C) 2023 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Dario Ramirez [committer] */
 
#ifndef CBMSTSANABASE_H
#define CBMSTSANABASE_H
 
#include "CbmCutMap.h"
#include "CbmDefs.h"
#include "CbmStsSetup.h"
#include "CbmStsUtils.h"
 
#include <FairRootManager.h>
#include <Logger.h>
 
#include <TCanvas.h>
#include <TClonesArray.h>
#include <TFile.h>
#include <TGraphErrors.h>
#include <TH1D.h>
#include <TH2D.h>
#include <TSystem.h>
 
#include <cstring>
#include <map>
#include <optional>
#include <unordered_set>
 
 
enum InputType
{
  kMCEventMode,
  kMCTimeMode,
  kEventMode,
  kTimeMode
};

class CbmStsAnaBase {
 public:
  CbmStsAnaBase()  = default;
  virtual ~CbmStsAnaBase() = default;

  void SetCutMap(CbmCutMap* map);

  void UserAlignment(const std::map<int32_t, std::vector<double>>& user_mat) { fUserAlignment = user_mat; }

  virtual void DrawResults() {}
 
 protected:
  uint entry_{0};
  std::unique_ptr<TFile> fReportFile;
  std::unordered_set<int32_t> fAddressBook;
  std::map<std::string, std::unique_ptr<TGraphErrors>> fG1D;  // Map of TGraphErrors objects.
  std::map<std::string, std::unique_ptr<TH1D>> fH1D;          // Map of TH1D objects.
  std::map<std::string, std::unique_ptr<TH2D>> fH2D;          // Map of TH2D objects.
  std::map<std::string, std::shared_ptr<TH2D>> fH2DShared;    // Map of TH2D objects.
  std::map<std::string, std::unique_ptr<TCanvas>> fCanvas;    // Map of TH2D objects.
 
  int nb_sts_station_{8};                                        // Number of STS stations.
  std::unordered_map<int32_t, std::vector<double>> fStsGeoInfo;  // Map of STS geometry information.
 
  std::map<int32_t, std::vector<double>> fUserAlignment;  // Stores the user-defined alignment information.
 
  std::unordered_map<int32_t, int> fFirstZStrip;
 
  CbmCutMap* fAnalysisCuts{nullptr};
 
  int fRunId{-1};

  void LoadSetup();

  void SaveToFile();
 
  ClassDef(CbmStsAnaBase, 1);
};
#endif