Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Menu
Open sidebar
ORNL Quantum Computing Institute
xacc
Commits
7c557f69
Commit
7c557f69
authored
Mar 15, 2018
by
Mccaskey, Alex
Browse files
adding setup.py for pip install integration
Signed-off-by:
Alex McCaskey
<
mccaskeyaj@ornl.gov
>
parent
ef1a69b9
Changes
1
Show whitespace changes
Inline
Side-by-side
setup.py
0 → 100644
View file @
7c557f69
#! /usr/bin/env python3
import
os
import
re
import
sys
import
sysconfig
import
platform
import
subprocess
import
multiprocessing
from
distutils.version
import
LooseVersion
from
setuptools
import
setup
,
Extension
,
find_packages
from
setuptools.command.build_ext
import
build_ext
from
setuptools.command.install
import
install
as
InstallCommandBase
from
setuptools.command.test
import
test
as
TestCommand
from
shutil
import
copyfile
,
copymode
import
shutil
env
=
os
.
environ
.
copy
()
class
CMakeExtension
(
Extension
):
def
__init__
(
self
,
name
,
sourcedir
=
''
):
Extension
.
__init__
(
self
,
name
,
sources
=
[])
self
.
sourcedir
=
os
.
path
.
abspath
(
sourcedir
)
openssl_root_dir
=
None
class
InstallCommand
(
InstallCommandBase
):
user_options
=
InstallCommandBase
.
user_options
+
[
(
'openssl-dir='
,
None
,
'Specify OPENSSL_ROOT_DIR'
)
]
def
initialize_options
(
self
):
InstallCommandBase
.
initialize_options
(
self
)
self
.
openssl_dir
=
'/usr/local/opt/openssl'
if
platform
.
system
()
==
'Darwin'
else
'/usr/lib64'
def
finalize_options
(
self
):
global
openssl_root_dir
print
(
'openssl '
,
self
.
openssl_dir
)
openssl_root_dir
=
self
.
openssl_dir
InstallCommandBase
.
finalize_options
(
self
)
def
run
(
self
):
global
install_directory
InstallCommandBase
.
run
(
self
)
self
.
do_egg_install
()
subdirs
=
[
name
for
name
in
os
.
listdir
(
self
.
install_lib
)
if
os
.
path
.
isdir
(
os
.
path
.
join
(
self
.
install_lib
,
name
))
and
'xacc-0.1.0'
in
name
and
'egg-info'
not
in
name
]
copyfile
(
env
[
'HOME'
]
+
'/.xacc/lib/python/pyxacc.so'
,
self
.
install_lib
+
subdirs
[
0
]
+
'/pyxacc.so'
)
class
CMakeBuild
(
build_ext
):
def
run
(
self
):
try
:
out
=
subprocess
.
check_output
([
'cmake'
,
'--version'
])
except
OSError
:
raise
RuntimeError
(
"CMake must be installed to build the following extensions: "
+
", "
.
join
(
e
.
name
for
e
in
self
.
extensions
))
if
platform
.
system
()
==
"Windows"
:
cmake_version
=
LooseVersion
(
re
.
search
(
r
'version\s*([\d.]+)'
,
out
.
decode
()).
group
(
1
))
if
cmake_version
<
'3.1.0'
:
raise
RuntimeError
(
"CMake >= 3.1.0 is required on Windows"
)
for
ext
in
self
.
extensions
:
self
.
build_extension
(
ext
)
def
build_extension
(
self
,
ext
):
extdir
=
os
.
path
.
abspath
(
os
.
path
.
dirname
(
self
.
get_ext_fullpath
(
ext
.
name
)))
cmake_args
=
[
'-DPYTHON_EXECUTABLE='
+
sys
.
executable
,
'-DOPENSSL_ROOT_DIR='
+
openssl_root_dir
,
'-DCMAKE_INSTALL_PREFIX='
+
env
[
'HOME'
]
+
'/.xacc'
]
cfg
=
'Debug'
if
self
.
debug
else
'Release'
build_args
=
[
'--config'
,
cfg
]
if
platform
.
system
()
==
"Windows"
:
cmake_args
+=
[
'-DCMAKE_LIBRARY_OUTPUT_DIRECTORY_{}={}'
.
format
(
cfg
.
upper
(),
extdir
)]
if
sys
.
maxsize
>
2
**
32
:
cmake_args
+=
[
'-A'
,
'x64'
]
build_args
+=
[
'--'
,
'/m'
]
else
:
cmake_args
+=
[
'-DCMAKE_BUILD_TYPE='
+
cfg
]
build_args
+=
[
'--'
,
'-j'
+
str
(
multiprocessing
.
cpu_count
())]
env
[
'CXXFLAGS'
]
=
'{} -DVERSION_INFO=
\\
"{}
\\
"'
.
format
(
env
.
get
(
'CXXFLAGS'
,
''
),
self
.
distribution
.
get_version
())
if
not
os
.
path
.
exists
(
self
.
build_temp
):
os
.
makedirs
(
self
.
build_temp
)
subprocess
.
check_call
([
'cmake'
,
ext
.
sourcedir
]
+
cmake_args
,
cwd
=
self
.
build_temp
,
env
=
env
)
subprocess
.
check_call
([
'cmake'
,
'--build'
,
'.'
]
+
build_args
,
cwd
=
self
.
build_temp
)
print
()
# Add an empty line for cleaner output
s
=
setup
(
name
=
'xacc'
,
version
=
'0.1.0'
,
author
=
'Alex McCaskey'
,
author_email
=
'xacc-dev@eclipse.org'
,
description
=
'Hardware-agnostic quantum programming framework'
,
long_description
=
'XACC provides a language and hardware agnostic programming framework for hybrid classical-quantum applications.'
,
scripts
=
[
'tools/plugins/xacc-install-plugins.py'
],
ext_modules
=
[
CMakeExtension
(
'.'
)],
cmdclass
=
{
'build_ext'
:
CMakeBuild
,
'install'
:
InstallCommand
},
zip_safe
=
False
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment