src/pydefx/parameters.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright (C) 2019  EDF R&D
   3 #
   4 # This library is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU Lesser General Public
   6 # License as published by the Free Software Foundation; either
   7 # version 2.1 of the License, or (at your option) any later version.
   8 #
   9 # This library is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 # Lesser General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Lesser General Public
  15 # License along with this library; if not, write to the Free Software
  16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17 #
  18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  19 #
  20 import tempfile
  21 from . import salome_proxy
  22 from . import configuration
  23
  24 class Parameters:
  25   def __init__(self, resource="localhost",
  26                nb_branches=None, salome_parameters=None):
  27     if salome_parameters is None :
  28       job_params = salome_proxy.createSalomeParameters()
  29       job_params.job_type = "yacs_file"
  30       job_params.resource_required.name = resource
  31       job_params.job_name = "idefix_job"
  32       job_params.wckey = configuration.defaultWckey(resource)
  33       job_params.work_directory = configuration.defaultWorkingDir(resource)
  34       if nb_branches is None:
  35         nb_branches = configuration.defaultNbBranches(resource)
  36       job_params.resource_required.nb_proc = nb_branches
  37       self.nb_branches = nb_branches
  38       self.salome_parameters = job_params
  39     else:
  40       if nb_branches is None:
  41         nb_branches = salome_parameters.resource_required.nb_proc
  42       self.nb_branches = nb_branches
  43       self.salome_parameters = salome_parameters
  44
  45   def configureResource(self, resource):
  46     self.salome_parameters.resource_required.name = resource
  47     self.salome_parameters.work_directory = configuration.defaultWorkingDir(
  48                                                                        resource)
  49     nb_branches = configuration.defaultNbBranches(resource)
  50     self.salome_parameters.resource_required.nb_proc = nb_branches
  51     self.nb_branches = nb_branches
  52     self.salome_parameters.wckey = configuration.defaultWckey(resource)
  53
  54   def createResultDirectory(self, result_base_dir):
  55     self.salome_parameters.result_directory = configuration.newResultDirectory(
  56                                                                 result_base_dir)
  57
  58   def createTmpResultDirectory(self):
  59     self.salome_parameters.result_directory = configuration.newResultDirectory(
  60                                                           tempfile.gettempdir())
  61
  62   # Specific deep copy function is needed because the default one does not work
  63   # for swig objects, when we are in no salome session mode.
  64   def __deepcopy__(self, memo):
  65     cls = self.__class__
  66     newobj = cls.__new__(cls)
  67     newobj.nb_branches = self.nb_branches
  68     newobj.salome_parameters = salome_proxy.createSalomeParameters()
  69     newobj.salome_parameters.job_name = self.salome_parameters.job_name
  70     newobj.salome_parameters.job_type = self.salome_parameters.job_type
  71     newobj.salome_parameters.job_file = self.salome_parameters.job_file
  72     newobj.salome_parameters.pre_command = self.salome_parameters.pre_command
  73     newobj.salome_parameters.env_file = self.salome_parameters.env_file
  74     newobj.salome_parameters.in_files = list(self.salome_parameters.in_files)
  75     newobj.salome_parameters.out_files = list(self.salome_parameters.out_files)
  76     newobj.salome_parameters.work_directory = self.salome_parameters.work_directory
  77     newobj.salome_parameters.local_directory = self.salome_parameters.local_directory
  78     newobj.salome_parameters.result_directory = self.salome_parameters.result_directory
  79     newobj.salome_parameters.maximum_duration = self.salome_parameters.maximum_duration
  80     newobj.salome_parameters.queue = self.salome_parameters.queue
  81     newobj.salome_parameters.partition = self.salome_parameters.partition
  82     newobj.salome_parameters.exclusive = self.salome_parameters.exclusive
  83     newobj.salome_parameters.mem_per_cpu = self.salome_parameters.mem_per_cpu
  84     newobj.salome_parameters.wckey = self.salome_parameters.wckey
  85     newobj.salome_parameters.extra_params = self.salome_parameters.extra_params
  86     newobj.salome_parameters.specific_parameters = self.salome_parameters.specific_parameters
  87     newobj.salome_parameters.resource_required.name = self.salome_parameters.resource_required.name
  88     newobj.salome_parameters.resource_required.hostname = self.salome_parameters.resource_required.hostname
  89     newobj.salome_parameters.resource_required.can_launch_batch_jobs = self.salome_parameters.resource_required.can_launch_batch_jobs
  90     newobj.salome_parameters.resource_required.can_run_containers = self.salome_parameters.resource_required.can_run_containers
  91     newobj.salome_parameters.resource_required.OS = self.salome_parameters.resource_required.OS
  92     newobj.salome_parameters.resource_required.nb_proc = self.salome_parameters.resource_required.nb_proc
  93     newobj.salome_parameters.resource_required.mem_mb = self.salome_parameters.resource_required.mem_mb
  94     newobj.salome_parameters.resource_required.cpu_clock = self.salome_parameters.resource_required.cpu_clock
  95     newobj.salome_parameters.resource_required.nb_node = self.salome_parameters.resource_required.nb_node
  96     newobj.salome_parameters.resource_required.nb_proc_per_node = self.salome_parameters.resource_required.nb_proc_per_node
  97
  98     return newobj