Use spanner job to render

unkaktus · Mar 7, 2023 · c7dc8a2 · c7dc8a2
1 parent 9f62569
commit c7dc8a2
Show file tree

Hide file tree

Showing 7 changed files with 37 additions and 37 deletions.
diff --git a/.dockerignore b/.dockerignore
@@ -1,5 +1,6 @@
 Dockerfile
+/aux
 /states
 /data
 /cache
-/.git
+/.git
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1 @@
+.aux
diff --git a/Dockerfile b/Dockerfile
@@ -47,4 +47,4 @@ RUN python3 -c 'import scri'
 WORKDIR /opt/rose
 COPY . .
 ENV PV_PLUGIN_PATH=/opt/rose/plugins
-ENTRYPOINT ["/bin/bash"]
+ENV PATH=${PATH}":/opt/rose/render"
diff --git a/render/begin.toml b/render/begin.toml
@@ -0,0 +1,29 @@
+Name = "rose"
+Type = "simple"
+
+Walltime = "4h"
+
+Nodes = 2
+TasksPerNode = 2
+
+Email = "batch@example.com"
+
+InitScript = [
+ "source /home/SPACK2023/share/spack/setup-env.sh",
+ "module load apptainer-1.0.3-gcc-12.2.0-aojy6ca"
+ ]
+
+Runtime = [
+ "apptainer",
+ "exec",
+ "--bind /work:/work",
+ "--bind /scratch:/scratch",
+ "--env ROSE_CACHE_DIR=/scratch/user/rose-cache"
+ ]
+Executable = "/home/user/apptainers/rose-v1.1.0.sif"
+Arguments = [
+ "render_state.py",
+ "--state={{.ConfigFilename}}",
+ "--total-task-number={{.TotalTaskNumber}}",
+ "--task-id={{.TaskID}}"
+ ]
diff --git a/render/job.sh b/render/job.sh
diff --git a/render/render.sh b/render/render.sh
diff --git a/render/render_state.py b/render/render_state.py
@@ -1,17 +1,17 @@
 #!/usr/bin/env -S python -u
 
-import paraview.simple as pv
-import numpy as np
 import argparse
-import os
 
-parser = argparse.ArgumentParser('rose')
+parser = argparse.ArgumentParser('render_state.py')
 parser.add_argument("--total-task-number", type=int, help="Total number of tasks")
 parser.add_argument("--task-id", type=int, help="Current task ID")
 parser.add_argument("--state", type=str, help="State filename")
 parser.add_argument("--output-dir", type=str, help="Path to output directory", default=None)
 args = parser.parse_args()
 
+import paraview.simple as pv
+import numpy as np
+import os
 
 output_dir = os.path.splitext(args.state)[0]
 if args.output_dir is not None: