Salome HOME
Fix regression in Launcher with ForEach-like batch jobs
[modules/yacs.git] / src / Launcher / Launcher.hxx
1 // Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License.
10 //
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 // Lesser General Public License for more details.
15 //
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22
23 #ifndef __LAUNCHER_HXX__
24 #define __LAUNCHER_HXX__
25
26 #include "Launcher_Utils.hxx"
27 #include "Launcher_Job.hxx"
28
29 #include "ResourcesManager.hxx"
30 #include <SALOME_ResourcesCatalog_Parser.hxx>
31
32 #include "SALOME_Launcher_Parser.hxx"
33
34 #include <string>
35 #include <vector>
36
37 #include <pthread.h>
38
39 class MpiImpl;
40
41 namespace Batch{
42   class BatchManager;
43   class Job;
44 }
45
46 struct batchParams{
47   std::string batch_directory;
48   std::string expected_during_time;
49   std::string mem;
50   unsigned long nb_proc;
51
52   // Parameters for COORM
53   std::string launcher_file;
54   std::string launcher_args;
55 };
56
57 class LAUNCHER_EXPORT Launcher_cpp
58 {
59
60 public:
61   Launcher_cpp();
62   virtual ~Launcher_cpp();
63
64   // Main interface
65   void         createJob(Launcher::Job * new_job);
66   void         launchJob(int job_id);
67   const char * getJobState(int job_id);
68   const char * getAssignedHostnames(int job_id); // Get names or ids of hosts assigned to the job
69   void         getJobResults(int job_id, std::string directory);
70   bool         getJobDumpState(int job_id, std::string directory);
71   void         stopJob(int job_id);
72   void         removeJob(int job_id);
73
74   // Useful methods
75   long createJobWithFile(std::string xmlExecuteFile, std::string clusterName);
76   std::map<int, Launcher::Job *> getJobs();
77   void createBatchManagerForJob(Launcher::Job * job);
78   void addJobDirectlyToMap(Launcher::Job * new_job, const std::string job_reference);
79
80   // Lib methods
81   void SetResourcesManager( ResourcesManager_cpp* rm ) {_ResManager = rm;}
82
83   // Used by SALOME_Launcher
84   ResourcesManager_cpp *_ResManager;
85 protected:
86
87   virtual void notifyObservers(const std::string & event_name, const std::string & event_data) {}
88
89   // Methods used by user interface methods
90 #ifdef WITH_LIBBATCH
91   Batch::BatchManager *FactoryBatchManager(ParserResourcesType& params);
92   std::map <int, Batch::BatchManager*> _batchmap;
93 #endif
94   ParserLauncherType ParseXmlFile(std::string xmlExecuteFile);
95
96   std::map <int, Launcher::Job *> _launcher_job_map;  
97   int _job_cpt; // job number counter
98   pthread_mutex_t * _job_cpt_mutex; // mutex for job counter
99 };
100
101 #endif