Mod: everything for db

anisotropy/anisotropy.py

@@ -110,10 +110,12 @@ def timer(func):
 
 class Anisotropy(object):
     def __init__(self):
-        self.db = self._setupDB()
+        self.env = env
+        self.db = None
         self.params = []
 
-        self.evalParameters(env)
+        #self.evalParameters(env)
+        self.setupDB()
 
     @staticmethod
     def version():
@@ -133,19 +135,17 @@ class Anisotropy(object):
 
         return "\n".join([ f"{ k }: { v }" for k, v in versions.items() ])
 
-    @staticmethod
+    def setupDB(self):
-    def _setupDB():
+        self.db = db.init(self.env["db_path"])
-        db.init(env["db_path"])
 
-        if not os.path.exists(env["db_path"]):
+        if not os.path.exists(self.env["db_path"]):
-            db.create_tables([Structure, Mesh])
+            self.db.create_tables([Structure, Mesh])
 
-        return db
+    def evalEnvParameters(self):
-
+        """ 'Uncompress' and eval environment parameters """
-    def evalParameters(self, _env: dict):
         from math import sqrt
 
-        structures = deepcopy(_env["structures"])
+        structures = deepcopy(self.env["structures"])
 
         for structure in structures:
             _theta = structure["geometry"]["theta"]
