about summary refs log tree commit diff
path: root/nixpkgs/pkgs/tools/graphics/fast-neural-doodle/fast-neural-doodle.sh
blob: a089d5a95061e60ec3f63498c233b93cd9025e97 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#! @bash@/bin/bash

declare -a args
c=1
flag=

for arg in "$@"; do
        if test "$arg" = "${arg#-}" && test "$arg" = "${arg#/}" && test -n "$flag"; then
                arg="$PWD/$arg"
                flag=
        elif (test "$arg" != "${arg%_image}" || test "$arg" == "-masks_hdf5") && test "$arg" != "${arg#-}"; then
                flag=1
        else
                flag=
        fi
        args[c]="$arg";
        c=$((c+1));
done

cd "@out@/lib/lua/fast_neural_doodle"

export LUA_PATH="$LUA_PATH${LUA_PATH:+;}@loadcaffe@/lua/?/init.lua;@loadcaffe@/lua/?.lua"
export LUA_PATH="$LUA_PATH${LUA_PATH:+;}@torch_hdf5@/lua/?/init.lua;@torch_hdf5@/lua/?.lua"
export LUA_CPATH="$LUA_CPATH${LUA_CPATH:+;}@loadcaffe@/lib/?.so"

set -x

@torch@/bin/th fast_neural_doodle.lua "${args[@]}"