-#!/usr/bin/env python-for-pytorch
+#!/usr/bin/env python
+
+#
+# flatland is a simple 2d physical simulator
+#
+# Copyright (c) 2016 Idiap Research Institute, http://www.idiap.ch/
+# Written by Francois Fleuret <francois.fleuret@idiap.ch>
+#
+# This file is part of flatland
+#
+# flatland is free software: you can redistribute it and/or modify it
+# under the terms of the GNU General Public License version 3 as
+# published by the Free Software Foundation.
+#
+# flatland is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with flatland. If not, see <http://www.gnu.org/licenses/>.
+#
import os
from torch.utils.ffi import create_extension
-this_file = os.path.dirname(__file__)
-
-print('__file__', __file__)
+abs_path = os.path.dirname(os.path.abspath(__file__))
ffi = create_extension(
- '_ext.mylib',
- headers = [ 'mylib.h' ],
- sources = [ 'mylib.c' ],
- extra_objects = [ '/home/fleuret/sources/python/flatland/flatland.so' ],
+ '_ext.flatland',
+ headers = [ 'flatland.h' ],
+ sources = [ 'flatland.c' ],
+ extra_objects = [ abs_path + '/flatland_generator.so' ],
libraries = [ ],
library_dirs = [ ],
define_macros = [ ],
- relative_to=this_file,
with_cuda = False
)