@@ -195,7 +195,7 @@ class Anisotropy(object):
 
 
                     path = os.path.join(
-                        _env["BUILD"],
+                        self.env["BUILD"],
                         structure["name"],
                         f"direction-{ direction }",
                         f"theta-{ theta }"
@@ -214,16 +214,25 @@ class Anisotropy(object):
                     mesh.update(
                         thickness = thicknessList[n]
                     )
-                    self.params.append({
+                    self.params.append(dict(
-                        "name": structure["name"],
+                        name = structure["name"],
-                        "path": path,
+                        path = path,
-                        "geometry": geometry,
+                        geometry = geometry,
-                        "mesh": mesh,
+                        mesh = mesh,
-                        "submesh": deepcopy(structure["submesh"])
+                        submesh = deepcopy(structure["submesh"])
-                    })
+                    ))
 
 
+    def getParams(structure, direction, theta):
+        for entry in self.params:
+            if entry["name"] == structure and
+                entry["geometry"]["direction"] == direction and
+                entry["geometry"]["theta"] == theta:
+                return entry
+
     # SELECT * FROM structure LEFT OUTER JOIN mesh ON mesh.structure_id = structure.id WHERE name = "faceCentered" AND direction = "[1, 1, 1]" AND theta = 0.12;
+    # Structure.select().join(Mesh, JOIN.LEFT_OUTER, on = (Mesh.structure_id == Structure.id)).where(Structure.name == "simple", Structure.direction == "[1, 0, 0]", Structure.theta == 0.13).dicts().get()
+    @timer
     def updateDB(self):
         for entry in self.params:
             query = (Structure
@@ -251,12 +260,38 @@ class Anisotropy(object):
                     mtab = Mesh.create(**m)
 
             else:
-                # TODO: not working update; incorrect update (all entries)
                 with self.db.atomic():
-                    stab = Structure.update(**s)
+                    (Structure.update(**s)
+                        .where(
+                            Structure.name == entry["name"],
+                            Structure.direction == str(entry["geometry"]["direction"]),
+                            Structure.theta == entry["geometry"]["theta"]
+                        )
+                        .execute())
 
-                    m.update(structure_id = stab)
+                    (Mesh.update(**m)
-                    mtab = Mesh.update(**m)
+                        .where(
+                            Mesh.structure_id == query.get().id
+                        )
+                        .execute())
+
+    @timer
+    def updateFromDB(self):
+        squery = Structure.select().order_by(Structure.id)
+        mquery = Mesh.select().order_by(Mesh.structure_id)
+
+        for s, m in zip(squery.dicts(), mquery.dicts()):
+            name = s.pop("name")
+            path = s.pop("path")
+
+            self.params.append(dict(
+                name = name,
+                path = path,
+                geometry = s,
+                mesh = m
+            ))
+
+        self.params = sorted(self.params, key = lambda entry: f"{ entry['name'] } { entry['geometry']['direction'] } { entry['geometry']['theta'] }")
 
     @timer
     def computeMesh(self):

anisotropy/models.py

@@ -17,7 +17,7 @@ class ListField(Field):
                 pass
 
             finally:
-                pval.append(ch.strip())
+                pval.append(ch.strip().replace("'", ""))
 
         return pval
 
@@ -34,7 +34,7 @@ class BaseModel(Model):
 
 class Structure(BaseModel):
     name = TextField()
-    direction = TextField()
+    direction = ListField()
     theta = FloatField()
 
     r0 = FloatField()

salomepl/genmesh.py

@@ -2,25 +2,37 @@
 #   This file executes inside salome environment
 #
 #   salome starts at user home directory
-#
-#   sys.argv = [ .., ROOT, case ]
 ##
 import os, sys
 import math
 
 import salome
 
-# get project path from args
-ROOT = sys.argv[1]
-CASE = sys.argv[2]
 
-sys.path.append(ROOT)
+@click.command()
-# site-packages from virtual env
+@click.argument("root")
-sys.path.append(os.path.join(ROOT, "env/lib/python3.9/site-packages"))
+@click.argument("name")
+@click.argument("direction")
+@click.argument("theta", type = click.FLOAT)
+def genmesh(root, name, direction, theta):
+    ###
+    #   Args
+    ##
+    direction = list(map(lambda num: float(num), direction[1:-1].split(",")))
+
+
+    ###
+    #   Modules
+    ##
+    sys.path.extend([
+        root,
+        os.path.join(root, "env/lib/python3.9/site-packages")
+    ])
 
-import toml
     import logging
-from anisotropy.utils import struct
+    import click
+
+    from anisotropy import Anisotropy
 
     from salomepl.simple import simple
     from salomepl.faceCentered import faceCentered
@@ -30,87 +42,119 @@ from salomepl.geometry import getGeom
     from salomepl.mesh import Mesh, Fineness, ExtrusionMethod, defaultParameters
 
 
-def genmesh(config):
+    ###
+    #   Model
+    ##
+    model = Anisotropy()
+    model.updateFromDB()
 
-    logger.info(f"""genmesh: 
+    p = model.getParams(name, direction, theta)
-    structure type:\t{ config.structure }
+
-    coefficient:\t{ config.geometry.theta }
+
-    fillet:\t{ config.geometry.fillet }
+    ###
-    flow direction:\t{ config.geometry.direction }""")
+    #   Logger
+    ##
+    logging.basicConfig(
+        level = logging.INFO,
+        format = model.env["logger"]["format"],
+        handlers = [
+            logging.StreamHandler(),
+            logging.FileHandler(
+                os.path.join(model.env["LOG"], model.env["logger"]["name"])
+            )
+        ]
+    )
+    logger = logging.getLogger(model.env["logger"]["name"])
+
+
+    ###
+    #   Entry
+    ##
+    logger.info("\n".join([
+        "genmesh:",
+        f"structure type:\t{ p['name'] }",
+        f"coefficient:\t{ p['geometry']['theta'] }",
+        f"fillet:\t{ p['geometry']['fillets'] }",
+        f"flow direction:\t{ p['geometry']['direction'] }"
+    ]))
 
     salome.salome_init()
 
+
     ###
     #   Shape
     ##
     geompy = getGeom()
-    structure = globals().get(config.structure)
+    structure = globals().get(p["name"])
-    shape, groups = structure(config.geometry.theta, config.geometry.fillet, config.geometry.direction)
+    shape, groups = structure(
+        p["geometry"]["theta"],
+        p["geometry"]["fillets"],
+        p["geometry"]["direction"]
+    )
+
     [length, surfaceArea, volume] = geompy.BasicProperties(shape, theTolerance = 1e-06)
 
-    logger.info(f"""shape:
+    logger.info("\n".join([
-    edges length:\t{ length }
+        "shape:"
-    surface area:\t{ surfaceArea }
+        f"edges length:\t{ length }",
-    volume:\t{ volume }""")
+        f"surface area:\t{ surfaceArea }",
+        f"volume:\t{ volume }"
+    ]))
+
 
     ###
     #   Mesh
     ##
