Commit dccf1a82 authored by Pierre-Antoine Rouby's avatar Pierre-Antoine Rouby
Browse files

Checker: Add study geometry checker.

Showing with 58 additions and 1 deletion
+58 -1
# -*- 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): ...@@ -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): ...@@ -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): ...@@ -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): ...@@ -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): ...@@ -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): ...@@ -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): ...@@ -36,6 +36,7 @@ class Study(Serializable):
def checkers(cls): def checkers(cls):
lst = [ lst = [
StudyNetworkReachChecker(), StudyNetworkReachChecker(),
StudyGeometryChecker(),
DummyOK(), DummyOK(),
DummyWARNING(), DummyWARNING(),
DummyERROR(), DummyERROR(),
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment