mirror of https://github.com/explosion/spaCy.git
103 lines
3.8 KiB
Python
103 lines
3.8 KiB
Python
#!/usr/bin/env python
|
|
import Cython.Distutils
|
|
from Cython.Distutils import Extension
|
|
import distutils.core
|
|
|
|
import sys
|
|
import os
|
|
import os.path
|
|
|
|
from os import path
|
|
from glob import glob
|
|
|
|
import numpy
|
|
|
|
|
|
def clean(ext):
|
|
for pyx in ext.sources:
|
|
if pyx.endswith('.pyx'):
|
|
c = pyx[:-4] + '.c'
|
|
cpp = pyx[:-4] + '.cpp'
|
|
so = pyx[:-4] + '.so'
|
|
html = pyx[:-4] + '.html'
|
|
if os.path.exists(so):
|
|
os.unlink(so)
|
|
if os.path.exists(c):
|
|
os.unlink(c)
|
|
elif os.path.exists(cpp):
|
|
os.unlink(cpp)
|
|
if os.path.exists(html):
|
|
os.unlink(html)
|
|
|
|
|
|
HERE = os.path.dirname(__file__)
|
|
virtual_env = os.environ.get('VIRTUAL_ENV', '')
|
|
compile_args = []
|
|
link_args = []
|
|
libs = []
|
|
|
|
includes = ['.', numpy.get_include()]
|
|
cython_includes = ['.']
|
|
|
|
|
|
if 'VIRTUAL_ENV' in os.environ:
|
|
includes += glob(path.join(os.environ['VIRTUAL_ENV'], 'include', 'site', '*'))
|
|
else:
|
|
# If you're not using virtualenv, set your include dir here.
|
|
pass
|
|
|
|
ext_args = {'language': "c++", "include_dirs": includes}
|
|
|
|
exts = [
|
|
Extension("spacy.strings", ["spacy/strings.pyx"], **ext_args),
|
|
Extension("spacy.lexeme", ["spacy/lexeme.pyx"], **ext_args),
|
|
Extension("spacy.vocab", ["spacy/vocab.pyx"], **ext_args),
|
|
Extension("spacy.tokens", ["spacy/tokens.pyx"], **ext_args),
|
|
Extension("spacy.morphology", ["spacy/morphology.pyx"], **ext_args),
|
|
Extension("spacy.tagger", ["spacy/tagger.pyx"], **ext_args),
|
|
Extension("spacy.tokenizer", ["spacy/tokenizer.pyx"], **ext_args),
|
|
Extension("spacy.en.lang", ["spacy/en/lang.pyx"], **ext_args),
|
|
Extension("spacy.en.pos", ["spacy/en/pos.pyx"], **ext_args),
|
|
Extension("spacy.syntax._state", ["spacy/syntax/_state.pyx"], **ext_args),
|
|
Extension("spacy.syntax.arc_eager", ["spacy/syntax/arc_eager.pyx"], **ext_args),
|
|
#Extension("spacy.syntax.parser", ["spacy/syntax/parser.pyx"], **ext_args),
|
|
#Extension("spacy.syntax.parser", ["spacy/syntax/parser.pyx"], language="c++",
|
|
# include_dirs=includes),
|
|
#Extension("spacy.syntax.arc_eager", ["spacy/syntax/arc_eager.pyx"], language="c++",
|
|
# include_dirs=includes),
|
|
#Extension("spacy.syntax._state", ["spacy/syntax/_state.pyx"], language="c++",
|
|
# include_dirs=includes),
|
|
# Extension("spacy.syntax._parse_features", ["spacy/syntax/_parse_features.pyx"], language="c++",
|
|
# include_dirs=includes),
|
|
|
|
#Extension("spacy.pos_feats", ["spacy/pos_feats.pyx"], language="c++", include_dirs=includes),
|
|
#Extension("spacy.ner._state", ["spacy/ner/_state.pyx"], language="c++", include_dirs=includes),
|
|
#Extension("spacy.ner.bilou_moves", ["spacy/ner/bilou_moves.pyx"], language="c++", include_dirs=includes),
|
|
#Extension("spacy.ner.io_moves", ["spacy/ner/io_moves.pyx"], language="c++", include_dirs=includes),
|
|
#Extension("spacy.ner.greedy_parser", ["spacy/ner/greedy_parser.pyx"], language="c++", include_dirs=includes),
|
|
#Extension("spacy.ner.pystate", ["spacy/ner/pystate.pyx"], language="c++", include_dirs=includes),
|
|
#Extension("spacy.ner.context", ["spacy/ner/context.pyx"], language="c++", include_dirs=includes),
|
|
#Extension("spacy.ner.feats", ["spacy/ner/feats.pyx"], language="c++", include_dirs=includes),
|
|
#Extension("spacy.ner.annot", ["spacy/ner/annot.pyx"], language="c++", include_dirs=includes),
|
|
]
|
|
|
|
|
|
if sys.argv[1] == 'clean':
|
|
print >> sys.stderr, "cleaning .c, .c++ and .so files matching sources"
|
|
map(clean, exts)
|
|
|
|
distutils.core.setup(
|
|
name='spacy',
|
|
packages=['spacy'],
|
|
author='Matthew Honnibal',
|
|
author_email='honnibal@gmail.com',
|
|
version='1.0',
|
|
package_data={"spacy": ["*.pxd"]},
|
|
cmdclass={'build_ext': Cython.Distutils.build_ext},
|
|
|
|
ext_modules=exts,
|
|
)
|
|
|
|
|
|
|