From d54b52ac43438b62956af40b0f8ecb3c806893f7 Mon Sep 17 00:00:00 2001
From: "P.-A. Loizeau" <p.-a.loizeau@gsi.de>
Date: Tue, 7 Nov 2023 15:13:51 +0100
Subject: [PATCH] [Evt dis] in TimesliceManager, add flags for RICH/MUCH ctrl
 in SetDisplayMcbm

---
 core/eventdisplay/CbmTimesliceManager.cxx        | 14 +++++++++-----
 core/eventdisplay/CbmTimesliceManager.h          |  2 +-
 macro/beamtime/mcbm2022/mcbm_digievent_display.C |  2 +-
 3 files changed, 11 insertions(+), 7 deletions(-)

diff --git a/core/eventdisplay/CbmTimesliceManager.cxx b/core/eventdisplay/CbmTimesliceManager.cxx
index 2a80ec2e1f..cef6e1aa86 100644
--- a/core/eventdisplay/CbmTimesliceManager.cxx
+++ b/core/eventdisplay/CbmTimesliceManager.cxx
@@ -95,19 +95,23 @@ void CbmTimesliceManager::SetDisplayCbmMuon()
   AddTask(drawTrack);
 }
 
-void CbmTimesliceManager::SetDisplayMcbm()
+void CbmTimesliceManager::SetDisplayMcbm(bool bLowRate, bool bMuch)
 {
-  CbmTimeslicePixelHitSetDraw* drawStsHit = new CbmTimeslicePixelHitSetDraw("StsHit", kBlue, kFullSquare);
-  //  CbmTimeslicePixelHitSetDraw* drawMuchHit     = new CbmTimeslicePixelHitSetDraw("MuchHit", kCyan, kFullSquare);
+  CbmTimeslicePixelHitSetDraw* drawStsHit  = new CbmTimeslicePixelHitSetDraw("StsHit", kBlue, kFullSquare);
+  CbmTimeslicePixelHitSetDraw* drawMuchHit = new CbmTimeslicePixelHitSetDraw("MuchHit", kCyan, kFullSquare);
   CbmTimeslicePixelHitSetDraw* drawTrdHit  = new CbmTimeslicePixelHitSetDraw("TrdHit", kYellow, kFullSquare);
   CbmTimeslicePixelHitSetDraw* drawTofHit  = new CbmTimeslicePixelHitSetDraw("TofHit", kRed, kFullCircle);
   CbmTimeslicePixelHitSetDraw* drawRichHit = new CbmTimeslicePixelHitSetDraw("RichHit", kOrange, kFullSquare);
 
   AddTask(drawStsHit);
-  //AddTask(drawMuchHit);
+  if (bMuch) {  //
+    AddTask(drawMuchHit);
+  }
   AddTask(drawTrdHit);
   AddTask(drawTofHit);
-  //AddTask(drawRichHit);
+  if (bLowRate) {  //
+    AddTask(drawRichHit);
+  }
 
   CbmTimesliceRecoTracks* drawTrack = new CbmTimesliceRecoTracks();
   AddTask(drawTrack);
diff --git a/core/eventdisplay/CbmTimesliceManager.h b/core/eventdisplay/CbmTimesliceManager.h
index 8cc9c8f810..367bc02a60 100644
--- a/core/eventdisplay/CbmTimesliceManager.h
+++ b/core/eventdisplay/CbmTimesliceManager.h
@@ -61,7 +61,7 @@ public:
   /**
    ** @brief Select/add default set of views and some Hit and Tracks drawing classes for the mCBM 2022+ setups.
    **/
-  void SetDisplayMcbm();
+  void SetDisplayMcbm(bool bLowRate = true, bool bMuch = false);
   /**
    ** @brief Select/add default set of views and some Hit and Tracks drawing classes for the HD 2023+ cosmics setups.
    **/
diff --git a/macro/beamtime/mcbm2022/mcbm_digievent_display.C b/macro/beamtime/mcbm2022/mcbm_digievent_display.C
index d7c0ada5e1..d53c3c81aa 100644
--- a/macro/beamtime/mcbm2022/mcbm_digievent_display.C
+++ b/macro/beamtime/mcbm2022/mcbm_digievent_display.C
@@ -1278,7 +1278,7 @@ Bool_t mcbm_digievent_display(UInt_t uRunId               = 2391,
 
   CbmTimesliceManager* fMan = new CbmTimesliceManager();
   fMan->SetXMLConfig(sXmlGeoConfig);
-  fMan->SetDisplayMcbm();
+  fMan->SetDisplayMcbm(false);
   // ---------------------------------------------------------------------------------------------------------------- //
   /* 
      CbmEvDisTracks* Tracks = new CbmEvDisTracks("TofTracks", 1, kFALSE,
-- 
GitLab