# Copyright (c) 2015-2016, NVIDIA CORPORATION.  All rights reserved.

sudo: required
dist: trusty

language: python
python: 2.7

env:
    matrix:
        - DIGITS_TEST_FRAMEWORK=caffe CAFFE_FORK=NVIDIA
        - DIGITS_TEST_FRAMEWORK=caffe CAFFE_FORK=BVLC
        - DIGITS_TEST_FRAMEWORK=torch
        - DIGITS_TEST_FRAMEWORK=none
        - DEB_BUILD=true

cache:
    apt: true
    directories:
        - ~/caffe
        - ~/torch

addons:
    apt:
        packages:
            - build-essential
            - cmake
            - cython
            - git
            - gfortran
            - graphviz
            - libboost-filesystem-dev
            - libboost-python-dev
            - libboost-system-dev
            - libboost-thread-dev
            - libgflags-dev
            - libgoogle-glog-dev
            - libhdf5-serial-dev
            - libleveldb-dev
            - liblmdb-dev
            - libopenblas-dev
            - libopencv-dev
            - libprotobuf-dev
            - libsnappy-dev
            - protobuf-compiler
            - python-dev
            - python-flask
            - python-gevent
            - python-gevent-websocket
            - python-gflags
            - python-h5py
            - python-matplotlib
            - python-mock
            - python-nose
            - python-numpy
            - python-pil
            - python-pip
            - python-protobuf
            - python-psutil
            - python-requests
            - python-scipy
            - python-six
            - python-skimage

services:
    - docker

before_install:
    - deactivate
    - virtualenv --system-site-packages ~/venv
    - source ~/venv/bin/activate

install:
    - ./scripts/travis/install-caffe.sh ~/caffe
    - travis_wait ./scripts/travis/install-torch.sh ~/torch
    - ./scripts/travis/install-python-requirements.sh
    - ./scripts/travis/install-digits.sh

script:
    - ./scripts/travis/test.sh
    - ./scripts/travis/deb-build.sh