LEC-5422-Public/Tomcat-Blender
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added opening to surface creation in 4D

Browse Source
This commit is contained in:
fische_r
2022-09-28 15:53:50 +02:00
parent ee33b89046
commit dbea5c6de4
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
01_extract_surfaces.py
View File

@@ -12,6 +12,7 @@ import trimesh
import numpy as np
from joblib import delayed, Parallel
import socket
from scipy import ndimage
host = socket.gethostname()
@@ -68,10 +69,12 @@ class mesh_maker:
with open(stlpath, 'w+') as file:
file.write(stl)
def time_4D_stl(self, ts, phase=1, phasename = 'phase_1'):
def time_4D_stl(self, ts, phase=1, phasename = 'phase_1', clean=False):
im = self.dyn_data['segmented'].sel(time=ts)==phase
if clean: im = ndimage.binary_opening(im)
if np.any(im):
im[:,:,:2] = 0
im[:,:,-3:] = 0
@@ -109,7 +112,7 @@ class mesh_maker:
self.dyn_data.close()
self.dyn_data.close()
def run2(self, phase, name):
def run2(self, phase, name, clean):
# laod the data
self.load_data()
# TODO:close ncfile if error
@@ -125,7 +128,7 @@ class mesh_maker:
steps = self.timesteps
if self.parallel:
Parallel(n_jobs=self.njobs, temp_folder=temp_folder)(delayed(self.time_4D_stl)(ts, phase, name) for ts in steps)
Parallel(n_jobs=self.njobs, temp_folder=temp_folder)(delayed(self.time_4D_stl)(ts, phase, name, clean) for ts in steps)
else:
for ts in steps:
if not ts == self.ref_ts: