|
|
|
#include <vector>#include <list>#include <string>#include <cmath>#include <TEveElement.h>#include <TEveGeoNode.h>#include <TGeoNode.h>#include "eve_macros.h"Go to the source code of this file.
Functions | |
| TEveGeoShape * | clone (const TEveElement *element, TEveElement *parent) |
| void | expand_node (TEveElement *element) |
| const char * | get_name (const TEveElement *element) |
| const TGeoShape * | get_shape (const TEveElement *element) |
| TGeoShape * | get_shape (TEveElement *element) |
| const char * | get_title (const TEveElement *element) |
| TGeoMatrix * | get_transform (const TEveElement *element) |
| void | hide_tracker_endcap (TEveElement *tracker) |
| bool | is_leaf_node (const TEveElement *element) |
| void | set_children_visibility (TEveElement *element, const std::string &node_name, const std::vector< std::string > &children_name, bool visibility) |
| void | set_color (TEveElement *element, Color_t color, float alpha, unsigned int levels) |
| void | set_tracker_endcap_visibility (TEveElement *tracker, bool visibility) |
| void | show_tracker_endcap (TEveElement *tracker) |
| TEveGeoShape* clone | ( | const TEveElement * | element, |
| TEveElement * | parent | ||
| ) |
Definition at line 135 of file eve_macros.cc.
References clone(), extract(), get_name(), get_shape(), get_title(), and get_transform().
{
TEveGeoShape* shape = new TEveGeoShape( get_name(element), get_title(element) );
std::auto_ptr<TGeoMatrix> matrix( get_transform(element) );
shape->SetTransMatrix( matrix.get() );
delete matrix;
TEveGeoShapeExtract extract; // FIXME put name and title here...
extract.SetShape( (TGeoShape *) get_shape(element)->Clone() );
extract.SetTrans( trans.Array() );
extract.SetRnrSelf( true );
extract.SetRnrElements( true );
TEveGeoShape * clone = TEveGeoShape::ImportShapeExtract( &extract, parent );
return clone;
}
| void expand_node | ( | TEveElement * | element | ) |
Definition at line 43 of file eve_macros.cc.
References python::Node::node.
{
// force a TEveGeoNode to load all its children
if (TEveGeoNode * node = dynamic_cast<TEveGeoNode *>( element )) {
if (node->GetNChildren() == 0 && node->GetNode()->GetVolume()->GetNdaughters() > 0) {
TIter next(node->GetNode()->GetVolume()->GetNodes());
TGeoNode* dnode;
while ((dnode = (TGeoNode*) next()) != 0) {
TEveGeoNode* node_re = new TEveGeoNode(dnode);
node->AddElement(node_re);
}
}
return;
}
// a TEveGeoShape is always exanded
//if (TEveGeoShape * shape __attribute__ ((unused)) = dynamic_cast<TEveGeoShape *>( element )) {
// return;
//}
// a generic TEveElement has no knwledge on children expansion
return;
}
| const char* get_name | ( | const TEveElement * | element | ) |
Definition at line 13 of file eve_macros.cc.
References python::Node::node.
{
// try as a TEveGeoNode or TEveGeoShape
if (const TEveGeoNode * node = dynamic_cast<const TEveGeoNode *>( element ))
return node->GetName();
if (const TEveGeoShape * shape = dynamic_cast<const TEveGeoShape *>( element ))
return shape->GetName();
// try to access the element as a generic named object
if (const TNamed * named = dynamic_cast<const TNamed *>( element ))
return named->GetName();
return 0;
}
| const TGeoShape* get_shape | ( | const TEveElement * | element | ) |
Definition at line 66 of file eve_macros.cc.
References python::Node::node.
{
// a TEveGeoNode, can look into its TGeoNode and retrieve the shape
if (const TEveGeoNode * node = dynamic_cast<const TEveGeoNode *>( element )) {
return node->GetNode()->GetVolume()->GetShape();
}
// a TEveGeoShape owns its shape
if (const TEveGeoShape * shape = dynamic_cast<const TEveGeoShape *>( element )) {
TEveGeoShape * nc_shape = const_cast<TEveGeoShape *>( shape );
return const_cast<const TGeoShape *>( nc_shape->GetShape() );
}
// a TEveElement is too generic, no way to get a shape
return 0;
}
| TGeoShape* get_shape | ( | TEveElement * | element | ) |
Definition at line 81 of file eve_macros.cc.
{
// a TEveGeoNode cannot modify its shape
//if (const TEveGeoNode * node __attribute__ ((unused)) = dynamic_cast<const TEveGeoNode *>( element )) {
// return 0;
//}
// a TEveGeoShape owns its shape, and can modifiy it
if (TEveGeoShape * shape = dynamic_cast<TEveGeoShape *>( element )) {
return shape->GetShape();
}
// a TEveElement is too generic, no way to get a shape
return 0;
}
| const char* get_title | ( | const TEveElement * | element | ) |
Definition at line 28 of file eve_macros.cc.
References python::Node::node.
Referenced by clone().
{
// try as a TEveGeoNode or TEveGeoShape
if (const TEveGeoNode * node = dynamic_cast<const TEveGeoNode *>( element ))
return node->GetTitle();
if (const TEveGeoShape * shape = dynamic_cast<const TEveGeoShape *>( element ))
return shape->GetTitle();
// try to access the element as a generic named object
if (const TNamed * named = dynamic_cast<const TNamed *>( element ))
return named->GetTitle();
return 0;
}
| TGeoMatrix* get_transform | ( | const TEveElement * | element | ) |
Definition at line 95 of file eve_macros.cc.
References i, and python::Node::node.
Referenced by clone().
{
if (const TEveGeoNode * node = dynamic_cast<const TEveGeoNode *>( element )) {
// a TEveGeoNode is a proxy to a TGeoNode, which knows its relative transformation wrt. its parent
// so we follow the TEveGeoNode hierarchy up to a TEveGeoTopNode, then jump to its TGeoManager, and go back down the branches to the TEveGeoNode's TGeoNode
std::vector< const TEveGeoNode * > nodes;
const TEveGeoTopNode * top = 0;
while ((top = dynamic_cast<const TEveGeoTopNode *>( node )) == 0) {
// save the current node
nodes.push_back(node);
// check that the node actually has any parents
TEveGeoNode * nc_node = const_cast<TEveGeoNode *>( node );
if (nc_node->BeginParents() == nc_node->EndParents())
return 0;
// assume the firt parent is the good one, and check that the parent type is correct
node = dynamic_cast<const TEveGeoNode *>( * nc_node->BeginParents() );
if (node == 0)
return 0;
}
// reached the top level node, start from its (optional) global transormation
TGeoHMatrix * matrix = new TGeoHMatrix();
(const_cast<TEveGeoTopNode *>(top))->RefGlobalTrans().SetGeoHMatrix( *matrix );
for (unsigned int i = 0; i < nodes.size(); ++i)
*matrix *= *(nodes[i]->GetNode()->GetMatrix());
return matrix;
}
if (const TEveGeoShape * shape = dynamic_cast<const TEveGeoShape *>( element )) {
// a TEveGeoShape knows the absolute transformation of its shape
TGeoHMatrix * matrix = new TGeoHMatrix();
(const_cast<TEveGeoShape *>(shape))->RefHMTrans().SetGeoHMatrix( *matrix );
return matrix;
}
return 0;
}
| void hide_tracker_endcap | ( | TEveElement * | tracker | ) |
Definition at line 244 of file eve_macros.cc.
References set_tracker_endcap_visibility().
{
set_tracker_endcap_visibility( tracker, false );
}
| bool is_leaf_node | ( | const TEveElement * | element | ) |
Definition at line 177 of file eve_macros.cc.
References python::Node::node.
{
// a TEveGeoNode can have unaccounted-for children
if (const TEveGeoNode * node = dynamic_cast<const TEveGeoNode *>( element )) {
return ((node->GetNChildren() == 0) and (node->GetNode()->GetVolume()->GetNdaughters() == 0));
}
// a TEveGeoShape always knows its children
if (const TEveGeoShape * shape = dynamic_cast<const TEveGeoShape *>( element )) {
return (shape->GetNChildren() == 0);
}
// default implementation
return (element->GetNChildren() == 0);
}
| void set_children_visibility | ( | TEveElement * | element, |
| const std::string & | node_name, | ||
| const std::vector< std::string > & | children_name, | ||
| bool | visibility | ||
| ) |
Definition at line 193 of file eve_macros.cc.
References get_name(), i, is_leaf_node(), j, mergeVDriftHistosByStation::name, and findQualityFiles::size.
{
// try to access the element as a named thingy
const char * name = get_name( element );
if (not name or strncmp(name, node_name.c_str(), node_name.size()))
// unnamed node, or wrong node
return;
for (std::list<TEveElement *>::iterator j = element->BeginChildren(); j != element->EndChildren(); ++j) {
TEveElement * child = *j;
name = get_name( child );
if (not name)
// unnamed node, ignore it
continue;
for (unsigned int i = 0; i < children_name.size(); ++i)
if (not strncmp(name, children_name[i].c_str(), children_name[i].size())) {
// change this child visibility
if (is_leaf_node( child )) {
child->SetRnrSelf( visibility );
child->SetRnrChildren( false );
} else {
child->SetRnrSelf( false );
child->SetRnrChildren( visibility );
}
break;
}
}
// notify the element that is had changed
element->ElementChanged(true, true);
}
| void set_color | ( | TEveElement * | element, |
| Color_t | color, | ||
| float | alpha, | ||
| unsigned int | levels | ||
| ) |
Definition at line 154 of file eve_macros.cc.
References expand_node(), i, and set_color().
{
if (not element)
return;
// set this node's color
element->SetMainColor( color );
if (alpha > 1.) alpha = 1.;
if (alpha < 0.) alpha = 0.;
unsigned char transparency = (unsigned char) roundf(100. - (alpha * 100.));
element->SetMainTransparency( transparency );
if (levels > 0) {
// set the node's children's color
expand_node( element );
for (std::list<TEveElement*>::iterator i = element->BeginChildren(); i != element->EndChildren(); ++i)
set_color( *i, color, alpha, levels - 1);
}
// notify the element that it has changed
element->ElementChanged(true, true);
}
| void set_tracker_endcap_visibility | ( | TEveElement * | tracker, |
| bool | visibility | ||
| ) |
Definition at line 226 of file eve_macros.cc.
References Reference_intrackfit_cff::endcap, and set_children_visibility().
{
std::vector<std::string> endcap;
endcap.push_back("tec:TEC");
endcap.push_back("tidf:TIDF");
endcap.push_back("tidb:TIDB");
endcap.push_back("pixfwd:PixelForwardZPlus");
endcap.push_back("pixfwd:PixelForwardZMinus");
set_children_visibility( tracker, "tracker:Tracker", endcap, visibility );
}
| void show_tracker_endcap | ( | TEveElement * | tracker | ) |
Definition at line 238 of file eve_macros.cc.
References set_tracker_endcap_visibility().
{
set_tracker_endcap_visibility( tracker, true );
}
1.7.3