src/SMDS/SMDS_VtkFace.cxx

   1 // Copyright (C) 2010-2013  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 #include "SMDS_VtkFace.hxx"
  21 #include "SMDS_MeshNode.hxx"
  22 #include "SMDS_Mesh.hxx"
  23 #include "SMDS_VtkCellIterator.hxx"
  24
  25 #include "utilities.h"
  26
  27 #include <vector>
  28
  29 using namespace std;
  30
  31 SMDS_VtkFace::SMDS_VtkFace()
  32 {
  33 }
  34
  35 SMDS_VtkFace::SMDS_VtkFace(const std::vector<vtkIdType>& nodeIds, SMDS_Mesh* mesh)
  36 {
  37   init(nodeIds, mesh);
  38 }
  39
  40 SMDS_VtkFace::~SMDS_VtkFace()
  41 {
  42 }
  43
  44 void SMDS_VtkFace::init(const std::vector<vtkIdType>& nodeIds, SMDS_Mesh* mesh)
  45 {
  46   SMDS_MeshFace::init();
  47   vtkUnstructuredGrid* grid = mesh->getGrid();
  48   myMeshId = mesh->getMeshId();
  49   vtkIdType aType = VTK_TRIANGLE;
  50   switch (nodeIds.size())
  51   {
  52     case 3:
  53       aType = VTK_TRIANGLE;
  54       break;
  55     case 4:
  56       aType = VTK_QUAD;
  57       break;
  58     case 6:
  59       aType = VTK_QUADRATIC_TRIANGLE;
  60       break;
  61     case 8:
  62       aType = VTK_QUADRATIC_QUAD;
  63       break;
  64     case 9:
  65       aType = VTK_BIQUADRATIC_QUAD;
  66       break;
  67     default:
  68       aType = VTK_POLYGON;
  69       break;
  70   }
  71   myVtkID = grid->InsertNextLinkedCell(aType, nodeIds.size(), (vtkIdType*) &nodeIds[0]);
  72   mesh->setMyModified();
  73   //MESSAGE("SMDS_VtkFace::init myVtkID " << myVtkID);
  74 }
  75
  76 void SMDS_VtkFace::initPoly(const std::vector<vtkIdType>& nodeIds, SMDS_Mesh* mesh)
  77 {
  78   SMDS_MeshFace::init();
  79   vtkUnstructuredGrid* grid = mesh->getGrid();
  80   myMeshId = mesh->getMeshId();
  81   myVtkID = grid->InsertNextLinkedCell(VTK_POLYGON, nodeIds.size(), (vtkIdType*) &nodeIds[0]);
  82   mesh->setMyModified();
  83 }
  84
  85 bool SMDS_VtkFace::ChangeNodes(const SMDS_MeshNode* nodes[], const int nbNodes)
  86 {
  87   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
  88   vtkIdType npts = 0;
  89   vtkIdType* pts = 0;
  90   grid->GetCellPoints(myVtkID, npts, pts);
  91   if (nbNodes != npts)
  92     {
  93       MESSAGE("ChangeNodes problem: not the same number of nodes " << npts << " -> " << nbNodes);
  94       return false;
  95     }
  96   for (int i = 0; i < nbNodes; i++)
  97     {
  98       pts[i] = nodes[i]->getVtkId();
  99     }
 100   SMDS_Mesh::_meshList[myMeshId]->setMyModified();
 101   return true;
 102 }
 103
 104 void SMDS_VtkFace::Print(std::ostream & OS) const
 105 {
 106   OS << "face <" << GetID() << "> : ";
 107 }
 108
 109 int SMDS_VtkFace::NbEdges() const
 110 {
 111   // TODO quadratic polygons ?
 112   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 113   vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
 114   int nbEdges = 3;
 115   switch (aVtkType)
 116   {
 117     case VTK_TRIANGLE:
 118     case VTK_QUADRATIC_TRIANGLE:
 119       nbEdges = 3;
 120       break;
 121     case VTK_QUAD:
 122     case VTK_QUADRATIC_QUAD:
 123     case VTK_BIQUADRATIC_QUAD:
 124       nbEdges = 4;
 125       break;
 126     case VTK_POLYGON:
 127     default:
 128       nbEdges = grid->GetCell(myVtkID)->GetNumberOfPoints();
 129       break;
 130   }
 131   return nbEdges;
 132 }
 133
 134 int SMDS_VtkFace::NbFaces() const
 135 {
 136   return 1;
 137 }
 138
 139 int SMDS_VtkFace::NbNodes() const
 140 {
 141   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 142   int nbPoints = grid->GetCell(myVtkID)->GetNumberOfPoints();
 143   return nbPoints;
 144 }
 145
 146 /*!
 147  * \brief Return node by its index
 148  * \param ind - node index
 149  * \retval const SMDS_MeshNode* - the node
 150  */
 151 const SMDS_MeshNode*
 152 SMDS_VtkFace::GetNode(const int ind) const
 153 {
 154   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 155   vtkIdType npts, *pts;
 156   grid->GetCellPoints( this->myVtkID, npts, pts );
 157   return SMDS_Mesh::_meshList[myMeshId]->FindNodeVtk( pts[ ind ]);
 158 }
 159
 160 bool SMDS_VtkFace::IsQuadratic() const
 161 {
 162   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 163   vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
 164   // TODO quadratic polygons ?
 165   switch (aVtkType)
 166   {
 167     case VTK_QUADRATIC_TRIANGLE:
 168     case VTK_QUADRATIC_QUAD:
 169     case VTK_BIQUADRATIC_QUAD:
 170       return true;
 171       break;
 172     default:
 173       return false;
 174   }
 175 }
 176
 177 bool SMDS_VtkFace::IsPoly() const
 178 {
 179   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 180   vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
 181   return (aVtkType == VTK_POLYGON);
 182 }
 183
 184 bool SMDS_VtkFace::IsMediumNode(const SMDS_MeshNode* node) const
 185 {
 186   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 187   vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
 188   int rankFirstMedium = 0;
 189   switch (aVtkType)
 190   {
 191     case VTK_QUADRATIC_TRIANGLE:
 192       rankFirstMedium = 3; // medium nodes are of rank 3,4,5
 193       break;
 194     case VTK_QUADRATIC_QUAD:
 195     case VTK_BIQUADRATIC_QUAD:
 196       rankFirstMedium = 4; // medium nodes are of rank 4,5,6,7
 197       break;
 198     default:
 199       //MESSAGE("wrong element type " << aVtkType);
 200       return false;
 201   }
 202   vtkIdType npts = 0;
 203   vtkIdType* pts = 0;
 204   grid->GetCellPoints(myVtkID, npts, pts);
 205   vtkIdType nodeId = node->getVtkId();
 206   for (int rank = 0; rank < npts; rank++)
 207     {
 208       if (pts[rank] == nodeId)
 209         {
 210           //MESSAGE("rank " << rank << " is medium node " << (rank < rankFirstMedium));
 211           if (rank < rankFirstMedium)
 212             return false;
 213           else
 214             return true;
 215         }
 216     }
 217   //throw SALOME_Exception(LOCALIZED("node does not belong to this element"));
 218   MESSAGE("======================================================");
 219   MESSAGE("= IsMediumNode: node does not belong to this element =");
 220   MESSAGE("======================================================");
 221   return false;
 222 }
 223
 224 int SMDS_VtkFace::NbCornerNodes() const
 225 {
 226   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 227   int       nbPoints = grid->GetCell(myVtkID)->GetNumberOfPoints();
 228   vtkIdType aVtkType = grid->GetCellType(myVtkID);
 229   if ( aVtkType != VTK_POLYGON )
 230     return nbPoints <= 4 ? nbPoints : nbPoints / 2;
 231   return nbPoints;
 232 }
 233
 234 SMDSAbs_EntityType SMDS_VtkFace::GetEntityType() const
 235 {
 236   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 237   vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
 238   return SMDS_MeshCell::toSmdsType( VTKCellType( aVtkType ));
 239 }
 240
 241 SMDSAbs_GeometryType SMDS_VtkFace::GetGeomType() const
 242 {
 243   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 244   vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
 245   switch ( aVtkType ) {
 246   case VTK_TRIANGLE:
 247   case VTK_QUADRATIC_TRIANGLE:
 248     return SMDSGeom_TRIANGLE;
 249   case VTK_QUAD:
 250   case VTK_QUADRATIC_QUAD:
 251   case VTK_BIQUADRATIC_QUAD:
 252     return SMDSGeom_QUADRANGLE;
 253   case VTK_POLYGON:
 254     return SMDSGeom_POLYGON;
 255   default:;
 256   }
 257   return SMDSGeom_NONE;
 258 }
 259
 260 vtkIdType SMDS_VtkFace::GetVtkType() const
 261 {
 262   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 263   vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
 264   return aVtkType;
 265 }
 266
 267 SMDS_ElemIteratorPtr SMDS_VtkFace::elementsIterator(SMDSAbs_ElementType type) const
 268 {
 269   switch (type)
 270   {
 271     case SMDSAbs_Node:
 272       return SMDS_ElemIteratorPtr(new SMDS_VtkCellIterator(SMDS_Mesh::_meshList[myMeshId], myVtkID, GetEntityType()));
 273     default:
 274       MESSAGE("ERROR : Iterator not implemented")
 275       ;
 276       return SMDS_ElemIteratorPtr((SMDS_ElemIterator*) NULL);
 277   }
 278 }
 279
 280 SMDS_ElemIteratorPtr SMDS_VtkFace::nodesIteratorToUNV() const
 281 {
 282   return SMDS_ElemIteratorPtr(new SMDS_VtkCellIteratorToUNV(SMDS_Mesh::_meshList[myMeshId], myVtkID, GetEntityType()));
 283 }
 284
 285 SMDS_ElemIteratorPtr SMDS_VtkFace::interlacedNodesElemIterator() const
 286 {
 287   return SMDS_ElemIteratorPtr(new SMDS_VtkCellIteratorToUNV(SMDS_Mesh::_meshList[myMeshId], myVtkID, GetEntityType()));
 288 }
 289
 290 //! change only the first node, used for temporary triangles in quadrangle to triangle adaptor
 291 void SMDS_VtkFace::ChangeApex(SMDS_MeshNode* node)
 292 {
 293   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 294   vtkIdType npts = 0;
 295   vtkIdType* pts = 0;
 296   grid->GetCellPoints(myVtkID, npts, pts);
 297   grid->RemoveReferenceToCell(pts[0], myVtkID);
 298   pts[0] = node->getVtkId();
 299   node->AddInverseElement(this),
 300   SMDS_Mesh::_meshList[myMeshId]->setMyModified();
 301 }