mirror of https://gitlab.com/pamhyr/pamhyr2
114 lines
2.9 KiB
Python
114 lines
2.9 KiB
Python
# Run.py -- Pamhyr
|
|
# Copyright (C) 2023-2024 INRAE
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
import logging
|
|
|
|
from queue import Queue
|
|
|
|
from PyQt5.QtCore import QProcess
|
|
|
|
from Scripts.AScript import AScript
|
|
from Model.Study import Study
|
|
|
|
logger = logging.getLogger()
|
|
|
|
|
|
class ScriptRun(AScript):
|
|
name = "Run"
|
|
description = "Run solver on Pamhyr2 a study"
|
|
|
|
def usage(self):
|
|
logger.info(
|
|
f"Usage : {self._args[0]} {self._args[1]} " +
|
|
"<SOLVER> <STUDY>"
|
|
)
|
|
|
|
def run(self):
|
|
if len(self._args) < 4:
|
|
return 1
|
|
|
|
command = self._args[1]
|
|
solver_name = self._args[2]
|
|
study_file = os.path.abspath(
|
|
self._args[3]
|
|
)
|
|
|
|
try:
|
|
solver = next(
|
|
filter(
|
|
lambda solver: solver.name == solver_name,
|
|
self._conf.solvers
|
|
)
|
|
)
|
|
except Exception as e:
|
|
logger.error(f"No solver found: {e}")
|
|
return 2
|
|
|
|
study = Study.open(study_file)
|
|
|
|
self._solver = solver
|
|
self._study = study
|
|
|
|
logger.info(
|
|
f"Run {solver.name} ({solver.type}) " +
|
|
"on study '{study.name}' ({study_file})"
|
|
)
|
|
|
|
# Workdir
|
|
workdir = os.path.join(
|
|
os.path.dirname(study.filename),
|
|
"_PAMHYR_",
|
|
study.name.replace(" ", "_"),
|
|
solver.name.replace(" ", "_")
|
|
)
|
|
os.makedirs(workdir, exist_ok=True)
|
|
logger.info(f"Set working dir to {workdir}")
|
|
|
|
# Preparate process
|
|
p = QProcess(None)
|
|
p.setWorkingDirectory(workdir)
|
|
|
|
self._q = Queue()
|
|
|
|
# Export and Run
|
|
logger.info(f"~Export~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~")
|
|
solver.export(study, workdir, qlog=self._q)
|
|
|
|
while self._q.qsize() != 0:
|
|
s = self._q.get()
|
|
logger.info(s)
|
|
|
|
if command == "run":
|
|
logger.info(f"~Run~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~")
|
|
solver.run(
|
|
study,
|
|
process=p,
|
|
output_queue=self._q
|
|
)
|
|
p.waitForFinished()
|
|
|
|
logger.info(f"~End~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~")
|
|
|
|
return 0
|
|
|
|
|
|
class ScriptExport(ScriptRun):
|
|
name = "Export"
|
|
description = "Export Pamhyr2 study for solver"
|