From 7fbadd4fcc47d932a2310c8b27f7e6abb33ad1a9 Mon Sep 17 00:00:00 2001
From: P-A Loizeau <p.-a.loizeau@gsi.de>
Date: Thu, 25 Jun 2020 17:14:26 +0200
Subject: [PATCH] Add support for mTOF vectors names (e.g. Calib Digi) in
 DigiManager

---
 core/base/CbmDigiManager.cxx | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/core/base/CbmDigiManager.cxx b/core/base/CbmDigiManager.cxx
index 1d0e7f4003..f868b40d5c 100644
--- a/core/base/CbmDigiManager.cxx
+++ b/core/base/CbmDigiManager.cxx
@@ -167,6 +167,41 @@ template <class Digi> void CbmDigiManager::SetBranch() {
       delete branch;
     }
 
+    // Special cases for mCBM TOF
+    if ( systemId == ECbmModuleId::kTof) {
+      if ( fBranches.find(systemId) == fBranches.end() ) {
+        branchName = "TofCalDigi";
+        branch = new CbmDigiBranch<Digi>(branchName.c_str());
+        if ( branch->ConnectToTree() ) {
+  	  LOG(info) << "DigiManager: Search branch " << branchName << " for class "
+	    	    << className << ": successful";
+	  fBranches[systemId] = branch;
+        } else {
+	  LOG(info) << "DigiManager: Search branch " << branchName << " for class "
+                    << className << ": failed";
+	  delete branch;
+        }
+      } 
+      if ( fBranches.find(systemId) == fBranches.end() ) {
+        branchName = "CbmTofDigi";
+        branch = new CbmDigiBranch<Digi>(branchName.c_str());
+        if ( branch->ConnectToTree() ) {
+	  LOG(info) << "DigiManager: Search branch " << branchName << " for class "
+	  	    << className << ": successful";
+	  fBranches[systemId] = branch;
+        } else {
+	  LOG(info) << "DigiManager: Search branch " << branchName << " for class "
+                    << className << ": failed";
+	  delete branch;
+        }
+      }
+      if ( fBranches.find(systemId) == fBranches.end() ) {
+    	branchName = "CbmTofCalDigi";
+        branch = new CbmDigiBranch<Digi>(branchName.c_str());
+        if ( branch->ConnectToTree() ) fBranches[systemId] = branch;
+        else delete branch;
+      }
+    }       
 }
 // -------------------------------------------------------------------------
 
-- 
GitLab