Belle2::EveGeometry Namespace Reference

Handles manipulation of detector geometry for the display. More...

Enumerations
enum	EType {   c_Full ,   c_Simplified }
	Type of geometry shown. More...

Functions
void	addGeometry (EType visMode)
	Add TGeo geometry to Eve (only needs to be done once.)
void	setVisualisationMode (EType visMode)
	switch to given visualisation mode.
void	enableVolume (const char *name, bool only_daughters=false, bool enable=true)
	enable/disable rendering of the volume 'name', or only its daughters if only_daughters is set.
void	disableVolume (const char *name, bool only_daughters=false)
	disable rendering of the volume 'name', or only its daughters if only_daughters is set.
void	setVolumeColor (const char *name, Color_t col)
	set fill color of the volume 'name' to 'col'.
void	setTransparency (int percent)
	Recursively set transparency of geometry (0: opaque, 100: fully transparent).
double	getMaxR ()
	find a point that is inside the top node.
void	saveExtract ()
	Save a geometry extract from the current state of the TGeo geometry.
void	setCustomExtractPath (const std::string &extractPath)
	Set custom path to the geometry extract (to change originally hard-coded value)
void	setHideVolumes (const std::vector< std::string > &volumes)
	List of volumes to be hidden (can be re-enabled in Eve panel / Geometry scene.
void	setDeleteVolumes (const std::vector< std::string > &volumes)
	List of volumes to be removed.
void	removeChildrenByRegExp (TEveElement *parent, const std::string &pattern)
	Recursive removal of volumes based on regular expression pattern.

Detailed Description

Handles manipulation of detector geometry for the display.

Enumeration Type Documentation

EType

enum EType

Type of geometry shown.

Enumerator
c_Full	Full geometry converted from Geant4 (use this for non-standard Belle II setups!).
c_Simplified	a simplified Belle II geometry.

Definition at line 25 of file EveGeometry.h.

               {
      c_Full, 
      c_Simplified 
    };

Function Documentation

addGeometry()

void addGeometry

(

EType

visMode

)

Add TGeo geometry to Eve (only needs to be done once.)

Definition at line 36 of file EveGeometry.cc.

{
  B2DEBUG(100, "Setting up geometry for TEve...");
  if (!gEve)
    B2FATAL("gEve must be set up before EveGeometry!");
 
  if (visMode == c_Full) {
    if (!gGeoManager) { //TGeo geometry not initialized, do it ourselves
      //convert geant4 geometry to TGeo geometry
      geometry::GeometryManager& geoManager = geometry::GeometryManager::getInstance();
      geoManager.createTGeoRepresentation();
      if (!gGeoManager) {
        B2FATAL("Couldn't create TGeo geometry! Please make sure you add the Geometry module before Display");
        return;
      }
    }
    //set colours by atomic mass number
    gGeoManager->DefaultColors();
 
    setTransparency(90);
    //set some nicer colors (at top level only)
    setVolumeColor("PXD.Envelope", getTColorID("Chameleon", 2));
    setVolumeColor("SVD.Envelope", getTColorID("Orange", 3));
    setVolumeColor("logical_ecl", getTColorID("Orange", 1));
    setVolumeColor("BKLM.EnvelopeLogical", getTColorID("Chameleon", 1));
    setVolumeColor("Endcap_1", getTColorID("Chameleon", 1));
    setVolumeColor("Endcap_2", getTColorID("Chameleon", 1));
 
    for (auto& volume : s_hideVolumes)
      disableVolume(volume.c_str(), false);
 
    TGeoNode* top_node = gGeoManager->GetTopNode();
    assert(top_node != NULL);
    s_eveTopNode = new TEveGeoTopNode(gGeoManager, top_node);
    s_eveTopNode->IncDenyDestroy();
    s_eveTopNode->SetVisLevel(2);
 
    gEve->AddGlobalElement(s_eveTopNode);
 
    //expand geometry in eve list (otherwise one needs three clicks to see that it has children)
    s_eveTopNode->ExpandIntoListTreesRecursively();
 
  }
  B2DEBUG(100, "Loading geometry projections...");
 
  const std::string extractPath = FileSystem::findFile(s_eveGeometryExtractPath);
  TFile* f = TFile::Open(extractPath.c_str(), "READ");
  TEveGeoShapeExtract* gse = dynamic_cast<TEveGeoShapeExtract*>(f->Get("Extract"));
  s_simplifiedShape = TEveGeoShape::ImportShapeExtract(gse, 0);
  s_simplifiedShape->SetRnrSelf(false);
  s_simplifiedShape->IncDenyDestroy();
  s_simplifiedShape->SetName("Minimal geometry extract");
  delete f;
 
  //I want to show full geo in unprojected view,
  //but I still need to add the extract to the geometry scene...
  gEve->AddGlobalElement(s_simplifiedShape);
 
  setVisualisationMode(visMode);
 
  // Allow deletion only for full geometry
  if (s_eveTopNode) {
    for (auto& volumeRegExp : s_deleteVolumes) {
      removeChildrenByRegExp(s_eveTopNode, volumeRegExp);
    }
  }
 
  B2DEBUG(100, "Done.");
}

disableVolume()

void disableVolume	(	const char *	name,
		bool	only_daughters = false
	)

disable rendering of the volume 'name', or only its daughters if only_daughters is set.

Definition at line 160 of file EveGeometry.cc.

{ enableVolume(name, only_daughters, false); }

enableVolume()

void enableVolume	(	const char *	name,
		bool	only_daughters = false,
		bool	enable = true
	)

enable/disable rendering of the volume 'name', or only its daughters if only_daughters is set.

Definition at line 147 of file EveGeometry.cc.

{
  if (!gGeoManager) return;
  TGeoVolume* vol = gGeoManager->FindVolumeFast(name);
  if (vol) {
    if (!only_daughters)
      vol->SetVisibility(enable);
    vol->SetVisDaughters(enable);
  } else {
    B2DEBUG(100, "Volume " << name << " not found?");
  }
}

getMaxR()

double getMaxR

(

)

find a point that is inside the top node.

Definition at line 184 of file EveGeometry.cc.

{
  if (gGeoManager && gGeoManager->GetTopNode()) {
    TGeoVolume* top_node = gGeoManager->GetTopNode()->GetVolume();
    double p[3] = { 380.0, 0.0, 0.0 }; //ok for normal Belle II geometry
    while (!top_node->Contains(p)) {
      p[0] *= 0.8;
    }
    return p[0];
  } else if (s_simplifiedShape) {
    s_simplifiedShape->ComputeBBox();
    const float* bbox = s_simplifiedShape->GetBBox();
    return std::min(380.f, std::max({std::abs(bbox[0]), bbox[1], std::abs(bbox[2]), bbox[2]}));
  }
  // fallback to something reasonable
  return 380.;
}

removeChildrenByRegExp()

void removeChildrenByRegExp	(	TEveElement *	parent,
		const std::string &	pattern
	)

Recursive removal of volumes based on regular expression pattern.

Leading '#' switches to removal of daughter only.

Definition at line 106 of file EveGeometry.cc.

{
  TPRegexp reAll(".*");
  bool onlyChildren = false;
  std::string regexp = pattern;
 
  // For patterns with leading '#', do delete only
  // children elements (and remove the '#' for regexp)
  if (pattern.substr(0, 1) == std::string("#")) {
    regexp = pattern.substr(1);
    onlyChildren = true;
  }
 
  TPRegexp reMatch(regexp.c_str());
 
  std::list<TEveElement*> children;
  parent->FindChildren(children, reAll);
 
  for (TEveElement* child : children) {
    if (reMatch.MatchB(child->GetElementName())) {
      if (onlyChildren) {
        B2INFO("Removing children of " << child->GetElementName());
        child->DestroyElements();
      } else {
        B2INFO("Removing " << child->GetElementName());
        child->Destroy();
      }
    } else {
      removeChildrenByRegExp(child, pattern);
    }
  }
}

saveExtract()

void saveExtract

(

)

Save a geometry extract from the current state of the TGeo geometry.

To actually create a reasonably sized extract, one needs to delete all objects in Top_1 that one doesn't want to save. Selecting them via their name and deleting them in a function should work, but really just crashes. This is all quite horrible, really.

Definition at line 202 of file EveGeometry.cc.

{
  TGeoManager* my_tgeomanager = gGeoManager;
  if (!s_eveTopNode) {
    B2ERROR("Couldn't find TEveGeoTopNode");
    return;
  }
  s_eveTopNode->ExpandIntoListTrees();
  s_eveTopNode->SaveExtract("geometry_extract.root", "Extract", false);
 
  //this doesn't work too well (i.e. crashes when geometry is drawn)
  //s_eveTopNode->ExpandIntoListTreesRecursively();
  //s_eveTopNode->SaveExtract("display_geometry_full.root", "Extract", false);
  gGeoManager = my_tgeomanager;
}

setCustomExtractPath()

void setCustomExtractPath

(

const std::string &

extractPath

)

Set custom path to the geometry extract (to change originally hard-coded value)

Definition at line 218 of file EveGeometry.cc.

{
  s_eveGeometryExtractPath = extractPath;
}

setDeleteVolumes()

void setDeleteVolumes

(

const std::vector< std::string > &

volumes

)

List of volumes to be removed.

The volume and all its daughters will be deleted. Leading '#' switches to removal of daughter only.

Definition at line 228 of file EveGeometry.cc.

{
  s_deleteVolumes = volumes;
}

setHideVolumes()

void setHideVolumes

(

const std::vector< std::string > &

volumes

)

List of volumes to be hidden (can be re-enabled in Eve panel / Geometry scene.

The volume and all its daughters will be hidden.

Definition at line 223 of file EveGeometry.cc.

{
  s_hideVolumes = volumes;
}

setTransparency()

void setTransparency

(

int

percent

)

Recursively set transparency of geometry (0: opaque, 100: fully transparent).

Definition at line 174 of file EveGeometry.cc.

{
  if (!gGeoManager) return;
  TObjArray* volumes = gGeoManager->GetListOfVolumes();
  for (int i = 0; i < volumes->GetEntriesFast(); i++) {
    TGeoVolume* volume = static_cast<TGeoVolume*>(volumes->At(i));
    volume->SetTransparency(percent);
  }
}

setVisualisationMode()

void setVisualisationMode

(

EType

visMode

)

switch to given visualisation mode.

Definition at line 139 of file EveGeometry.cc.

{
  bool fullgeo = (visMode == c_Full);
  if (s_eveTopNode)
    s_eveTopNode->SetRnrSelfChildren(fullgeo, fullgeo);
  s_simplifiedShape->SetRnrSelfChildren(false, !fullgeo);
}

setVolumeColor()

void setVolumeColor	(	const char *	name,
		Color_t	col
	)

set fill color of the volume 'name' to 'col'.

Definition at line 162 of file EveGeometry.cc.

{
  if (!gGeoManager) return;
  TGeoVolume* vol = gGeoManager->FindVolumeFast(name);
  if (vol) {
    //while TGeoVolume derives from TAttFill, the line color actually is important here
    vol->SetLineColor(col);
  } else {
    B2DEBUG(100, "Volume " << name << " not found?");
  }
}