G4OpenInventorXtExaminerViewer.hh

Go to the documentation of this file.

//
// ********************************************************************
// * License and Disclaimer                                           *
// *                                                                  *
// * The  Geant4 software  is  copyright of the Copyright Holders  of *
// * the Geant4 Collaboration.  It is provided  under  the terms  and *
// * conditions of the Geant4 Software License,  included in the file *
// * LICENSE and available at  http://cern.ch/geant4/license .  These *
// * include a list of copyright holders.                             *
// *                                                                  *
// * 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.  Please see the license in the file  LICENSE  and URL above *
// * for the full disclaimer and the limitation of liability.         *
// *                                                                  *
// * This  code  implementation is the result of  the  scientific and *
// * technical work of the GEANT4 collaboration.                      *
// * By using,  copying,  modifying or  distributing the software (or *
// * any work based  on the software)  you  agree  to acknowledge its *
// * use  in  resulting  scientific  publications,  and indicate your *
// * acceptance of all terms of the Geant4 Software license.          *
// ********************************************************************
//
// Open Inventor Xt Extended Viewer - 30 Oct 2012
// Rastislav Ondrasek, Pierre-Luc Gagnon, Frederick Jones TRIUMF

#ifndef HookEventProcState_H
#define HookEventProcState_H 1
#include "G4VStateDependent.hh"

class G4OpenInventorXtExaminerViewer;
class HookEventProcState : public G4VStateDependent
{
private:
    G4OpenInventorXtExaminerViewer *viewer;
public:
    HookEventProcState(G4OpenInventorXtExaminerViewer*);
    ~HookEventProcState();

    virtual G4bool Notify(G4ApplicationState requiredState);
};
#endif /* HookEventProcState_H */


#ifndef G4OPENINVENTORXTEXAMINERVIEWER_HH
#define G4OPENINVENTORXTEXAMINERVIEWER_HH

#include <map>
#include <vector>
#include <fstream>
#include <Inventor/SbLinear.h>
#include <Inventor/nodes/SoLineSet.h>
#include <Inventor/Xt/viewers/SoXtExaminerViewer.h>
#include <Inventor/events/SoKeyboardEvent.h>

class SoCoordinate3;
class SoFont;
class SoText2;
class SoPointSet;

class G4OpenInventorXtExaminerViewer : public SoXtExaminerViewer {
private:
  Widget prevViewPtButton, nextViewPtButton;
  Widget menuBar, fileMenu, openFileDialog, newFileDialog,
  loadRefCoordsDialog, saveRefCoordsDialog,
  loadSceneGraphDialog, saveSceneGraphDialog,
  viewPtSelection, listsDialog, myShellDialog, myViewPtList, myElementList;

  static G4OpenInventorXtExaminerViewer *viewer;
  void (*escapeCallback)(void *);
  void * examinerObject;
  SbBool lshiftdown, rshiftdown, lctrldown, rctrldown;
  
public:

  // Same constructor as the ExaminerViewer
  G4OpenInventorXtExaminerViewer(Widget parent = NULL,
         const char *name = NULL,
         SbBool embed = TRUE, 
         SoXtFullViewer::BuildFlag flag = BUILD_ALL,
         SoXtViewer::Type type = BROWSER);

  ~G4OpenInventorXtExaminerViewer();
  template <class T> void parseString(T &t, const std::string &s, bool &error);
  Widget addMenu(std::string name);
  void addButton(Widget menu, std::string name, XtCallbackProc);
  Widget getMenuBar() { return menuBar; }
  Widget getMenu() { return fileMenu; }
  void warningMsgDialog(std::string, String, XtCallbackProc);
  bool warningFlag;

  std::string saveScenegraphFileName;
  Widget saveScenegraphWidget;
  std::string saveRefCoordsFileName;
  Widget saveRefCoordsWidget;

  Widget createScale(Widget, char *, int, float);
  void addEscapeCallback(void (*cb)(void *), void *);
  bool abbrOutputFlag;
  bool pickRefPathFlag;
  bool viewingBeforePickRef;
             
protected:
  // Same constructor as the ExaminerViewer 
  G4OpenInventorXtExaminerViewer(Widget parent,
         const char *name,
         SbBool embed,
         SoXtFullViewer::BuildFlag flag,
         SoXtViewer::Type type,
         SbBool build);

  // Overloaded for adding the MenuBar
  Widget buildWidget(Widget parent);
  // Overloaded so additional buttons can be added
  virtual void createViewerButtons (Widget parent, SbPList * buttonlist);
  // Overloaded for catching various keyboard events  
  virtual SbBool processSoEvent(const SoEvent * const event);
  void moveCamera(float dist = 0, bool lookdown = false);
  std::string curEltName;
  SbVec3f camUpVec;
  SbVec3f camDir;
  void rotateCamera();
  void updateViewParams(SoKeyboardEvent::Key);
  bool loadViewPts();
  virtual void afterRealizeHook();

private:
  // Each constructor calls this generic constructor
  void constructor(const SbBool build);

