EVE/EveBase/AliEveHFEditor.cxx

   1 // @(#)root/eve:$Id$
   2 // Main author: Davide Caffarri 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
  10 #include "AliEveHFEditor.h"
  11 #include "AliEveHF.h"
  12
  13 #include "TVirtualPad.h"
  14 #include "TColor.h"
  15
  16 #include <TEveTrack.h>
  17
  18 // Cleanup these includes:
  19 #include "TGLabel.h"
  20 #include "TGButton.h"
  21 #include "TGNumberEntry.h"
  22 #include "TGColorSelect.h"
  23 #include "TGDoubleSlider.h"
  24
  25
  26 //______________________________________________________________________________
  27 // GUI editor for AliEveHF.
  28 //
  29
  30 ClassImp(AliEveHFEditor)
  31
  32 //______________________________________________________________________________
  33 AliEveHFEditor::AliEveHFEditor(const TGWindow *p, Int_t width, Int_t height,
  34                                UInt_t options, Pixel_t back) :
  35   TGedFrame(p, width, height, options | kVerticalFrame, back),
  36   fM(0),
  37   fnProng(0),
  38   fInfoLabel0(0),
  39   fInfoLabel1(0),
  40   fInfoLabel2(0),
  41   fXButton(0)
  42   // Initialize widget pointers to 0
  43 {
  44   // Constructor.
  45
  46   MakeTitle("AliEveHF");
  47
  48   fInfoLabel0 = new TGLabel(this);
  49   fInfoLabel0->SetTextJustify(kTextLeft);
  50   AddFrame(fInfoLabel0, new TGLayoutHints(kLHintsLeft|kLHintsExpandX,
  51                                           8, 0, 2, 0));
  52
  53   fInfoLabel1 = new TGLabel(this);
  54   fInfoLabel1->SetTextJustify(kTextLeft);
  55   AddFrame(fInfoLabel1, new TGLayoutHints(kLHintsLeft|kLHintsExpandX,
  56                                           8, 0, 2, 0));
  57
  58   fInfoLabel2 = new TGLabel(this);
  59   fInfoLabel2->SetTextJustify(kTextLeft);
  60   AddFrame(fInfoLabel2, new TGLayoutHints(kLHintsLeft|kLHintsExpandX,
  61                                           8, 0, 2, 0));
  62
  63   fXButton = new TGTextButton(this, "Detailed View");
  64   AddFrame(fXButton, new TGLayoutHints(kLHintsLeft|kLHintsExpandX, 1, 1, 0, 0));
  65   fXButton->Connect("Clicked()", "AliEveHFEditor", this, "DisplayDetailed()");
  66 }
  67
  68 /******************************************************************************/
  69
  70 //______________________________________________________________________________
  71 void AliEveHFEditor::SetModel(TObject* obj)
  72 {
  73   // Set model object.
  74
  75   fM = dynamic_cast<AliEveHF*>(obj);
  76
  77   // Set values of widgets
  78   fInfoLabel0->SetText(Form("CosPointingAngle = %f",  fM->GetCosPointingAngle()));
  79   fInfoLabel1->SetText(Form("Pt = %f", fM->GetPt()));
  80
  81   fInfoLabel2->SetText(Form("PDG Invariant Mass = %f", TDatabasePDG::Instance()->GetParticle(421)->Mass()));
  82
  83 }
  84
  85 /******************************************************************************/
  86
  87 // Implements callback/slot methods
  88
  89 //______________________________________________________________________________
  90 // void AliEveHFEditor::DoXYZZ()
  91 // {
  92 //    // Slot for XYZZ.
  93 //
  94 //    fM->SetXYZZ(fXYZZ->GetValue());
  95 //    Update();
  96 // }
  97
  98 #include <TEveManager.h>
  99 #include <TEveWindow.h>
 100 #include <TEveViewer.h>
 101 #include <TEveScene.h>
 102 #include <TEveGeoNode.h>
 103 #include <TEveProjectionManager.h>
 104
 105 #include <TGLCamera.h>
 106 #include <TGLViewer.h>
 107 #include "TGLCameraOverlay.h"
 108
 109 #include <TLatex.h>
 110 #include <TRootEmbeddedCanvas.h>
 111 #include <TInterpreter.h>
 112
 113 void AliEveHFEditor::DisplayDetailed()
 114 {
 115   TEveWindowSlot *slot = TEveWindow::CreateWindowMainFrame();
 116   TEveWindowPack *pack = slot->MakePack();
 117   pack->SetShowTitleBar(kFALSE);
 118   pack->SetHorizontal();
 119
 120   //
 121   // This part is for getting the different objects to display
 122   //
 123   char displayInfo[100] = {0};
 124   sprintf(displayInfo,"pt = %.3f",fM->GetPt());
 125   TEveLine *lhfTransverseMomentumDirection = new TEveLine(displayInfo);
 126   lhfTransverseMomentumDirection->SetLineColor(kOrange+8);
 127   lhfTransverseMomentumDirection->SetLineWidth(2);
 128   lhfTransverseMomentumDirection->SetLineStyle(2);
 129   lhfTransverseMomentumDirection->SetLineWidth(2);
 130   Float_t scalePt = 100.; // this needs to be available as a ruler
 131   lhfTransverseMomentumDirection->SetPoint(0,fM->fRecDecayHF.fX, fM->fRecDecayHF.fY, fM->fRecDecayHF.fZ);
 132   lhfTransverseMomentumDirection->SetPoint(1,scalePt*fM->fRecDecayP_HF.fX, scalePt*fM->fRecDecayP_HF.fY,0);
 133
 134   TEvePointSet *pvlocation = new TEvePointSet("PV location");
 135   pvlocation->SetNextPoint(fM->fRecBirthHF.fX, fM->fRecBirthHF.fY, fM->fRecBirthHF.fZ);
 136   pvlocation->SetTitle("pv location");
 137   pvlocation->SetMarkerStyle(4);
 138   pvlocation->SetMarkerSize(2.5);
 139   pvlocation->SetMarkerColor(7);
 140
 141   TEvePointSet *hflocation = new TEvePointSet("HF location");
 142   hflocation->SetNextPoint(fM->fRecDecayHF.fX, fM->fRecDecayHF.fY, fM->fRecDecayHF.fZ);
 143   hflocation->SetTitle("HF location");
 144   hflocation->SetMarkerStyle(4);
 145   hflocation->SetMarkerSize(2.5);
 146   hflocation->SetMarkerColor(kOrange+8);
 147
 148   TEveUtil::LoadMacro("clusters_from_index.C");
 149
 150   TEveTrack *negTrack =  fM->GetNegTrack();
 151   TEveTrack *posTrack =  fM->GetPosTrack();
 152
 153
 154   char macroWithIndex[100] = {0};
 155   Int_t daughterIndex = 0;
 156   TEvePointSet *negDaughterCluster = 0;
 157   TEvePointSet *posDaughterCluster = 0;
 158
 159   daughterIndex = negTrack->GetIndex();
 160   sprintf(macroWithIndex,"clusters_from_index(%d)",daughterIndex);
 161   Long_t negResult = gInterpreter->ProcessLine(macroWithIndex);
 162   if (negResult) {
 163     negDaughterCluster = reinterpret_cast<TEvePointSet*>(negResult);
 164     if (negDaughterCluster){
 165       negDaughterCluster->SetMarkerStyle(4);
 166       negDaughterCluster->SetMarkerSize(1.5);
 167       negDaughterCluster->SetMarkerColor(kBlue+3);
 168     }
 169   }
 170   else
 171   {
 172     Warning("DisplayDetailed", "Import of negative daughter's clusters failed.");
 173   }
 174
 175   daughterIndex = posTrack->GetIndex();
 176   sprintf(macroWithIndex,"clusters_from_index(%d)",daughterIndex);
 177   Long_t posResult = gInterpreter->ProcessLine(macroWithIndex);
 178   if (posResult) {
 179     posDaughterCluster = reinterpret_cast<TEvePointSet*>(posResult);
 180     if (posDaughterCluster){
 181       posDaughterCluster->SetMarkerStyle(4);
 182       posDaughterCluster->SetMarkerSize(1.5);
 183       posDaughterCluster->SetMarkerColor(kRed+3);
 184     }
 185   }
 186   else
 187   {
 188     Warning("DisplayDetailed", "Import of positive daughter's clusters failed.");
 189   }
 190
 191   //
 192   // This part is for the bending plane view
 193   //
 194   pack->NewSlot()->MakeCurrent();
 195   TEveViewer *bpViewer = gEve->SpawnNewViewer("HF bending plane View");
 196   TEveScene  *bpScene  = gEve->SpawnNewScene("HF bending plane Scene");
 197
 198   TEveUtil::LoadMacro("geom_gentle.C");
 199   Long_t result = gInterpreter->ProcessLine("geom_gentle_rphi()");
 200   if (result)
 201   {
 202     TEveGeoShape *geomRPhi = reinterpret_cast<TEveGeoShape*>(result);
 203     geomRPhi->IncDenyDestroy();
 204     TEveProjectionManager *projMgr = new TEveProjectionManager();
 205     projMgr->ImportElements(geomRPhi, bpScene);
 206   }
 207   else
 208   {
 209     Warning("DisplayDetailed", "Import of R-Phi geometry failed.");
 210   }
 211
 212   bpViewer->AddScene(bpScene);
 213   bpScene->AddElement(fM);
 214   bpScene->AddElement(lhfTransverseMomentumDirection);
 215   bpScene->AddElement(pvlocation);
 216   bpScene->AddElement(hflocation);
 217
 218   if (negDaughterCluster) bpScene->AddElement(negDaughterCluster);
 219   if (posDaughterCluster) bpScene->AddElement(posDaughterCluster);
 220
 221   // This is the to-do list for the bending plane:
 222   // 1. fix the view to orthographic XOY (no rotation allowed but moving the center ok) ->done!
 223   // 2. show axis and tickles along X and Y ->done!
 224   //       -> note for the projection the cartesian scales are not very useful
 225   //       -> propose a phi and R scale which rotates with a reset at 0;
 226   //       -> propose a transformation for an eta scale (keep the z one);
 227   // 3. show the center, the main vertex and the detectors for this view ->done!
 228   // 4. show V0 direction in the bending plane with arrow length proportional to pT ->done!
 229   // 5. show angles with respect to axis (phi angle) ->almost.
 230   // 6. show clusters in the ITS and in the TPC associated with the daughter tracks
 231   //       -> include a radius cut for plotting only ITS and TPC ->done!
 232   bpViewer->GetGLViewer()->SetCurrentCamera(TGLViewer::kCameraOrthoXOY);
 233   bpViewer->GetGLViewer()->ResetCamerasAfterNextUpdate();
 234   TGLViewer *lbpGLViewer = bpViewer->GetGLViewer();
 235   TGLCameraOverlay* co = lbpGLViewer->GetCameraOverlay();
 236   co->SetShowOrthographic(true); //(false);
 237   co->SetOrthographicMode(TGLCameraOverlay::kAxis); // ::kPlaneIntersect or ::kBar
 238   // end of the bending plane part
 239
 240   //
 241   // This part is for the decay plane view
 242   //
 243   pack->NewSlot()->MakeCurrent();
 244   TEveViewer *dpViewer = gEve->SpawnNewViewer("HF decay plane View");
 245   TEveScene  *dpScene  = gEve->SpawnNewScene("HF decay plane Scene");
 246
 247   dpViewer->AddScene(dpScene);
 248
 249   result = gInterpreter->ProcessLine("geom_gentle(kFALSE)");
 250   if (result)
 251   {
 252     TEveGeoShape *geom = reinterpret_cast<TEveGeoShape*>(result);
 253     geom->IncDenyDestroy();
 254     geom->FindChild("TRD+TOF")->SetRnrState(kFALSE);
 255     geom->FindChild("PHOS")   ->SetRnrState(kFALSE);
 256     geom->FindChild("HMPID")  ->SetRnrState(kFALSE);
 257     dpScene->AddElement(geom);
 258   }
 259   else
 260   {
 261     Warning("DisplayDetailed", "Import of 3D geometry failed.");
 262   }
 263
 264   dpScene->AddElement(fM);
 265   dpScene->AddElement(lhfTransverseMomentumDirection);
 266   dpScene->AddElement(pvlocation);
 267   dpScene->AddElement(hflocation);
 268   if (negDaughterCluster) dpScene->AddElement(negDaughterCluster);
 269   if (posDaughterCluster) dpScene->AddElement(posDaughterCluster);
 270
 271   // This is the to-do list for the decay plane:
 272   // 1. fix the view to decay plane (no rotation allowed but moving the center ok)
 273   // 2. show V0 direction with a vertical arrow length proportional to pT -> done!
 274   // 3. show the center, the main vertex and the detectors for this view -> done!
 275   // 4. show the x,y and z axis and the different angles
 276   //       -> this needs a referential object that we can move around
 277   //          or fix to a selected point (origin being the default)
 278   // 5. draw the dca between daughters and the extrapolation to the main vertex.
 279   //       -> this is an issue since we only store the distance: check with J.Belikov
 280   // 6. show clusters in the ITS and in the TPC associated with the daughter tracks
 281   //       -> include a radius cut for plotting only ITS and TPC ->done!
 282   dpViewer->GetGLViewer()->ResetCamerasAfterNextUpdate();
 283   TGLCamera& dpCam = dpViewer->GetGLViewer()->CurrentCamera();
 284   dpCam.SetExternalCenter(kTRUE);
 285   dpCam.SetCenterVec(fM->fRecDecayHF.fX, fM->fRecDecayHF.fY, fM->fRecDecayHF.fZ);
 286   dpCam.RotateRad(0,-TMath::Pi()/2.); // RotateRad rotates in radians (hRotate,vRotate)
 287   //  Here rotate the _view_ (not the camera) by (fM->GetPhi() - TMath::Pi()/2.)
 288
 289   // In the end maybe truck and rotate properly...
 290   //  dpCam.Truck(0,200);// Truck center wrt the view panel: (x=0 pixels, y pixels)
 291   //  dpCam.Rotate(0,50,0,0); // Rotate in pixels (xDelta,yDelta)
 292
 293   // end of the decay plane part
 294
 295   //
 296   // This part is for displaying the information
 297   //
 298   slot = pack->NewSlot();
 299
 300   TEveWindowFrame *frame = slot->MakeFrame(new TRootEmbeddedCanvas());
 301   frame->SetElementName("Details");
 302
 303   // Print and show detailed information about the HF
 304   // Calculation of the invariant mass with the max prob PID hypothesis first
 305   // pseudorapidity, phi angle, pt, radius, dcas
 306   char info[100] = {0};
 307
 308   sprintf(info,"Cos Pointing angle = %.3f ",fM->GetCosPointingAngle());
 309   TLatex* ltx = new TLatex(0.05, 0.9, info);
 310   ltx->SetTextSize(0.08);
 311   ltx->DrawLatex(0.05, 0.8, info);
 312
 313   sprintf(info,"p_{T} = %.3f [GeV/c]",fM->GetPt());
 314   ltx->DrawLatex(0.05, 0.7, info);
 315
 316   sprintf(info,"#eta = - ln( tan(#theta/2) ) = %.3f",fM->GetEta());
 317   ltx->DrawLatex(0.05, 0.6, info);
 318
 319   sprintf(info, "D^{0} inv. mass = %.3f", fM->GetInvariantMassPart());
 320   ltx->DrawLatex(0.05, 0.5, info);
 321
 322  sprintf(info, "D^{0}bar inv. mass = %.3f", fM->GetInvariantMassAntiPart());
 323   ltx->DrawLatex(0.05, 0.4, info);
 324
 325   gEve->Redraw3D();
 326 }