122 lines
3.5 KiB
Python
122 lines
3.5 KiB
Python
from collections import namedtuple
|
|
import os, sys
|
|
import logging
|
|
from pyquaternion import Quaternion
|
|
|
|
ROOT = "/home/nafaryus/projects/anisotrope-cube"
|
|
sys.path.append(ROOT)
|
|
|
|
LOG = os.path.join(ROOT, "logs")
|
|
|
|
import salome
|
|
|
|
from simpleCubic import simpleCubic
|
|
from faceCenteredCubic import faceCenteredCubic
|
|
from bodyCenteredCubic import bodyCenteredCubic
|
|
|
|
from src import geometry_utils
|
|
from src import mesh_utils
|
|
|
|
def genMesh(stype, theta, flowdirection, saveto):
|
|
_G = globals()
|
|
|
|
structure = _G.get(stype)
|
|
|
|
if structure:
|
|
salome.salome_init()
|
|
|
|
#grains, cubic, rhombohedron = structure(theta)
|
|
#fd = namedtuple("fd", ["x", "y", "z"])
|
|
grains, geometry1, geometry2 = structure(theta)
|
|
geometry = geometry1
|
|
|
|
if flowdirection == [1, 1, 1]:
|
|
geometry = geometry2
|
|
norm = [-1, -1, 1]
|
|
bcount = 6
|
|
|
|
# initial angle
|
|
angle = math.pi / 6
|
|
vec = Quaternion(0, flowdirection[0], flowdirection[1], flowdirection[2])
|
|
ax = Quaternion(
|
|
math.cos(angle * 0.5),
|
|
math.sin(angle * 0.5) * norm[0],
|
|
math.sin(angle * 0.5) * norm[1],
|
|
math.sin(angle * 0.5) * norm[2])
|
|
qvec = (ax * vec * ax.inverse).vector
|
|
norm = [qvec[0], qvec[1], qvec[2]]
|
|
#direction = fd([1, 1, 1], [1, -1, 1], [1, -1, -1])
|
|
|
|
#else:
|
|
# geometry = cubic
|
|
|
|
if flowdirection == [1, 0, 0]:
|
|
norm = [0, 0, 1]
|
|
bcount = 4
|
|
#direction = fd([1, 1, 0], [1, -1, 0], [0, 0, 1])
|
|
|
|
if flowdirection == [0, 0, 1]:
|
|
norm = [1, 1, 0]
|
|
bcount = 4
|
|
#direction = fd([0, 0, 1], [1, -1, 0], [1, 1, 0])
|
|
|
|
#boundary = geometry_utils.boundaryCreate(geometry, direction, grains)
|
|
boundary = geometry_utils.createBoundary(geometry, bcount, flowdirection, norm, grains)
|
|
|
|
fineness = 3
|
|
mesh = mesh_utils.meshCreate(geometry, boundary, fineness)
|
|
mesh_utils.meshCompute(mesh)
|
|
|
|
#path = os.path.join(saveto,
|
|
# stype,
|
|
# "theta-%s" % theta,
|
|
# "direction-{}{}{}".format(flowdirection[0], flowdirection[1], flowdirection[2]))
|
|
|
|
#if not os.path.exists(path):
|
|
# logging.info("Creating directory: {}".format(path))
|
|
# os.makedirs(path)
|
|
|
|
mesh_utils.meshExport(mesh, saveto) # os.path.join(path, "mesh.unv"))
|
|
|
|
salome.salome_close()
|
|
|
|
else:
|
|
raise Exception("Unknown sample function")
|
|
|
|
if __name__ == "__main__":
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="%(levelname)s: %(message)s",
|
|
handlers = [
|
|
logging.StreamHandler(),
|
|
logging.FileHandler("{}/cubic.log".format(LOG))
|
|
])
|
|
|
|
fancyline = "--------------------------------------------------------------------------------"
|
|
logging.info(fancyline)
|
|
|
|
stype = str(sys.argv[1])
|
|
theta = float(sys.argv[2])
|
|
|
|
#ignore = "[], "
|
|
flowdirection = [int(coord) for coord in sys.argv[3]]
|
|
|
|
#for sym in str(sys.argv[3]):
|
|
# if sym not in list(ignore):
|
|
# flowdirection.append(int(sym))
|
|
|
|
saveto = str(sys.argv[4])
|
|
|
|
logging.info("""Args:
|
|
structure type:\t{}
|
|
coefficient:\t{}
|
|
flow direction:\t{}
|
|
export path:\t{}\n""".format(stype, theta, flowdirection, saveto))
|
|
|
|
#print(flowdirection)
|
|
|
|
genMesh(stype, theta, flowdirection, saveto)
|
|
|
|
logging.info(fancyline)
|