]> SALOME platform Git repositories - tools/medcoupling.git/blob - src/MEDCoupling_Swig/MEDCouplingRemapperTest.py
Salome HOME
Merge from V6_main_20120808 08Aug12
[tools/medcoupling.git] / src / MEDCoupling_Swig / MEDCouplingRemapperTest.py
1 #  -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2012  CEA/DEN, EDF R&D
3 #
4 # This library is free software; you can redistribute it and/or
5 # modify it under the terms of the GNU Lesser General Public
6 # License as published by the Free Software Foundation; either
7 # version 2.1 of the License.
8 #
9 # This library is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12 # Lesser General Public License for more details.
13 #
14 # You should have received a copy of the GNU Lesser General Public
15 # License along with this library; if not, write to the Free Software
16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
17 #
18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 #
20
21 from MEDCouplingRemapper import *
22 from math import *
23 import unittest
24
25 class MEDCouplingBasicsTest(unittest.TestCase):
26     def testRemapper1(self):
27         sourceMesh=self.build2DSourceMesh_1();
28         targetMesh=self.build2DTargetMesh_1();
29         remapper=MEDCouplingRemapper()
30         remapper.setPrecision(1e-12);
31         remapper.setIntersectionType(Triangulation);
32         self.failUnless(remapper.prepare(sourceMesh,targetMesh,"P0P0")==1);
33         srcField=MEDCouplingFieldDouble.New(ON_CELLS);
34         srcField.setNature(ConservativeVolumic);
35         srcField.setMesh(sourceMesh);
36         array=DataArrayDouble.New();
37         ptr=sourceMesh.getNumberOfCells()*[None]
38         for i in xrange(sourceMesh.getNumberOfCells()):
39             ptr[i]=float(i+7)
40             pass
41         array.setValues(ptr,sourceMesh.getNumberOfCells(),1);
42         srcField.setArray(array);
43         trgfield=remapper.transferField(srcField,4.57);
44         values=trgfield.getArray().getValues();
45         valuesExpected=[7.5 ,7. ,7.,8.,7.5];
46         for i in xrange(targetMesh.getNumberOfCells()):
47             self.failUnless(abs(values[i]-valuesExpected[i])<1e-12);
48             pass
49         self.failUnless(1==trgfield.getArray().getNumberOfComponents());
50         pass
51
52     def testPrepareEx1(self):
53         sourceMesh=self.build2DSourceMesh_1();
54         targetMesh=self.build2DTargetMesh_3();
55         #
56         remapper=MEDCouplingRemapper();
57         remapper.setPrecision(1e-12);
58         remapper.setIntersectionType(Triangulation);
59         srcFt=MEDCouplingFieldTemplate.New(ON_CELLS);
60         trgFt=MEDCouplingFieldTemplate.New(ON_CELLS);
61         srcFt.setMesh(sourceMesh);
62         trgFt.setMesh(targetMesh);
63         self.assertEqual(1,remapper.prepareEx(srcFt,trgFt));
64         srcField=MEDCouplingFieldDouble.New(ON_CELLS);
65         srcField.setNature(ConservativeVolumic);
66         srcField.setMesh(sourceMesh);
67         array=DataArrayDouble.New();
68         ptr=sourceMesh.getNumberOfCells()*[None]
69         for i in xrange(sourceMesh.getNumberOfCells()):
70             ptr[i]=float(i+7);
71             pass
72         array.setValues(ptr,sourceMesh.getNumberOfCells(),1);
73         srcField.setArray(array);
74         trgfield=remapper.transferField(srcField,4.220173);
75         values=trgfield.getArray().getValues();
76         valuesExpected=[7.75, 7.0625, 4.220173,8.0]
77         self.assertEqual(4,trgfield.getArray().getNumberOfTuples());
78         self.assertEqual(1,trgfield.getArray().getNumberOfComponents());
79         for i0 in xrange(4):
80             self.assertAlmostEqual(valuesExpected[i0],values[i0],12);
81             pass
82         pass
83
84     def testPartialTransfer1(self):
85         sourceMesh=self.build2DSourceMesh_1();
86         targetMesh=self.build2DTargetMesh_3();
87         #
88         remapper=MEDCouplingRemapper();
89         remapper.setPrecision(1e-12);
90         remapper.setIntersectionType(Triangulation);
91         srcFt=MEDCouplingFieldTemplate.New(ON_CELLS);
92         trgFt=MEDCouplingFieldTemplate.New(ON_CELLS);
93         srcFt.setMesh(sourceMesh);
94         trgFt.setMesh(targetMesh);
95         self.assertEqual(1,remapper.prepareEx(srcFt,trgFt));
96         srcField=MEDCouplingFieldDouble.New(ON_CELLS);
97         srcField.setNature(ConservativeVolumic);
98         srcField.setMesh(sourceMesh);
99         array=DataArrayDouble.New();
100         ptr=sourceMesh.getNumberOfCells()*[None]
101         for i in xrange(sourceMesh.getNumberOfCells()):
102             ptr[i]=float(i+7);
103             pass
104         array.setValues(ptr,sourceMesh.getNumberOfCells(),1);
105         srcField.setArray(array);
106         trgfield=MEDCouplingFieldDouble.New(ON_CELLS);
107         trgfield.setNature(ConservativeVolumic);
108         trgfield.setMesh(targetMesh);
109         array=DataArrayDouble.New();
110         ptr=targetMesh.getNumberOfCells()*[None]
111         for i in xrange(targetMesh.getNumberOfCells()):
112             ptr[i]=4.220173;
113             pass
114         array.setValues(ptr,targetMesh.getNumberOfCells(),1);
115         trgfield.setArray(array);
116         remapper.partialTransfer(srcField,trgfield);
117         values=trgfield.getArray().getValues();
118         valuesExpected=[7.75, 7.0625, 4.220173,8.0]
119         self.assertEqual(4,trgfield.getArray().getNumberOfTuples());
120         self.assertEqual(1,trgfield.getArray().getNumberOfComponents());
121         for i0 in xrange(4):
122             self.assertAlmostEqual(valuesExpected[i0],values[i0],12);
123             pass
124         pass
125     
126     def build2DSourceMesh_1(self):
127         sourceCoords=[-0.3,-0.3, 0.7,-0.3, -0.3,0.7, 0.7,0.7]
128         sourceConn=[0,3,1,0,2,3]
129         sourceMesh=MEDCouplingUMesh.New("my name of mesh 2D",2)
130         sourceMesh.allocateCells(2);
131         sourceMesh.insertNextCell(NORM_TRI3,3,sourceConn[0:3]);
132         sourceMesh.insertNextCell(NORM_TRI3,3,sourceConn[3:6]);
133         sourceMesh.finishInsertingCells();
134         myCoords=DataArrayDouble.New();
135         myCoords.setValues(sourceCoords,4,2);
136         sourceMesh.setCoords(myCoords);
137         return sourceMesh;
138     
139     def build2DTargetMesh_1(self):
140         targetCoords=[-0.3,-0.3, 0.2,-0.3, 0.7,-0.3, -0.3,0.2, 0.2,0.2, 0.7,0.2, -0.3,0.7, 0.2,0.7, 0.7,0.7 ]
141         targetConn=[0,3,4,1, 1,4,2, 4,5,2, 6,7,4,3, 7,8,5,4]
142         targetMesh=MEDCouplingUMesh.New();
143         targetMesh.setMeshDimension(2);
144         targetMesh.allocateCells(5);
145         targetMesh.insertNextCell(NORM_QUAD4,4,targetConn[0:4]);
146         targetMesh.insertNextCell(NORM_TRI3,3,targetConn[4:7]);
147         targetMesh.insertNextCell(NORM_TRI3,3,targetConn[7:10]);
148         targetMesh.insertNextCell(NORM_QUAD4,4,targetConn[10:14]);
149         targetMesh.insertNextCell(NORM_QUAD4,4,targetConn[14:18]);
150         targetMesh.finishInsertingCells();
151         myCoords=DataArrayDouble.New();
152         myCoords.setValues(targetCoords,9,2);
153         targetMesh.setCoords(myCoords);
154         return targetMesh;
155
156     def build2DTargetMesh_3(self):
157         targetCoords=[-0.6,-0.4, -0.1,-0.4, 1.1,-0.4, 2.1,-0.4, -0.6,0.1,  -0.1,0.1,  1.1,0.1,  2.1,0.1, -0.6,1.1,  -0.1,1.1]
158         targetConn=[0,4,5,1, 1,5,6,2, 2,6,7,3, 4,8,9,5]
159         targetMesh=MEDCouplingUMesh.New();
160         targetMesh.setMeshDimension(2);
161         targetMesh.allocateCells(4);
162         for i in xrange(4):
163             targetMesh.insertNextCell(NORM_QUAD4,4,targetConn[4*i:4*(i+1)])
164             pass
165         targetMesh.finishInsertingCells();
166         myCoords=DataArrayDouble.New();
167         myCoords.setValues(targetCoords,10,2);
168         targetMesh.setCoords(myCoords);
169         return targetMesh;
170         pass
171     
172     def setUp(self):
173         pass
174     pass
175
176 unittest.main()