143 lines
3.5 KiB
Python
143 lines
3.5 KiB
Python
###
|
|
# 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)
|
|
# site-packages from virtual env
|
|
sys.path.append(os.path.join(ROOT, "env/lib/python3.9/site-packages"))
|
|
|
|
import toml
|
|
import logging
|
|
from anisotropy.utils import struct
|
|
|
|
from salomepl.simple import simple
|
|
from salomepl.faceCentered import faceCentered
|
|
from salomepl.bodyCentered import bodyCentered
|
|
|
|
from salomepl.geometry import getGeom
|
|
from salomepl.mesh import Mesh, Fineness, ExtrusionMethod, defaultParameters
|
|
|
|
|
|
def genmesh(config):
|
|
|
|
logger.info(f"""genmesh:
|
|
structure type:\t{ config.structure }
|
|
coefficient:\t{ config.geometry.theta }
|
|
fillet:\t{ config.geometry.fillet }
|
|
flow direction:\t{ config.geometry.direction }""")
|
|
|
|
salome.salome_init()
|
|
|
|
###
|
|
# Shape
|
|
##
|
|
geompy = getGeom()
|
|
structure = globals().get(config.structure)
|
|
shape, groups = structure(config.geometry.theta, config.geometry.fillet, config.geometry.direction)
|
|
[length, surfaceArea, volume] = geompy.BasicProperties(shape, theTolerance = 1e-06)
|
|
|
|
logger.info(f"""shape:
|
|
edges length:\t{ length }
|
|
surface area:\t{ surfaceArea }
|
|
volume:\t{ volume }""")
|
|
|
|
###
|
|
# Mesh
|
|
##
|
|
config = dict(config)
|
|
|
|
mconfig = defaultParameters(**config["mesh"])
|
|
|
|
lengths = [ geompy.BasicProperties(edge)[0] for edge in geompy.SubShapeAll(shape, geompy.ShapeType["EDGE"]) ]
|
|
meanSize = sum(lengths) / len(lengths)
|
|
mconfig["maxSize"] = meanSize
|
|
mconfig["minSize"] = meanSize * 1e-1
|
|
mconfig["chordalError"] = mconfig["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)
|
|
|
|
if mconfig["viscousLayers"]:
|
|
mesh.ViscousLayers(**mconfig)
|
|
|
|
config["mesh"].update(mconfig)
|
|
smconfigs = config["mesh"]["submesh"]
|
|
|
|
for name in smconfigs.keys():
|
|
for group in groups:
|
|
if group.GetName() == name:
|
|
subshape = group
|
|
|
|
smconfig = defaultParameters(**smconfigs[name])
|
|
smconfig["maxSize"] = meanSize * 1e-1
|
|
smconfig["minSize"] = meanSize * 1e-3
|
|
smconfig["chordalError"] = smconfig["minSize"] * 1e+1
|
|
|
|
mesh.Triangle(subshape, **smconfig)
|
|
config["mesh"]["submesh"][name].update(smconfig)
|
|
|
|
|
|
returncode, errors = mesh.compute()
|
|
|
|
if not returncode:
|
|
config["status"]["mesh"] = True
|
|
|
|
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"))
|
|
|
|
stats = ""
|
|
for k, v in mesh.stats().items():
|
|
stats += f"{ k }:\t\t{ v }\n"
|
|
|
|
logger.info(f"mesh stats:\n{ stats[ :-1] }")
|
|
|
|
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)
|
|
|
|
|