#!/bin/bash

set -e

echo "=== RK1808 NPU Docker Build Script ==="

# 清理之前的构建
echo "=== Cleaning previous build ==="
rm -rf lib
mkdir -p lib/usr/lib
mkdir -p lib/usr/bin

# 只复制 Rockchip 特定的 NPU 库（系统没有的）
echo "=== Copying NPU-specific libraries ==="
cp /usr/lib/librknn_api.so lib/usr/lib/
cp /usr/lib/librknn_runtime.so lib/usr/lib/
cp /usr/lib/librockx.so lib/usr/lib/
cp /usr/lib/libVSC.so lib/usr/lib/
cp /usr/lib/libGAL.so lib/usr/lib/
cp /usr/lib/libArchModelSw.so lib/usr/lib/
cp /usr/lib/libNN*.so lib/usr/lib/ 2>/dev/null || true
cp /usr/lib/libOpenCL.so* lib/usr/lib/ 2>/dev/null || true
cp /usr/lib/libOpenVX.so* lib/usr/lib/ 2>/dev/null || true
cp /usr/lib/librockchip_mpp.so* lib/usr/lib/ 2>/dev/null || true
cp /usr/lib/librga.so* lib/usr/lib/ 2>/dev/null || true

# 复制 ffmpeg（如果宿主机有特殊版本）
cp /usr/bin/ffmpeg lib/usr/bin/

echo "=== Copied files ==="
ls -la lib/usr/lib/

# 构建 Docker 镜像
echo "=== Building Docker image ==="
docker build -t fire_leave .

echo "=== Build successful! ==="