src/Launcher/Launcher_Job_SALOME.cxx

   1 // Copyright (C) 2009-2014  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 // Author: André RIBES - EDF R&D
  21 //
  22 #include "Launcher_Job_SALOME.hxx"
  23 #include "Basics_DirUtils.hxx"
  24
  25 #ifdef WITH_LIBBATCH
  26 #include <libbatch/Constants.hxx>
  27 #endif
  28
  29 #ifdef WIN32
  30 #include <io.h>
  31 #define _chmod chmod
  32 #endif
  33
  34 Launcher::Job_SALOME::Job_SALOME() {}
  35
  36 Launcher::Job_SALOME::~Job_SALOME() {}
  37
  38 void
  39 Launcher::Job_SALOME::setResourceDefinition(const ParserResourcesType & resource_definition)
  40 {
  41   // Check resource_definition
  42   if (resource_definition.AppliPath == "")
  43   {
  44     std::string mess = "Resource definition must define an application path !, resource name is: " + resource_definition.Name;
  45     throw LauncherException(mess);
  46   }
  47   Launcher::Job::setResourceDefinition(resource_definition);
  48 }
  49
  50 void
  51 Launcher::Job_SALOME::update_job()
  52 {
  53 #ifdef WITH_LIBBATCH
  54   Batch::Parametre params = common_job_params();
  55   params[Batch::EXECUTABLE] = buildSalomeScript(params);
  56   _batch_job->setParametre(params);
  57 #endif
  58 }
  59
  60 #ifdef WITH_LIBBATCH
  61 std::string
  62 Launcher::Job_SALOME::buildSalomeScript(Batch::Parametre params)
  63 {
  64   // parameters
  65   std::string work_directory = params[Batch::WORKDIR].str();
  66
  67   std::string launch_script = Kernel_Utils::GetTmpDir() + "runSalome_" + _job_file_name + "_" + _launch_date + ".sh";
  68   std::ofstream launch_script_stream;
  69   launch_script_stream.open(launch_script.c_str(),
  70                             std::ofstream::out
  71 #ifdef WIN32
  72  | std::ofstream::binary   //rnv: to avoid CL+RF end of line on windows
  73 #endif
  74                            );
  75
  76   // Begin of script
  77   launch_script_stream << "#!/bin/sh -f" << std::endl;
  78   launch_script_stream << "cd " << work_directory << std::endl;
  79   launch_script_stream << "export PYTHONPATH=" << work_directory << ":$PYTHONPATH" << std::endl;
  80   launch_script_stream << "export PATH=" << work_directory << ":$PATH" << std::endl;
  81   if (_env_file != "")
  82   {
  83     std::string::size_type last = _env_file.find_last_of("/");
  84     launch_script_stream << ". " << _env_file.substr(last+1) << std::endl;
  85   }
  86   launch_script_stream << "export SALOME_TMP_DIR=" << work_directory << "/logs" << std::endl;
  87
  88   // -- Generates Catalog Resources
  89   std::string resource_protocol = _resource_definition.getClusterInternalProtocolStr();
  90   launch_script_stream << "if [ \"x$LIBBATCH_NODEFILE\" != \"x\" ]; then " << std::endl;
  91   launch_script_stream << "CATALOG_FILE=" << "CatalogResources_" << _launch_date << ".xml" << std::endl;
  92   launch_script_stream << "export USER_CATALOG_RESOURCES_FILE=" << "$CATALOG_FILE" << std::endl;
  93   launch_script_stream << "echo '<!DOCTYPE ResourcesCatalog>'  > $CATALOG_FILE" << std::endl;
  94   launch_script_stream << "echo '<resources>'                 >> $CATALOG_FILE" << std::endl;
  95   launch_script_stream << "cat $LIBBATCH_NODEFILE | sort | uniq -c | while read nbproc host"  << std::endl;
  96   launch_script_stream << "do"                                                  << std::endl;
  97   // Full name doesn't work. eg: sagittaire-7 instead of sagittaire-7.lyon.grid5000.fr
  98   launch_script_stream << "host_basename=$(echo $host | cut -f1 -d.)"                                                  << std::endl;
  99   launch_script_stream << "echo '<machine name='\\\"$host_basename\\\" >> $CATALOG_FILE" << std::endl;
 100   launch_script_stream << "echo '         hostname='\\\"$host_basename\\\" >> $CATALOG_FILE" << std::endl;
 101   launch_script_stream << "echo '         type=\"single_machine\"' >> $CATALOG_FILE" << std::endl;
 102   launch_script_stream << "echo '         protocol=\"" << resource_protocol               << "\"' >> $CATALOG_FILE" << std::endl;
 103   launch_script_stream << "echo '         userName=\"" << _resource_definition.UserName   << "\"' >> $CATALOG_FILE" << std::endl;
 104   launch_script_stream << "echo '         appliPath=\"" << _resource_definition.AppliPath << "\"' >> $CATALOG_FILE" << std::endl;
 105   launch_script_stream << "echo '         mpi=\"" << _resource_definition.getMpiImplTypeStr() << "\"' >> $CATALOG_FILE" << std::endl;
 106   launch_script_stream << "echo '         nbOfNodes='\\\"$nbproc\\\" >> $CATALOG_FILE" << std::endl;
 107   launch_script_stream << "echo '         nbOfProcPerNode=\"1\"' >> $CATALOG_FILE" << std::endl;
 108   launch_script_stream << "echo '         canRunContainers=\"true\"' >> $CATALOG_FILE" << std::endl;
 109   launch_script_stream << "echo '/>'                                                              >> $CATALOG_FILE" << std::endl;
 110   launch_script_stream << "done"                                 << std::endl;
 111   launch_script_stream << "echo '</resources>' >> $CATALOG_FILE" << std::endl;
 112   launch_script_stream << "fi" << std::endl;
 113
 114   // Create file for ns-port-log
 115   launch_script_stream << "NS_PORT_FILE_PATH=`mktemp " << _resource_definition.AppliPath << "/USERS/nsport_XXXXXX` &&\n";
 116   launch_script_stream << "NS_PORT_FILE_NAME=`basename $NS_PORT_FILE_PATH` &&\n";
 117
 118   // Launch SALOME with an appli
 119   launch_script_stream << _resource_definition.AppliPath << "/runAppli --terminal --ns-port-log=$NS_PORT_FILE_NAME --server-launch-mode=fork ";
 120   launch_script_stream << "> logs/salome_" << _launch_date << ".log 2>&1 &&" << std::endl;
 121   launch_script_stream << "current=0 &&\n"
 122                        << "stop=20 &&\n"
 123                        << "while ! test -s $NS_PORT_FILE_PATH\n"
 124                        << "do\n"
 125                        << "  sleep 2\n"
 126                        << "  current=$((current+1))\n"
 127                        << "  if [ \"$current\" -eq \"$stop\" ] ; then\n"
 128                        << "    echo Error Naming Service failed ! >&2\n"
 129                        << "    exit\n"
 130                        << "  fi\n"
 131                        << "done &&\n"
 132                        << "appli_port=`cat $NS_PORT_FILE_PATH` &&\n"
 133                        << "rm $NS_PORT_FILE_PATH &&\n";
 134
 135   // Call real job type
 136   addJobTypeSpecificScript(launch_script_stream);
 137
 138   // End
 139   launch_script_stream << _resource_definition.AppliPath << "/runSession -p $appli_port shutdownSalome.py" << std::endl;
 140   launch_script_stream << "sleep 10" << std::endl;
 141
 142   // Return
 143   launch_script_stream.flush();
 144   launch_script_stream.close();
 145   chmod(launch_script.c_str(), 0x1ED);
 146   return launch_script;
 147 }
 148 #endif
 149