Checker: Add study geometry checker.

This commit is contained in:
Pierre-Antoine Rouby committed 2023-06-19 10:35:34 +02:00
1 parent acaa369e77
commit dccf1a82ac
3 files changed
+58 -1

No files matched your search

+54 -1
View File
@@ -1,5 +1,7 @@
# -*- coding: utf-8 -*-
import time
from PyQt5.QtCore import QCoreApplication
from Checker.Checker import AbstractModelChecker, STATUS
@@ -26,7 +28,8 @@ class StudyNetworkReachChecker(AbstractModelChecker):
self._summary = "no_river_found"
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._summary = "no_reach_defined"
return False
@@ -35,6 +38,48 @@ class StudyNetworkReachChecker(AbstractModelChecker):
self._status = STATUS.OK
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):
def __init__(self):
@@ -44,8 +89,11 @@ class DummyOK(AbstractModelChecker):
self._description = _translate("Checker", "Dummy ok")
def run(self, study):
time.sleep(1)
self._summary = "ok"
self._status = STATUS.OK
return True
@@ -57,8 +105,11 @@ class DummyWARNING(AbstractModelChecker):
self._description = _translate("Checker", "Dummy warning")
def run(self, study):
time.sleep(1)
self._summary = "Warning detected"
self._status = STATUS.WARNING
return True
class DummyERROR(AbstractModelChecker):
@@ -69,6 +120,8 @@ class DummyERROR(AbstractModelChecker):
self._description = _translate("Checker", "Dummy error")
def run(self, study):
time.sleep(1)
self._summary = "Error detected"
self._status = STATUS.ERROR
return True