[caffe-contrib] 260/362: add bash completion script for caffe
Zhou Mo
cdluminate-guest at moszumanska.debian.org
Tue May 3 09:24:40 UTC 2016
This is an automated email from the git hooks/post-receive script.
cdluminate-guest pushed a commit to branch master
in repository caffe-contrib.
commit bf2dc076cd20eac6b33401a7c65b4e4b52254739
Author: Zhou Mo <cdluminate at gmail.com>
Date: Wed Mar 9 04:34:14 2016 +0000
add bash completion script for caffe
---
debian/caffe-cpu.install | 3 +++
debian/caffe-cuda.install | 3 +++
debian/script/caffe | 68 +++++++++++++++++++++++++++++++++++++++++++++++
3 files changed, 74 insertions(+)
diff --git a/debian/caffe-cpu.install b/debian/caffe-cpu.install
index 74116ca..6079625 100644
--- a/debian/caffe-cpu.install
+++ b/debian/caffe-cpu.install
@@ -8,3 +8,6 @@ debian/tmp/caffe_cpu/bin/convert_mnist_siamese_data usr/bin
debian/tmp/caffe_cpu/bin/upgrade_net_proto_text usr/bin
debian/tmp/caffe_cpu/bin/convert_cifar_data usr/bin
debian/tmp/caffe_cpu/bin/upgrade_net_proto_binary usr/bin
+
+# Package Maintainer's bash-completion script
+debian/script/caffe usr/share/bash-completion/completions/
diff --git a/debian/caffe-cuda.install b/debian/caffe-cuda.install
index ddc188a..0a99f34 100644
--- a/debian/caffe-cuda.install
+++ b/debian/caffe-cuda.install
@@ -8,3 +8,6 @@ debian/tmp/caffe_cuda/bin/convert_mnist_siamese_data usr/bin
debian/tmp/caffe_cuda/bin/upgrade_net_proto_text usr/bin
debian/tmp/caffe_cuda/bin/convert_cifar_data usr/bin
debian/tmp/caffe_cuda/bin/upgrade_net_proto_binary usr/bin
+
+# Package Maintainer's bash-completion script
+debian/script/caffe usr/share/bash-completion/completions/
diff --git a/debian/script/caffe b/debian/script/caffe
new file mode 100644
index 0000000..d766607
--- /dev/null
+++ b/debian/script/caffe
@@ -0,0 +1,68 @@
+# bash completion for Caffe's command line utility -*- shell-script -*-
+# COPYRIGHT (C) 2015,2016 Zhou Mo <cdluminate at gmail.com>
+# License: BSD-2-Clause
+# Originally appeard at https://github.com/BVLC/caffe/issues/3149
+
+_caffe()
+{
+ local cur prev words cword
+ _init_completion -s || return
+
+ local prototxts='@(prototxt)'
+ local caffemodels='@(caffemodel)'
+ local solverstates='@(solverstate)'
+ local caffefiles='@(prototxt|caffemodel|solverstate)'
+
+ local flags='-gpu -iterations -model -snapshot -solver -weights -sighup_effect -sigint_effect'
+
+ if [[ $cword -eq 1 ]]; then
+ COMPREPLY=( $( compgen -W 'train test time device_query' -- "$cur" ) )
+ return 0
+ fi
+
+ if [[ $cword -eq 2 ]]; then
+ case ${words[1]} in
+ train|test|device_query|time)
+ COMPREPLY=( $( compgen -W "$flags" -- "$cur") )
+ return 0
+ ;;
+ *)
+ return 0
+ ;;
+ esac
+ fi
+
+ case $prev in
+ -gpu|-iterations|-version)
+ return 0
+ ;;
+ -solver|-model)
+ _filedir $prototxts
+ return 0
+ ;;
+ -weights)
+ _filedir $caffemodels
+ return 0
+ ;;
+ -snapshot)
+ _filedir $solverstates
+ return 0
+ ;;
+ -sighup_effect|-sigint_effect)
+ COMPREPLY=( $( compgen -W 'snapshot stop none' -- "$cur") )
+ return 0
+ ;;
+ *)
+ COMPREPLY=( $( compgen -W "$flags" -- "$cur") )
+ return 0
+ ;;
+ esac
+
+ # file completion on relevant files
+ _filedir "$caffefiles"
+
+ return 0
+}
+complete -F _caffe caffe
+
+# vim
--
Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-science/packages/caffe-contrib.git
More information about the debian-science-commits
mailing list