EVE/alice-macros/tpc_clusters.C

   1 // $Id$
   2 // Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007
   3
   4 /**************************************************************************
   5  * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
   6  * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
   7  * full copyright notice.                                                 *
   8  **************************************************************************/
   9
  10 #ifdef __CINT__
  11
  12 class TEveElement;
  13 class TEvePointSet;
  14
  15 #else
  16
  17 #include <TEveManager.h>
  18 #include <TEvePointSet.h>
  19 #include <EveBase/AliEveEventManager.h>
  20
  21 #include <AliRunLoader.h>
  22 #include <AliCluster.h>
  23 #include <TPC/AliTPCClustersRow.h>
  24
  25 #endif
  26
  27 TEvePointSet* tpc_clusters(TEveElement* cont=0, Float_t maxR=270)
  28 {
  29   const Int_t kMaxCl=100*160;
  30
  31   AliEveEventManager::AssertGeometry();
  32
  33   TEvePointSet* clusters = new TEvePointSet(kMaxCl);
  34   clusters->SetOwnIds(kTRUE);
  35
  36   AliRunLoader* rl = AliEveEventManager::AssertRunLoader();
  37   rl->LoadRecPoints("TPC");
  38
  39   AliTPCClustersRow *clrow=new AliTPCClustersRow();
  40   clrow->SetClass("AliTPCclusterMI");
  41   clrow->SetArray(kMaxCl);
  42
  43   TTree *cTree = rl->GetTreeR("TPC", false);
  44   cTree->SetBranchAddress("Segment", &clrow);
  45
  46   Float_t maxRsqr = maxR*maxR;
  47   TClonesArray *cl=clrow->GetArray();
  48   Int_t nentr=(Int_t)cTree->GetEntries();
  49   for (Int_t i=0; i<nentr; i++) {
  50     if (!cTree->GetEvent(i)) continue;
  51
  52     Int_t ncl=cl->GetEntriesFast();
  53
  54     while (ncl--) {
  55       AliCluster *c=(AliCluster*)cl->UncheckedAt(ncl);
  56       Float_t g[3]; //global coordinates
  57       c->GetGlobalXYZ(g);
  58       if (g[0]*g[0]+g[1]*g[1] < maxRsqr)
  59       {
  60         clusters->SetNextPoint(g[0], g[1], g[2]);
  61         AliCluster *atp = new AliCluster(*c);
  62         clusters->SetPointId(atp);
  63       }
  64     }
  65     cl->Clear();
  66   }
  67
  68   delete clrow;
  69
  70   if(clusters->Size() == 0 && gEve->GetKeepEmptyCont() == kFALSE) {
  71     Warning("tpc_clusters", "No TPC clusters");
  72     delete clusters;
  73     return 0;
  74   }
  75
  76   clusters->SetMarkerStyle(2);
  77   clusters->SetMarkerSize(0.2);
  78   clusters->SetMarkerColor(4);
  79
  80   char form[1000];
  81   sprintf(form,"TPC Clusters");
  82   clusters->SetName(form);
  83
  84   char tip[1000];
  85   sprintf(tip,"N=%d", clusters->Size());
  86   clusters->SetTitle(tip);
  87   gEve->AddElement(clusters, cont);
  88   gEve->Redraw3D();
  89
  90   return clusters;
  91 }