From Alexandru: new/improved TRD macros.

[u/mrichter/AliRoot.git] / EVE / alice-macros / primary_vertex.C

// $Id$
// Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007

/**************************************************************************
 * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
 * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
 * full copyright notice.                                                 *
 **************************************************************************/

TPolyMarker3D* make_vertex_marker(AliESDVertex* v, const Text_t* name )
{
  Double_t x[3], e[3];
  v->GetXYZ(x);
  v->GetSigmaXYZ(e);

  printf("%16s: %f %f %f   -   %f %f %f\n", name,
	 x[0], x[1], x[2], e[0], e[1], e[2]);

  TPolyMarker3D* m = new TPolyMarker3D(1);
  m->SetName(name);
  m->SetPoint(0, x[0], x[1], x[2]);

  return m;
}

TEveBoxSet* make_vertex_boxes(AliESDVertex* v)
{
  Double_t x[3], e[3];
  v->GetXYZ(x);
  v->GetSigmaXYZ(e);

  TEveBoxSet* bs;

  bs = new TEveBoxSet("+- 10 x 10 x 20mm");
  bs->SetRenderMode(TEveBoxSet::RM_TEveLine);
  bs->AddBox(Box(-1, x[0], x[1], x[2], 1, 1, 2));
  bs->SetMainColor((Color_t) 2);
  gEve->AddElement(bs);

  bs = new TEveBoxSet("+- 30 sigma_r x 10 sigma_z");
  bs->SetRenderMode(TEveBoxSet::RM_TEveLine);
  bs->AddBox(Box(-1, x[0], x[1], x[2], 30*e[0], 30*e[1], 10*e[2]));
  bs->SetMainColor((Color_t) 3);
  gEve->AddElement(bs);

  gEve->Redraw3D();
}

void register_vertex_marker(TPolyMarker3D* m)
{
  Color_t* colp = TEveUtil::FindColorVar(m, "fMarkerColor");
  TEveElementObjectPtr* rnrEl = new TEveElementObjectPtr(m, *colp);
  gEve->AddElement(rnrEl);
  gEve->Redraw3D();
}

void primary_vertex_primitive(Bool_t showSPD=kTRUE, Bool_t showBoxes=kFALSE)
{
  AliESDEvent* esd = AliEveEventManager::AssertESD();

  AliESDVertex*  pv  = esd->GetPrimaryVertex();
  TPolyMarker3D* pvm = make_vertex_marker(pv, "Primary Vertex");
  pvm->SetMarkerStyle(5);
  pvm->SetMarkerColor(5);
  pvm->SetMarkerSize(1.4);
  register_vertex_marker(pvm);

  if (showSPD) {
    AliESDVertex*  spdv  = esd->GetPrimaryVertexSPD();
    TPolyMarker3D* spdvm = make_vertex_marker(spdv, "SPD Vertex");
    spdvm->SetMarkerStyle(2);
    spdvm->SetMarkerColor(6);
    register_vertex_marker(spdvm);
  }

  if (showBoxes)
    make_vertex_boxes(pv);
}

/******************************************************************************/

TEveStraightLineSet* make_vertex_lineset(AliESDVertex* v, const Text_t* name)
{
  Double_t x[3], e[3];
  v->GetXYZ(x); v->GetSigmaXYZ(e);
  printf("%16s: %f %f %f   -   %f %f %f\n", name,
	 x[0], x[1], x[2], e[0], e[1], e[2]);

  // dimensions
  TEveStraightLineSet* ls = new TEveStraightLineSet();
  ls->SetName(name);
  ls->AddLine(e[0], 0,    0,   -e[0], 0,    0);
  ls->AddLine(0,    e[1], 0,    0,   -e[1], 0);
  ls->AddLine(0,    0,    e[2], 0,    0,   -e[2]);

  ls->RefMainTrans().SetPos(x);
  return ls;
}

void make_vertex_ellipses(TEveStraightLineSet* ls, AliESDVertex* v, Bool_t ellipseUseSigma)
{
  Double_t x[3], e[3];
  v->GetXYZ(x); v->GetSigmaXYZ(e);

  if (ellipseUseSigma)
  {
    e[0] *= 30; e[1] *= 30; e[2] *= 10;
    ls->SetMarkerStyle(5);
    ls->SetMarkerColor(5);
    ls->SetMarkerSize(1.4);
    ls->SetLineColor(7);
    ls->SetTitle("+- 30 sigma_r x 10 sigma_z");
  }
  else
  {
    e[0] = 1; e[1] = 1; e[2] = 2;
    ls->SetMarkerStyle(2);
    ls->SetMarkerColor(6);
    ls->SetLineColor(6);
    ls->SetTitle("+- 10 x 10 x 20mm");
  }

  Int_t N = 32;
  Float_t S = 2*TMath::Pi()/N;
  Float_t b, a, phi;

  a = e[0]; b = e[1];
  for (Int_t i = 0; i<N; i++)
    ls->AddLine(a*TMath::Cos(i*S)  , b*TMath::Sin(i*S)  , 0,
		a*TMath::Cos(i*S+S), b*TMath::Sin(i*S+S), 0);

  a = e[0]; b = e[2];
  for (Int_t i = 0; i<N; i++)
    ls->AddLine(a*TMath::Cos(i*S)  , 0, b*TMath::Sin(i*S),
		a*TMath::Cos(i*S+S), 0, b*TMath::Sin(i*S+S));

  a = e[1]; b = e[2];
  for (Int_t i = 0; i<N; i++)
    ls->AddLine(0, a*TMath::Cos(i*S)  ,  b*TMath::Sin(i*S),
		0, a*TMath::Cos(i*S+S),  b*TMath::Sin(i*S+S));
}

void primary_vertex(Bool_t showSPD=kTRUE, Bool_t rnrEllipse=kTRUE)
{
  AliESDEvent* esd = AliEveEventManager::AssertESD();
  TEveStraightLineSet* ls;

  AliESDVertex* PV  =  esd->GetPrimaryVertex();
  ls = make_vertex_lineset(PV, "Primary Vertex");
  if (rnrEllipse) make_vertex_ellipses(ls, PV, kTRUE);
  gEve->AddElement(ls);

  if (showSPD)
  {
    AliESDVertex*  SPDV  = esd->GetPrimaryVertexSPD();
    ls = make_vertex_lineset(SPDV, "SPD Vertex");
    if (rnrEllipse) make_vertex_ellipses(ls, SPDV, kFALSE);
    gEve->AddElement(ls);
  }
}