from setuptools import setup, Distribution

class BinaryDistribution(Distribution):
    def has_ext_modules(foo):
        return True

packages=['paddle',
          'paddle.proto',
          'paddle.trainer',
          'paddle.trainer_config_helpers',
          'paddle.utils',
          'paddle.v2',
          'paddle.v2.dataset',
          'paddle.v2.reader',
          'paddle.v2.master',
          'paddle.v2.plot',
          'paddle.v2.framework',
          'paddle.v2.framework.proto',
          'py_paddle']

setup_requires=["requests",
                "numpy>=1.12",
                "protobuf==3.1",
                "recordio",
                "matplotlib",
                "rarfile",
                "scipy>=0.19.0",
                "Pillow",
                "nltk>=3.2.2"]

if '${CMAKE_SYSTEM_PROCESSOR}' not in ['arm', 'armv7-a', 'aarch64']:
    setup_requires+=["opencv-python"]

setup(name='paddlepaddle',
      version='${PADDLE_VERSION}',
      description='Parallel Distributed Deep Learning',
      install_requires=setup_requires,
      packages=packages,
      package_data={
        'paddle.v2.master': ['libpaddle_master.so'],
        'paddle.v2.framework': ['core.so'],
        'py_paddle':['*.py','_swig_paddle.so']
      },
      package_dir={
          '': '${CMAKE_CURRENT_SOURCE_DIR}',
          # The paddle.v2.framework.proto will be generated while compiling.
          # So that package points to other directory.
          'paddle.v2.framework.proto': '${PROJ_BINARY_ROOT}/paddle/framework',
          'py_paddle': '${PROJ_ROOT}/paddle/py_paddle'
      },
      scripts=['${PROJ_BINARY_ROOT}/paddle/scripts/paddle'],
      distclass=BinaryDistribution,
      data_files=[('/usr/local/opt/paddle/bin',
                       ['${PROJ_BINARY_ROOT}/paddle/scripts/paddle_usage',
                        '${PROJ_BINARY_ROOT}/paddle/trainer/paddle_trainer',
                        '${PROJ_BINARY_ROOT}/paddle/trainer/paddle_merge_model',
                        '${PROJ_BINARY_ROOT}/paddle/pserver/paddle_pserver_main'])]
)