|
15 | 15 |
|
16 | 16 | parser = argparse.ArgumentParser() |
17 | 17 | parser.add_argument('--libdirs', dest='libdirs', action='store',nargs='*',default=[]) |
| 18 | +parser.add_argument('--incdirs', dest='incdirs', action='store',nargs='*',default=[]) |
18 | 19 | args,unknown = parser.parse_known_args() |
19 | 20 | sys.argv = [sys.argv[0]] + unknown |
20 | 21 |
|
| 22 | +if args.libdirs: |
| 23 | + library_dirs=args.libdirs |
| 24 | + extra_link_args=["-Wl,-rpath,%s" %s for s in args.libdirs] |
| 25 | +else: |
| 26 | + library_dirs=[] |
| 27 | + extra_link_args=["-Wl,-rpath,/usr/local/lib"] |
| 28 | +if args.incdirs: |
| 29 | + include_dirs=[np.get_include()]+args.incdirs |
| 30 | +else: |
| 31 | + include_dirs=[np.get_include(),"../include"] |
| 32 | + |
21 | 33 | setup(name='PFNET', |
22 | | - version='1.2.8', |
| 34 | + version='1.2.9', |
23 | 35 | license='BSD 2-clause license', |
24 | 36 | description='Power Flow Network Library', |
25 | 37 | author='Tomas Tinoco De Rubira', |
|
31 | 43 | 'pfnet.constraints'], |
32 | 44 | ext_modules=cythonize([Extension("pfnet.cpfnet", |
33 | 45 | [os.path.join("pfnet","cpfnet.pyx")], |
34 | | - library_dirs=args.libdirs, |
| 46 | + include_dirs=include_dirs, |
| 47 | + library_dirs=library_dirs, |
35 | 48 | libraries=["pfnet"], |
36 | 49 | extra_compile_args=[], |
37 | | - extra_link_args=['-Wl,-rpath,%s' %s for s in args.libdirs]+["-Wl,-rpath,/usr/local/lib"], |
38 | | - include_dirs=["../include",np.get_include()])])) |
| 50 | + extra_link_args=extra_link_args)])) |
| 51 | + |
0 commit comments