PoMesh2DVec.h

/*=======================================================================
 *** THE CONTENT OF THIS WORK IS PROPRIETARY TO FEI S.A.S, (FEI S.A.S.),            ***
 ***              AND IS DISTRIBUTED UNDER A LICENSE AGREEMENT.                     ***
 ***                                                                                ***
 ***  REPRODUCTION, DISCLOSURE,  OR USE,  IN WHOLE OR IN PART,  OTHER THAN AS       ***
 ***  SPECIFIED  IN THE LICENSE ARE  NOT TO BE  UNDERTAKEN  EXCEPT WITH PRIOR       ***
 ***  WRITTEN AUTHORIZATION OF FEI S.A.S.                                           ***
 ***                                                                                ***
 ***                        RESTRICTED RIGHTS LEGEND                                ***
 ***  USE, DUPLICATION, OR DISCLOSURE BY THE GOVERNMENT OF THE CONTENT OF THIS      ***
 ***  WORK OR RELATED DOCUMENTATION IS SUBJECT TO RESTRICTIONS AS SET FORTH IN      ***
 ***  SUBPARAGRAPH (C)(1) OF THE COMMERCIAL COMPUTER SOFTWARE RESTRICTED RIGHT      ***
 ***  CLAUSE  AT FAR 52.227-19  OR SUBPARAGRAPH  (C)(1)(II)  OF  THE RIGHTS IN      ***
 ***  TECHNICAL DATA AND COMPUTER SOFTWARE CLAUSE AT DFARS 52.227-7013.             ***
 ***                                                                                ***
 ***                   COPYRIGHT (C) 1996-2014 BY FEI S.A.S,                        ***
 ***                        BORDEAUX, FRANCE                                        ***
 ***                      ALL RIGHTS RESERVED                                       ***
**=======================================================================*/
/*=======================================================================
** Author      : VSG (MMM YYYY)
**=======================================================================*/
 
#ifndef  _PO_MESH2DVEC_
#define  _PO_MESH2DVEC_
 
#include <Inventor/SbLinear.h>
#include <MeshViz/3Ddata/PoMesh2D.h>
 
class PiArrow;
class PbCartesianGrid2D;
class PbGrid2D;
class PbIndexedMesh2D;
class PbMesh2D;
class PbParalCartesianGrid2D;
class PbRegularCartesianGrid2D;
class PbPolarGrid2D;
 
class  PoMesh2DVec : public PoMesh2D {
 
  SO_KIT_HEADER(PoMesh2DVec) ;
 
  /* Define fields for new parts */
  SO_KIT_CATALOG_ENTRY_HEADER(bodySep);
  SO_KIT_CATALOG_ENTRY_HEADER(bodyApp);
  SO_KIT_CATALOG_ENTRY_HEADER(bodyGroup);
 
  SO_KIT_CATALOG_ENTRY_HEADER(startArrowSep);
  SO_KIT_CATALOG_ENTRY_HEADER(startArrowApp);
  SO_KIT_CATALOG_ENTRY_HEADER(startArrowGroup);
 
  SO_KIT_CATALOG_ENTRY_HEADER(endArrowSep);
  SO_KIT_CATALOG_ENTRY_HEADER(endArrowApp);
  SO_KIT_CATALOG_ENTRY_HEADER(endArrowGroup);
 
 
 public:
 
  PoMesh2DVec() ;
 
  virtual void   rebuild() ;
 
  enum EliminationStatus {
    NONE,
    TOO_SMALL,
    TOO_LONG,
    TOO_SMALL_OR_LONG
  } ;
 
 
 
  enum BodyShape {
    NO_BODY,
    LINE,
    CYLINDER
  } ;
 
 
 
  enum BodyLengthType {
    CONSTANT_LENGTH,
    RELATIVE_LENGTH
  } ;
 
 
 
  enum BodyRadiusType {
    CONSTANT_RADIUS,
    RELATIVE_RADIUS
  } ;
 
 
 
  enum ArrowShape {
    NO_SHAPE,
    POINT,
    CHEVRON,
    TRIANGLE, 
    RECTANGLE,
    CONE,
    BOX,
    SPHERE,
    INDIRECT_CHEVRON,
    INDIRECT_TRIANGLE,
    INDIRECT_CONE
  } ;
 
 
 
  enum ArrowHeightType {
    CONSTANT_HEIGHT,
    RELATIVE_HEIGHT
  } ;
 
 
 
