EVE/alice-macros/esd_kink_points.C

   1 // $Id$
   2 // Main authors: Paraskevi Ganoti: 2009
   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 #if !defined(__CINT__) || defined(__MAKECINT__)
  10 #include <TMath.h>
  11 #include <TVector3.h>
  12 #include <TEvePointSet.h>
  13 #include <TEveManager.h>
  14
  15 #include <AliESDEvent.h>
  16 #include <AliESDtrack.h>
  17 #include <AliESDkink.h>
  18 #include <AliEveEventManager.h>
  19 #endif
  20
  21 void esd_kink_fill_pointset(TEvePointSet* ps)
  22 {
  23   AliESDEvent* esd = AliEveEventManager::AssertESD();
  24
  25   for (Int_t n=0; n<esd->GetNumberOfTracks(); ++n)
  26   {
  27       AliESDtrack* track = esd->GetTrack(n);
  28       if(track->GetKinkIndex(0)<0){
  29
  30           AliESDkink *kink = esd->GetKink(TMath::Abs(track->GetKinkIndex(0))-1);
  31           const TVector3 Position(kink->GetPosition());
  32           ps->SetNextPoint(Position.X(), Position.Y(), Position.Z());
  33           ps->SetPointId(kink);
  34       }
  35   }
  36
  37 }
  38
  39 TEvePointSet* esd_kink_points()
  40 {
  41   TEvePointSet* points = new TEvePointSet("Kink vertex locations");
  42
  43   esd_kink_fill_pointset(points);
  44
  45   points->SetTitle(Form("N=%d", points->Size()));
  46   points->SetMarkerStyle(4);
  47   points->SetMarkerSize(1.5);
  48   points->SetMarkerColor(kOrange+8);
  49
  50   gEve->AddElement(points);
  51   gEve->Redraw3D();
  52
  53   return points;
  54 }