Salome HOME
c9ddc4cb95952c656ce31c7acc6541084f576882
[tools/medcoupling.git] / src / INTERP_KERNEL / PointLocator3DIntersectorP1P1.txx
1 // Copyright (C) 2007-2020  CEA/DEN, EDF R&D
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 // Author : Anthony Geay (CEA/DEN)
20 #ifndef __POINTLOCATOR3DINTERSECTORP1P1_TXX__
21 #define __POINTLOCATOR3DINTERSECTORP1P1_TXX__
22
23 #include "PointLocator3DIntersectorP1P1.hxx"
24 #include "Intersector3DP1P1.txx"
25 #include "MeshUtils.hxx"
26
27 namespace INTERP_KERNEL
28 {
29
30   /**
31    * Constructor creating object from target cell global number 
32    * 
33    * @param targetMesh  mesh containing the target elements
34    * @param srcMesh     mesh containing the source elements
35    */
36   template<class MyMeshType, class MyMatrix>
37   PointLocator3DIntersectorP1P1<MyMeshType,MyMatrix>::PointLocator3DIntersectorP1P1(const MyMeshType& targetMesh, const MyMeshType& srcMesh, double precision):
38     Intersector3DP1P1<MyMeshType,MyMatrix>(targetMesh,srcMesh),_precision(precision)
39   {
40   }
41
42   template<class MyMeshType, class MyMatrix>
43   PointLocator3DIntersectorP1P1<MyMeshType,MyMatrix>::~PointLocator3DIntersectorP1P1()
44   {
45   }
46
47   /**
48    * @param targetCell in C mode.
49    * @param srcCells in C mode.
50    */
51   template<class MyMeshType, class MyMatrix>
52   void PointLocator3DIntersectorP1P1<MyMeshType,MyMatrix>::intersectCells(ConnType targetCell, const std::vector<ConnType>& srcCells, MyMatrix& res)
53   {
54     std::vector<double> CoordsT;
55     Intersector3DP1P1<MyMeshType,MyMatrix>::getRealTargetCoordinates(OTT<ConnType,numPol>::indFC(targetCell),CoordsT);
56     std::size_t nbOfNodesT=CoordsT.size()/SPACEDIM;
57     const double *coordsS=Intersector3DP1P1<MyMeshType,MyMatrix>::_src_mesh.getCoordinatesPtr();
58     for(typename std::vector<ConnType>::const_iterator iterCellS=srcCells.begin();iterCellS!=srcCells.end();iterCellS++)
59       {
60         NormalizedCellType tS=Intersector3DP1P1<MyMeshType,MyMatrix>::_src_mesh.getTypeOfElement(OTT<ConnType,numPol>::indFC(*iterCellS));
61         if(tS!=NORM_TETRA4)
62           throw INTERP_KERNEL::Exception("Invalid source cell detected for meshdim==3. Only TETRA4 supported !");
63         const CellModel& cmTypeS=CellModel::GetCellModel(tS);
64         const ConnType *startOfCellNodeConnT=Intersector3DP1P1<MyMeshType,MyMatrix>::getStartConnOfTargetCell(targetCell);
65         for(std::size_t nodeIdT=0;nodeIdT<nbOfNodesT;nodeIdT++)
66           {
67             typename MyMatrix::value_type& resRow=res[OTT<ConnType,numPol>::ind2C(startOfCellNodeConnT[nodeIdT])];
68             std::vector<ConnType> connOfCurCellS;
69             Intersector3DP1P1<MyMeshType,MyMatrix>::getConnOfSourceCell(OTT<ConnType,numPol>::indFC(*iterCellS),connOfCurCellS);
70             if( PointLocatorAlgos<MyMeshType>::isElementContainsPointAlg3D(&CoordsT[nodeIdT*SPACEDIM],&connOfCurCellS[0],ToConnType(connOfCurCellS.size()),coordsS,cmTypeS,_precision) )
71               {
72                 double resLoc[4];
73                 std::vector<double> localCoordsS;
74                 Intersector3DP1P1<MyMeshType,MyMatrix>::getRealSourceCoordinates(OTT<ConnType,numPol>::indFC(*iterCellS),localCoordsS);
75                 std::vector<const double*> eap(4);
76                 eap[0]=&localCoordsS[0]; eap[1]=&localCoordsS[3]; eap[2]=&localCoordsS[6]; eap[3]=&localCoordsS[9];
77                 barycentric_coords(eap,&CoordsT[nodeIdT*SPACEDIM],resLoc);
78                 const ConnType *startOfCellNodeConnS=Intersector3DP1P1<MyMeshType,MyMatrix>::getStartConnOfSourceCell(*iterCellS);
79                 for(int nodeIdS=0;nodeIdS<4;nodeIdS++)
80                   {
81                     if(fabs(resLoc[nodeIdS])>_precision)
82                       {
83                         ConnType curNodeSInCmode=OTT<ConnType,numPol>::coo2C(startOfCellNodeConnS[nodeIdS]);
84                         typename MyMatrix::value_type::const_iterator iterRes=resRow.find(OTT<ConnType,numPol>::indFC(curNodeSInCmode));
85                         if(iterRes==resRow.end())
86                           resRow.insert(std::make_pair(OTT<ConnType,numPol>::indFC(curNodeSInCmode),resLoc[nodeIdS]));
87                         else
88                           {
89                             double val=(*iterRes).second+resLoc[nodeIdS];
90                             resRow.erase(OTT<ConnType,numPol>::indFC(curNodeSInCmode));
91                             resRow.insert(std::make_pair(OTT<ConnType,numPol>::indFC(curNodeSInCmode),val));
92                           }
93                       }
94                   }
95               }
96           }
97       }
98   }
99 }
100
101 #endif