PbGrid3D.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-2020 BY FEI S.A.S,                        ***
 ***                        BORDEAUX, FRANCE                                        ***
 ***                      ALL RIGHTS RESERVED                                       ***
**=======================================================================*/
/*=======================================================================
** Author      : VSG (MMM YYYY)
**=======================================================================*/
 
#ifndef  _PB_GRID3D_
#define  _PB_GRID3D_
 
#include <MeshViz/3Ddata/PbMesh3D.h>
 
class  PbGrid3D : public PbMesh3D {
 
 
 public:
 
  PbGrid3D(SbBool is_data_duplicate=TRUE) : PbMesh3D(is_data_duplicate) {};
 
  PbGrid3D(const PbGrid3D &mesh);
 
  ~PbGrid3D();
 
  PbGrid3D& operator=(const PbGrid3D &mesh) ;
 
  void setGeometry(int num_x, int num_y, int num_z, const float *x, const float *y, const float *z);
 
  void getGeometry(int &num_x, int &num_y, int &num_z, const float* &x, const float* &y, const float* &z) const;
 
  virtual SbVec3f getNodeCoord(int nod_index) const=0;
 
  virtual SbVec3f getNodeCoord(int i, int j, int k) const=0;
 
  void getDim(int &num_x, int &num_y, int &num_z) const { num_x = numX; num_y = numY; num_z = numZ;};
 
  virtual void getNodeOwnerCellsInd(int nod_index, PbArrayOfInt &owner_cells) const;
 
  virtual void getAdjacentCellsIndByNode(int cell_index, PbArrayOfInt &adjacent_cells) const;
 
  virtual void getAdjacentCellsIndByFacet(int cell_index, PbArrayOfInt &adjacent_cells) const;
 
  void  getNodeIndices(int nod_index, int &i, int &j, int &k) const;
 
  void  getCellIndices(int cell_index, int &i, int &j, int &k) const;
 
  /*----------------------------------------------------------------------------*/
 private:
  void meshSkin(PoMeshSkin *_PoMeshSkin) const;
 
  virtual int evaluateSkinFacesNum() const { 
    return 2*((numZ-1)*(numY-1) + (numZ-1)*(numX-1) + (numX-1)*(numY-1));
  }
 
  int * getCellNodeIds(int i, int j, int k) const;
  int * getCellNodeIds(int cell_index) const;
 
  void print(FILE *fp) const;
 
 private:
  virtual void grid3DSkin(PoMeshSkin *_PoMeshSkin) const = 0;
 
  PbGrid3D(int num_x, int num_y, int num_z, SbBool isDataDuplicate=TRUE) ;
  void setDim(int num_x, int num_y, int num_z);
  virtual void setXYZGeometry(int num_x, int num_y, int num_z, const float *x, const float *y, const float *z)=0;
  virtual void getXYZGeometry(const float* &x, const float* &y, const float* &z) const=0;
  int   numX,numY,numZ;
 
  // node ids of the mutable private cell
  // each time the getCell or getTopoCell is called, m_cellNodeIds is updated
  mutable int m_cellNodeIds[8]; 
 
 private:
  void copy(const PbGrid3D &mesh) ;
  void destroy();
} ;
 
/*---------------------------------------------------------------------------*/
 
// calculate indices i,j,k that verify    nod_index = (i*numY + j)*numZ + k 
inline void
PbGrid3D::getNodeIndices(int nod_index, int &i, int &j, int &k) const
{
  int tmp;
  k = nod_index % numZ;   
  tmp = (nod_index-k) / numZ;           // tmp = i*numY + j
  j = tmp % numY;
  i = (tmp-j) / numY;
}
 
// calculate indices i,j,k that verify    cell_index = (i*(numY-1) + j)*(numZ-1) + k 
inline void
PbGrid3D::getCellIndices(int cell_index, int &i, int &j, int &k) const
{
  int ny=numY-1, nz=numZ-1;
  int tmp;
  k = cell_index % nz;   
  tmp = (cell_index-k) / nz;           // tmp = i*ny + j
  j = tmp % ny;
  i = (tmp-j) / ny;
}
 
// set dimensions of the grid
inline void
PbGrid3D::setDim(int num_x, int num_y, int num_z)
{
  numX = num_x;
  numY = num_y;
  numZ = num_z;
  numMeshNodes = numX * numY * numZ;
  m_numMeshCells = (numX-1) * (numY-1) * (numZ-1);
  updateTopologyId();
}
 
inline int * 
PbGrid3D::getCellNodeIds(int cell_index) const
{
  int i, j, k;
  getCellIndices(cell_index, i, j, k);
  return getCellNodeIds(i,j,k);
}
 
#define NSI_(i,j,k)  numZ*((j)+(i)*numY) + (k)
inline int * 
PbGrid3D::getCellNodeIds(int i, int j, int k) const
{
  m_cellNodeIds[4] = (m_cellNodeIds[0] = NSI_(i  ,j  ,k  )) + 1;
  m_cellNodeIds[5] = (m_cellNodeIds[1] = NSI_(i+1,j  ,k  )) + 1;
  m_cellNodeIds[6] = (m_cellNodeIds[2] = NSI_(i+1,j+1,k  )) + 1;
  m_cellNodeIds[7] = (m_cellNodeIds[3] = NSI_(i  ,j+1,k  )) + 1;
  return m_cellNodeIds;
}
 
#endif /* _PB_GRID3D_ */