getCellsContainingPoints is sensitive to 2D quadratic static cells.

[tools/medcoupling.git] / src / MEDCoupling / MEDCouplingNormalizedUnstructuredMesh.txx

diff --git a/src/MEDCoupling/MEDCouplingNormalizedUnstructuredMesh.txx b/src/MEDCoupling/MEDCouplingNormalizedUnstructuredMesh.txx
index 4ad2677b6e1a4e5c7c85c537f368f4c7159d2f05..111b6186cb3776cf513ac447c6d1a7bef740cd51 100644 (file)
--- a/src/MEDCoupling/MEDCouplingNormalizedUnstructuredMesh.txx
+++ b/src/MEDCoupling/MEDCouplingNormalizedUnstructuredMesh.txx
@@ -1,9 +1,9 @@
-// Copyright (C) 2007-2012  CEA/DEN, EDF R&D
+// Copyright (C) 2007-2016  CEA/DEN, EDF R&D
 //
 // This library is free software; you can redistribute it and/or
 // modify it under the terms of the GNU Lesser General Public
 // License as published by the Free Software Foundation; either
-// version 2.1 of the License.
+// version 2.1 of the License, or (at your option) any later version.
 //
 // This library is distributed in the hope that it will be useful,
 // but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ -16,18 +16,22 @@
 //
 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
 //
+// Author : Anthony Geay (CEA/DEN)
+
 #ifndef __MEDCOUPLINGNORMALIZEDUNSTRUCTUREDMESH_TXX__
 #define __MEDCOUPLINGNORMALIZEDUNSTRUCTUREDMESH_TXX__
 
 #include "MEDCouplingNormalizedUnstructuredMesh.hxx"
+#include "InterpKernelAssert.hxx"
 
 #include "MEDCouplingUMesh.hxx"
+#include "MEDCoupling1GTUMesh.hxx"
 #include "MEDCouplingMemArray.hxx"
 
 #include <limits>
 
 template<int SPACEDIM,int MESHDIM>
-MEDCouplingNormalizedUnstructuredMesh<SPACEDIM,MESHDIM>::MEDCouplingNormalizedUnstructuredMesh(const ParaMEDMEM::MEDCouplingUMesh *mesh):_mesh(mesh)
+MEDCouplingNormalizedUnstructuredMesh<SPACEDIM,MESHDIM>::MEDCouplingNormalizedUnstructuredMesh(const MEDCoupling::MEDCouplingPointSet *mesh):_mesh(mesh)
 {
   if(_mesh)
     _mesh->incrRef();
@@ -42,7 +46,7 @@ void MEDCouplingNormalizedUnstructuredMesh<SPACEDIM,MESHDIM>::getBoundingBox(dou
       boundingBox[i]=std::numeric_limits<double>::max();
       boundingBox[SPACEDIM+i]=-std::numeric_limits<double>::max();
     }
-  const ParaMEDMEM::DataArrayDouble *array=_mesh->getCoords();
+  const MEDCoupling::DataArrayDouble *array=_mesh->getCoords();
   const double *ptr=array->getConstPointer();
   int nbOfPts=array->getNbOfElems()/SPACEDIM;
   for(int j=0;j<SPACEDIM;j++)
@@ -65,19 +69,19 @@ INTERP_KERNEL::NormalizedCellType MEDCouplingNormalizedUnstructuredMesh<SPACEDIM
 }
 
 template<int SPACEDIM,int MESHDIM>
-unsigned char MEDCouplingNormalizedUnstructuredMesh<SPACEDIM,MESHDIM>::getNumberOfNodesOfElement(int eltId) const
+int MEDCouplingNormalizedUnstructuredMesh<SPACEDIM,MESHDIM>::getNumberOfNodesOfElement(int eltId) const
 {
   return _mesh->getNumberOfNodesInCell(eltId);
 }
 
 template<int SPACEDIM,int MESHDIM>
-unsigned long MEDCouplingNormalizedUnstructuredMesh<SPACEDIM,MESHDIM>::getNumberOfElements() const
+int MEDCouplingNormalizedUnstructuredMesh<SPACEDIM,MESHDIM>::getNumberOfElements() const
 {
   return _mesh->getNumberOfCells();
 }
 
 template<int SPACEDIM,int MESHDIM>
-unsigned long MEDCouplingNormalizedUnstructuredMesh<SPACEDIM,MESHDIM>::getNumberOfNodes() const
+int MEDCouplingNormalizedUnstructuredMesh<SPACEDIM,MESHDIM>::getNumberOfNodes() const
 {
   return _mesh->getNumberOfNodes();
 }
@@ -91,7 +95,7 @@ const int *MEDCouplingNormalizedUnstructuredMesh<SPACEDIM,MESHDIM>::getConnectiv
 template<int SPACEDIM,int MESHDIM>
 const double *MEDCouplingNormalizedUnstructuredMesh<SPACEDIM,MESHDIM>::getCoordinatesPtr() const
 {
-  const ParaMEDMEM::DataArrayDouble *array=_mesh->getCoords();
+  const MEDCoupling::DataArrayDouble *array=_mesh->getCoords();
   return array->getConstPointer();
 }
 
