From 27fcda1ba2e0d6677dc349c63760cde0f718b7ef Mon Sep 17 00:00:00 2001
From: Carlos Mastalli <carlos.mastalli@gmail.com>
Date: Tue, 16 Apr 2019 18:35:10 +0200
Subject: [PATCH] [module] Created a module that can be reused later (e.g.
 loading functions)

---
 CMakeLists.txt                                        |  6 ++++++
 python/CMakeList.txt                                  | 10 ++++++++++
 python/__init__.py                                    |  2 ++
 unittest/unittest_utils.py => python/robots_loader.py |  0
 unittest/test_load.py                                 |  2 +-
 5 files changed, 19 insertions(+), 1 deletion(-)
 create mode 100644 python/CMakeList.txt
 create mode 100644 python/__init__.py
 rename unittest/unittest_utils.py => python/robots_loader.py (100%)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 543659b..0c75238 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -2,6 +2,7 @@ CMAKE_MINIMUM_REQUIRED(VERSION 2.8)
 
 INCLUDE(cmake/base.cmake)
 INCLUDE(cmake/test.cmake)
+INCLUDE(cmake/python.cmake)
 
 SET(PROJECT_NAMESPACE gepetto)
 SET(PROJECT_NAME example-robot-data)
@@ -11,9 +12,14 @@ SET(PROJECT_URL https://gepgitlab.laas.fr/${PROJECT_NAMESPACE}/${PROJECT_NAME})
 
 SETUP_PROJECT()
 
+ADD_REQUIRED_DEPENDENCY("pinocchio >= 2.1.0")
+
+FINDPYTHON()
+
 INSTALL(DIRECTORY hyq_description DESTINATION share/${PROJECT_NAME})
 INSTALL(DIRECTORY talos_data DESTINATION share/${PROJECT_NAME})
 
+ADD_SUBDIRECTORY(python)
 ADD_SUBDIRECTORY(unittest)
 
 SETUP_PROJECT_FINALIZE()
diff --git a/python/CMakeList.txt b/python/CMakeList.txt
new file mode 100644
index 0000000..cba92ed
--- /dev/null
+++ b/python/CMakeList.txt
@@ -0,0 +1,10 @@
+SET(${PROJECT_NAME}_PYTHON_FILES
+  robots_loader.py
+  display.py
+  __init__.py
+  )
+
+FOREACH(python ${${PROJECT_NAME}_PYTHON_FILES})
+  PYTHON_BUILD(. ${python})
+  INSTALL(FILES ${python} DESTINATION ${PYTHON_SITELIB}/examples_robot_data)
+ENDFOREACH(python ${${PROJECT_NAME}_PYTHON_FILES})
diff --git a/python/__init__.py b/python/__init__.py
new file mode 100644
index 0000000..3d3f63d
--- /dev/null
+++ b/python/__init__.py
@@ -0,0 +1,2 @@
+from robots_loader import (getModelPath, loadHyQ, loadICub, loadTalos, loadTalosArm, loadTiago, loadTiagoNoHand,
+                           readParamsFromSrdf)
diff --git a/unittest/unittest_utils.py b/python/robots_loader.py
similarity index 100%
rename from unittest/unittest_utils.py
rename to python/robots_loader.py
diff --git a/unittest/test_load.py b/unittest/test_load.py
index 0654710..8e1f4e0 100755
--- a/unittest/test_load.py
+++ b/unittest/test_load.py
@@ -2,7 +2,7 @@
 
 import unittest
 
-from unittest_utils import loadHyQ, loadICub, loadTalos, loadTalosArm, loadTiago, loadTiagoNoHand
+from example_robot_data import loadHyQ, loadICub, loadTalos, loadTalosArm, loadTiago, loadTiagoNoHand
 
 
 class RobotTestCase(unittest.TestCase):
-- 
GitLab