Files
geant4/source/visualization/OpenInventor/src/SoCounterAction.cc
T
2016-06-09 11:11:55 +02:00

103 lines
4.1 KiB
C++

//
// ********************************************************************
// * DISCLAIMER *
// * *
// * The following disclaimer summarizes all the specific disclaimers *
// * of contributors to this software. The specific disclaimers,which *
// * govern, are listed with their locations in: *
// * http://cern.ch/geant4/license *
// * *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work make any representation or warranty, express or implied, *
// * regarding this software system or assume any liability for its *
// * use. *
// * *
// * This code implementation is the intellectual property of the *
// * GEANT4 collaboration. *
// * By copying, distributing or modifying the Program (or any work *
// * based on the Program) you indicate your acceptance of this *
// * statement, and all its terms. *
// ********************************************************************
//
#ifdef G4VIS_BUILD_OI_DRIVER
/*----------------------------HEPVis----------------------------------------*/
/* */
/* Node: SoCounterAction */
/* Author: Guy Barrand */
/* */
/*--------------------------------------------------------------------------*/
// this :
#include <HEPVis/actions/SoCounterAction.h>
#include <Inventor/nodes/SoNode.h>
#include <Inventor/nodes/SoGroup.h>
#include <Inventor/nodes/SoSwitch.h>
#include <Inventor/nodekits/SoBaseKit.h>
#include <Inventor/elements/SoSwitchElement.h>
SO_ACTION_SOURCE(SoCounterAction);
void SoCounterAction::initClass(void){
SO_ACTION_INIT_CLASS(SoCounterAction,SoAction);
SO_ENABLE(SoCounterAction,SoSwitchElement);
SO_ACTION_ADD_METHOD(SoNode,SoCounterAction::actionMethod);
}
SoCounterAction::SoCounterAction()
:fCount(0),fLookFor(NODE),fCheckDerived(TRUE){
SO_ACTION_CONSTRUCTOR(SoCounterAction);
}
SoCounterAction::~SoCounterAction(){}
void SoCounterAction::beginTraversal(SoNode* node){
fCount = 0;
SoAction::beginTraversal(node);
}
void SoCounterAction::actionMethod(SoAction* aThis,SoNode* aNode) {
//printf("debug : begin : %s %s\n",aNode->getName().getString(),
//aNode->getTypeId().getName().getString());
SoCounterAction* This = (SoCounterAction*)aThis;
if(This->fLookFor==NODE) {
This->fCount++;
} else if(This->fLookFor==TYPE) {
if(This->fCheckDerived==TRUE) {
if(aNode->getTypeId().isDerivedFrom(This->fType)) This->fCount++;
} else {
if(aNode->getTypeId()==This->fType) This->fCount++;
}
} else if(This->fLookFor==NAME) {
if(This->fName==aNode->getName()) This->fCount++;
}
if(aNode->isOfType(SoSwitch::getClassTypeId())) {
SoSwitch* sw = (SoSwitch*)aNode;
SbBool flag = sw->whichChild.enableNotify(FALSE);
int old = sw->whichChild.getValue();
sw->whichChild.setValue(SO_SWITCH_ALL);
aNode->doAction(aThis);
sw->whichChild.setValue(old);
sw->whichChild.enableNotify(flag);
} else if(aNode->isOfType(SoGroup::getClassTypeId())) {
aNode->doAction(aThis);
} else if(aNode->isOfType(SoBaseKit::getClassTypeId())) {
aNode->doAction(aThis);
}
}
void SoCounterAction::setLookFor(LookFor aLookFor) {
fLookFor = aLookFor;
}
void SoCounterAction::setType(const SoType aType,SbBool aCheckDerived) {
fType = aType;
fCheckDerived = aCheckDerived;
}
void SoCounterAction::setName(const SbName aName){
fName = aName;
}
int SoCounterAction::getCount() const {
return fCount;
}
#endif