HLTcalo module

[u/mrichter/AliRoot.git] / HLT / BASE / AliHLTSpacePointContainer.cxx

diff --git a/HLT/BASE/AliHLTSpacePointContainer.cxx b/HLT/BASE/AliHLTSpacePointContainer.cxx
index e764f94b605604cd8471958ea6599dbd2777a7d0..eda14439f732236cac6f8de2c8b41067d4e4cef4 100644 (file)
--- a/HLT/BASE/AliHLTSpacePointContainer.cxx
+++ b/HLT/BASE/AliHLTSpacePointContainer.cxx
@@ -35,6 +35,7 @@
 #include <memory>
 #include <algorithm>
 #include <iostream>
+#include <iomanip>
 
 /** ROOT macro for the implementation of ROOT specific class methods */
 ClassImp(AliHLTSpacePointContainer)
@@ -401,11 +402,11 @@ TTree* AliHLTSpacePointContainer::FillTree(const char* name, const char* title)
   std::auto_ptr<TTree> tree(new TTree(treename, treetitle));
   if (!tree.get()) return NULL;
 
-  const unsigned dimension=5;
+  const unsigned dimension=8;
   float values[dimension];
   memset(values, 0, sizeof(values));
   const char* names[dimension]={
-    "x", "y", "z", "charge", "alpha"
+    "x", "y", "z", "sigmaY2", "sigmaZ2", "charge", "qmax", "alpha"
   };
 
   for (unsigned i=0; i<dimension; i++) {
@@ -421,7 +422,10 @@ TTree* AliHLTSpacePointContainer::FillTree(const char* name, const char* title)
     values[pos++]=GetX(*clusterID);
     values[pos++]=GetY(*clusterID);
     values[pos++]=GetZ(*clusterID);
+    values[pos++]=GetYWidth(*clusterID);
+    values[pos++]=GetZWidth(*clusterID);
     values[pos++]=GetCharge(*clusterID);
+    values[pos++]=GetMaxSignal(*clusterID);
     values[pos++]=GetPhi(*clusterID);
 
     tree->Fill();
@@ -435,3 +439,15 @@ ostream& operator<<(ostream &out, const AliHLTSpacePointContainer& c)
   c.Print(out);
   return out;
 }
+
+ostream& operator<<(ostream &out, const AliHLTSpacePointContainer::AliHLTSpacePointProperties& p)
+{
+  // print
+  cout << p.fId;
+  return out;
+}
+
+bool operator==(const AliHLTSpacePointContainer::AliHLTSpacePointProperties& a,
+               const AliHLTSpacePointContainer::AliHLTSpacePointProperties& b) {
+  return a.fId==b.fId;
+}