void KVSpectroDetector::UpdateVolumeAndNodeNames(){
// Update the names of Volumes and the names of the Nodes of this
// detector.
// The name of the volume representing the detector (returned
// by GetAbsGeoVolume()) is DET_<detector name>.
// The name of the active volumes is ACTIVE_<detector name>_<material name>.
// The name of the other volumes is <detector name>_<material name>.
GetAbsGeoVolume()->SetName( Form("DET_%s", GetName() ) );
TObjArray *nodes = GetAbsGeoVolume()->GetNodes();
TGeoNode *node = NULL;
TGeoVolume *vol = NULL;
TIter next( nodes );
while( (node = (TGeoNode *)next()) ){
TString name, nname;
vol = node->GetVolume();
name.Form("%s_%s", GetName(), vol->GetMaterial()->GetName());
if( GetActiveVolumes()->Contains( vol ) )
name.Prepend("ACTIVE_");
vol->SetName( name.Data() );
nname = name;
Int_t i=0;
while( nodes->FindObject( nname.Data() ) )
nname.Form("%s_%d",name.Data(), ++i);
node->SetName( nname.Data() );
node->SetNumber( i );
}
}
void fw_simGeo_set_volume_color_by_material(const char* material_re, Bool_t use_names, Color_t color, Char_t transparency=-1)
{
// Note: material_re is a perl regexp!
// If you want exact match, enclose in begin / end meta characters (^ / $):
// set_volume_color_by_material("^materials:Silicon$", kRed);
TPMERegexp re(material_re, "o");
TGeoMaterial *m;
TIter it(FWGeometryTableViewManager_GetGeoManager()->GetListOfMaterials());
while ((m = (TGeoMaterial*) it()) != 0)
{
if (re.Match(use_names ? m->GetName() : m->GetTitle()))
{
if (transparency != -1)
{
m->SetTransparency(transparency);
}
TGeoVolume *v;
TIter it2(FWGeometryTableViewManager_GetGeoManager()->GetListOfVolumes());
while ((v = (TGeoVolume*) it2()) != 0)
{
if (v->GetMaterial() == m)
{
v->SetLineColor(color);
}
}
}
}
}
void set_volume_color_by_material(const char* material_re, Color_t color, Char_t transparency=-1)
{
// Note: material_re is a perl regexp!
// If you want exact match, enclose in begin / end meta characters (^ / $):
// set_volume_color_by_material("^materials:Silicon$", kRed);
TPMERegexp re(material_re, "o");
TGeoMaterial *m;
TIter it(gGeoManager->GetListOfMaterials());
while ((m = (TGeoMaterial*) it()) != 0)
{
if (re.Match(m->GetName()))
{
if (transparency != -1)
{
m->SetTransparency(transparency);
}
TGeoVolume *v;
TIter it2(gGeoManager->GetListOfVolumes());
while ((v = (TGeoVolume*) it2()) != 0)
{
if (v->GetMaterial() == m)
{
v->SetLineColor(color);
}
}
}
}
full_update();
}
void visibility_volume_by_material(const char* material_re, Bool_t vis_state)
{
TPMERegexp re(material_re, "o");
TGeoMaterial *m;
TIter it(gGeoManager->GetListOfMaterials());
while ((m = (TGeoMaterial*) it()) != 0)
{
if (re.Match(m->GetName()))
{
TGeoVolume *v;
TIter it2(gGeoManager->GetListOfVolumes());
while ((v = (TGeoVolume*) it2()) != 0)
{
if (v->GetMaterial() == m)
{
v->SetVisibility(vis_state);
}
}
}
}
full_update();
}
drawFGT(TString volName="")
{
gSystem->Load("libGeom");
gSystem->Load("libGdml");
TGeoManager::Import("fgt.gdml");
gGeoManager->DefaultColors();
bool allSolid=true;
char topVol[] ="volWorld";
//char topVol[] ="volDetEnclosure";
//char topVol[] ="volDetector";
//char topVol[] ="volMuIDDownstream";
//char topVol[] ="volMuIDBarrel";
//char topVol[] ="volRPCMod";
//char topVol[] ="volRPCTray_End";
//char topVol[] ="volECALDownstream";
//char topVol[] ="volECALUpstream";
//char topVol[] ="volBarrelECAL";
//char topVol[] ="volSBPlane";
//char topVol[] ="volSTT";
//char topVol[] ="volSTPlaneTarget";
//char topVol[] ="volSTPlaneRadiator";
//char topVol[] ="volTargetPlaneArgon";
gGeoManager->GetVolume("volDetEnclosure")->SetLineColor(kGray);
gGeoManager->GetVolume("volDetEnclosure")->SetVisibility(1);
gGeoManager->GetVolume("volDetEnclosure")->SetTransparency(20);
gGeoManager->GetVolume("volDirtLayer")->SetTransparency(20);
gGeoManager->GetVolume("volServiceBuilding")->SetLineColor(kGray);
gGeoManager->GetVolume("volServiceBuilding")->SetVisibility(1);
gGeoManager->GetVolume("volServiceBuilding")->SetTransparency(20);
gGeoManager->GetVolume("volSky")->SetLineColor(kWhite);
gGeoManager->GetVolume("volSky")->SetVisibility(1);
gGeoManager->GetVolume("volSky")->SetTransparency(20);
gGeoManager->GetVolume("volMagnet")->SetLineColor(kGreen-1);
gGeoManager->GetVolume("volMagnet")->SetVisibility(1);
if(!allSolid) gGeoManager->GetVolume("volMagnet")->SetTransparency(10);
gGeoManager->GetVolume("volECALBarrelMod")->SetLineColor(kRed);
gGeoManager->GetVolume("volECALBarrelMod")->SetVisibility(1);
if(!allSolid) gGeoManager->GetVolume("volECALBarrelMod")->SetTransparency(75);
gGeoManager->GetVolume("volSBPlane")->SetLineColor(kRed-3);
gGeoManager->GetVolume("volSBPlane")->SetVisibility(1);
if(!allSolid) gGeoManager->GetVolume("volSBPlane")->SetTransparency(80);
/*
gGeoManager->GetVolume("volECALUpstream")->SetLineColor(kYellow-3);
gGeoManager->GetVolume("volECALUpstream")->SetVisibility(1);
gGeoManager->GetVolume("volECALUpstream")->SetTransparency(20);
gGeoManager->GetVolume("volECALDownstream")->SetLineColor(kYellow-3);
gGeoManager->GetVolume("volECALDownstream")->SetVisibility(1);
gGeoManager->GetVolume("volECALDownstream")->SetTransparency(20);
*/
TObjArray* va = gGeoManager->GetListOfVolumes();
int nv = va->GetEntries();
for (int i=0; i<nv; ++i) {
TGeoVolume* v = (TGeoVolume*)va->At(i);
std::string m(v->GetMaterial()->GetName());
//cout << v->GetMaterial()->GetName() << endl;
int lc, vi, tr, vd;
if (m == "Scintillator") {
lc = kGreen-7 ; vi = 1; tr = 0; vd = 1;
v->SetLineColor(lc);
v->SetVisibility(vi);
v->VisibleDaughters(vd);
v->SetTransparency(tr);
}
//else {
// continue;
//std::cout << "'" << m << "' has no defaults" << std::endl;
//lc = kOrange; vi = 0; tr = 50; vd = 1;
//}
}
//gGeoManager->GetTopNode();
gGeoManager->CheckOverlaps(1e-5,"d");
gGeoManager->PrintOverlaps();
//gGeoManager->FindVolumeFast(topVol)->CheckOverlaps(1e-5,"d");
//gGeoManager->FindVolumeFast(topVol)->GetNode(0)->PrintOverlaps();
gGeoManager->SetMaxVisNodes(70000);
gGeoManager->FindVolumeFast(topVol)->Draw("ogl");
TFile *tf = new TFile("drawFGT.root", "RECREATE");
gGeoManager->Write();
tf->Close();
}
//____________________________________________________________________________
void get_mass(Double_t length_unit, Double_t density_unit)
{
//tables of Z and A
const Int_t lcin_Z = 150;
const Int_t lcin_A = 300;
// calc unit conversion factors
Double_t density_unit_to_SI = density_unit / units::kg_m3;
Double_t length_unit_to_SI = length_unit / units::m;
Double_t volume_unit_to_SI = TMath::Power(length_unit_to_SI, 3.);
#ifdef _debug_
cout << "Input density unit --> kg/m^3 : x" << density_unit_to_SI << endl;
cout << "Input length unit --> m : x" << length_unit_to_SI << endl;
#endif
// get materials in geometry
TList *matlist = gGeoManager->GetListOfMaterials();
if (!matlist ) {
cout << "Null list of materials!" << endl;
return;
} else {
#ifdef _debug_
matlist->Print();
#endif
}
int max_idx = 0; // number of mixtures in geometry
Int_t nmat = matlist->GetEntries();
for( Int_t imat = 0; imat < nmat; imat++ )
{
Int_t idx = gGeoManager->GetMaterial(imat)->GetIndex();
max_idx = TMath::Max(max_idx, idx);
}
//check if material index is unique
Int_t * checkindex = new Int_t[max_idx+1];
for( Int_t i = 0; i<max_idx+1; i++ ) checkindex[i] = 0;
for( Int_t imat = 0; imat < nmat; imat++ )
{
if( !checkindex[imat] ) checkindex[imat] = 1;
else
{
cout << "material index is not unique" << endl;
return;
}
}
#ifdef _debug_
cout << "max_idx = " << max_idx << endl;
cout << "nmat = " << nmat << endl;
#endif
TGeoVolume * topvol = gGeoManager->GetTopVolume(); //get top volume
if (!topvol) {
cout << "volume does not exist" << endl;
return;
}
TGeoIterator NodeIter(topvol);
TGeoNode *node;
NodeIter.SetType(0); // include all daughters
Double_t * volume = new Double_t[max_idx+1];
Double_t * mass = new Double_t[max_idx+1];
for( Int_t i = 0; i<max_idx+1; i++ ){ volume[i]=0.; mass[i]=0.; } // IMPORTANT! force empty arrays, allows repated calls without ending ROOT session
volume[ topvol->GetMaterial()->GetIndex() ] = topvol->Capacity() * volume_unit_to_SI; //iterator does not include topvolume
while ( (node=NodeIter()) )
{
Int_t momidx = node->GetMotherVolume()->GetMaterial()->GetIndex() ;
Int_t idx = node->GetVolume() ->GetMaterial()->GetIndex() ;
Double_t node_vol = node->GetVolume()->Capacity() * volume_unit_to_SI;
volume[ momidx ] -= node_vol; //substract subvolume from mother
volume[ idx ] += node_vol;
}
Double_t larr_MassIsotopes[lcin_Z][lcin_A] = {0.}; //[Z][A], no map in pure ROOT
Double_t larr_VolumeIsotopes[lcin_Z][lcin_A] = {0.}; //[Z][A], no map in pure ROOT
for( Int_t i=0; i<gGeoManager->GetListOfMaterials()->GetEntries(); i++ )
{
TGeoMaterial *lgeo_Mat = gGeoManager->GetMaterial(i);
Int_t idx = gGeoManager->GetMaterial(i)->GetIndex();
if( lgeo_Mat->IsMixture() )
{
TGeoMixture * lgeo_Mix = dynamic_cast <TGeoMixture*> ( lgeo_Mat );
Int_t lint_Nelements = lgeo_Mix->GetNelements();
for ( Int_t j=0; j<lint_Nelements; j++)
{
Int_t lint_Z = TMath::Nint( (Double_t) lgeo_Mix->GetZmixt()[j] );
Int_t lint_A = TMath::Nint( (Double_t) lgeo_Mix->GetAmixt()[j] );
Double_t ldou_Fraction = lgeo_Mix->GetWmixt()[j];
Double_t ldou_Density = lgeo_Mix->GetDensity() * density_unit_to_SI;
larr_MassIsotopes[ lint_Z ][ lint_A ] += volume[idx] * ldou_Fraction * ldou_Density;
larr_VolumeIsotopes[ lint_Z ][ lint_A ] += volume[idx] * ldou_Fraction;
}
}
}
//
// print out volume/mass for each `material'
//
Double_t ldou_MinimumVolume = 1e-20;
cout << endl
<< " Geometry: \"" << gFileName << "\"" << endl
<< " TopVolume: \"" << topvol->GetName() << "\""
<< endl;
cout <<endl << "materials:" << endl;
cout << setw(5) << "index"
<< setw(15) << "name"
<< setprecision(6)
<< setw(14) << "volume (m^3)"
<< setw(14) << "mass (kg)"
<< setw(14) << "mass (%)"
<< endl;
double total_mass_materials = 0;
for( Int_t i=0; i<gGeoManager->GetListOfMaterials()->GetEntries(); i++ )
{
Int_t idx = gGeoManager->GetMaterial(i)->GetIndex();
Double_t density = gGeoManager->GetMaterial(i)->GetDensity() * density_unit_to_SI;
Double_t mass_material = density * volume[idx];
if ( volume[idx] > ldou_MinimumVolume ) {
total_mass_materials += mass_material;
}
}
for( Int_t i=0; i<gGeoManager->GetListOfMaterials()->GetEntries(); i++ )
{
Int_t idx = gGeoManager->GetMaterial(i)->GetIndex();
Double_t density = gGeoManager->GetMaterial(i)->GetDensity() * density_unit_to_SI;
mass[idx] = density * volume[idx];
if( volume[idx] > ldou_MinimumVolume ) {
cout << setw(5) << i
<< setw(15) << gGeoManager->GetMaterial(i)->GetName()
<< setprecision(6)
<< setw(14) << volume[idx]
<< setw(14) << mass[idx]
<< setw(14) << mass[idx]*100./total_mass_materials
<< endl;
}
}
//
// print out mass contribution for each nuclear target
//
PDGLibrary* pdglib = PDGLibrary::Instance();
cout <<endl << "isotopes:" << endl;
cout << setw(4) << "Z"
<< setw(4) << "A"
<< setw(14) << "PDG isotope"
<< setw(5) << " "
<< setprecision(6)
<< setw(14) << "volume (m^3)"
<< setw(14) << "mass (kg)"
<< setw(10) << "mass (%)"
<< endl;
double total_mass_isotopes = 0;
for( Int_t i=0; i<lcin_Z; i++ ) {
for( Int_t j=0; j<lcin_A; j++ ) {
if( larr_VolumeIsotopes[ i ][ j ] > ldou_MinimumVolume ) {
total_mass_isotopes += larr_MassIsotopes[ i ][ j ];
}
}
}
for( Int_t i=0; i<lcin_Z; i++ )
{
for( Int_t j=0; j<lcin_A; j++ )
{
if( larr_VolumeIsotopes[ i ][ j ] > ldou_MinimumVolume ) {
int pdgcode = 1000000000 + i*10000 + j*10;
cout << setw(4) << i
<< setw(4)<< j
<< setw(14) << pdgcode
<< setw(5) << pdglib->Find(pdgcode)->GetName()
<< setprecision(6)
<< setw(14) << larr_VolumeIsotopes[ i ][ j ]
<< setw(14) << larr_MassIsotopes[ i ][ j ]
<< setw(10) << larr_MassIsotopes[ i ][ j ]*100.0/total_mass_isotopes
<< endl;
}
else if ( larr_VolumeIsotopes[ i ][ j ] < -ldou_MinimumVolume ) {
cout << "negative volume, check geometry " << larr_VolumeIsotopes[ i ][ j ] << endl;
}
}
}
cout << endl << " mass totals: " << total_mass_materials << " " << total_mass_isotopes
<< endl << endl;
delete [] volume;
delete [] mass;
}
