mirror of https://gitlab.com/pamhyr/pamhyr2
Checker: Add study geometry checker.
parent
acaa369e77
commit
dccf1a82ac
|
|
@ -1,5 +1,7 @@
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
import time
|
||||||
|
|
||||||
from PyQt5.QtCore import QCoreApplication
|
from PyQt5.QtCore import QCoreApplication
|
||||||
|
|
||||||
from Checker.Checker import AbstractModelChecker, STATUS
|
from Checker.Checker import AbstractModelChecker, STATUS
|
||||||
|
|
@ -26,7 +28,8 @@ class StudyNetworkReachChecker(AbstractModelChecker):
|
||||||
self._summary = "no_river_found"
|
self._summary = "no_river_found"
|
||||||
return False
|
return False
|
||||||
|
|
||||||
if len(river.edges()) == 0:
|
edges = list(filter(lambda e: e.is_enable(), river.edges()))
|
||||||
|
if len(edges) == 0:
|
||||||
self._status = STATUS.ERROR
|
self._status = STATUS.ERROR
|
||||||
self._summary = "no_reach_defined"
|
self._summary = "no_reach_defined"
|
||||||
return False
|
return False
|
||||||
|
|
@ -35,6 +38,48 @@ class StudyNetworkReachChecker(AbstractModelChecker):
|
||||||
self._status = STATUS.OK
|
self._status = STATUS.OK
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
class StudyGeometryChecker(AbstractModelChecker):
|
||||||
|
def __init__(self):
|
||||||
|
super(StudyGeometryChecker, self).__init__()
|
||||||
|
|
||||||
|
self._name = _translate("Checker", "Study geometry checker")
|
||||||
|
self._description = _translate("Checker", "Check if exists geometry for each reach of study")
|
||||||
|
self._reachs = []
|
||||||
|
|
||||||
|
def run(self, study):
|
||||||
|
ok = True
|
||||||
|
nerror = 0
|
||||||
|
summary = "ok"
|
||||||
|
status = STATUS.OK
|
||||||
|
|
||||||
|
if study is None:
|
||||||
|
self._status = STATUS.ERROR
|
||||||
|
self._summary = "invalid_study"
|
||||||
|
return False
|
||||||
|
|
||||||
|
river = study.river
|
||||||
|
if river is None:
|
||||||
|
self._status = STATUS.ERROR
|
||||||
|
self._summary = "no_river_found"
|
||||||
|
return False
|
||||||
|
|
||||||
|
edges = list(filter(lambda e: e.is_enable(), river.edges()))
|
||||||
|
if len(edges) == 0:
|
||||||
|
self._status = STATUS.ERROR
|
||||||
|
self._summary = "no_reach_defined"
|
||||||
|
return False
|
||||||
|
|
||||||
|
for edge in edges:
|
||||||
|
if edge.is_enable() and len(edge.reach.profiles) < 3:
|
||||||
|
summary = f"no_geometry_defined"
|
||||||
|
status = STATUS.ERROR
|
||||||
|
ok = False
|
||||||
|
self._reachs.append(edge)
|
||||||
|
|
||||||
|
self._summary = summary
|
||||||
|
self._status = status
|
||||||
|
return ok
|
||||||
|
|
||||||
|
|
||||||
class DummyOK(AbstractModelChecker):
|
class DummyOK(AbstractModelChecker):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
|
|
@ -44,8 +89,11 @@ class DummyOK(AbstractModelChecker):
|
||||||
self._description = _translate("Checker", "Dummy ok")
|
self._description = _translate("Checker", "Dummy ok")
|
||||||
|
|
||||||
def run(self, study):
|
def run(self, study):
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
self._summary = "ok"
|
self._summary = "ok"
|
||||||
self._status = STATUS.OK
|
self._status = STATUS.OK
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -57,8 +105,11 @@ class DummyWARNING(AbstractModelChecker):
|
||||||
self._description = _translate("Checker", "Dummy warning")
|
self._description = _translate("Checker", "Dummy warning")
|
||||||
|
|
||||||
def run(self, study):
|
def run(self, study):
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
self._summary = "Warning detected"
|
self._summary = "Warning detected"
|
||||||
self._status = STATUS.WARNING
|
self._status = STATUS.WARNING
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
class DummyERROR(AbstractModelChecker):
|
class DummyERROR(AbstractModelChecker):
|
||||||
|
|
@ -69,6 +120,8 @@ class DummyERROR(AbstractModelChecker):
|
||||||
self._description = _translate("Checker", "Dummy error")
|
self._description = _translate("Checker", "Dummy error")
|
||||||
|
|
||||||
def run(self, study):
|
def run(self, study):
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
self._summary = "Error detected"
|
self._summary = "Error detected"
|
||||||
self._status = STATUS.ERROR
|
self._status = STATUS.ERROR
|
||||||
return True
|
return True
|
||||||
|
|
|
||||||
|
|
@ -53,6 +53,9 @@ class Edge(object):
|
||||||
def name(self):
|
def name(self):
|
||||||
return self._name if self._name != "" else f"{self.node1.name} -> {self.node2.name}"
|
return self._name if self._name != "" else f"{self.node1.name} -> {self.node2.name}"
|
||||||
|
|
||||||
|
def is_enable(self):
|
||||||
|
return self._enable
|
||||||
|
|
||||||
def enable(self, enable=True):
|
def enable(self, enable=True):
|
||||||
self._enable = enable
|
self._enable = enable
|
||||||
self._status.modified()
|
self._status.modified()
|
||||||
|
|
|
||||||
|
|
@ -36,6 +36,7 @@ class Study(Serializable):
|
||||||
def checkers(cls):
|
def checkers(cls):
|
||||||
lst = [
|
lst = [
|
||||||
StudyNetworkReachChecker(),
|
StudyNetworkReachChecker(),
|
||||||
|
StudyGeometryChecker(),
|
||||||
DummyOK(),
|
DummyOK(),
|
||||||
DummyWARNING(),
|
DummyWARNING(),
|
||||||
DummyERROR(),
|
DummyERROR(),
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue