From 145452b0aec87ee4ad918f32a5ebcc2c3a3ef3ad Mon Sep 17 00:00:00 2001
From: Mathieu Broillet <mathieu@broillet.ch>
Date: Mon, 5 Aug 2024 11:32:18 +0200
Subject: [PATCH] revert shared libraries stripping, messes with tensorflow and
 use tensorflow v1 instead of v2 on cpu (smaller size)

---
 build_and_setup_ultimatealvr.sh | 29 +++++++++++++++++++----------
 1 file changed, 19 insertions(+), 10 deletions(-)

diff --git a/build_and_setup_ultimatealvr.sh b/build_and_setup_ultimatealvr.sh
index 5d4a3b6..64534e5 100755
--- a/build_and_setup_ultimatealvr.sh
+++ b/build_and_setup_ultimatealvr.sh
@@ -1,5 +1,7 @@
 #!/bin/bash
 
+deactivate 2>/dev/null
+
 # Function to create virtual environment, install the wheel, and copy assets and libs
 install_and_setup() {
     echo "Creating virtual environment at the root..."
@@ -55,7 +57,7 @@ prompt_auto_setup() {
     esac
 }
 
-# Directories
+# Variables
 ROOT_DIR=$(pwd)
 BUILD_DIR="$ROOT_DIR/tmp-build-env"
 SDK_ZIP_URL="https://github.com/DoubangoTelecom/ultimateALPR-SDK/archive/8130c76140fe8edc60fe20f875796121a8d22fed.zip"
@@ -69,7 +71,12 @@ cd "$BUILD_DIR" || exit
 
 # Clone SDK
 echo "Downloading SDK..."
-wget "$SDK_ZIP_URL" -O "$SDK_ZIP" >/dev/null 2>&1
+if [ -f "$SDK_ZIP" ]; then
+    echo "SDK zip already exists."
+    rm -R "$SDK_DIR"
+else
+    wget "$SDK_ZIP_URL" -O "$SDK_ZIP" >/dev/null 2>&1
+fi
 if [ $? -ne 0 ]; then
     echo "Failed to download SDK."
     exit 1
@@ -121,26 +128,30 @@ read -r -p "Do you want TensorFlow for CPU or GPU? (cpu/gpu): " tf_choice
 mkdir -p "$BIN_DIR/tensorflow"
 if [ "$tf_choice" == "gpu" ]; then
     echo "Downloading TensorFlow GPU..."
-    wget https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-2.6.0.tar.gz >/dev/null 2>&1
+    wget https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-2.6.0.tar.gz >/dev/null 2>&1 # Use 2.6 for newer GPU support
     if [ $? -ne 0 ]; then
         echo "Failed to download TensorFlow GPU."
         exit 1
     fi
     echo "Extracting TensorFlow GPU..."
     tar -xf libtensorflow-gpu-linux-x86_64-2.6.0.tar.gz -C "$BIN_DIR/tensorflow" >/dev/null 2>&1
+
+    mv "$BIN_DIR/tensorflow/lib/libtensorflow.so.1" "$BIN_DIR/libs/libtensorflow.so.1"
+    mv "$BIN_DIR/tensorflow/lib/libtensorflow_framework.so.2.6.0" "$BIN_DIR/libs/libtensorflow_framework.so.2"
+
 else
     echo "Downloading TensorFlow CPU..."
-    wget https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.6.0.tar.gz >/dev/null 2>&1
+    #wget https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.6.0.tar.gz >/dev/null 2>&1
+    wget https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-1.14.0.tar.gz >/dev/null 2>&1 # Use 1.14 as it's smaller in size
     if [ $? -ne 0 ]; then
         echo "Failed to download TensorFlow CPU."
         exit 1
     fi
     echo "Extracting TensorFlow CPU..."
-    tar -xf libtensorflow-cpu-linux-x86_64-2.6.0.tar.gz -C "$BIN_DIR/tensorflow" >/dev/null 2>&1
-fi
+    tar -xf libtensorflow-cpu-linux-x86_64-1.14.0.tar.gz -C "$BIN_DIR/tensorflow" >/dev/null 2>&1
 
-mv "$BIN_DIR/tensorflow/lib/libtensorflow.so.1" "$BIN_DIR/libs/libtensorflow.so.1"
-mv "$BIN_DIR/tensorflow/lib/libtensorflow_framework.so.2.6.0" "$BIN_DIR/libs/libtensorflow_framework.so.2"
+    mv "$BIN_DIR/tensorflow/lib/"* "$BIN_DIR/libs/"
+fi
 
 # Build the wheel
 echo "Building the wheel..."
@@ -155,8 +166,6 @@ mv "$BIN_DIR/dist/"*.whl "$BUILD_DIR"
 mv "$BIN_DIR/libs" "$BUILD_DIR"
 mv "$BIN_DIR/plugins.xml" "$BUILD_DIR/libs"
 
-strip "$BUILD_DIR/libs"/*.so*
-
 # Move the assets to the root directory
 mv "$SDK_DIR/assets" "$BUILD_DIR/assets"