[caffe-contrib] 260/362: add bash completion script for caffe

[Zhou Mo]

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