# https://hub.docker.com/r/cwaffles/openpose
FROM ubuntu:20.04

#get deps
RUN apt-get update
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends python3-dev python3-pip python3-setuptools 
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends git wget 
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends g++ 
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends make 
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends libprotobuf-dev 
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends protobuf-compiler 
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends libopencv-dev 
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends libgoogle-glog-dev 
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends libboost-all-dev 
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends libcaffe-cuda-dev 
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends libhdf5-dev 
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends libatlas-base-dev

#for python api
RUN pip3 install --upgrade pip
RUN pip3 install numpy opencv-python 

#replace cmake as old version has CUDA variable bugs
RUN wget https://github.com/Kitware/CMake/releases/download/v3.16.0/cmake-3.16.0-Linux-x86_64.tar.gz && \
tar xzf cmake-3.16.0-Linux-x86_64.tar.gz -C /opt && \
rm cmake-3.16.0-Linux-x86_64.tar.gz
ENV PATH="/opt/cmake-3.16.0-Linux-x86_64/bin:${PATH}"

#get openpose
WORKDIR /openpose
RUN git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose.git .

#build it
WORKDIR /openpose/build
RUN cmake -DBUILD_PYTHON=ON .. && make -j `nproc`
WORKDIR /openpose