  enum ArrowColoringType {
    CONSTANT_COLOR,
    MODULE_MAPPING_COLOR,
    SCALAR_MAPPING_COLOR
  } ;
 
 
 
  /* Fields */
 
  SoSFInt32  density;                
 
  SoSFEnum   eliminationStatus;      
 
  SoSFFloat  minLength;              
 
  SoSFFloat  maxLength;              
 
 
 
  /* Body representation */
 
  SoSFEnum   bodyShape;              
 
  SoSFEnum   bodyLengthType;         
 
  SoSFFloat  bodyLengthFactor;       
 
  SoSFEnum   bodyRadiusType;         
 
  SoSFFloat  bodyRadiusFactor;       
 
  SoSFEnum   bodyColoringType;       
 
 
 
  /* Start arrow representation */
 
  SoSFEnum   startArrowShape;        
 
  SoSFEnum   startArrowHeightType;   
 
  SoSFFloat  startArrowHeightFactor; 
 
  SoSFFloat  startArrowRadiusFactor; 
 
  SoSFEnum   startArrowColoringType; 
 
 
 
  /* End arrow representation */
 
  SoSFEnum   endArrowShape;        
 
  SoSFEnum   endArrowHeightType;   
 
  SoSFFloat  endArrowHeightFactor; 
 
  SoSFFloat  endArrowRadiusFactor; 
 
  SoSFEnum   endArrowColoringType; 
 
  /*----------------------------------------------------------------------------*/
 private:
  static void initClass() ;
    static void exitClass() ;
 
  void indexedMesh2DVec(PbIndexedMesh2D *mesh) ;
  void cartGrid2DVec(PbCartesianGrid2D *mesh) ;
  void paralCartGrid2DVec(PbParalCartesianGrid2D *mesh) ;
  void regularCartGrid2DVec(PbRegularCartesianGrid2D *mesh) ;
  void polarGrid2DVec(PbPolarGrid2D *mesh) ;
 
 private:
  virtual ~PoMesh2DVec() ;  
  // Destructor
 
  // Methods
  virtual SbBool setUpConnections(SbBool onOff, SbBool doItAlways = FALSE) ;
  virtual void   setDefaultOnNonWritingFields();
  virtual void addElementsAllCaches() ;
  virtual void preRebuild();
 
  void addVector(const SbVec3f &orig, const SbVec3f &vec);
  void startVectorField();
  void finishVectorField();
 
  SoGroup *m_bodyGroupNode, *m_startArrowGroupNode, *m_endArrowGroupNode;
 
  PiArrow *m_arrow;
  const float *m_scalarDataSet;
  int m_vecIndexToDraw, m_vecIndex;
 
  int   m_allocVecPoints, m_allocModuleColr, m_allocScalarColr, 
    m_allocModuleTransparencies, m_allocScalarTransparencies,
    m_numVecP, m_numVec;
  float (*m_vecPoints)[3];
  float (*m_moduleColr)[3];
  float (*m_scalarColr)[3];
  float *m_moduleTransparencies, *m_scalarTransparencies;
 
  /*----------------------------------------------------------------------------*/
 private:
 
  void drawVectorLine(int nod_index);
 
  void drawVectorLineScalarColr(int nod_index);
  void drawVectorLineScalarTranspColr(int nod_index);
 
  void drawVectorLineModuleColr(int nod_index);
  void drawVectorLineModuleTranspColr(int nod_index);
 
  void drawVectorLineScalarModuleColr(int nod_index);
  void drawVectorLineScalarModuleTranspColr(int nod_index);
 
  void drawVectorElaborated(int nod_index);
 
  void finishQuickVectorField();
  void finishElaboratedVectorField();
  void buildVectorFieldBodies();
  void buildVectorFieldStartArrow();
  void buildVectorFieldEndArrow();
 
  void (PoMesh2DVec::*drawVector)(int nod_index);
  void (PoMesh2DVec::*finishVectorFieldPtr)();
 
  // Sensors
  FieldSensorList m_fieldSensorList ;
 
  // List of fields defined in this class
  SoFieldList     m_fieldList ;
 
  SbVec3f m_arrowPoints[2];
  float   m_vecLength, m_arrowLength;
 
  ArrowColoringType m_bodyColoringType, m_startArrowColoringType, m_endArrowColoringType;
} ;
 
/*----------------------------------------------------------------------------*/
 
#endif /* _PO_MESH2DVEC_ */