rewrite the sh

3 files changed, 42 insertions, 12 deletions

diff --git a/.gitignore b/.gitignore
index 018a3de08144..370f82349600 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,3 +2,4 @@
 !PKGBUILD
 !.SRCINFO
 !.gitignore
+!cain-ncnn-vulkan
diff --git a/PKGBUILD b/PKGBUILD
index af387e9af6b9..342950f7e435 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -1,31 +1,31 @@
 # Maintainer: PumpkinCheshire <sollyonzou@gmail.com>
 
-_name=cain-ncnn-vulkan
 pkgname=cain-ncnn-vulkan-bin
 pkgver=20210210
-pkgrel=1
+pkgrel=2
 pkgdesc="CAIN, Channel Attention Is All You Need for Video Frame Interpolation implemented with ncnn library"
 arch=('x86_64' 'i686')
 url="https://github.com/nihui/cain-ncnn-vulkan"
 license=('MIT')
 depends=('vulkan-icd-loader' 'vulkan-headers')
-makedepends=()
+makedepends=('sh')
 provides=("cain-ncnn-vulkan")
 conflicts=("cain-ncnn-vulkan" "cain-ncnn-vulkan-git")
-source=("https://github.com/nihui/$_name/releases/download/$pkgver/$_name-$pkgver-ubuntu.zip")
-sha256sums=('25b5a02a876e1a9a78d9a049298989e4ac6c8eb50b60c464ff0052a340f095d2')
+source=("https://github.com/nihui/${pkgname%bin}/releases/download/$pkgver/${pkgname%-bin}-$pkgver-ubuntu.zip"
+	"${pkgname%-bin}")
+sha256sums=('25b5a02a876e1a9a78d9a049298989e4ac6c8eb50b60c464ff0052a340f095d2'
+	'002f5b77d7d697f01d2a360242a026464599401bf90a040764e839bb53f76e12')
 
 package() {
-	cd "${srcdir}/$_name-$pkgver-ubuntu"
+	install -Dm755 ${pkgname%-bin} "${pkgdir}/usr/bin/${pkgname%-bin}"
+
+	cd "${srcdir}/${pkgname%-bin}-$pkgver-ubuntu"
 	install -Dm644 LICENSE "${pkgdir}/usr/share/licenses/${pkgname}/LICENSE"
 
-	install -Dm755 "${srcdir}/$_name-$pkgver-ubuntu/$_name" "${pkgdir}/usr/share/$_name/$_name"
+	install -Dm755 "${srcdir}/${pkgname%-bin}-$pkgver-ubuntu/${pkgname%-bin}" "${pkgdir}/usr/share/${pkgname%-bin}/${pkgname%-bin}"
 
-	cd "${srcdir}/$_name-$pkgver-ubuntu/"
+	cd "${srcdir}/${pkgname%-bin}-$pkgver-ubuntu/"
 	for f in cain/*; do
-		install -Dm 644 "$f" ${pkgdir}/usr/share/$_name/"$f"
+		install -Dm 644 "$f" ${pkgdir}/usr/share/${pkgname%-bin}/"$f"
 	done
-
-	printf '#!/bin/sh\n/usr/share/cain-ncnn-vulkan/cain-ncnn-vulkan "$@"' >$_name
-	install -Dm755 $_name "${pkgdir}/usr/bin/$_name"
 }
diff --git a/cain-ncnn-vulkan b/cain-ncnn-vulkan
new file mode 100755
index 000000000000..f2e662ad4811
--- /dev/null
+++ b/cain-ncnn-vulkan
@@ -0,0 +1,29 @@
+#!/bin/sh
+
+default_model_path="/usr/share/cain-ncnn-vulkan/"
+readonly default_model_path
+
+args=""
+
+while getopts "hv0:1:i:o:m:g:j:f:" arg; do
+    case $arg in
+    h) args="$args -$arg" ;;
+    v) args="$args -$arg" ;;
+    1) args="$args -$arg $OPTARG" ;;
+    0) args="$args -$arg $OPTARG" ;;
+    i) args="$args -$arg $OPTARG" ;;
+    o) args="$args -$arg $OPTARG" ;;
+    m)
+        if ! [ -d $OPTARG ]; then
+            args="$args -m $default_model_path$OPTARG"
+        else
+            args="$args -$arg $OPTARG"
+        fi
+        ;;
+    g) args="$args -$arg $OPTARG" ;;
+    j) args="$args -$arg $OPTARG" ;;
+    f) args="$args -$arg $OPTARG" ;;
+    esac
+done
+
+/usr/share/cain-ncnn-vulkan/cain-ncnn-vulkan $args