@@ -121,24 +125,54 @@ MEDCouplingNormalizedUnstructuredMesh<SPACEDIM,MESHDIM>::~MEDCouplingNormalizedU
 template<int SPACEDIM,int MESHDIM>
 void MEDCouplingNormalizedUnstructuredMesh<SPACEDIM,MESHDIM>::prepare()
 {
-  int nbOfCell=_mesh->getNumberOfCells();
-  int initialConnSize=_mesh->getNodalConnectivity()->getNbOfElems();
-  _conn_for_interp=new int[initialConnSize-nbOfCell];
-  _conn_index_for_interp=new int[nbOfCell+1];
-  _conn_index_for_interp[0]=0;
-  const int *work_conn=_mesh->getNodalConnectivity()->getConstPointer()+1;
-  const int *work_conn_index=_mesh->getNodalConnectivityIndex()->getConstPointer();
-  int *work_conn_for_interp=_conn_for_interp;
-  int *work_conn_index_for_interp=_conn_index_for_interp;
-  for(int i=0;i<nbOfCell;i++)
+  IKAssert(_mesh->getSpaceDimension()==SPACEDIM);
+  const MEDCoupling::MEDCouplingUMesh *m1(dynamic_cast<const MEDCoupling::MEDCouplingUMesh *>(_mesh));
+  if(m1)
+    {
+      int nbOfCell=m1->getNumberOfCells();
+      int initialConnSize=m1->getNodalConnectivity()->getNbOfElems();
+      _conn_for_interp=new int[initialConnSize-nbOfCell];
+      _conn_index_for_interp=new int[nbOfCell+1];
+      _conn_index_for_interp[0]=0;
+      const int *work_conn=m1->getNodalConnectivity()->getConstPointer()+1;
+      const int *work_conn_index=m1->getNodalConnectivityIndex()->getConstPointer();
+      int *work_conn_for_interp=_conn_for_interp;
+      int *work_conn_index_for_interp=_conn_index_for_interp;
+      for(int i=0;i<nbOfCell;i++)
+        {
+          int nbOfValsToCopy=work_conn_index[1]-work_conn_index[0]-1;
+          work_conn_for_interp=std::copy(work_conn,work_conn+nbOfValsToCopy,work_conn_for_interp);
+          work_conn_index_for_interp[1]=work_conn_index_for_interp[0]+nbOfValsToCopy;
+          work_conn_index++;
+          work_conn+=nbOfValsToCopy+1;
+          work_conn_index_for_interp++;
+        }
+      return ;
+    }
+  const MEDCoupling::MEDCoupling1DGTUMesh *m2(dynamic_cast<const MEDCoupling::MEDCoupling1DGTUMesh *>(_mesh));
+  if(m2)
+    {
+      int nbOfCell(m2->getNumberOfCells());
+      _conn_index_for_interp=new int[nbOfCell+1];
+      const int *conni(m2->getNodalConnectivityIndex()->begin());
+      std::copy(conni,conni+nbOfCell+1,_conn_index_for_interp);
+      _conn_for_interp=new int[m2->getNodalConnectivity()->getNumberOfTuples()];
+      std::copy(m2->getNodalConnectivity()->begin(),m2->getNodalConnectivity()->end(),_conn_for_interp);
+      return ;
+    }
+  const MEDCoupling::MEDCoupling1SGTUMesh *m3(dynamic_cast<const MEDCoupling::MEDCoupling1SGTUMesh *>(_mesh));
+  if(m3)
     {
-      int nbOfValsToCopy=work_conn_index[1]-work_conn_index[0]-1;
-      work_conn_for_interp=std::copy(work_conn,work_conn+nbOfValsToCopy,work_conn_for_interp);
-      work_conn_index_for_interp[1]=work_conn_index_for_interp[0]+nbOfValsToCopy;
-      work_conn_index++;
-      work_conn+=nbOfValsToCopy+1;
-      work_conn_index_for_interp++;
+      int nbOfCell(m3->getNumberOfCells()),nbNodesPerCell(m3->getNumberOfNodesPerCell());
+      _conn_index_for_interp=new int[nbOfCell+1]; _conn_index_for_interp[0]=0;
+      int *work(_conn_index_for_interp);
+      for(int i=0;i<nbOfCell;i++,work++)
+        work[1]=work[0]+nbNodesPerCell;
+      _conn_for_interp=new int[m3->getNodalConnectivity()->getNumberOfTuples()];
+      std::copy(m3->getNodalConnectivity()->begin(),m3->getNodalConnectivity()->end(),_conn_for_interp);
+      return ;
     }
+  throw INTERP_KERNEL::Exception("MEDCouplingNormalizedUnstructuredMesh::prepare : Unrecognized unstructured mesh ! Type must be in MEDCouplingUMesh, MEDCoupling1DGTUMesh, MEDCoupling1SGTUMesh !");
 }
 
 #endif