# syntax=docker/dockerfile:1

#-- Build configuration --#
# ARCH_CFLAGS is supplied as a compile option
ARG ARCH_CFLAGS="-O2 -mthumb -mfpu=neon -mfloat-abi=hard -mcpu=cortex-a9 -fomit-frame-pointer"
# AXIS_ARCH is the AXIS platform descriptor
ARG AXIS_ARCH=armv7hf
# BUILD_ROOT defines where in the build containers the building takes place
ARG BUILD_ROOT=/build-root
# DOCKERHUB_ARCH is the DockerHub platform descriptor
ARG DOCKERHUB_ARCH=arm32v7
# OPENCV_MODULES defines what OpenCV modules to build
ARG OPENCV_MODULES=core,imgproc,imgcodecs,videoio,objdetect,python3,video
# SDK_ROOT_DIR defines the root directory of the final SDK images
ARG SDK_ROOT_DIR=/axis/
# UBUNTU_ARCH is the Ubuntu platform descriptor
ARG UBUNTU_ARCH=armhf
# TARGET_TOOLCHAIN is the name of the compilation toolchain for the target platform
ARG TARGET_TOOLCHAIN=arm-linux-gnueabihf
#CMAKE_TOOLCHAIN_FOLDER is the name of the CMAKE toolchain folder
ARG CMAKE_TOOLCHAIN_FOLDER=arm-gnueabi
# TARGET_ROOT defines where in the build containers the resulting application is put
ARG TARGET_ROOT=/target-root
# UBUNTU_VERSION defines the ubuntu version of the build and SDK containers
ARG UBUNTU_VERSION=22.04
# UBUNTU_CODENAME should be the ubuntu codename of the UBUNTU_VERSION used, e.g., focal, hirsute, ..
ARG UBUNTU_CODENAME=jammy

#-- Versions of installed packages defined as repository tags --#
ARG NUMPY_VERSION=v1.17.3
ARG OPENBLAS_VERSION=v0.3.14
ARG OPENCV_VERSION=4.5.1
ARG PYTHON_VERSION=3.8.8
ARG PYTESSERACT_VERSION=0.3.7
ARG SCIPY_VERSION=v1.7.1
ARG TESSERACT_VERSION=4.1.1
ARG TFSERVING_VERSION=2.9.0

#-- Build parallelization  --#
ARG OPENBLAS_BUILD_CORES=16
ARG OPENCV_BUILD_CORES=16
ARG PYTHON_BUILD_CORES=16
ARG NUMPY_BUILD_CORES=16
ARG SCIPY_BUILD_CORES=16
ARG TESSERACT_BUILD_CORES=16
ARG TENSORFLOW_BUILD_CORES=16
#-- ACAP SDK configuration --#
ARG REPO=axisecp
ARG ACAP_SDK_IMAGE=acap-native-sdk
ARG ACAP_SDK_UBUNTU_VERSION=22.04
ARG ACAP_SDK_VERSION=1.6_rc1
ARG ACAP_SDK_TAG=${ACAP_SDK_VERSION}-${AXIS_ARCH}-ubuntu${ACAP_SDK_UBUNTU_VERSION}

# The ACAP Native SDK is used to get certain libs and headers
FROM ${REPO}/${ACAP_SDK_IMAGE}:${ACAP_SDK_TAG} AS acap-native-sdk
# libs: /opt/axis/acapsdk/sysroots/$AXIS_ARCH/usr/lib/
# headers: /opt/axis/acapsdk/sysroots/$AXIS_ARCH/usr/include/

# Create a emulated base image with build tools, env vars, etc.,
FROM $DOCKERHUB_ARCH/ubuntu:${UBUNTU_VERSION} as build-base-arm
ARG BUILD_ROOT
ENV DEBIAN_FRONTEND=noninteractive
ARG PYTHON_VERSION
ARG TARGET_TOOLCHAIN
ARG TARGET_ROOT
ARG UBUNTU_ARCH
ARG http_proxy
ARG https_proxy

# To support DOCKER_BUILDKIT=0, base ARGs are converted to ENVs to allow propagation
ENV BUILD_ROOT=$BUILD_ROOT
ENV TARGET_TOOLCHAIN=$TARGET_TOOLCHAIN
ENV TARGET_ROOT=$TARGET_ROOT
ENV UBUNTU_ARCH=$UBUNTU_ARCH
ENV http_proxy=$http_proxy
ENV https_proxy=$https_proxy

