From c8c7b4585f01090c3e688533e1bbd82e712cb2f3 Mon Sep 17 00:00:00 2001
From: Hideo TERADA <h.tee2006@gmail.com>
Date: Tue, 11 Jun 2024 11:00:58 +0900
Subject: [PATCH] Dev:

---
 Dockerfile   | 24 ++++++++++++++++++++++--
 compose.yaml |  4 ++++
 2 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 314585a..c4b1bb8 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,11 +1,12 @@
-#FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
-FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
+FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
+#FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
 #FROM nvidia/cuda:12.5.0-devel-ubuntu22.04
 
 RUN apt update -y
 RUN apt install -y wget
 
 RUN apt install -y build-essential
+RUN apt-get install -y cmake
 RUN apt install -y python3 python3-pip python-is-python3
 
 ## Miniconda
@@ -24,4 +25,23 @@ RUN rm -rf /root/gaussian-splatting
 
 RUN apt-get install -y git
 
+ENV TZ=Asia/Tokyo
+ENV DEBIAN_FRONTEND=noninteractive
+RUN apt-get update -y --fix-missing
+RUN apt-get install -y libglew-dev libassimp-dev libboost-all-dev libgtk-3-dev libopencv-dev
+RUN apt-get install -y libglfw3-dev libavdevice-dev libavcodec-dev libeigen3-dev libxxf86vm-dev libembree-dev ninja-build
+
+# SIBR viewer
+COPY SIBR_viewers/CMakeLists.txt /root/SIBR_viewers/CMakeLists.txt
+COPY SIBR_viewers/cmake /root/SIBR_viewers/cmake
+COPY SIBR_viewers/src /root/SIBR_viewers/src
+RUN cd /root/SIBR_viewers && \
+    cmake -Bbuild . -DCMAKE_BUILD_TYPE=Release && \
+    cmake --build build -j24 --target install
+
+ENV LIBGL_ALWAYS_INDIRECT=1
+
+RUN apt-get install -y mesa-utils
+RUN apt-get install -y libglew-dev
+
 CMD ["/bin/bash"]
\ No newline at end of file
diff --git a/compose.yaml b/compose.yaml
index 07bf68b..d555d1a 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -5,6 +5,8 @@ services:
       context: .
       dockerfile: Dockerfile
     container_name: gsplat-dev-container
+    environment:
+      - DISPLAY=${DISPLAY}
     image: gsplat:1.0
     # user: 1010:1010 #terada
     volumes:
@@ -13,6 +15,8 @@ services:
       - .:/root/gaussian-splatting
       - ~/dataset/tandt_db:/root/dataset/tandt_db
       - ~/.ssh:/root/.ssh
+      # X11 forwarding
+      - ~/.Xauthority:/root/.Xauthority
       # ports:
       #   - "8087:8088"
     working_dir: /root/gaussian-splatting