DISPLAY/AliDisplayClusters.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15 /////////////////////////////////////////////////////////////////////////
  16 // ALICE DISPLAY CLUSTERS CLASS                                        //
  17 // Author: Mayeul   ROUSSELET                                          //
  18 // e-mail: Mayeul.Rousselet@cern.ch                                    //
  19 // Last update:26/08/2003                                              //
  20 /////////////////////////////////////////////////////////////////////////
  21
  22 #include <Riostream.h>
  23 #include <TFile.h>
  24 #include <TPolyMarker3D.h>
  25
  26 #include "AliClusters.h"
  27 #include "AliDisplay2.h"
  28 #include "AliDisplayClusters.h"
  29 #include "AliITSLoader.h"
  30 #include "AliITSclusterV2.h"
  31 #include "AliITSgeom.h"
  32 #include "AliModuleInfo.h"
  33 #include "AliRun.h"
  34 #include "AliRunLoader.h"
  35 #include "AliTPCLoader.h"
  36 #include "AliTPCParam.h"
  37 #include "AliTPCclusterMI.h"
  38
  39 ClassImp(AliDisplayClusters)
  40
  41 //_____________________________________________________________
  42 AliDisplayClusters::AliDisplayClusters()
  43 {
  44   //Default constructor
  45   fPoints = new TPolyMarker3D[gAliDisplay2->GetNbModules()];
  46   fName = new char*[gAliDisplay2->GetNbModules()];
  47   fNb=0;
  48   for(Int_t i=0;i<gAliDisplay2->GetNbModules();i++){
  49     fPoints[i].SetMarkerSize(0.2);
  50     fPoints[i].SetMarkerColor(2);
  51     fPoints[i].SetMarkerStyle(1);
  52   }
  53 }
  54
  55 //_____________________________________________________________
  56 AliDisplayClusters::~AliDisplayClusters()
  57 {
  58   // Destructor
  59   delete [] fPoints;
  60 }
  61
  62 //_____________________________________________________________
  63 Int_t AliDisplayClusters::GetNbClusters()
  64 {
  65   // Returns the number of clusters
  66   Int_t r=0;
  67   for(Int_t i=0;i<fNb;i++){
  68      if(gAliDisplay2->GetModuleInfo()->IsEnabled(fName[i])) r+=fPoints[i].GetN();
  69   }
  70   return r;
  71 }
  72
  73 //_____________________________________________________________
  74 void AliDisplayClusters::LoadClusters(const char *name,Int_t nevent)
  75 {
  76   // Loads ITS and TPC clusters
  77   if(strstr(name,"ITS")) LoadITSClusters(nevent);
  78   if(strstr(name,"TPC")) LoadTPCClusters(nevent);
  79 }
  80
  81 //_____________________________________________________________
  82 void AliDisplayClusters::LoadITSClusters(Int_t nevent)
  83 {
  84   // Loads ITS clusters
  85   fName[fNb]=new char[strlen("ITS")];
  86   strcpy(fName[fNb],"ITS");
  87   AliRunLoader *rl = AliRunLoader::Open("galice.root");
  88   if(!rl) {
  89     cerr<<"Can't open galice.root";
  90     return;
  91   }
  92   AliITSLoader *itsl = (AliITSLoader*)rl->GetLoader("ITSLoader");
  93
  94   rl->GetEvent(nevent);
  95   itsl->LoadRecPoints();
  96   TTree *cTree=itsl->TreeR();
  97   if(!cTree)
  98   {
  99     cerr<<"Error occured during ITS clusters load";
 100     return;
 101   }
 102
 103   TDirectory * olddir = gDirectory;
 104   rl->CdGAFile();
 105   AliITSgeom *geom = (AliITSgeom*)gDirectory->Get("AliITSgeom");
 106   olddir->cd();
 107   Int_t count = 0;
 108
 109   TClonesArray *clusters=new TClonesArray("AliITSclusterV2",10000);
 110   TBranch *branch=cTree->GetBranch("Clusters");
 111   branch->SetAddress(&clusters);
 112   Int_t nentr=(Int_t)cTree->GetEntries();
 113   for (Int_t i=0; i<nentr; i++) {
 114        if (!cTree->GetEvent(i)) continue;
 115
 116         Double_t rot[9];
 117         geom->GetRotMatrix(i,rot);
 118         Int_t lay,lad,det;
 119         geom->GetModuleId(i,lay,lad,det);
 120         Float_t tx,ty,tz;
 121         geom->GetTrans(lay,lad,det,tx,ty,tz);
 122
 123         Double_t r=-tx*rot[1]+ty*rot[0];
 124         if (lay==1) r=-r;
 125         Double_t phi=TMath::ATan2(rot[1],rot[0]);
 126         if (lay==1) phi-=3.1415927;
 127         Double_t cp=TMath::Cos(phi), sp=TMath::Sin(phi);
 128
 129         Int_t ncl=clusters->GetEntriesFast();
 130         while (ncl--) {
 131                 AliITSclusterV2 *c=(AliITSclusterV2*)clusters->UncheckedAt(ncl);
 132                 Double_t g[3];
 133                 g[0]= r*cp + c->GetY()*sp; //
 134                 g[1]=-r*sp + c->GetY()*cp; //
 135                 g[2]=c->GetZ();
 136                 fPoints[fNb].SetPoint(count,g[0],g[1],g[2]);
 137                 count++;
 138         }
 139    }
 140   fNb++;
 141 }
 142
 143 //_____________________________________________________________
 144 void AliDisplayClusters::LoadTPCClusters(Int_t nevent)
 145 {
 146   // Loads TPC clusters
 147   fName[fNb]=new char[strlen("TPC")];
 148   strcpy(fName[fNb],"TPC");
 149   TFile *file = TFile::Open("galice.root");
 150   AliTPCParam *dig=(AliTPCParam *)file->Get("75x40_100x60_150x60");
 151   if (!dig) {cerr<<"TPC parameters have not been found !\n";}
 152   file->Close();
 153
 154   AliRunLoader *rl = AliRunLoader::Open("galice.root");
 155   if(!rl) {
 156     cerr<<"Can't open galice.root";
 157     return;
 158   }
 159   AliTPCLoader *itsl = (AliTPCLoader*)rl->GetLoader("TPCLoader");
 160   if(!itsl){
 161     cerr<<"Can't find Loader";
 162     return;
 163   }
 164
 165   rl->GetEvent(nevent);
 166   itsl->LoadRecPoints();
 167   TTree *cTree=itsl->TreeR();
 168   if(!cTree)
 169   {
 170     cerr<<"Error during TPC clusters load";
 171     return;
 172   }
 173
 174   Int_t count = 0;
 175   Float_t noiseth = 10;
 176
 177    AliClusters *clusters=new AliClusters();
 178    clusters->SetClass("AliTPCclusterMI");
 179
 180    cTree->SetBranchAddress("Segment",&clusters);
 181
 182    Int_t nrows=Int_t(cTree->GetEntries());
 183    for (Int_t n=0; n<nrows; n++) {
 184        cTree->GetEvent(n);
 185        Int_t sec,row;
 186        dig->AdjustSectorRow(clusters->GetID(),sec,row);
 187        TClonesArray &clrow=*clusters->GetArray();
 188        Int_t ncl=clrow.GetEntriesFast();
 189        while (ncl--) {
 190            AliTPCclusterMI *cl=(AliTPCclusterMI*)clrow[ncl];
 191            Double_t x=dig->GetPadRowRadii(sec,row), y=cl->GetY(), z=cl->GetZ();
 192            if (cl->GetQ()<noiseth) continue;
 193            Float_t cs, sn, tmp;
 194            dig->AdjustCosSin(sec,cs,sn);
 195            tmp = x*cs-y*sn; y= x*sn+y*cs; x=tmp;
 196            fPoints[fNb].SetPoint(count,x,y,z);
 197            count++;
 198        }
 199        clrow.Clear();
 200
 201    }
 202    delete cTree;
 203    delete dig;
 204    fNb++;
 205
 206 }
 207
 208 //_____________________________________________________________
 209 void AliDisplayClusters::Draw()
 210 {
 211   // Draws clusters
 212   for(Int_t i=0;i<fNb;i++){
 213     if(gAliDisplay2->GetModuleInfo()->IsEnabled(fName[i])) fPoints[i].Draw();
 214   }
 215 }
 216