Changeset dc883b4 in git for external/ExRootAnalysis

Timestamp:

Mar 4, 2021, 2:36:16 PM (4 years ago)

Author:

Pavel Demin <pavel.demin@…>

Branches:

master

Children:

4658ca4

Parents:

f9517a5

Message:

add info to TreeWriter

Location:

external/ExRootAnalysis

Files:

• ExRootTreeReader.cc (modified) (2 diffs)
• ExRootTreeReader.h (modified) (1 diff)
• ExRootTreeWriter.cc (modified) (2 diffs)
• ExRootTreeWriter.h (modified) (1 diff)

external/ExRootAnalysis/ExRootTreeReader.cc

@@ -12 +12 @@
 #include "TBranchElement.h"
 #include "TCanvas.h"
+#include "TParameter.h"
 #include "TClonesArray.h"
 #include "TH2.h"
@@ -119 +120 @@
 //------------------------------------------------------------------------------
 
+Double_t ExRootTreeReader::GetInfo(const char *name)
+{
+  TTree *tree = NULL;
+  TParameter<Double_t> *param = NULL;
+  Double_t result = -999.9;
+  if(fChain) tree = fChain->GetTree();
+  if(tree) param = static_cast<TParameter<Double_t> *>(tree->GetUserInfo()->FindObject(name));
+  if(param) result = param->GetVal();
+  return result;
+}
+
+//------------------------------------------------------------------------------
+
 Bool_t ExRootTreeReader::Notify()
 {

external/ExRootAnalysis/ExRootTreeReader.h

@@ -29 +29 @@
 
   TClonesArray *UseBranch(const char *branchName);
+  Double_t GetInfo(const char *name);
 
 private:

external/ExRootAnalysis/ExRootTreeWriter.cc

@@ -11 +11 @@
 #include "ExRootAnalysis/ExRootTreeBranch.h"
 
+#include "TParameter.h"
 #include "TClonesArray.h"
 #include "TFile.h"
@@ -48 +49 @@
   fBranches.insert(branch);
   return branch;
+}
+
+//------------------------------------------------------------------------------
+
+void ExRootTreeWriter::AddInfo(const char *name, Double_t value)
+{
+  if(!fTree) fTree = NewTree();
+  fTree->GetUserInfo()->Add(new TParameter<Double_t>(name, value));
 }
 

external/ExRootAnalysis/ExRootTreeWriter.h

@@ -29 +29 @@
 
   ExRootTreeBranch *NewBranch(const char *name, TClass *cl);
+  void AddInfo(const char *name, Double_t value);
 
   void Clear();