# qemu is used to emulate arm
COPY --from=multiarch/qemu-user-static:x86_64-arm-6.1.0-8 /usr/bin/qemu-arm-static /usr/bin/

RUN <<EOF
apt-get update
apt-get install -y --no-install-recommends \
    build-essential \
    ca-certificates \
    curl \
    gfortran \
    git \
    pkg-config
update-ca-certificates
EOF

RUN mkdir -p ${TARGET_ROOT}
RUN mkdir -p ${BUILD_ROOT}



# Crosscompile OpenCV with AXIS VDO bindings
FROM build-base-arm AS build-opencv
ARG ARCH_CFLAGS
ARG AXIS_ARCH
ARG OPENCV_BUILD_CORES
ARG OPENCV_MODULES
ARG OPENCV_VERSION
ARG CMAKE_TOOLCHAIN_FOLDER

# Get required packages for both build env and target env
RUN mkdir -p $BUILD_ROOT/opencv_deps
RUN apt-get install --reinstall --download-only -o=dir::cache=$BUILD_ROOT/opencv_deps -y -f libglib2.0-dev:$UBUNTU_ARCH libsystemd0:$UBUNTU_ARCH libgfortran5:$UBUNTU_ARCH
RUN for f in $BUILD_ROOT/opencv_deps/archives/*.deb; do dpkg -x $f $TARGET_ROOT; done
RUN for f in $BUILD_ROOT/opencv_deps/archives/*.deb; do dpkg -x $f $TARGET_ROOT/build; done
RUN apt-get install -y libglib2.0-dev libsystemd0 libglib2.0-dev:$UBUNTU_ARCH libsystemd0:$UBUNTU_ARCH
#RUN cp -r /usr/lib/x86_64-linux-gnu/glib-2.0 /usr/lib/
RUN apt-get install python3 python3-pip -y
# Get AXIS specific files for the build environment
RUN mkdir -p /usr/lib/$TARGET_TOOLCHAIN/pkgconfig
ENV PKG_CONFIG_LIBDIR=/usr/share/pkgconfig:/usr/lib/$TARGET_TOOLCHAIN/pkgconfig
COPY opencv/capaxis/$TARGET_TOOLCHAIN/include/ /usr/include/vdo
COPY opencv/capaxis/$TARGET_TOOLCHAIN/lib/ /usr/lib/$TARGET_TOOLCHAIN
COPY opencv/capaxisvdo.pc /usr/lib/$TARGET_TOOLCHAIN/pkgconfig/
COPY --from=acap-native-sdk /opt/axis/acapsdk/sysroots/$AXIS_ARCH/usr/include/vdo/* /usr/include/vdo/


# Get AXIS specific files for the runtime
# The stubs added below should perhaps be separated from lib/ to avoid confusion vs full lib counterparts
COPY opencv/capaxis/$TARGET_TOOLCHAIN/stubs/ $TARGET_ROOT/build/usr/lib/
COPY opencv/capaxis/$TARGET_TOOLCHAIN/lib/ $TARGET_ROOT/build/usr/lib/
COPY --from=acap-native-sdk /opt/axis/acapsdk/sysroots/$AXIS_ARCH/usr/lib/libfido* $TARGET_ROOT/build/usr/lib/
COPY --from=acap-native-sdk /opt/axis/acapsdk/sysroots/$AXIS_ARCH/usr/lib/libvdo* $TARGET_ROOT/build/usr/lib/
COPY --from=acap-native-sdk /opt/axis/acapsdk/sysroots/$AXIS_ARCH/usr/lib/libfido* /usr/lib/$TARGET_TOOLCHAIN/
COPY --from=acap-native-sdk /opt/axis/acapsdk/sysroots/$AXIS_ARCH/usr/lib/libvdo* /usr/lib/$TARGET_TOOLCHAIN/

# Setup build directory
COPY opencv/capaxis/$TARGET_TOOLCHAIN/stubs/ $TARGET_ROOT/usr/lib/
COPY opencv/capaxis/$TARGET_TOOLCHAIN/lib/ $TARGET_ROOT/usr/lib/
COPY --from=acap-native-sdk /opt/axis/acapsdk/sysroots/$AXIS_ARCH/usr/lib/libfido* $TARGET_ROOT/usr/lib/
COPY --from=acap-native-sdk /opt/axis/acapsdk/sysroots/$AXIS_ARCH/usr/lib/libvdo* $TARGET_ROOT/usr/lib/

# https://github.com/opencv/opencv-python#manual-builds
RUN git clone --recursive https://github.com/opencv/opencv-python.git
WORKDIR /opencv-python/opencv
RUN git checkout $OPENCV_VERSION
COPY opencv/cap_axis_vdo.cpp modules/videoio/src
COPY opencv/detect_axis_vdo.cmake modules/videoio/cmake
COPY opencv/axisvdo-4.5.1.patch .
RUN patch -p 1 < axisvdo-${OPENCV_VERSION}.patch
WORKDIR /opencv-python
ENV CMAKE_ARGS='-D CMAKE_TOOLCHAIN_FILE="opencv/platforms/linux/${CMAKE_TOOLCHAIN_FOLDER}.toolchain.cmake" \
    -D CMAKE_BUILD_TYPE=RELEASE \
    # This path will contain all opencv header files
    -D CMAKE_INSTALL_PREFIX=$TARGET_ROOT/usr \
    -D INSTALL_C_EXAMPLES=OFF \
    -D CMAKE_C_FLAGS="$ARCH_CFLAGS --sysroot=$TARGET_ROOT/build" \
    -D CMAKE_CXX_FLAGS="$ARCH_CFLAGS --sysroot=$TARGET_ROOT/build" \
    -D OpenBLAS_LIB=$TARGET_ROOT/build/usr/lib/libopenblas.so \
    -D OpenBLAS_INCLUDE_DIR=$TARGET_ROOT/build/usr/include \
    -D BLAS_LIBRARIES=$TARGET_ROOT/build/usr/lib/libopenblas.so \
    # ON:
    -D WITH_TBB=ON \
    -D WITH_OPENGL=ON \
    -D BUILD_JPEG=ON \
    -D ENABLE_NEON=ON \
    -D CPU_BASELINE=NEON,VFPV3 \
    -D ENABLE_VFPV3=ON \
    -D WITH_AXIS_VDO=ON \
    -D CMAKE_SKIP_RPATH=ON \
    -D OPENCV_SKIP_PYTHON_LOADER=ON \
    -D OPENCV_GENERATE_PKGCONFIG=ON \
    # ON end.
    -D WITH_OPENEXR=OFF \
    -D WITH_IPP=OFF \
    -D WITH_WEBP=OFF \
    -D ENABLE_PRECOMPILED_HEADERS=OFF \
    -D WITH_1394=OFF \
    -D WITH_LIBV4L=OFF \
    -D WITH_V4L=OFF \
    -D WITH_QT=OFF \
    -D WITH_GTK=OFF \
    -D WITH_FFMPEG=OFF \
    -D WITH_GSTREAMER=OFF \
    -D WITH_GSTREAMER_0_10=OFF \
    -D BUILD_LIST=$OPENCV_MODULES \
    -D BUILD_EXAMPLES=OFF \
    -D BUILD_OPENCV_DNN=OFF \
    -D BUILD_OPENCV_APPS=OFF \
    -D BUILD_DOCS=OFF \
    -D BUILD_opencv_python=OFF \
    -D BUILD_PNG=OFF \
    -D BUILD_IPP_IWD=OFF \
    -D BUILD_QUIRC=OFF \
    -D BUILD_OPENEXR=OFF \
    -D WITH_JASPER=OFF \
    -D BUILD_TIFF=OFF \
    -D BUILD_PERF_TESTS=OFF \
    -D HAVE_IPP_ICV=OFF \
    -D BUILD_IPP=OFF \
    -D BUILD_ZLIB=OFF \
    -D BUILD_TESTS=OFF \
    -D BUILD_IPP_ICV=OFF \
    -D BUILD_PROTOBUF=OFF \
    -D WITH_CUDA=OFF \
    -D HAVE_WEBP=OFF \'
RUN pip wheel . --verbose