您最多选择25个主题
主题必须以中文或者字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
50 行
1.8 KiB
50 行
1.8 KiB
# variables
|
|
|
|
# GRPC-TOOLS required. Install with `nuget install Grpc.Tools`.
|
|
# Then un-comment and replace [DIRECTORY] with location of files.
|
|
# For example, on macOS, you might have something like:
|
|
# COMPILER=Grpc.Tools.1.14.1/tools/macosx_x64
|
|
# COMPILER=[DIRECTORY]
|
|
|
|
SRC_DIR=proto/mlagents/envs/communicator_objects
|
|
DST_DIR_C=../UnitySDK/Assets/ML-Agents/Scripts/Grpc/CommunicatorObjects
|
|
DST_DIR_P=../ml-agents-envs
|
|
PROTO_PATH=proto
|
|
PYTHON_PACKAGE=mlagents/envs/communicator_objects
|
|
|
|
# clean
|
|
rm -rf $DST_DIR_C
|
|
rm -rf $DST_DIR_P/$PYTHON_PACKAGE
|
|
mkdir -p $DST_DIR_C
|
|
mkdir -p $DST_DIR_P/$PYTHON_PACKAGE
|
|
|
|
# generate proto objects in python and C#
|
|
|
|
$COMPILER/protoc --proto_path=proto --csharp_out=$DST_DIR_C $SRC_DIR/*.proto
|
|
$COMPILER/protoc --proto_path=proto --python_out=$DST_DIR_P --mypy_out=$DST_DIR_P $SRC_DIR/*.proto
|
|
|
|
# grpc
|
|
|
|
GRPC=unity_to_external.proto
|
|
|
|
$COMPILER/protoc --proto_path=proto --csharp_out $DST_DIR_C --grpc_out $DST_DIR_C $SRC_DIR/$GRPC --plugin=protoc-gen-grpc=$COMPILER/grpc_csharp_plugin
|
|
python3 -m grpc_tools.protoc --proto_path=proto --python_out=$DST_DIR_P --grpc_python_out=$DST_DIR_P $SRC_DIR/$GRPC
|
|
|
|
|
|
# Generate the init file for the python module
|
|
# rm -f $DST_DIR_P/$PYTHON_PACKAGE/__init__.py
|
|
for FILE in $DST_DIR_P/$PYTHON_PACKAGE/*.py
|
|
do
|
|
FILE=${FILE##*/}
|
|
# echo from .$(basename $FILE) import \* >> $DST_DIR_P/$PYTHON_PACKAGE/__init__.py
|
|
echo from .${FILE%.py} import \* >> $DST_DIR_P/$PYTHON_PACKAGE/__init__.py
|
|
done
|
|
|
|
# Surround UnityToExternal.cs file with macro
|
|
echo "#if UNITY_EDITOR || UNITY_STANDALONE_WIN || UNITY_STANDALONE_OSX || UNITY_STANDALONE_LINUX
|
|
`cat $DST_DIR_C/UnityToExternalGrpc.cs`
|
|
#endif" > $DST_DIR_C/UnityToExternalGrpc.cs
|
|
|
|
# Remove the __init__.py file since it is not needed
|
|
rm $DST_DIR_P/$PYTHON_PACKAGE/__init__.py
|
|
touch $DST_DIR_P/$PYTHON_PACKAGE/__init__.py
|