-Created new libarey AliHLTEve with processor classes for the HLT online display

[u/mrichter/AliRoot.git] / HLT / EVE / AliHLTEveITS.cxx

diff --git a/HLT/EVE/AliHLTEveITS.cxx b/HLT/EVE/AliHLTEveITS.cxx
new file mode 100644 (file)
index 0000000..9e93742
--- /dev/null
+++ b/HLT/EVE/AliHLTEveITS.cxx
@@ -0,0 +1,129 @@
+/**************************************************************************
+ * This file is property of and copyright by the ALICE HLT Project        *
+ * ALICE Experiment at CERN, All rights reserved.                         *
+ *                                                                        *
+ * Primary Authors: Svein Lindal <slindal@fys.uio.no   >                  *
+ *                  for The ALICE HLT Project.                            *
+ *                                                                        *
+ * Permission to use, copy, modify and distribute this software and its   *
+ * documentation strictly for non-commercial purposes is hereby granted   *
+ * without fee, provided that the above copyright notice appears in all   *
+ * copies and that both the copyright notice and this permission notice   *
+ * appear in the supporting documentation. The authors make no claims     *
+ * about the suitability of this software for any purpose. It is          *
+ * provided "as is" without express or implied warranty.                  *
+ **************************************************************************/
+
+/// @file   AliHLTEvePhos.cxx
+/// @author Svein Lindal <slindal@fys.uio.no>
+/// @brief  base class for the ITS elements in the HLT EVE display
+
+#include "AliHLTEveITS.h"
+#include "AliHLTHOMERBlockDesc.h"
+#include "TCanvas.h"
+#include "AliEveHOMERManager.h"
+#include "TEveManager.h"
+#include "TH1F.h"
+#include "TH2F.h"
+#include "TEvePointSet.h"
+#include "ITS/AliHLTITSClusterDataFormat.h"
+#include "AliITSRecPoint.h"
+#include "TCanvas.h"
+
+ClassImp(AliHLTEveITS)
+
+AliHLTEveITS::AliHLTEveITS(TString name) : 
+AliHLTEveBase(),
+  fName(name), 
+  fPointSet(NULL)
+{
+  // Constructor.
+}
+
+AliHLTEveITS::~AliHLTEveITS()
+{
+  //Destructor
+  if(fPointSet)
+    delete fPointSet;
+  fPointSet = NULL;
+}
+
+void AliHLTEveITS::ProcessBlock(AliHLTHOMERBlockDesc * block) {
+  //See header file for documentation
+ 
+  if ( block->GetDataType().CompareTo("ROOTHIST") == 0 ) {
+    if(!fCanvas)  fCanvas = CreateCanvas(Form("%s QA",fName.Data()), Form("%s QA", fName.Data()));
+    AddHistogramsToCanvas( block , fCanvas, fHistoCount);
+  } 
+
+  else if ( block->GetDataType().CompareTo("CLUSTERS") == 0 ) {
+    if(!fPointSet) {
+      fPointSet = CreatePointSet(fName);
+      fEventManager->GetEveManager()->AddElement(fPointSet);
+    }
+    ProcessClusters(block, fPointSet);
+  }
+}
+
+
+
+void AliHLTEveITS::UpdateElements() {
+  //See header file for documentation
+  if(fCanvas) fCanvas->Update();
+  if(fPointSet) fPointSet->ElementChanged();
+}
+
+void AliHLTEveITS::ResetElements() {
+  //See header file for documentation
+  fHistoCount = 0;
+  if(fPointSet) fPointSet->Reset();
+}
+
+TEvePointSet * AliHLTEveITS::CreatePointSet(TString name) {
+  //See header file for documentation
+  TEvePointSet * ps = new TEvePointSet(name.Data());
+  SetUpPointSet(ps);
+  return ps;
+} 
+
+void AliHLTEveITS::SetUpPointSet(TEvePointSet * ps ) {
+  //See header file for documentation
+  ps->SetMainColor(kBlack);
+  ps->SetMarkerStyle((Style_t)kFullDotMedium);
+}
+
+void AliHLTEveITS::ProcessClusters(AliHLTHOMERBlockDesc * block, TEvePointSet * cont ) {
+  //See header file for documentation
+  AliHLTITSClusterData *cd = reinterpret_cast<AliHLTITSClusterData*> (block->GetData());
+  UChar_t *data            = reinterpret_cast<UChar_t*> (cd->fSpacePoints);
+  
+  if ( cd->fSpacePointCnt != 0 ) {
+    for (UInt_t iter = 0; iter < cd->fSpacePointCnt; ++iter, data += sizeof(AliHLTITSSpacePointData)) {
+      AliHLTITSSpacePointData *sp = reinterpret_cast<AliHLTITSSpacePointData*> (data);
+  
+      Int_t lab[4]   = {0,0,0,0};
+      Float_t hit[6] = {0,0,0,0,0,0};
+      Int_t info[3]  = {0,0,0};
+                                
+      lab[0]  = sp->fTracks[0];
+      lab[1]  = sp->fTracks[1];
+      lab[2]  = sp->fTracks[2];
+      lab[3]  = sp->fIndex;
+      hit[0]  = sp->fY;
+      hit[1]  = sp->fZ;
+      hit[2]  = sp->fSigmaY2;
+      hit[3]  = sp->fSigmaZ2;
+      hit[4]  = sp->fQ;
+      hit[5]  = sp->fSigmaYZ;
+      info[0] = sp->fNy;
+      info[1] = sp->fNz;
+      info[2] = sp->fLayer;
+      
+      Float_t xyz[3];
+      AliITSRecPoint recpoint(lab,hit,info);
+      recpoint.GetGlobalXYZ(xyz);
+
+      cont->SetNextPoint(xyz[0], xyz[1], xyz[2]);
+    }
+  }
+}