#!/bin/bash

#function:
#   script used to generate caffe_pb2.py from caffe.proto using protoc
#

PROTOC=`which protoc`
if [[ -z $PROTOC ]];then
    echo "not found protoc, you should first install it following this[https://github.com/google/protobuf/releases]"
    exit 1
fi

WORK_ROOT=$1
PY_NAME="$WORK_ROOT/caffe_pb2.py"
$PROTOC --proto_path=$WORK_ROOT --python_out=$WORK_ROOT $WORK_ROOT/caffe.proto
ret=$?

if [ -e "$PY_NAME" ];then
    echo "succeed to generate [$PY_NAME]"
    exit 0
else
    echo "failed to generate [$PY_NAME]"
fi
exit $ret