surmeh01 | bceff2f | 2018-03-29 16:29:27 +0100 | [diff] [blame] | 1 | #!/bin/sh |
| 2 | # |
| 3 | # Copyright © 2017 Arm Ltd. All rights reserved. |
David Beck | ecb56cd | 2018-09-05 12:52:57 +0100 | [diff] [blame] | 4 | # SPDX-License-Identifier: MIT |
surmeh01 | bceff2f | 2018-03-29 16:29:27 +0100 | [diff] [blame] | 5 | # |
| 6 | |
| 7 | THIS_SCRIPT=$0 |
| 8 | OUTPUT_DIR=$1 |
| 9 | PROTOBUF_INSTALL_DIR=$2 |
| 10 | |
| 11 | usage() |
| 12 | { |
| 13 | echo |
| 14 | echo "Usage: ${THIS_SCRIPT} <OUTPUT_DIR> [PROTOBUF_INSTALL_DIR]" |
| 15 | echo |
| 16 | echo " <OUTPUT_DIR> is the location where the generated files will be placed" |
| 17 | echo " [PROTOBUF_INSTALL_DIR] the location of the protobuf installation" |
| 18 | echo |
| 19 | } |
| 20 | |
| 21 | if [ "x$OUTPUT_DIR" = "x" ] |
| 22 | then |
| 23 | usage |
| 24 | exit 1 |
| 25 | fi |
| 26 | |
| 27 | mkdir -p ${OUTPUT_DIR} |
| 28 | ERR=$? |
| 29 | if [ $ERR -ne 0 ] |
| 30 | then |
| 31 | echo |
| 32 | echo "Cannot create output dir: ${OUTPUT_DIR}" |
| 33 | echo "mkdir returned: $ERR" |
| 34 | echo |
| 35 | usage |
| 36 | exit 1 |
| 37 | fi |
| 38 | |
| 39 | |
| 40 | if [ "x${PROTOBUF_INSTALL_DIR}" = "x" ] |
| 41 | then |
| 42 | PROTOBUF_INSTALL_DIR=/usr/local |
| 43 | fi |
| 44 | |
| 45 | if [ ! -x "${PROTOBUF_INSTALL_DIR}/bin/protoc" ] |
| 46 | then |
| 47 | echo |
| 48 | echo "No usable protocol buffer (protoc) compiler found in ${PROTOBUF_INSTALL_DIR}/bin/" |
| 49 | echo "You can specify the location of the protobuf installation as the second" |
| 50 | echo "argument of ${THIS_SCRIPT}." |
| 51 | usage |
| 52 | exit 1 |
| 53 | fi |
| 54 | |
| 55 | |
| 56 | TF_PROTO_FILES=tensorflow/contrib/makefile/tf_proto_files.txt |
| 57 | if [ -r $TF_PROTO_FILES ] |
| 58 | then |
| 59 | OLD_LD_LIBRARY_PATH=$LD_LIBRARY_PATH |
| 60 | for i in `cat $TF_PROTO_FILES` |
| 61 | do |
| 62 | LD_LIBRARY_PATH=$OLD_LD_LIBRARY_PATH:${PROTOBUF_INSTALL_DIR}/lib \ |
| 63 | $PROTOBUF_INSTALL_DIR/bin/protoc $i \ |
| 64 | --proto_path=. \ |
| 65 | --proto_path=${PROTOBUF_INSTALL_DIR}/include \ |
| 66 | --cpp_out $OUTPUT_DIR |
| 67 | done |
| 68 | else |
| 69 | echo "Couldn't find $TF_PROTO_FILES. This script should be run from the" |
| 70 | echo "tensorflow source directory." |
| 71 | exit 1 |
| 72 | fi |
| 73 | |