  static G4OpenInventorXtExaminerViewer *getObject();

  HookEventProcState *hookBeamOn;
  friend class HookEventProcState;
  bool newEvents;
  static void sceneChangeCB(void *, SoSensor *);
    
  void setViewPt();
  void writeViewPtIdx();
  void cleanUpAfterPrevFile();
  
  void popUpFileSelDialog(Widget&, std::string, std::string, XtCallbackProc);
  static void cancelFileSelDialogCB(Widget, XtPointer, XtPointer);
  static void openViewPtFileCB(Widget, XtPointer, XtPointer);
  static void viewPtFileSelectedCB(Widget, XtPointer, XtPointer);
  static void newViewPtFileCB(Widget, XtPointer, XtPointer);
  static void createNewVPFileCB(Widget, XtPointer, XtPointer);
  static void overwriteFileCB(Widget, XtPointer, XtPointer);
  static void loadRefCoordsDialogCB(Widget, XtPointer, XtPointer);  //pop file dialog
  static void loadRefCoordsCB(Widget, XtPointer, XtPointer);            //execute loading
  static void saveRefCoordsDialogCB(Widget, XtPointer, XtPointer);  //pop file dialog
  static void saveRefCoordsCB(Widget, XtPointer, XtPointer);            //execute saving
  static void saveRefCoordsOverWriteCB(Widget, XtPointer, XtPointer);
  static void loadSceneGraphDialogCB(Widget, XtPointer, XtPointer);
  static void loadSceneGraphCB(Widget, XtPointer, XtPointer);
  static void saveSceneGraphDialogCB(Widget, XtPointer, XtPointer);
  static void saveSceneGraphCB(Widget, XtPointer, XtPointer);
  static void saveSceneGraphOverWriteCB(Widget, XtPointer, XtPointer);
  static void mouseoverCB(void *aThis, SoEventCallback *eventCB);
  static void pickingCB(void *aThis, SoEventCallback *eventCB);

             
  // Viewpoint operations
  void addViewPoints();
  static void closeListsDialogCB(Widget, XtPointer, XtPointer);
  static void loadBookmarkCB(Widget, XtPointer, XtPointer);
  static void renameBookmarkCB(Widget, XtPointer, XtPointer);
  void renameViewPt(char *vpName);
  static void sortBookmarksCB(Widget, XtPointer, XtPointer);
  void sortViewPts(std::vector<std::string>);
  static void deleteBookmarkCB(Widget, XtPointer, XtPointer);
  static void deleteViewPtCB(Widget, XtPointer, XtPointer);
  void deleteViewPt(char *vpName = NULL);
    
  // Animation
  static void animateRefParticleCB(Widget, XtPointer, XtPointer);
  static void animateSensorCB(void *, SoSensor *);
  static void animateSensorRotationCB(void *, SoSensor *);
  void animateRefParticle();
  void saveCurCamera();
  void restoreCamera();
  double animateBtwPtsPeriod, speedStep;
  void incSpeed();
  void decSpeed();
  
  SoTimerSensor *animateSensor;
  SoTimerSensor *animateSensorRotation;
  SoNodeSensor *sceneChangeSensor;
  SbVec3f camStartPos, camEndPos;
  SbRotation camStartOrient, camEndOrient;
  
  static void prevViewPtCB(Widget, XtPointer, XtPointer);
  static void nextViewPtCB(Widget, XtPointer, XtPointer);
  static void saveViewPtCB(Widget, XtPointer, XtPointer);
  static void abbrOutputCB(Widget, XtPointer, XtPointer);
  static void pickRefPathCB(Widget, XtPointer, XtPointer);
  static void constructListsDialog(Widget, XtPointer, XtPointer);
  void saveViewPt(char *name);
  

  static void lookAtSceneElementCB(Widget, XtPointer, XtPointer);
  static void cancelSceneElementSelectionCB(Widget, XtPointer, XtPointer);

  void setReferencePath(SoLineSet*, SoCoordinate3*, bool append = false);
  void setReferencePathZPos();
  void findAndSetRefPath();
  SoCoordinate3 *getCoordsNode(SoFullPath *path);
  void getSceneElements(); // reads elements from the scene graph
  float sqrlen(const SbVec3f&);
  void distanceToTrajectory(const SbVec3f&, float&, SbVec3f&, int&);
  void sortElements();
  void createElementsList(Widget);
  static void closeMainWindowCB(Widget, XtPointer, XtPointer);
  void evenOutRefParticlePts();

  static void gotoRefPathStartCB(Widget, XtPointer, XtPointer);
  void gotoRefPathStart();
  static void invertRefPathCB(Widget, XtPointer, XtPointer);
  void invertRefPath();

