setup.py

#   Copyright (c) 2018, Xilinx, Inc.
#   All rights reserved.
# 
#   Redistribution and use in source and binary forms, with or without 
#   modification, are permitted provided that the following conditions are met:
#
#   1.  Redistributions of source code must retain the above copyright notice, 
#       this list of conditions and the following disclaimer.
#
#   2.  Redistributions in binary form must reproduce the above copyright 
#       notice, this list of conditions and the following disclaimer in the 
#       documentation and/or other materials provided with the distribution.
#
#   3.  Neither the name of the copyright holder nor the names of its 
#       contributors may be used to endorse or promote products derived from 
#       this software without specific prior written permission.
#
#   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
#   AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, 
#   THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 
#   PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR 
#   CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 
#   EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 
#   PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
#   OR BUSINESS INTERRUPTION). HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, 
#   WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 
#   OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF 
#   ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.


from setuptools import setup, find_packages
import shutil
import subprocess
import sys
import os
from datetime import datetime

__author__ = "Naveen Purushotham"
__copyright__ = "Copyright 2018, Xilinx"
__email__ = "npurusho@xilinx.com"

GIT_DIR = os.path.dirname(os.path.realpath(__file__))


# Install packages
def install_packages():
    subprocess.check_call(['apt-get', '--yes', '--force-yes', 'install'])
    subprocess.check_call(['pip3.6', 'install'])
    print("Installing packages done ...")


# Notebook delivery
def fill_notebooks():
    src_nb = GIT_DIR + '/notebooks'
    dst_nb_dir = '/home/xilinx/jupyter_notebooks/pynqDL'
    if os.path.exists(dst_nb_dir):
        shutil.rmtree(dst_nb_dir)
    shutil.copytree(src_nb, dst_nb_dir)

    print("Filling notebooks done ...")


# Images delivery
def fill_images():
    src_nb = GIT_DIR + '/notebooks/images'
    dst_nb_dir = '/home/xilinx/jupyter_notebooks/pynqDL/images'
    if os.path.exists(dst_nb_dir):
        shutil.rmtree(dst_nb_dir)
    shutil.copytree(src_nb, dst_nb_dir)

    print("Filling notebooks done ...")


# Overlays delivery
def fill_overlays_darius():
    src_nb = GIT_DIR + '/darius/overlays'
    dst_nb_dir = '/home/xilinx/pynq/overlays/darius'
    if os.path.exists(dst_nb_dir):
        shutil.rmtree(dst_nb_dir)
    shutil.copytree(src_nb, dst_nb_dir)

    print("Filling overlays done ...")

    # Overlays delivery
def fill_overlays_resize():
    src_nb = GIT_DIR + '/resize/overlays'
    dst_nb_dir = '/home/xilinx/pynq/overlays/resize'
    if os.path.exists(dst_nb_dir):
        shutil.rmtree(dst_nb_dir)
    shutil.copytree(src_nb, dst_nb_dir)

    print("Filling overlays done ...")

# Overlays delivery
def fill_lib():
    src_nb = GIT_DIR + '/darius/lib'
    dst_nb_dir = '/home/xilinx/pynq/lib/darius'
    if os.path.exists(dst_nb_dir):
        shutil.rmtree(dst_nb_dir)
    shutil.copytree(src_nb, dst_nb_dir)

    print("Filling overlays done ...")


if len(sys.argv) > 1 and sys.argv[1] == 'install':
    install_packages()
    fill_notebooks()
    fill_images()
    fill_overlays_darius()
    fill_overlays_resize()
    fill_lib()


def package_files(directory):
    paths = []
    for (path, directories, file_names) in os.walk(directory):
        for file_name in file_names:
            paths.append(os.path.join('..', path, file_name))
    return paths


extra_files = package_files('pynqDL')

setup(name='pynqDL',
      version='1.0',
      description='Xilinx Deep Learning IP using PYNQ Framework',
      author='Xilinx',
      author_email='npurusho@xilinx.com',
      url='https://github.com/Xilinx/PYNQ-DL.git',
      packages=find_packages(),
      download_url='https://github.com/Xilinx/PYNQ-DL.git',
      package_data={
          '': extra_files,
      }
      )