-    config = dict(config)
+    mp = p["mesh"]
 
-    mconfig = defaultParameters(**config["mesh"])
+    lengths = [
-
+        geompy.BasicProperties(edge)[0] for edge in geompy.SubShapeAll(shape, geompy.ShapeType["EDGE"]) 
-    lengths = [ geompy.BasicProperties(edge)[0] for edge in geompy.SubShapeAll(shape, geompy.ShapeType["EDGE"]) ]
+    ]
     meanSize = sum(lengths) / len(lengths)
-    mconfig["maxSize"] = meanSize
+    mp["maxSize"] = meanSize
-    mconfig["minSize"] = meanSize * 1e-1
+    mp["minSize"] = meanSize * 1e-1
-    mconfig["chordalError"] = mconfig["maxSize"] / 2
+    mp["chordalError"] = mp["maxSize"] / 2
 
     faces = []
     for group in groups:
         if group.GetName() in mconfig["facesToIgnore"]:
             faces.append(group)
 
-    mconfig["faces"] = faces
 
     mesh = Mesh(shape)
-    mesh.Tetrahedron(**mconfig)
+    mesh.Tetrahedron(**mp)
 
-    if mconfig["viscousLayers"]:
+    if mp["viscousLayers"]:
-        mesh.ViscousLayers(**mconfig)
+        mesh.ViscousLayers(**mp, faces = faces)
 
-    config["mesh"].update(mconfig)
+    smp = p["submesh"]
-    smconfigs = config["mesh"]["submesh"]
 
-    for name in smconfigs.keys():
+    for name in smp.keys():
         for group in groups:
             if group.GetName() == name:
                 subshape = group
 
-        smconfig = defaultParameters(**smconfigs[name])
+        smp["maxSize"] = meanSize * 1e-1
-        smconfig["maxSize"] = meanSize * 1e-1
+        smp["minSize"] = meanSize * 1e-3
-        smconfig["minSize"] = meanSize * 1e-3
+        smp["chordalError"] = smp["minSize"] * 1e+1
-        smconfig["chordalError"] = smconfig["minSize"] * 1e+1
 
-        mesh.Triangle(subshape, **smconfig)
+        mesh.Triangle(subshape, **smp)
-        config["mesh"]["submesh"][name].update(smconfig)
 
 
     returncode, errors = mesh.compute()
 
     if not returncode:
-        config["status"]["mesh"] = True
+        # TODO: MeshResult
+        pass
 
     else:
         logger.error(errors)
 
-    with open(CONFIG, "w") as io:
-        toml.dump(config, io)
 
     mesh.removePyramids()
     mesh.assignGroups()
 
-    mesh.exportUNV(os.path.join(CASE, "mesh.unv"))
+    mesh.exportUNV(os.path.join(p["path"], "mesh.unv"))
 
     stats = ""
     for k, v in mesh.stats().items():
@@ -120,23 +164,3 @@ def genmesh(config):
 
     salome.salome_close()
 
-
-if __name__ == "__main__":
-    CONFIG = os.path.join(CASE, "task.toml")
-    config = struct(toml.load(CONFIG))
-
-    LOG = os.path.join(ROOT, "logs")
-
-    logging.basicConfig(
-        level = logging.INFO,
-        format = config.logger.format,
-        handlers = [
-            logging.StreamHandler(),
-            logging.FileHandler(f"{ LOG }/{ config.logger.name }.log")
-        ]
-    )
-    logger = logging.getLogger(config.logger.name)
-
-    genmesh(config)
-
-    

salomepl/utils.py

@@ -40,14 +40,14 @@ def runSalome(port: int, scriptpath: str, root: str, logpath: str = None, *args)
     if not logpath:
         logpath = "/tmp/salome.log"
 
+    fullargs = list(args)
+    fullargs.extend([ root, logpath ])
     fmtargs = "args:{}".format(", ".join([ str(arg) for arg in args ]))
     cmdargs = [
         "start", "-t",
         "--shutdown-servers=1",
         "--port", str(port),
         scriptpath,
-        root,
-        logpath,
         fmtargs
     ]