src/SMDS/SMDS_VtkFace.cxx

   1 // Copyright (C) 2010-2014  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, or (at your option) any later version.
   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     case 7:
  68       aType = VTK_BIQUADRATIC_TRIANGLE;
  69       break;
  70     default:
  71       aType = VTK_POLYGON;
  72       break;
  73   }
  74   myVtkID = grid->InsertNextLinkedCell(aType, nodeIds.size(), (vtkIdType*) &nodeIds[0]);
  75   mesh->setMyModified();
  76   //MESSAGE("SMDS_VtkFace::init myVtkID " << myVtkID);
  77 }
  78
  79 void SMDS_VtkFace::initPoly(const std::vector<vtkIdType>& nodeIds, SMDS_Mesh* mesh)
  80 {
  81   SMDS_MeshFace::init();
  82   vtkUnstructuredGrid* grid = mesh->getGrid();
  83   myMeshId = mesh->getMeshId();
  84   myVtkID = grid->InsertNextLinkedCell(VTK_POLYGON, nodeIds.size(), (vtkIdType*) &nodeIds[0]);
  85   mesh->setMyModified();
  86 }
  87
  88 bool SMDS_VtkFace::ChangeNodes(const SMDS_MeshNode* nodes[], const int nbNodes)
  89 {
  90   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
  91   vtkIdType npts = 0;
  92   vtkIdType* pts = 0;
  93   grid->GetCellPoints(myVtkID, npts, pts);
  94   if (nbNodes != npts)
  95     {
  96       MESSAGE("ChangeNodes problem: not the same number of nodes " << npts << " -> " << nbNodes);
  97       return false;
  98     }
  99   for (int i = 0; i < nbNodes; i++)
 100     {
 101       pts[i] = nodes[i]->getVtkId();
 102     }
 103   SMDS_Mesh::_meshList[myMeshId]->setMyModified();
 104   return true;
 105 }
 106
 107 void SMDS_VtkFace::Print(std::ostream & OS) const
 108 {
 109   OS << "face <" << GetID() << "> : ";
 110 }
 111
 112 int SMDS_VtkFace::NbEdges() const
 113 {
 114   // TODO quadratic polygons ?
 115   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 116   vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
 117   int nbEdges = 3;
 118   switch (aVtkType)
 119   {
 120     case VTK_TRIANGLE:
 121     case VTK_QUADRATIC_TRIANGLE:
 122     case VTK_BIQUADRATIC_TRIANGLE:
 123       nbEdges = 3;
 124       break;
 125     case VTK_QUAD:
 126     case VTK_QUADRATIC_QUAD:
 127     case VTK_BIQUADRATIC_QUAD:
 128       nbEdges = 4;
 129       break;
 130     case VTK_POLYGON:
 131     default:
 132       nbEdges = grid->GetCell(myVtkID)->GetNumberOfPoints();
 133       break;
 134   }
 135   return nbEdges;
 136 }
 137
 138 int SMDS_VtkFace::NbFaces() const
 139 {
 140   return 1;
 141 }
 142
 143 int SMDS_VtkFace::NbNodes() const
 144 {
 145   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 146   int nbPoints = grid->GetCell(myVtkID)->GetNumberOfPoints();
 147   return nbPoints;
 148 }
 149
 150 /*!
 151  * \brief Return node by its index
 152  * \param ind - node index
 153  * \retval const SMDS_MeshNode* - the node
 154  */
 155 const SMDS_MeshNode*
 156 SMDS_VtkFace::GetNode(const int ind) const
 157 {
 158   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 159   vtkIdType npts, *pts;
 160   grid->GetCellPoints( this->myVtkID, npts, pts );
 161   return SMDS_Mesh::_meshList[myMeshId]->FindNodeVtk( pts[ ind ]);
 162 }
 163
 164 /*!
 165  * \brief Check if a node belongs to the element
 166  * \param node - the node to check
 167  * \retval int - node index within the element, -1 if not found
 168  */
 169 int SMDS_VtkFace::GetNodeIndex( const SMDS_MeshNode* node ) const
 170 {
 171   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 172   vtkIdType npts, *pts;
 173   grid->GetCellPoints( this->myVtkID, npts, pts );
 174   for ( vtkIdType i = 0; i < npts; ++i )
 175     if ( pts[i] == node->getVtkId() )
 176       return i;
 177   return -1;
 178 }
 179
 180 bool SMDS_VtkFace::IsQuadratic() const
 181 {
 182   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 183   vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
 184   // TODO quadratic polygons ?
 185   switch (aVtkType)
 186   {
 187     case VTK_QUADRATIC_TRIANGLE:
 188     case VTK_QUADRATIC_QUAD:
 189     case VTK_BIQUADRATIC_QUAD:
 190     case VTK_BIQUADRATIC_TRIANGLE:
 191       return true;
 192       break;
 193     default:
 194       return false;
 195   }
 196 }
 197
 198 bool SMDS_VtkFace::IsPoly() const
 199 {
 200   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 201   vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
 202   return (aVtkType == VTK_POLYGON);
 203 }
 204
 205 bool SMDS_VtkFace::IsMediumNode(const SMDS_MeshNode* node) const
 206 {
 207   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 208   vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
 209   int rankFirstMedium = 0;
 210   switch (aVtkType)
 211   {
 212     case VTK_QUADRATIC_TRIANGLE:
 213     case VTK_BIQUADRATIC_TRIANGLE:
 214       rankFirstMedium = 3; // medium nodes are of rank 3,4,5
 215       break;
 216     case VTK_QUADRATIC_QUAD:
 217     case VTK_BIQUADRATIC_QUAD:
 218       rankFirstMedium = 4; // medium nodes are of rank 4,5,6,7
 219       break;
 220     default:
 221       //MESSAGE("wrong element type " << aVtkType);
 222       return false;
 223   }
 224   vtkIdType npts = 0;
 225   vtkIdType* pts = 0;
 226   grid->GetCellPoints(myVtkID, npts, pts);
 227   vtkIdType nodeId = node->getVtkId();
 228   for (int rank = 0; rank < npts; rank++)
 229     {
 230       if (pts[rank] == nodeId)
 231         {
 232           //MESSAGE("rank " << rank << " is medium node " << (rank < rankFirstMedium));
 233           if (rank < rankFirstMedium)
 234             return false;
 235           else
 236             return true;
 237         }
 238     }
 239   //throw SALOME_Exception(LOCALIZED("node does not belong to this element"));
 240   MESSAGE("======================================================");
 241   MESSAGE("= IsMediumNode: node does not belong to this element =");
 242   MESSAGE("======================================================");
 243   return false;
 244 }
 245
 246 int SMDS_VtkFace::NbCornerNodes() const
 247 {
 248   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 249   int       nbPoints = grid->GetCell(myVtkID)->GetNumberOfPoints();
 250   vtkIdType aVtkType = grid->GetCellType(myVtkID);
 251   if ( aVtkType != VTK_POLYGON )
 252     return nbPoints <= 4 ? nbPoints : nbPoints / 2;
 253   return nbPoints;
 254 }
 255
 256 SMDSAbs_EntityType SMDS_VtkFace::GetEntityType() const
 257 {
 258   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 259   vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
 260   return SMDS_MeshCell::toSmdsType( VTKCellType( aVtkType ));
 261 }
 262
 263 SMDSAbs_GeometryType SMDS_VtkFace::GetGeomType() const
 264 {
 265   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 266   vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
 267   switch ( aVtkType ) {
 268   case VTK_TRIANGLE:
 269   case VTK_QUADRATIC_TRIANGLE:
 270   case VTK_BIQUADRATIC_TRIANGLE: return SMDSGeom_TRIANGLE;
 271
 272   case VTK_QUAD:
 273   case VTK_QUADRATIC_QUAD:
 274   case VTK_BIQUADRATIC_QUAD: return SMDSGeom_QUADRANGLE;
 275
 276   case VTK_POLYGON: return SMDSGeom_POLYGON;
 277   default:;
 278   }
 279   return SMDSGeom_NONE;
 280 }
 281
 282 vtkIdType SMDS_VtkFace::GetVtkType() const
 283 {
 284   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 285   vtkIdType aVtkType = grid->GetCellType(this->myVtkID);
 286   return aVtkType;
 287 }
 288
 289 SMDS_ElemIteratorPtr SMDS_VtkFace::elementsIterator(SMDSAbs_ElementType type) const
 290 {
 291   switch (type)
 292   {
 293     case SMDSAbs_Node:
 294       return SMDS_ElemIteratorPtr(new SMDS_VtkCellIterator(SMDS_Mesh::_meshList[myMeshId], myVtkID, GetEntityType()));
 295     default:
 296       MESSAGE("ERROR : Iterator not implemented")
 297       ;
 298       return SMDS_ElemIteratorPtr((SMDS_ElemIterator*) NULL);
 299   }
 300 }
 301
 302 SMDS_NodeIteratorPtr SMDS_VtkFace::nodesIteratorToUNV() const
 303 {
 304   return SMDS_NodeIteratorPtr(new SMDS_VtkCellIteratorToUNV(SMDS_Mesh::_meshList[myMeshId], myVtkID, GetEntityType()));
 305 }
 306
 307 SMDS_NodeIteratorPtr SMDS_VtkFace::interlacedNodesIterator() const
 308 {
 309   return nodesIteratorToUNV();
 310 }
 311
 312 //! change only the first node, used for temporary triangles in quadrangle to triangle adaptor
 313 void SMDS_VtkFace::ChangeApex(SMDS_MeshNode* node)
 314 {
 315   vtkUnstructuredGrid* grid = SMDS_Mesh::_meshList[myMeshId]->getGrid();
 316   vtkIdType npts = 0;
 317   vtkIdType* pts = 0;
 318   grid->GetCellPoints(myVtkID, npts, pts);
 319   grid->RemoveReferenceToCell(pts[0], myVtkID);
 320   pts[0] = node->getVtkId();
 321   node->AddInverseElement(this),
 322   SMDS_Mesh::_meshList[myMeshId]->setMyModified();
 323 }