def clean_mesh_files(self, path_prefix="foam/constant"):
system_settings = bpy.context.scene.ODS_CFD.system
solver_properties = bpy.context.scene.ODS_CFD.solver
control_properties = bpy.context.scene.ODS_CFD.control
project_id = system_settings.project_id
task_id = system_settings.task_id
# Get the file list
files = GenericViewSet(f'/api/task/{task_id}/file/').list()
# Get the processor folder paths using regex
mesh_files = set()
for _file in files:
paths = re.findall(f'{path_prefix}/polyMesh/.*', _file.get("file"))
paths = [i for i in paths if not i.endswith('blockMeshDict')]
mesh_files = mesh_files.union(set(paths))
# Get the first level files and folders
for _path in list(mesh_files):
folder_path = re.findall(".*/polyMesh/.*/", _path)
if folder_path:
mesh_files.remove(_path)
mesh_files = mesh_files.union(set(folder_path))
print(f"Got mesh file and folder paths: {mesh_files}")
# Delete each of the paths
for path in mesh_files:
response = GenericViewSet(f'/api/task/{task_id}/file/').delete(
path)
print(f"Deleted mesh file path: {path}")
time.sleep(0.05) # Sleep for a bit to avoid throttling
def run_solver(self):
system_settings = bpy.context.scene.ODS_CFD.system
solver_properties = bpy.context.scene.ODS_CFD.solver
control_properties = bpy.context.scene.ODS_CFD.control
project_id = system_settings.project_id
task_id = system_settings.task_id
# Get or create the mesh task
mesh_task = self.get_or_create_task(project_id, 'mesh', task_id)
# Get or create an empty setup task
solver_task = self.get_or_create_task(project_id,
'solution',
task_id,
dependent_on=mesh_task['uid'])
# Dispatch to the mesh task
solver_task = GenericViewSet(
f'/api/project/{project_id}/task/').update(
solver_task['uid'], {
'status': "pending",
'config': {
'task_type':
'cfd',
'cmd':
'pipeline',
'commands':
[solver_properties.name, "reconstructPar -noZero"],
'cpus': [i for i in system_settings.decompN],
'iterations':
control_properties.endTime
}
})
return solver_task
def write_mesh_files(self):
""" Upload the geometry and dispatch the task in one operation
"""
system_settings = bpy.context.scene.ODS_CFD.system
mesh_properties = bpy.context.scene.ODS_CFD.mesh
solver_properties = bpy.context.scene.ODS_CFD.solver
project_id = system_settings.project_id
task_id = system_settings.task_id
# Get or create an empty setup task
setup_task = self.get_or_create_task(
project_id,
'setup',
task_id,
)
# Write the mesh files
setup_task = GenericViewSet(f'/api/project/{project_id}/task/').update(
setup_task['uid'], {
'status': "pending",
'config': {
'task_type': 'magpy',
'cmd': 'cfd.io.tasks.write_mesh',
'base_mesh': mesh_properties.domain_json(),
'snappyhex_mesh': mesh_properties.snappy_json()
}
})
return setup_task
def get_or_create_project_and_task(self):
system_settings = bpy.context.scene.ODS_CFD.system
project_name = system_settings.project_name.strip()
project_number = system_settings.project_number
task_name = system_settings.task_name.strip()
# Get the list of projects
if not project_number:
project_list = GenericViewSet('/api/project/').list()
project_number = len(project_list) or 1
# Get or create the project
project = GenericViewSet('/api/project/').get_or_create(
{'name': project_name},
create_params={'number': project_number or None},
create=True)
if project:
system_settings.project_name = project.get('name')
system_settings.project_number = str(project.get('number')) or ""
system_settings.project_id = project.get('uid')
system_settings.project_data = json.dumps(project)
# We won't get here unless the getting of project succeded
task = GenericViewSet(
f"/api/project/{project['uid']}/task/").get_or_create(
{'name': task_name},
create_params={
'config': {
'case_dir': 'foam',
'task_type': 'parent'
}
},
create=True)
if task:
system_settings.task_name = task.get('name')
system_settings.task_id = task.get('uid')
system_settings.task_data = json.dumps(task)
return {'project': project, 'task': task}
def clean_processor_dirs(self, path_prefix="foam"):
system_settings = bpy.context.scene.ODS_CFD.system
solver_properties = bpy.context.scene.ODS_CFD.solver
control_properties = bpy.context.scene.ODS_CFD.control
project_id = system_settings.project_id
task_id = system_settings.task_id
# Get the file list
files = GenericViewSet(f'/api/task/{task_id}/file/').list()
# Get the processor folder paths using regex
processor_paths = set()
for _file in files:
processors = re.findall(f'{path_prefix}/processor[0-9]*/',
_file.get("file"))
processor_paths = processor_paths.union(set(processors))
print(f"Got processor folder paths: {processor_paths}")
# Delete each of the paths
for path in processor_paths:
response = GenericViewSet(f'/api/task/{task_id}/file/').delete(
path)
print(f"Deleted processor path: {path}")
time.sleep(0.05) # Sleep for a bit to avoid throttling
def upload_separate_surfaces(self):
system_settings = bpy.context.scene.ODS_CFD.system
task_id = system_settings.task_id
print("Writing refinement regions")
separate_objects = [
obj for obj in bpy.context.visible_objects
if self._separate_stl(obj)
]
for obj in separate_objects:
name = foamUtils.formatObjectName(obj.name)
print(f" - Writing separate stl region: {name}")
geometry = io.StringIO()
asciiSTLExport.writeObjectsToFile(geometry, objects=[obj])
geometry.seek(0)
print(" -- Cast refinement regions and cell sets to STL format")
response = GenericViewSet(
f'/api/task/{task_id}/file/foam/constant/triSurface/{name}.stl/'
).update(None, geometry.read().encode('utf8'), raw=True)
def get_or_create_task(self,
project_id,
name,
parent_task_id,
dependent_on=None,
config=None):
query_params = {
'name': name,
'parent': parent_task_id,
}
if dependent_on is not None:
query_params.update({'dependent_on': dependent_on})
if not config:
config = {'task_type': 'empty'}
action_task = GenericViewSet(
f'/api/project/{project_id}/task/').get_or_create(
query_params, {'config': config}, create=True)
return action_task
def upload_setset(self):
system_settings = bpy.context.scene.ODS_CFD.system
task_id = system_settings.task_id
cellset_objects = [
obj for obj in bpy.context.visible_objects
if obj.ODS_CFD.mesh.makeCellSet
]
if not cellset_objects:
return None
setset_str = ""
for obj in cellset_objects:
name = foamUtils.formatObjectName(obj.name)
setset_str += f'cellSet {name} new surfaceToCell "constant/triSurface/{name}.stl" ((0 0 0)) true true false 0 0\n'
print(f" - Writing separate setSet file:\n\n {setset_str}")
response = GenericViewSet(
f'/api/task/{task_id}/file/foam/cellSets.setSet/').update(
None, setset_str.encode('utf8'), raw=True)
def upload_geometry(self):
system_settings = bpy.context.scene.ODS_CFD.system
task_id = system_settings.task_id
geometry_objects = [
obj for obj in bpy.context.visible_objects
if not self._separate_stl(obj)
]
geometry = io.StringIO()
asciiSTLExport.writeObjectsToFile(geometry, objects=geometry_objects)
geometry.seek(0)
print("Cast geometry to STL format")
response = GenericViewSet(
f'/api/task/{task_id}/file/foam/constant/triSurface/cfdGeom.stl/'
).update(None, geometry.read().encode('utf8'), raw=True)
self.upload_separate_surfaces()
return response
def run_mesh_pipeline(self):
system_settings = bpy.context.scene.ODS_CFD.system
project_id = system_settings.project_id
task_id = system_settings.task_id
# Get or create an empty setup task
setup_task = self.get_or_create_task(project_id, 'setup', task_id)
# Get or create the mesh task
mesh_task = self.get_or_create_task(project_id,
'mesh',
task_id,
dependent_on=setup_task['uid'])
commands = [
'blockMesh', "snappyHexMesh -overwrite",
"reconstructParMesh -constant -mergeTol 1e-6",
"!checkMesh -writeSets vtk",
"foamToSurface -constant surfaceMesh.obj"
]
cellset_objects = [
obj for obj in bpy.context.visible_objects
if obj.ODS_CFD.mesh.makeCellSet
]
if len(cellset_objects) > 0:
commands.append("!setSet -batch zones.setSet")
# Get or create the mesh task
mesh_task = GenericViewSet(f'/api/project/{project_id}/task/').update(
mesh_task['uid'], {
'status': "pending",
'config': {
'task_type': 'cfd',
'cmd': 'pipeline',
'cpus': [i for i in system_settings.decompN],
'commands': commands
}
})
return mesh_task
def write_solver_files(self):
system_settings = bpy.context.scene.ODS_CFD.system
solver_properties = bpy.context.scene.ODS_CFD.solver
project_id = system_settings.project_id
task_id = system_settings.task_id
# Get or create an empty setup task
setup_task = self.get_or_create_task(project_id, 'setup', task_id)
# the action to create the CFD files
setup_task = GenericViewSet(f'/api/project/{project_id}/task/').update(
setup_task['uid'], {
'status': "pending",
'config': {
'task_type': 'magpy',
'cmd': 'cfd.io.tasks.write_solution',
'solution': solver_properties.to_json()
}
})
return setup_task
def probe_selected(self):
system_settings = bpy.context.scene.ODS_CFD.system
solver_properties = bpy.context.scene.ODS_CFD.solver
postproc_properties = bpy.context.scene.ODS_CFD.postproc
project_id = system_settings.project_id
task_id = system_settings.task_id
# Get or create an empty setup task
sub_task = self.get_or_create_task(project_id, 'PostProcess', task_id)
fields = [
i.strip() for i in postproc_properties.probe_fields.split(',')
]
config = {
'task_type': 'cfd',
"cmd": "pipeline",
"case_dir": f"{postproc_properties.task_case_dir}/",
"cpus": [1, 1, 1],
"commands":
["write_sample_set", "!postProcess -func internalCloud"],
"fields": fields,
"sets": get_sets_from_selected()
}
# the action to create the CFD files
setup_task = GenericViewSet(f'/api/project/{project_id}/task/').update(
sub_task['uid'], {
'status': "pending",
'config': config
})
# Start polling for the probe results
if postproc_properties.auto_load_probes:
self.load_selected_probes()
return setup_task