init

4 files changed, 86 insertions, 0 deletions

diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..2abd6285bd3e
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,19 @@
+pkgbase = srmd-ncnn-vulkan-bin
+	pkgdesc = SRMD super resolution implemented with ncnn library
+	pkgver = 20210210
+	pkgrel = 1
+	url = https://github.com/nihui/srmd-ncnn-vulkan
+	arch = x86_64
+	license = MIT
+	depends = vulkan-icd-loader
+	depends = vulkan-headers
+	provides = srmd-ncnn-vulkan
+	conflicts = srmd-ncnn-vulkan
+	conflicts = srmd-ncnn-vulkan-git
+	source = https://github.com/nihui/srmd-ncnn-vulkan/releases/download/20210210/srmd-ncnn-vulkan-20210210-ubuntu.zip
+	source = srmd-ncnn-vulkan
+	sha256sums = ea4690e128e278517d8b288c8b3597d4f2df7bd862c97b2f7575719979269d1c
+	sha256sums = 68aab0058035d57bc285a5af2713ea48ebf11667434780fc82e1b0df9da25009
+
+pkgname = srmd-ncnn-vulkan-bin
+
diff --git a/.gitignore b/.gitignore
new file mode 100644
index 000000000000..9da8cc83f71e
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,5 @@
+*
+!PKGBUILD
+!.SRCINFO
+!.gitignore
+!srmd-ncnn-vulkan
diff --git a/PKGBUILD b/PKGBUILD
new file mode 100644
index 000000000000..1d27b8e3e4f2
--- /dev/null
+++ b/PKGBUILD
@@ -0,0 +1,31 @@
+# Maintainer: PumpkinCheshire <sollyonzou@gmail.com>
+
+pkgname=srmd-ncnn-vulkan-bin
+pkgver=20210210
+pkgrel=1
+pkgdesc="SRMD super resolution implemented with ncnn library"
+arch=('x86_64')
+url="https://github.com/nihui/srmd-ncnn-vulkan"
+license=('MIT')
+depends=("vulkan-icd-loader" "vulkan-headers")
+makedepends=()
+provides=("srmd-ncnn-vulkan")
+conflicts=("srmd-ncnn-vulkan" "srmd-ncnn-vulkan-git")
+source=("https://github.com/nihui/${pkgname%-bin}/releases/download/$pkgver/${pkgname%-bin}-$pkgver-ubuntu.zip"
+	"srmd-ncnn-vulkan")
+sha256sums=('ea4690e128e278517d8b288c8b3597d4f2df7bd862c97b2f7575719979269d1c'
+	'68aab0058035d57bc285a5af2713ea48ebf11667434780fc82e1b0df9da25009')
+
+package() {
+	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}/${pkgname%-bin}-$pkgver-ubuntu/${pkgname%-bin}" "${pkgdir}/usr/share/${pkgname%-bin}/${pkgname%-bin}"
+
+	cd "${srcdir}/${pkgname%-bin}-$pkgver-ubuntu/"
+	for f in models-*/*; do
+		install -Dm 644 "$f" ${pkgdir}/usr/share/${pkgname%-bin}/"$f"
+	done
+}
diff --git a/srmd-ncnn-vulkan b/srmd-ncnn-vulkan
new file mode 100755
index 000000000000..cc53af3b8a81
--- /dev/null
+++ b/srmd-ncnn-vulkan
@@ -0,0 +1,31 @@
+#!/bin/sh
+
+default_model_path="/usr/share/srmd-ncnn-vulkan/"
+readonly default_model_path
+
+args=""
+
+while getopts "hvi:o:n:s:t:m:g:j:xf:" arg; do
+    case $arg in
+    h) args="$args -$arg" ;;
+    v) args="$args -$arg" ;;
+    i) args="$args -$arg $OPTARG" ;;
+    o) args="$args -$arg $OPTARG" ;;
+    n) args="$args -$arg $OPTARG" ;;
+    s) args="$args -$arg $OPTARG" ;;
+    t) 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" ;;
+    x) args="$args -$arg" ;;
+    f) args="$args -$arg $OPTARG" ;;
+    esac
+done
+
+/usr/share/srmd-ncnn-vulkan/srmd-ncnn-vulkan $args