-
Notifications
You must be signed in to change notification settings - Fork 6
/
materialDisplay.C
51 lines (49 loc) · 1.58 KB
/
materialDisplay.C
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
void materialDisplay(TString filename,Bool_t checkoverlaps=kFALSE,TString label="geom-test.root" )
{
Int_t PriKolor[] = { 2, 3, 4, 5, 6, 7, 8, 9, 28, 30, 38, 40, 41, 42, 46 };
Int_t PriIndex = 0;
std::map<TString,Int_t> Kolor;
Kolor["Steel"] = kGreen;
Kolor["Copper"] = kYellow;
Kolor["Aluminum"] = kRed;
Kolor["FR4"] = kGray;
TGeoManager *geo2 = TGeoManager::Import(filename, "geo2");
if ( checkoverlaps )
{
geo2->CheckOverlaps(1e-5,"d");
geo2->CheckOverlaps(1e-5,"s10000000");
geo2->PrintOverlaps();
}
geo2->SetVisLevel(20);
TGeoVolume *volume = NULL;
TObjArray *volumes = geo2->GetListOfVolumes();
Int_t nvolumes = volumes->GetEntries();
for ( int i = 0; i < nvolumes; i++ )
{
volume = (TGeoVolume*)volumes->At(i);
volume->SetVisContainers(kTRUE);
if ( TString(volume->GetName()).Contains("DetEnclosure"))
{
volume->SetVisibility(kFALSE);
continue;
}
if ( TString(volume->GetMaterial()->GetName()).Contains("Air"))
{
volume->SetVisibility(kFALSE);
continue;
}
Int_t daughters = volume->GetNdaughters();
cout << volume->GetName() << " NDaughters = " << volume->GetMaterial()->GetName() << " " << daughters << endl;
volume->SetLineColor(Kolor[volume->GetMaterial()->GetName()]);
switch ( daughters )
{
case 0: volume->SetTransparency(100);
case 1: volume->SetTransparency(80);
case 2: volume->SetTransparency(60);
case 3: volume->SetTransparency(40);
default: volume->SetTransparency(10);
}
}
geo2->Export(label);
geo2->GetTopVolume()->Draw("ogl");
}