Provide return value.

[u/mrichter/AliRoot.git] / EVE / macros / reve_quad_test.C

diff --git a/EVE/macros/reve_quad_test.C b/EVE/macros/reve_quad_test.C
index 414c5d4f54ac76ebb540abd7a9f9fa704691c6c0..c99cc051d7f531f529e25398c2fd3a8c7db56247 100644 (file)
--- a/EVE/macros/reve_quad_test.C
+++ b/EVE/macros/reve_quad_test.C
@@ -1,20 +1,126 @@
 // $Header$
 
-void reve_quad_test()
+Reve::QuadSet* reve_quad_test(Float_t x=0, Float_t y=0, Float_t z=0,
+                             Int_t num=100)
 {
   TRandom r(0);
 
   gStyle->SetPalette(1, 0);
 
-  Reve::QuadSet* q = new Reve::QuadSet("Pepe");
-  q->Reset(Reve::QuadSet::QT_AxisAligned, kFALSE, 32);
-  for (Int_t i=0; i<128; ++i) {
+  Reve::QuadSet* q = new Reve::QuadSet("RectangleXY");
+  q->Reset(Reve::QuadSet::QT_RectangleXY, kFALSE, 32);
+  for (Int_t i=0; i<num; ++i) {
     q->AddQuad(r.Uniform(-10, 10), r.Uniform(-10, 10), r.Uniform(-10, 10),
-              r.Uniform(-1, 1), r.Uniform(-1, 1));
+              r.Uniform(0.2, 1), r.Uniform(0.2, 1));
     q->QuadValue(r.Uniform(0, 130));
   }
   q->RefitPlex();
 
+  Reve::ZTrans& t = q->RefHMTrans();
+  t.SetPos(x, y, z);
+
   gReve->AddRenderElement(q);
   gReve->Redraw3D();
+
+  return q;
+}
+
+Reve::QuadSet* reve_quad_test_circ()
+{
+  TRandom r(0);
+
+  gStyle->SetPalette(1, 0);
+
+  Reve::QuadSet* q = new Reve::QuadSet("Pepe");
+  q->Reset(Reve::QuadSet::QT_RectangleXY, kFALSE, 32);
+
+  Float_t R = 10, dW = 1, dH = .5;
+  for (Int_t i=0; i<12; ++i) {
+    Float_t x = R * TMath::Cos(TMath::TwoPi()*i/12);
+    Float_t y = R * TMath::Sin(TMath::TwoPi()*i/12);
+    q->AddQuad(x-dW, y-dH, r.Uniform(-1, 1), 2*dW, 2*dH);
+    q->QuadValue(r.Uniform(0, 130));
+  }
+  q->RefitPlex();
+
+  Reve::ZTrans& t = q->RefHMTrans();
+  t.SetPos(0, 0, 300);
+
+  gReve->AddRenderElement(q);
+  gReve->Redraw3D();
+
+  return q;
+}
+
+Reve::QuadSet* reve_quad_test_hex(Float_t x=0, Float_t y=0, Float_t z=0,
+                                 Int_t num=100)
+{
+  TRandom r(0);
+
+  gStyle->SetPalette(1, 0);
+
+  {
+    Reve::QuadSet* q = new Reve::QuadSet("HexagonXY");
+    q->Reset(Reve::QuadSet::QT_HexagonXY, kFALSE, 32);
+    for (Int_t i=0; i<num; ++i) {
+      q->AddHexagon(r.Uniform(-10, 10), r.Uniform(-10, 10), r.Uniform(-10, 10),
+                   r.Uniform(0.2, 1));
+      q->QuadValue(r.Uniform(0, 120));
+    }
+    q->RefitPlex();
+
+    Reve::ZTrans& t = q->RefHMTrans();
+    t.SetPos(x, y, z);
+
+    gReve->AddRenderElement(q);
+    gReve->Redraw3D();
+  }
+
+  {
+    Reve::QuadSet* q = new Reve::QuadSet("HexagonYX");
+    q->Reset(Reve::QuadSet::QT_HexagonYX, kFALSE, 32);
+    for (Int_t i=0; i<num; ++i) {
+      q->AddHexagon(r.Uniform(-10, 10), r.Uniform(-10, 10), r.Uniform(-10, 10),
+                   r.Uniform(0.2, 1));
+      q->QuadValue(r.Uniform(0, 120));
+    }
+    q->RefitPlex();
+
+    Reve::ZTrans& t = q->RefHMTrans();
+    t.SetPos(x, y, z);
+
+    gReve->AddRenderElement(q);
+    gReve->Redraw3D();
+  }
+
+  return q;
+}
+
+Reve::QuadSet* reve_quad_test_hexid(Float_t x=0, Float_t y=0, Float_t z=0,
+                                   Int_t num=100)
+{
+  TRandom r(0);
+
+  gStyle->SetPalette(1, 0);
+
+  {
+    Reve::QuadSet* q = new Reve::QuadSet("HexagonXY");
+    q->SetOwnIds(kTRUE);
+    q->Reset(Reve::QuadSet::QT_HexagonXY, kFALSE, 32);
+    for (Int_t i=0; i<num; ++i) {
+      q->AddHexagon(r.Uniform(-10, 10), r.Uniform(-10, 10), r.Uniform(-10, 10),
+                   r.Uniform(0.2, 1));
+      q->QuadValue(r.Uniform(0, 120));
+      q->QuadId(new TNamed(Form("Quad with idx=%d", i), "This title is not confusing."));
+    }
+    q->RefitPlex();
+
+    Reve::ZTrans& t = q->RefHMTrans();
+    t.SetPos(x, y, z);
+
+    gReve->AddRenderElement(q);
+    gReve->Redraw3D();
+  }
+
+  return q;
 }