From ba1962ce7f5f4e58eb6f0a8b42249ebe042fe265 Mon Sep 17 00:00:00 2001
From: Christian Langenbacher <clangenb+gh@protonmail.ch>
Date: Tue, 5 Oct 2021 12:03:56 +0200
Subject: [PATCH] parametrize benchmark-script

---
 scripts/benchmark_all_pallets.sh | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/scripts/benchmark_all_pallets.sh b/scripts/benchmark_all_pallets.sh
index 61c73f2..6b32d95 100755
--- a/scripts/benchmark_all_pallets.sh
+++ b/scripts/benchmark_all_pallets.sh
@@ -2,10 +2,16 @@
 
 # Create `WeightInfo` implementations for all the pallets and store it in the weight module of the `runtime`.
 
-RUNTIME_WEIGHT_DIR=runtime/src/weights
-NODE=./target/release/integritee-node
+NODE=${1:-target/release/integritee-node}
+CHAIN_SPEC=${2:-integritee-solo-fresh}
+WEIGHT_OUTPUT_DIR=${3:-runtime/src/weights}
 
-mkdir -p $RUNTIME_WEIGHT_DIR
+echo "Running benchmarks for all pallets:"
+echo "NODE:               ${NODE}"
+echo "CHAIN_SPEC:         ${CHAIN_SPEC}"
+echo "WEIGHT_OUTPUT_DIR:  ${WEIGHT_OUTPUT_DIR}"
+
+mkdir -p "$WEIGHT_OUTPUT_DIR"
 
 pallets=(
   "frame_system" \
@@ -24,7 +30,7 @@ for pallet in ${pallets[*]}; do
 
   $NODE \
   benchmark \
-  --chain=integritee-solo-fresh \
+  --chain="$CHAIN_SPEC" \
   --steps=50 \
   --repeat=20 \
   --pallet="$pallet" \
@@ -32,6 +38,6 @@ for pallet in ${pallets[*]}; do
   --execution=wasm \
   --wasm-execution=compiled \
   --heap-pages=4096 \
-  --output=./$RUNTIME_WEIGHT_DIR/"$pallet".rs \
+  --output="$WEIGHT_OUTPUT_DIR"/"$pallet".rs \
 
 done