zebra/pyzebra
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add algorithm property

Browse Source
This commit is contained in:
usov_i 2020-07-15 15:29:26 +02:00
parent def7074a23
commit 71347327ab
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pyzebra/anatric.py
View File

@ -15,6 +15,8 @@ REFLECTION_PRINTER_FORMATS = (
"oksana",
)
ALGORITHMS = ("adaptivemaxcog", "adaptivedynamic")
def anatric(config_file):
subprocess.run(["anatric", config_file], check=True)
@ -30,7 +32,6 @@ class AnatricConfig:
self._tree = tree
alg_elem = tree.find("Algorithm")
self.algorithm = alg_elem.attrib["implementation"]
if self.algorithm == "adaptivemaxcog":
self.threshold = float(alg_elem.find("threshold").attrib["value"])
self.shell = float(alg_elem.find("shell").attrib["value"])
@ -212,5 +213,16 @@ class AnatricConfig:
self._tree.find("ReflectionPrinter").attrib["format"] = value
@property
def algorithm(self):
return self._tree.find("Algorithm").attrib["implementation"]
@algorithm.setter
def algorithm(self, value):
if value not in ALGORITHMS:
raise ValueError("Unknown algorithm.")
self._tree.find("Algorithm").attrib["implementation"] = value
def save_as(self, filename):
self._tree.write(filename)