boxset_cones.C

Go to the documentation of this file.

/// \file
/// \ingroup tutorial_eve
/// Demonstrates usage of 'cone' mode in REveBoxSet class.
///
/// \image html eve_boxset_cones.png
/// \macro_code
///
/// \author Alja Mrak-Tadel
 
using namespace ROOT::Experimental;
 
REveBoxSet* elliptic_boxset_cones(Float_t x=0, Float_t y=0, Float_t z=0,
                                  Int_t num=100)
{
   auto eveMng = REveManager::Create();
 
   using namespace TMath;
 
   REveManager::Create();
 
   auto lines = new REveStraightLineSet("StraightLines");
   lines->SetLineColor(kYellow);
   lines->SetLineWidth(2);
 
   TRandom r(0);
 
   auto cones = new REveBoxSet("EllipticConeSet");
   bool valIsColor = true;
   cones->Reset(REveBoxSet::kBT_InstancedScaledRotated, valIsColor, 64);
 
   cones->SetPickable(kTRUE);
 
   Float_t a = 40; // max distance between cones
   REveVector dir, pos;
   Float_t theta, phi, height, rad;
   for (Int_t i=0; i<num; ++i) {
      theta  = r.Uniform(0,TMath::Pi());
      phi    = r.Uniform (-TMath::Pi(), TMath::Pi());
      height = r.Uniform(5, 15);
      rad    = r.Uniform(3, 5);
      dir.Set(Cos(phi)*Cos(theta), Sin(phi)*Cos(theta), Sin(theta));
      dir *= height;
      pos.Set(r.Uniform(-a,a), r.Uniform(-a, a), r.Uniform(-a, a));
 
      cones->AddEllipticCone(pos, dir, rad, 0.5*rad, r.Uniform(0,360));
      if (i %2 == 0)
      cones->DigitColor(kRed);
      else
      cones->DigitColor(kGreen);
      // draw axis line 30% longer than cone height
      REveVector end = pos + dir*1.3f;
      lines->AddLine(pos.fX, pos.fY, pos.fZ, end.fX, end.fY, end.fZ);
   }
 
   // by default cone cap not drawn
   if (r.Integer(2)>0)  cones->SetDrawConeCap(kTRUE);
 
   cones->RefitPlex();
   REveTrans& t = cones->RefMainTrans();
   t.SetPos(x, y, z);
   cones->SetPickable(1);
   cones->SetAlwaysSecSelect(1);
 
   eveMng->GetEventScene()->AddElement(cones);
   eveMng->GetEventScene()->AddElement(lines);
 
   REveViewer* v = (REveViewer*)eveMng->GetViewers()->FirstChild();
   v->SetAxesType(REveViewer::kAxesOrigin);
 
   eveMng->Show();
   return cones;
}
 
 
REveBoxSet* boxset_cones(Float_t x=0, Float_t y=0, Float_t z=0,
                         Int_t num=100)
{
   auto eveMng = REveManager::Create();
 
   using namespace TMath;
 
   auto lines = new REveStraightLineSet("StraightLines");
   lines->SetLineColor(kYellow);
   lines->SetLineWidth(2);
 
   TRandom r(0);
   auto pal   = new REveRGBAPalette(0, 500);
   auto cones = new REveBoxSet("ConeSet");
   cones->SetPalette(pal);
   bool valIsColor = false;
   cones->Reset(REveBoxSet::kBT_InstancedScaledRotated, valIsColor, 64);
 
   Float_t a = 40; // max distance between cones
   REveVector dir, pos;
   Float_t theta, phi, height, rad;
   for (Int_t i=0; i<num; ++i) {
      theta  = r.Uniform(0,TMath::Pi());
      phi    = r.Uniform (-TMath::Pi(), TMath::Pi());
      height = r.Uniform(5, 15);
      rad    = r.Uniform(3, 5);
      dir.Set(Cos(phi)*Cos(theta), Sin(phi)*Cos(theta), Sin(theta));
      dir *= height;
      pos.Set(r.Uniform(-a,a), r.Uniform(-a, a), r.Uniform(-a, a));
 
      cones->AddCone(pos, dir, rad);
      cones->DigitValue(r.Uniform(0, 500));
 
      // draw axis line 30% longer than cone height
      REveVector end = pos + dir*1.3f;
      lines->AddLine(pos.fX, pos.fY, pos.fZ, end.fX, end.fY, end.fZ);
   }
 
   cones->SetPickable(1);
   cones->SetAlwaysSecSelect(1);
   // by default cone cap not drawn
   if (r.Integer(2)>0)  cones->SetDrawConeCap(kTRUE);
 
   cones->RefitPlex();
   REveTrans& t = cones->RefMainTrans();
   t.SetPos(x, y, z);
 
   eveMng->GetEventScene()->AddElement(cones);
   eveMng->GetEventScene()->AddElement(lines);
   eveMng->GetEventScene()->AddElement(pal);
   eveMng->Show();
 
   return cones;
}