albaAvatar3D.h

Go to the documentation of this file.

/*=========================================================================
 
 Program: ALBA (Agile Library for Biomedical Applications)
 Module: albaAvatar3D
 Authors: Marco Petrone
 
 Copyright (c) BIC
 All rights reserved. See Copyright.txt or
 
 
 This software is distributed WITHOUT ANY WARRANTY; without even
 the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
 PURPOSE.  See the above copyright notice for more information.
 
=========================================================================*/
 
#ifndef __albaAvatar3D_h
#define __albaAvatar3D_h
 
#include "albaAvatar.h"
#include "albaMatrix.h"
 
//----------------------------------------------------------------------------
// forward declarations :
//----------------------------------------------------------------------------
class albaInteractor6DOF;
class albaDeviceButtonsPadTracker;
class albaTransform;
class albaCameraTransform;
class albaView;
class albaEventBase;
class vtkALBARayCast3DPicker;
class vtkTextMapper;
class vtkActor2D;
class vtkOutlineSource;
class vtkTransform;
class vtkCellPicker;
class vtkProp3D;
class vtkActor;
class vtkAssemblyPath;;
class vtkRenderer;
class vtkRenderWindowInteractor;
 
class ALBA_EXPORT albaAvatar3D : public albaAvatar
{
public:  
  //----------------------------------------------------------------------------
  //    GUI Constants
  //----------------------------------------------------------------------------
  enum AVATAR_3D_GUI_ID
  {
    ID_FITTING_COMBO = albaAvatar::ID_LAST,
    ID_WBOX_BOOL,
    ID_DEBUG_TEXT,
    ID_DEBUG_TEXT_POSITION,
    ID_COORDS_COMBO,
    ID_LAST
  };
 
  enum AVATAR_COORDS_TYPE
  {
    TRACKER_COORDS=0,
    CANONICAL_COORDS,
    WORLD_COORDS,
  };
 
  enum AVATAR_COORDS_SCALING
  {
    NO_SCALE=0,
    TRACKER_TO_WORLD_SCALE,
    CANONICAL_TO_WORLD_SCALE
  };
 
  albaAbstractTypeMacro(albaAvatar3D,albaAvatar);
 
  virtual void SetRendererAndView(vtkRenderer *ren, albaView *view);
  
  virtual void Hide();
 
  virtual void Show();
  
  vtkALBARayCast3DPicker *GetPicker3D();
 
  vtkCellPicker *GetPicker2D();
 
  virtual vtkAbstractPropPicker *GetPicker();
  
  void SetDisplayWorkingBox(int vis);
  int GetDisplayWorkingBox(){ return m_DisplayWorkingBox;}
  void DisplayWorkingBoxOn() {SetDisplayWorkingBox(true);}
  void DisplayWorkingBoxOff() {SetDisplayWorkingBox(false);}
 
  void SetDisplayDebugText(int vis);
  int GetDisplayDebugText() {return m_DisplayDebugText;}
 
  void SetDebugTextPosition(double *pos) {SetDebugTextPosition(pos[0],pos[1]);}
  void SetDebugTextPosition(double posx,double posy);
  double *GetDebugTextPosition() {return m_DebugTextPosition;}
  
  void SetCoordsFrame(int frame) {m_CoordsFrame=frame;}
  int GetCoordsFrame() {return m_CoordsFrame;}
  
 
  virtual void OnEvent(albaEventBase *event);
 
  virtual void SetTracker(albaDeviceButtonsPadTracker *tracker);
 
  void CanonicalToWorld(const albaMatrix &pose,albaMatrix &dest);
  void CanonicalToWorld(albaMatrix &pose) {CanonicalToWorld(pose,pose);}
  void CanonicalToWorld(albaTransform *trans);
 
  void WorldToCanonical(const albaMatrix &pose,albaMatrix &dest);
  void WorldToCanonical(albaMatrix &pose) {WorldToCanonical(pose,pose);}
  void WorldToCanonical(albaTransform *trans);
 
 
  void TrackerToWorld(albaMatrix &tracker_pose,albaMatrix &world_pose,int use_scale=1,int use_rot=1, int use_trans=1);
  void WorldToTracker(albaMatrix &world_pose,albaMatrix &tracker_pose,int use_scale=1,int use_rot=1, int use_trans=1);
 
  
  void TrackerToDisplay(albaMatrix &tracker_pose,double xy[2]);
 
  void WorldToDisplay(albaMatrix &world_pose,double xy[2]);
 
  void WorldToNormalizedDisplay(albaMatrix &world_pose,double xy[2]);
  
 
  albaCameraTransform *GetCanonicalToWorldTransform() {return m_CanonicalToWorldTransform;}
 
  void SetFittingMode(int type);
 
  int GetFittingMode() {return m_FittingMode;}
 
  albaMatrix &GetLastPoseMatrix() {return m_LastPoseMatrix;}
 
  virtual void CreateGui();
 
protected:
  albaAvatar3D();
  virtual ~albaAvatar3D();
 
  void SetLastPoseMatrix(albaMatrix &matrix);
 
  virtual int InternalStore(albaStorageElement *node);
  virtual int InternalRestore(albaStorageElement *node);
 
  virtual int InternalInitialize();
 
  virtual void InternalShutdown();
 
  virtual void UpdateDebugText(const char *title, albaMatrix &pose);
 
  virtual void OnMove3DEvent(albaEventInteraction *e);
  
  void OnUpdateBoundsEvent(albaDeviceButtonsPadTracker *tracker);
 
  virtual void OnPostResetCamera(albaEventBase *event);
 
  void SetPicker3D(vtkALBARayCast3DPicker *picker);
 
  vtkProp3D*          m_WorkingBoxActor; 
  vtkOutlineSource*   m_WorkingBox; 
 
  vtkTextMapper*      m_DebugTextMapper;
  vtkActor2D*         m_DebugTextActor;
 
  albaMatrix           m_LastPoseMatrix;
 
  vtkALBARayCast3DPicker* m_Picker3D; 
  vtkCellPicker*      m_Picker2D;
 
 
  albaCameraTransform* m_CanonicalToWorldTransform;
  
  int                 m_FittingMode; 
  
  int                 m_DisplayWorkingBox; 
  int                 m_DisplayDebugText; 
  int                 m_CoordsFrame; 
 
  double*              m_DebugTextPosition;
 
private:
  albaAvatar3D(const albaAvatar3D&);  // Not implemented.
  void operator=(const albaAvatar3D&);  // Not implemented.
};
 
#endif