  enum CameraType {
    PERSPECTIVE,
    ORTHOGRAPHIC
  };

    
  enum State {
    GENERAL,
    BEAMLINE,
    VIEWPOINT,
    ANIMATION,
    REVERSED_ANIMATION,
    PAUSED_ANIMATION,
    ROTATING
  };

  // For storing the view point
  struct viewPtData {
      char* viewPtName;
      int viewportMapping;
      SbVec3f position;
      SbRotation orientation;
      float aspectRatio;
      float nearDistance;
      float farDistance;
      float focalDistance;
      CameraType camType;
      float height;
  };
  
  struct sceneElement {
      std::string name;
      SoFullPath * path;
      SbVec3f center;
      float closestPointZCoord;

      sceneElement operator=(sceneElement const &rhs) const {
          sceneElement el;
          el.name = rhs.name;
          el.path = rhs.path;
          el.center = rhs.center;
          el.closestPointZCoord = rhs.closestPointZCoord;

          return el;
      }
  };

   struct elementForSorting {
      float closestPointZCoord;
      SbVec3f closestPoint;
      float smallestDistance;
      float distanceToBeamlineStart;
      std::string name;

      bool operator<(elementForSorting const &other) const
      {
         if (closestPointZCoord < other.closestPointZCoord)
            return true;
         if (closestPointZCoord > other.closestPointZCoord)
            return false;

         // otherwise closestPointZCoord == other.closestPointZCoord.
         // Compare the distances from the center of the element to
         // the start of the beamline.
         if (distanceToBeamlineStart < other.distanceToBeamlineStart)
            return true;
         if (distanceToBeamlineStart > other.distanceToBeamlineStart)
            return false;

         // In case both closestPointZCoord and smallestDistance are
         // equal, we have two exactly overlapping elements, if so
         // the order doesn't matter
         return true;
      }
   };

  bool zcoordSetFlag;

  std::vector<sceneElement> sceneElements;
  std::vector<viewPtData> viewPtList;
  std::string fileName;
  std::ifstream fileIn;
  std::ofstream fileOut; 
  int viewPtIdx;
  int MAX_VP_IDX;
  int MAX_VP_NAME;
  
  // For storing coordinate points of the reference particle
  std::vector<SbVec3f> refParticleTrajectory;
  // For displaying distance during anim and beamline modes
  std::vector<float> refZPositions;
  int refParticleIdx;
  int prevRefIdx;
  float distance;
  State currentState, prevState, beforePausing;
  char *curViewPtName;
  
  int step;
  SbVec3f prevPt;
  SbVec3f prevParticleDir;
  void * prevColorField;
  
  viewPtData camB4Animation;
  bool returnToSaveVP;
  bool returnToAnim;
  SoCamera *myCam;
  void setStartingPtForAnimation(); 
  float left_right, up_down;    
  SbVec3f rotAxis; // For 90 degree rotations
  int rotCnt;  // # of steps rotation is split into

  static void getViewPtNameCB(Widget, XtPointer, XtPointer);
  std::string viewPtAutoName();

  SoSearchAction * searcher;

  SoNode * superimposition;
  SoCoordinate3 * sgeometry;
  SoScale * sscale;
  
  SoTranslation * stranslation;
  SoTranslation * curInfoTrans;
  SoTranslation * mouseOverTransSolid;
  SoTranslation * mouseOverTransMaterial;
  SoTranslation * mouseOverTransLogName;
  SoTranslation * mouseOverTransZPos;

  // Used for 2 similar purposes: 1. Displays z position during animation
  //                              2. Displays name of the current viewpoint
  SoText2 * curInfoText;
  /* Need to use many different fields for mouseover
   * because newlines are ignored when the scene is rendered */
  SoText2 * mouseOverTextSolid;
  SoText2 * mouseOverTextMaterial;
  SoText2 * mouseOverTextLogName;
  SoText2 * mouseOverTextZPos;
   
  SoFont * curInfoFont;
  SoFont * mouseOverFontSolid;
  SoFont * mouseOverFontMaterial;
  SoFont * mouseOverFontLogName;
  SoFont * mouseOverFontZPos;
  SoSwitch * axisSwitch;
  SoSwitch * animSpeedOutlineSwitch; 
  SoSwitch * animSpeedSwitch;
  SoSwitch * curInfoSwitch;
    
  SoNode * getSuperimpositionNode(SoNode *, const char * name);

  void superimpositionEvent(SoAction * action);
  static void superimpositionCB(void * closure, SoAction * action);

  virtual void actualRedraw(void);
  void updateSpeedIndicator(void);

  float maxSpeed; 

  // FWJ added for Ortho camera
  float defaultHeight;
  float defaultHeightAngle;
  
  // Used by G4 app during element rotations, stores previous view
  SbVec3f upVector, offsetFromCenter, center;   
  bool rotUpVec;

  SoSeparator * newSceneGraph;


};
#endif /* G4OPENINVENTORXTEXAMINERVIEWER_HH */