Changeset 5175 for BOL

Timestamp:

Sep 9, 2024, 11:33:12 AM (3 months ago)

Author:

abarral

Message:

Automatically set root_dir via setup.sh

Location:

BOL/LMDZ_Setup_amaury

Files:

• lmdz_env.sh (modified) (6 diffs)
• setup.sh (modified) (14 diffs)

BOL/LMDZ_Setup_amaury/lmdz_env.sh

@@ -7 +7 @@
 # See the end of <set_env> for the explanation of each
 ########################################################
+
+# <root_dir> will be set by sed by setup.sh here
+root_dir=/home/abarral/PycharmProjects/installLMDZ/LMDZ_Setup
 
 function get_hostname {
@@ -39 +42 @@
       #        alors $STORE, $WORK etc vont designer les espaces de "newproj")
       account="lmd"  # $(idrproj | grep active | awk '{ print $1}') doesn't work on compute nodes
-      root_dir="$WORK/LMDZ_Setup";
       ARCH="X64_JEANZAY"
       SIMRUNBASEDIR="$SCRATCH/"
@@ -57 +59 @@
       module load cdo/2.3.0
 
-      root_dir="/data/abarral/LMDZ_SETUP_ROOT/LMDZ_Setup";
       ARCH="X64_MESOIPSL-GNU"
       SIMRUNBASEDIR="$SCRATCH/"
@@ -93 +94 @@
 
       account=$(/usr/sbin/my_project.py -l 2>&1 | head -1 | cut -d " " -f 3- | cut -c 5-)
-      root_dir="$WORKDIR/LMDZ_Setup";
       ARCH="X64_ADASTRA-GNU"
       SIMRUNBASEDIR="$SCRATCHDIR/"
@@ -108 +108 @@
       ;;
     *) echo "WARNING: RUNNING THIS SCRIPT ON A LOCAL COMPUTER IS DISCOURAGED (lackluster cpus and memory)"
-      root_dir="/home/abarral/PycharmProjects/installLMDZ/LMDZ_Setup"; # Where you have extracted LMDZ_Setup. Can't use $(pwd) since this script gets copied and ran from several locations
       ARCH="local-gfortran-parallel"  # The arch file to use
       SIMRUNBASEDIR="/tmp/SCRATCH/"  # Where the simulations will be executed ($SIMRUNBASEDIR/LMDZ_Setup/...)
@@ -128 +127 @@
 set_env
 
-if [[ ! (-d $root_dir && -f $root_dir/lmdz_env.sh) ]]; then
-  echo "STOP: root_dir $root_dir not found, either you are running on an unsupported cluster, or you haven't edited lmdz_env.sh properly"; exit 1
+if [[ ! (-d $root_dir && -f $root_dir/.lmdz_setup_root_dir && -f $root_dir/lmdz_env.sh) ]]; then
+  echo "STOP: root_dir $root_dir not found, either you are running on an unsupported cluster, or the initialisation failed midway"; exit 1
 fi

BOL/LMDZ_Setup_amaury/setup.sh

@@ -7 +7 @@
 # EXPERT-LEVEL CHOICES, only available in setup.sh, not in main.sh :
 ####################################################################
-function define_expert_options {
+function define_expert_options() {
   # optim: either "" or "-debug" to compile in debug mode (slower but better diagnosis of segfaults)
   optim=""
@@ -50 +50 @@
 # /!\ DO NOT EDIT BELOW UNLESS YOU KNOW WHAT YOU ARE DOING /!\
 
-function enable_platform {  # In job scripts, sed platform-specific headers
+function enable_platform() {  # In job scripts, sed platform-specific headers
   local file="$1"
   local platform
@@ -64 +64 @@
 }
 
-function load_install_lib {
+function load_install_lib() {
   # Fetch and source install_lmdz.sh to get `myget`
   if [[ ! -f "install_lmdz.sh" ]]; then
@@ -74 +74 @@
 }
 
-function set_default_params {
+function set_default_params() {
   # Default value of script parameters
   SIM=$(basename "$local")CTL  # name
@@ -107 +107 @@
 }
 
-function read_cmdline_args {
+function read_cmdline_args() {
   while (($# > 0)); do
     case $1 in
@@ -188 +188 @@
 }
 
-function ensure_correct_option_combinations {
+function ensure_correct_option_combinations() {
   # AVOID COMBINATIONS OF OPTIONS THAT DON'T WORK in user choices
   if [[ $ok_guide = y && $climato = 1 ]]; then
@@ -213 +213 @@
 }
 
-function install_model {
+function install_model() {
   mkdir -p "$LMDZD"
 
@@ -249 +249 @@
 }
 
-function setup_def {  # modify various .def in ./DEF (+ xios xml as needed)
+function setup_def() {  # modify various .def in ./DEF (+ xios xml as needed)
   cd "$local"
 
@@ -359 +359 @@
 }
 
-function setup_ce0l { # Verification de l'existance de l'état initial, compilation eventuelle pour sa creation
+function setup_ce0l() { # Verification de l'existance de l'état initial, compilation eventuelle pour sa creation
   if [[ ! -d $INIT ]]; then
     if [[ $init = 0 ]]; then
@@ -379 +379 @@
 }
 
-function setup_simu {
+function setup_simu() {
   SIMRUNTOPDIR="$SIMRUNBASEDIR/$(basename "$local")"
   SIMRUNDIR=$SIMRUNTOPDIR
@@ -461 +461 @@
 }
 
-function fetch_simu_init_files {
+function fetch_simu_init_files() {
   #####################################################################
   echo "Recuperation eventuelle de certains fichiers sur $LMDZ_INIT"
@@ -542 +542 @@
 }
 
-function run_sim_or_init {
+function run_sim_or_init() {
   cd "$local"
 
@@ -715 +715 @@
 }
 
-function message_post_submit {
+function message_post_submit() {
   if [[ $ok_guide = "y" && $init = 1 ]]; then
     cd "$local"
@@ -731 +731 @@
 }
 
-# Set up the appropriate environment
-source lmdz_env.sh
+function setup_and_load_lmdz_env() {
+  if [[ ! -f .lmdz_setup_root_dir ]]; then echo "STOP: setup.sh is not located in the root dir ??!!"; exit 1; fi
+  # sed root_dir in lmdz_env.sh
+  sed -i'' "s<root_dir=.*<root_dir=$local<" lmdz_env.sh
+
+  # Set up the appropriate environment
+  source lmdz_env.sh
+}
 
 local=$(pwd)
 
+setup_and_load_lmdz_env
 load_install_lib
 define_expert_options