]> SALOME platform Git repositories - tools/adao_interface.git/blob - TestAdaoExchange.cxx
Salome HOME
First implementation
[tools/adao_interface.git] / TestAdaoExchange.cxx
1 // Copyright (C) 2019 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.
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, anthony.geay@edf.fr, EDF R&D
20
21 #include "TestAdaoExchange.hxx"
22
23 #include "AdaoExchangeLayer4Quintet.hxx"
24 #include "AdaoExchangeLayerException.hxx"
25 #include "AdaoModelKeyVal.hxx"
26 #include "PyObjectRAII.hxx"
27
28 #include "py2cpp/py2cpp.hxx"
29
30 #include <vector>
31 #include <iterator>
32
33 std::vector<double> func(const std::vector<double>& vec)
34 {
35   return {vec[0],2.*vec[1],3.*vec[2],vec[0]+2.*vec[1]+3.*vec[2]};
36 }
37
38 PyObject *multiFunc(PyObject *inp)
39 {
40   PyGILState_STATE gstate(PyGILState_Ensure());
41   PyObjectRAII iterator(PyObjectRAII::FromNew(PyObject_GetIter(inp)));
42   if(iterator.isNull())
43     throw AdaoExchangeLayerException("Input object is not iterable !");
44   PyObject *item(nullptr);
45   PyObjectRAII numpyModule(PyObjectRAII::FromNew(PyImport_ImportModule("numpy")));
46   if(numpyModule.isNull())
47     throw AdaoExchangeLayerException("Failed to load numpy");
48   PyObjectRAII ravelFunc(PyObjectRAII::FromNew(PyObject_GetAttrString(numpyModule,"ravel")));
49   std::vector< PyObjectRAII > pyrets;
50   while( item = PyIter_Next(iterator) )
51     {
52       PyObjectRAII item2(PyObjectRAII::FromNew(item));
53       {
54         PyObjectRAII args(PyObjectRAII::FromNew(PyTuple_New(1)));
55         PyTuple_SetItem(args,0,item2.retn());
56         PyObjectRAII npyArray(PyObjectRAII::FromNew(PyObject_CallObject(ravelFunc,args)));
57         // Waiting management of npy arrays into py2cpp
58         PyObjectRAII lolistFunc(PyObjectRAII::FromNew(PyObject_GetAttrString(npyArray,"tolist")));
59         PyObjectRAII listPy;
60         {
61           PyObjectRAII args2(PyObjectRAII::FromNew(PyTuple_New(0)));
62           listPy=PyObjectRAII::FromNew(PyObject_CallObject(lolistFunc,args2));
63           }
64         std::vector<double> vect;
65         {
66           py2cpp::PyPtr listPy2(listPy.retn());
67           py2cpp::fromPyPtr(listPy2,vect);
68         }
69         //
70         PyGILState_Release(gstate);
71         std::vector<double> res(func(vect));
72         gstate=PyGILState_Ensure();
73         //
74         py2cpp::PyPtr resPy(py2cpp::toPyPtr(res));
75         PyObjectRAII resPy2(PyObjectRAII::FromBorrowed(resPy.get()));
76         pyrets.push_back(resPy2);
77       }
78     }
79   std::size_t len(pyrets.size());
80   PyObjectRAII ret(PyObjectRAII::FromNew(PyList_New(len)));
81   for(std::size_t i=0;i<len;++i)
82     {
83       PyList_SetItem(ret,i,pyrets[i].retn());
84     }
85   //PyObject *tmp(PyObject_Repr(ret));
86   // std::cerr << PyUnicode_AsUTF8(tmp) << std::endl;
87   PyGILState_Release(gstate);
88   return ret.retn();
89 }
90
91 PyObjectRAII NumpyToListWaitingForPy2CppManagement(PyObject *npObj)
92 {
93   PyObjectRAII func(PyObjectRAII::FromNew(PyObject_GetAttrString(npObj,"tolist")));
94   if(func.isNull())
95     throw AdaoExchangeLayerException("input pyobject does not have tolist attribute !");
96   PyObjectRAII args(PyObjectRAII::FromNew(PyTuple_New(0)));
97   PyObjectRAII ret(PyObjectRAII::FromNew(PyObject_CallObject(func,args)));
98   return ret;
99 }
100
101 void AdaoExchangeTest::setUp()
102 {
103 }
104
105 void AdaoExchangeTest::tearDown()
106 {
107 }
108
109 void AdaoExchangeTest::cleanUp()
110 {
111 }
112
113 using namespace AdaoModel;
114
115 void AdaoExchangeTest::test3DVar()
116 {
117   MainModel mm;
118   AdaoExchangeLayer4Quintet adao;
119   adao.init(&mm);
120   {
121     std::string sciptPyOfModelMaker(mm.pyStr());
122     std::cerr << sciptPyOfModelMaker << std::endl;
123   }
124   adao.execute();
125   PyObject *listOfElts( nullptr );
126   while( adao.next(listOfElts) )
127     {
128       PyObject *resultOfChunk(multiFunc(listOfElts));
129       adao.setResult(resultOfChunk);
130     }
131   PyObject *res(adao.getResult());
132   PyObjectRAII optimum(PyObjectRAII::FromNew(res));
133   PyObjectRAII optimum_4_py2cpp(NumpyToListWaitingForPy2CppManagement(optimum));
134   std::vector<double> vect;
135   {
136     py2cpp::PyPtr obj(optimum_4_py2cpp);
137     py2cpp::fromPyPtr(obj,vect);
138   }
139   CPPUNIT_ASSERT_EQUAL(3,(int)vect.size());
140   CPPUNIT_ASSERT_DOUBLES_EQUAL(2.,vect[0],1e-7);
141   CPPUNIT_ASSERT_DOUBLES_EQUAL(3.,vect[1],1e-7);
142   CPPUNIT_ASSERT_DOUBLES_EQUAL(4.,vect[2],1e-7);
143 }
144
145 void AdaoExchangeTest::testBlue()
146 {
147   class TestBlueVisitor : public RecursiveVisitor
148   {
149   public:
150     void visit(GenericKeyVal *obj)
151     {
152       EnumAlgoKeyVal *objc(dynamic_cast<EnumAlgoKeyVal *>(obj));
153       if(objc)
154         objc->setVal(EnumAlgo::Blue);
155     }
156     void enterSubDir(DictKeyVal *subdir) { }
157     void exitSubDir(DictKeyVal *subdir) { }
158   };
159
160   MainModel mm;
161   //
162   TestBlueVisitor vis;
163   mm.visitAll(&vis);
164   //
165   AdaoExchangeLayer4Quintet adao;
166   adao.init(&mm);
167   {
168     std::string sciptPyOfModelMaker(mm.pyStr());
169     std::cerr << sciptPyOfModelMaker << std::endl;
170   }
171   adao.execute();
172     PyObject *listOfElts( nullptr );
173     while( adao.next(listOfElts) )
174       {
175         PyObject *resultOfChunk(multiFunc(listOfElts));
176         adao.setResult(resultOfChunk);
177       }
178     PyObject *res(adao.getResult());
179     PyObjectRAII optimum(PyObjectRAII::FromNew(res));
180     PyObjectRAII optimum_4_py2cpp(NumpyToListWaitingForPy2CppManagement(optimum));
181     std::vector<double> vect;
182     {
183       py2cpp::PyPtr obj(optimum_4_py2cpp);
184       py2cpp::fromPyPtr(obj,vect);
185     }
186     CPPUNIT_ASSERT_EQUAL(3,(int)vect.size());
187     CPPUNIT_ASSERT_DOUBLES_EQUAL(2.,vect[0],1e-7);
188     CPPUNIT_ASSERT_DOUBLES_EQUAL(3.,vect[1],1e-7);
189     CPPUNIT_ASSERT_DOUBLES_EQUAL(4.,vect[2],1e-7);
190 }
191
192 void AdaoExchangeTest::testNonLinearLeastSquares()
193 {
194   class TestNonLinearLeastSquaresVisitor : public RecursiveVisitor
195   {
196   public:
197     void visit(GenericKeyVal *obj)
198     {
199       EnumAlgoKeyVal *objc(dynamic_cast<EnumAlgoKeyVal *>(obj));
200       if(objc)
201         objc->setVal(EnumAlgo::NonLinearLeastSquares);
202     }
203     void enterSubDir(DictKeyVal *subdir) { }
204     void exitSubDir(DictKeyVal *subdir) { }
205   };
206
207   MainModel mm;
208   //
209   TestNonLinearLeastSquaresVisitor vis;
210   mm.visitAll(&vis);
211   //
212   AdaoExchangeLayer4Quintet adao;
213   adao.init(&mm);
214   {
215     std::string sciptPyOfModelMaker(mm.pyStr());
216     std::cerr << sciptPyOfModelMaker << std::endl;
217   }
218   adao.execute();
219   PyObject *listOfElts( nullptr );
220   while( adao.next(listOfElts) )
221     {
222       PyObject *resultOfChunk(multiFunc(listOfElts));
223       adao.setResult(resultOfChunk);
224     }
225   PyObject *res(adao.getResult());
226   PyObjectRAII optimum(PyObjectRAII::FromNew(res));
227   PyObjectRAII optimum_4_py2cpp(NumpyToListWaitingForPy2CppManagement(optimum));
228   std::vector<double> vect;
229   {
230     py2cpp::PyPtr obj(optimum_4_py2cpp);
231     py2cpp::fromPyPtr(obj,vect);
232   }
233   CPPUNIT_ASSERT_EQUAL(3,(int)vect.size());
234   CPPUNIT_ASSERT_DOUBLES_EQUAL(2.,vect[0],1e-7);
235   CPPUNIT_ASSERT_DOUBLES_EQUAL(3.,vect[1],1e-7);
236   CPPUNIT_ASSERT_DOUBLES_EQUAL(4.,vect[2],1e-7);
237 }
238
239 CPPUNIT_TEST_SUITE_REGISTRATION( AdaoExchangeTest );
240
241 #include <cppunit/CompilerOutputter.h>
242 #include <cppunit/TestResult.h>
243 #include <cppunit/TestResultCollector.h>
244 #include <cppunit/TextTestProgressListener.h>
245 #include <cppunit/BriefTestProgressListener.h>
246 #include <cppunit/extensions/TestFactoryRegistry.h>
247 #include <cppunit/TestRunner.h>
248 #include <cppunit/TextTestRunner.h>
249
250 int main(int argc, char* argv[])
251 {
252   // --- Create the event manager and test controller
253   CPPUNIT_NS::TestResult controller;
254
255   // ---  Add a listener that collects test result
256   CPPUNIT_NS::TestResultCollector result;
257   controller.addListener( &result );        
258
259   // ---  Add a listener that print dots as test run.
260 #ifdef WIN32
261   CPPUNIT_NS::TextTestProgressListener progress;
262 #else
263   CPPUNIT_NS::BriefTestProgressListener progress;
264 #endif
265   controller.addListener( &progress );      
266
267   // ---  Get the top level suite from the registry
268
269   CPPUNIT_NS::Test *suite =
270     CPPUNIT_NS::TestFactoryRegistry::getRegistry().makeTest();
271
272   // ---  Adds the test to the list of test to run
273
274   CPPUNIT_NS::TestRunner runner;
275   runner.addTest( suite );
276   runner.run( controller);
277
278   // ---  Print test in a compiler compatible format.
279   std::ofstream testFile;
280   testFile.open("test.log", std::ios::out | std::ios::app);
281   testFile << "------ ADAO exchange test log:" << std::endl;
282   CPPUNIT_NS::CompilerOutputter outputter( &result, testFile );
283   outputter.write(); 
284
285   // ---  Run the tests.
286
287   bool wasSucessful = result.wasSuccessful();
288   testFile.close();
289
290   // ---  Return error code 1 if the one of test failed.
291
292   return wasSucessful ? 0 : 1;
293 }