Logo ROOT   6.30.04
Reference Guide
 All Namespaces Files Pages
csgdemo.C
Go to the documentation of this file.
1 /// \file
2 /// \ingroup tutorial_geom
3 /// Combinatorial Solid Geometry example.
4 ///
5 /// \macro_code
6 ///
7 /// \author Andrei Gheata
8 
9 Bool_t raytracing = kTRUE;
10 
11 #include "TGeoManager.h"
12 //______________________________________________________________________________
13 void csgdemo ()
14 {
15  gSystem->Load("libGeom");
16  TControlBar* bar = new TControlBar("vertical", "TGeo composite shapes",20,20);
17  bar->AddButton("How to run ","help()","Instructions ");
18  bar->AddButton("Union ", "s_union()", "A + B ");
19  bar->AddButton("Intersection ", "s_intersection()","A * B ");
20  bar->AddButton("Difference ", "s_difference()","A - B ");
21  //These two buttons are disabled cause of bugs
22  bar->AddButton("Complex composite", "complex_1()","(A * B) + (C - D)");
23  bar->Show();
24 
25  gROOT->SaveContext();
26 }
27 
28 //______________________________________________________________________________
29 void MakePicture()
30 {
31  gGeoManager->GetGeomPainter()->SetRaytracing(kFALSE);
32  gPad->Modified();
33  gPad->Update();
34 }
35 
36 
37 void s_union()
38 {
39  gROOT->GetListOfCanvases()->Delete();
40  TCanvas *c = new TCanvas("composite shape", "Union boolean operation", 700, 1000);
41 
42  c->Divide(1,2,0,0);
43  c->cd(2);
44  gPad->SetPad(0,0,1,0.4);
45  c->cd(1);
46  gPad->SetPad(0,0.4,1,1);
47 
48  if (gGeoManager) delete gGeoManager;
49 
50  new TGeoManager("xtru", "poza12");
51  TGeoMaterial *mat = new TGeoMaterial("Al", 26.98,13,2.7);
52  TGeoMedium *med = new TGeoMedium("MED",1,mat);
53  TGeoVolume *top = gGeoManager->MakeBox("TOP",med,100,100,100);
54  gGeoManager->SetTopVolume(top);
55 
56  // define shape components with names
57  TGeoPgon *pgon = new TGeoPgon("pg",0.,360.,6,2);
58  pgon->DefineSection(0,0,0,20);
59  pgon->DefineSection(1, 30,0,20);
60 
61  TGeoSphere *sph = new TGeoSphere("sph", 40., 45., 5., 175., 0., 340.);
62  // define named geometrical transformations with names
63  TGeoTranslation *tr = new TGeoTranslation(0., 0., 45.);
64  tr->SetName("tr");
65  // register all used transformations
66  tr->RegisterYourself();
67  // create the composite shape based on a Boolean expression
68  TGeoCompositeShape *cs = new TGeoCompositeShape("mir", "sph:tr + pg");
69 
70  TGeoVolume *vol = new TGeoVolume("COMP1",cs);
71  top->AddNode(vol,1);
72  gGeoManager->CloseGeometry();
73  gGeoManager->SetNsegments(100);
74  top->Draw();
75  MakePicture();
76 
77  c->cd(2);
78  TPaveText *pt = new TPaveText(0.01,0.01,0.99,0.99);
79  pt->SetLineColor(1);
80  TText *text = pt->AddText("TGeoCompositeShape - composite shape class");
81  text->SetTextColor(2);
82  pt->AddText("----- It's an example of boolean union operation : A + B");
83  pt->AddText("----- A == part of sphere (5-175, 0-340), B == pgon");
84  pt->AddText(" ");
85  pt->SetAllWith("-----","color",4);
86  pt->SetAllWith("-----","font",72);
87  pt->SetAllWith("-----","size",0.04);
88  pt->SetTextAlign(12);
89  pt->SetTextSize(.044);
90  pt->Draw();
91  c->cd(1);
92 }
93 
94 void s_intersection()
95 {
96  gROOT->GetListOfCanvases()->Delete();
97  TCanvas *c = new TCanvas("composite shape", "Intersection boolean operation", 700, 1000);
98 
99  c->Divide(1,2,0,0);
100  c->cd(2);
101  gPad->SetPad(0,0,1,0.4);
102  c->cd(1);
103  gPad->SetPad(0,0.4,1,1);
104 
105  if (gGeoManager) delete gGeoManager;
106 
107  new TGeoManager("xtru", "poza12");
108  TGeoMaterial *mat = new TGeoMaterial("Al", 26.98,13,2.7);
109  TGeoMedium *med = new TGeoMedium("MED",1,mat);
110  TGeoVolume *top = gGeoManager->MakeBox("TOP",med,100,100,100);
111  gGeoManager->SetTopVolume(top);
112 
113  // define shape components with names
114  TGeoBBox *box = new TGeoBBox("bx", 40., 40., 40.);
115  TGeoSphere *sph = new TGeoSphere("sph", 40., 45.);
116  // define named geometrical transformations with names
117  TGeoTranslation *tr = new TGeoTranslation(0., 0., 45.);
118  tr->SetName("tr");
119  // register all used transformations
120  tr->RegisterYourself();
121  // create the composite shape based on a Boolean expression
122  TGeoCompositeShape *cs = new TGeoCompositeShape("mir", "sph:tr * bx");
123 
124  TGeoVolume *vol = new TGeoVolume("COMP2",cs);
125  top->AddNode(vol,1);
126  gGeoManager->CloseGeometry();
127  gGeoManager->SetNsegments(100);
128  top->Draw();
129  MakePicture();
130 
131  c->cd(2);
132 
133  TPaveText *pt = new TPaveText(0.01,0.01,0.99,0.99);
134 
135  pt->SetLineColor(1);
136 
137  TText *text = pt->AddText("TGeoCompositeShape - composite shape class");
138 
139  text->SetTextColor(2);
140  pt->AddText("----- Here is an example of boolean intersection operation : A * B");
141  pt->AddText("----- A == sphere (with inner radius non-zero), B == box");
142  pt->AddText(" ");
143  pt->SetAllWith("-----","color",4);
144  pt->SetAllWith("-----","font",72);
145  pt->SetAllWith("-----","size",0.04);
146  pt->SetTextAlign(12);
147  pt->SetTextSize(0.044);
148  pt->Draw();
149  c->cd(1);
150 }
151 
152 void s_difference()
153 {
154  gROOT->GetListOfCanvases()->Delete();
155  TCanvas *c = new TCanvas("composite shape", "Difference boolean operation", 700, 1000);
156 
157  c->Divide(1,2,0,0);
158  c->cd(2);
159  gPad->SetPad(0,0,1,0.4);
160  c->cd(1);
161  gPad->SetPad(0,0.4,1,1);
162 
163  if (gGeoManager) delete gGeoManager;
164 
165  new TGeoManager("xtru", "poza12");
166  TGeoMaterial *mat = new TGeoMaterial("Al", 26.98,13,2.7);
167  TGeoMedium *med = new TGeoMedium("MED",1,mat);
168  TGeoVolume *top = gGeoManager->MakeBox("TOP",med,100,100,100);
169  gGeoManager->SetTopVolume(top);
170 
171  // define shape components with names
172  TGeoTorus *tor = new TGeoTorus("tor", 45., 15., 20., 45., 145.);
173  TGeoSphere *sph = new TGeoSphere("sph", 20., 45., 0., 180., 0., 270.);
174  // create the composite shape based on a Boolean expression
175  TGeoCompositeShape *cs = new TGeoCompositeShape("mir", "sph - tor");
176 
177  TGeoVolume *vol = new TGeoVolume("COMP3",cs);
178  top->AddNode(vol,1);
179  gGeoManager->CloseGeometry();
180  gGeoManager->SetNsegments(60);
181  top->Draw();
182  MakePicture();
183 
184  c->cd(2);
185 
186  TPaveText *pt = new TPaveText(.01, .01, .99, .99);
187 
188  pt->SetLineColor(1);
189 
190  TText *text = pt->AddText("TGeoCompositeShape - composite shape class");
191 
192  text->SetTextColor(2);
193 
194  pt->AddText("----- It's an example of boolean difference: A - B");
195  pt->AddText("----- A == part of sphere (0-180, 0-270), B == partial torus (45-145)");
196  pt->AddText(" ");
197  pt->SetAllWith("-----","color",4);
198  pt->SetAllWith("-----","font",72);
199  pt->SetAllWith("-----","size",0.04);
200  pt->SetTextAlign(12);
201  pt->SetTextSize(0.044);
202  pt->Draw();
203  c->cd(1);
204 }
205 
206 void complex_1()
207 {
208  gROOT->GetListOfCanvases()->Delete();
209  TCanvas *c = new TCanvas("composite shape", "A * B - C", 700, 1000);
210 
211  c->Divide(1,2,0,0);
212  c->cd(2);
213  gPad->SetPad(0,0,1,0.4);
214  c->cd(1);
215  gPad->SetPad(0,0.4,1,1);
216 
217  if (gGeoManager) delete gGeoManager;
218 
219  new TGeoManager("xtru", "poza12");
220  TGeoMaterial *mat = new TGeoMaterial("Al", 26.98,13,2.7);
221  TGeoMedium *med = new TGeoMedium("MED",1,mat);
222  TGeoVolume *top = gGeoManager->MakeBox("TOP",med,100,100,100);
223  gGeoManager->SetTopVolume(top);
224 
225  // define shape components with names
226  TGeoBBox *box = new TGeoBBox("box", 20., 20., 20.);
227  TGeoBBox *box1 = new TGeoBBox("box1", 5., 5., 5.);
228  TGeoSphere *sph = new TGeoSphere("sph", 5., 25.);
229  TGeoSphere *sph1 = new TGeoSphere("sph1", 1., 15.);
230  // create the composite shape based on a Boolean expression
231  TGeoTranslation *tr = new TGeoTranslation(0., 30., 0.);
232  TGeoTranslation *tr1 = new TGeoTranslation(0., 40., 0.);
233  TGeoTranslation *tr2 = new TGeoTranslation(0., 30., 0.);
234  TGeoTranslation *tr3 = new TGeoTranslation(0., 30., 0.);
235  tr->SetName("tr");
236  tr1->SetName("tr1");
237  tr2->SetName("tr2");
238  tr3->SetName("tr3");
239  // register all used transformations
240  tr->RegisterYourself();
241  tr1->RegisterYourself();
242  tr2->RegisterYourself();
243  tr3->RegisterYourself();
244 
245  TGeoCompositeShape *cs = new TGeoCompositeShape("mir", "(sph * box) + (sph1:tr - box1:tr1)");
246 
247  TGeoVolume *vol = new TGeoVolume("COMP4",cs);
248 // vol->SetLineColor(randomColor());
249  top->AddNode(vol,1);
250  gGeoManager->CloseGeometry();
251  gGeoManager->SetNsegments(80);
252  top->Draw();
253  MakePicture();
254 
255  c->cd(2);
256  TPaveText *pt = new TPaveText(0.01,0.01,0.99,0.99);
257  pt->SetLineColor(1);
258  TText *text = pt->AddText("TGeoCompositeShape - composite shape class");
259  text->SetTextColor(2);
260  pt->AddText("----- (sphere * box) + (sphere - box) ");
261 
262  pt->AddText(" ");
263  pt->SetAllWith("-----","color",4);
264  pt->SetAllWith("-----","font",72);
265  pt->SetAllWith("-----","size",0.04);
266  pt->SetTextAlign(12);
267  pt->SetTextSize(0.044);
268  pt->Draw();
269  c->cd(1);
270 
271 }
272 
273 //______________________________________________________________________________
274 void AddText(TPaveText *pave, const char *datamember, Double_t value, const char *comment)
275 {
276  char line[128];
277  for (Int_t i=0; i<128; i++) line[i] = ' ';
278  memcpy(&line[0], datamember, strlen(datamember));
279  line[10] = '=';
280  char number[20];
281  sprintf(number, "%5.2f", value);
282  memcpy(&line[12], number, strlen(number));
283  line[26] = '=';
284  line[27] = '>';
285  sprintf(&line[30], "%s",comment);
286  TText *text = pave->AddText(line);
287  text->SetTextAlign(12);
288 }
289 
290 //______________________________________________________________________________
291 void AddText(TPaveText *pave, const char *datamember, Int_t value, const char *comment)
292 {
293  char line[128];
294  for (Int_t i=0; i<128; i++) line[i] = ' ';
295  memcpy(&line[0], datamember, strlen(datamember));
296  line[10] = '=';
297  char number[20];
298  sprintf(number, "%5i", value);
299  memcpy(&line[12], number, strlen(number));
300  line[26] = '=';
301  line[27] = '>';
302  sprintf(&line[30], "%s",comment);
303  TText *text = pave->AddText(line);
304  text->SetTextAlign(12);
305 }
306 
307 //______________________________________________________________________________
308 void AddText(TPaveText *pave, TObject *pf, Int_t iaxis)
309 {
310  char line[128];
311  TGeoPatternFinder *finder = (TGeoPatternFinder*)pf;
312  if (!pave || !pf) return;
313  for (Int_t i=0; i<128; i++) line[i] = ' ';
314  TGeoVolume *volume = finder->GetVolume();
315  TGeoShape *sh = volume->GetShape();
316  sprintf(line, "Division of %s on axis %d (%s)", volume->GetName(), iaxis,sh->GetAxisName(iaxis));
317  TText *text = pave->AddText(line);
318  text->SetTextColor(3);
319  text->SetTextAlign(12);
320  AddText(pave, "fNdiv",finder->GetNdiv(),"number of divisions");
321  AddText(pave, "fStart",finder->GetStart(),"start divisioning position");
322  AddText(pave, "fStep",finder->GetStep(),"division step");
323 }
324 
325 //______________________________________________________________________________
326 void raytrace() {
327  if (!gGeoManager) return;
328 
329  TVirtualGeoPainter *painter = gGeoManager->GetGeomPainter();
330 
331  if (!painter) return;
332 
333  painter->SetRaytracing(raytracing);
334 
335  if (!gPad) return;
336 
337  gPad->Modified();
338  gPad->Update();
339 }
340 
341 //______________________________________________________________________________
342 void help() {
343  //
344 
345  new TCanvas("chelp","Help to run demos",200,10,700,600);
346 
347  TPaveText* welcome = new TPaveText(.1,.8,.9,.97);
348  welcome->AddText("Welcome to the new geometry package");
349  welcome->SetTextFont(32);
350  welcome->SetTextColor(4);
351  welcome->SetFillColor(24);
352  welcome->Draw();
353 
354  TPaveText* hdemo = new TPaveText(.05,.05,.95,.7);
355  hdemo->SetTextAlign(12);
356  hdemo->SetTextFont(52);
357  hdemo->AddText("- Demo for building TGeo composite shapes");
358  hdemo->AddText(" ");
359  hdemo->SetAllWith("....","color",2);
360  hdemo->SetAllWith("....","font",72);
361  hdemo->SetAllWith("....","size",0.03);
362 
363  hdemo->Draw();
364 }
365