sbosse/simndt2b
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Mon 15 Sep 09:44:42 CEST 2025

Browse Source
This commit is contained in:
sbosse 2025-09-15 09:46:31 +02:00
parent 04616cdeb3
commit 08bd4de1b5
1 changed files with 22 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/SimNDT/run_setup/snapshots.py
View File

@ -18,6 +18,7 @@ class SnapShots:
enableImages=False,
enableMaterial=False,
enableNumPy=False,
enableCsv=False,
enableVolume=False,
enableView=False,
sensorShape = [],
@ -42,6 +43,7 @@ class SnapShots:
self.enableImages = enableImages
self.enableMaterial = enableMaterial
self.enableNumPy = enableNumPy
self.enableCsv = enableCsv
self.enableVolume = enableVolume
self.enableView = enableView
self.sensorShape = sensorShape
@ -51,7 +53,7 @@ class SnapShots:
self.volS = []
self.volP = []
print("SnapShots Field="+str(Field)+" enableNumPy="+str(enableNumPy)+" enableVolume="+str(enableVolume)+ " enableSignals="+str(enableSignals)+ " enableMaterial="+str(enableMaterial))
print("SnapShots Field="+str(Field)+" enableNumPy="+str(enableNumPy)+" enableCsv="+str(enableCsv)+" enableVolume="+str(enableVolume)+ " enableSignals="+str(enableSignals)+ " enableMaterial="+str(enableMaterial))
# if self.File_path does not exist, create it
if not os.path.exists(self.File_path):
os.makedirs(self.File_path)
@ -113,15 +115,15 @@ class SnapShots:
VT[ind] = 1e-30
FILE = self.Filename + "-materials-Rho"
FILE_PATH = os.path.join(self.File_path, FILE)
print('Saving simulation material distribution matrix ('+FILE+') ['+str(Rho.min())+','+str(Rho.max())+']')
print('Saving simulation material distribution matrix ('+FILE+') ['+str(Rho.shape[0])+','+str(Rho.shape[1])+'] (min='+str(Rho.min())+',max='+str(Rho.max())+')')
np.save(FILE_PATH,Rho)
FILE = self.Filename + "-materials-VL"
FILE_PATH = os.path.join(self.File_path, FILE)
print('Saving simulation material distribution matrix ('+FILE+') ['+str(VL.min())+','+str(VL.max())+']')
print('Saving simulation material distribution matrix ('+FILE+') ['+str(VL.shape[0])+','+str(VL.shape[1])+'] (min='+str(VL.min())+',max='+str(VL.max())+')')
np.save(FILE_PATH,VL)
FILE = self.Filename + "-materials-VT"
FILE_PATH = os.path.join(self.File_path, FILE)
print('Saving simulation material distribution matrix ('+FILE+') ['+str(VT.min())+','+str(VT.max())+']')
print('Saving simulation material distribution matrix ('+FILE+') ['+str(VT.shape[0])+','+str(VT.shape[1])+'] (min='+str(VT.min())+',max='+str(VT.max())+')')
np.save(FILE_PATH,VT)
@ -161,6 +163,10 @@ class SnapShots:
FILE = self.Filename + ("-signal-"+Label+"-%05d" % (int(n / self.Step)))
FILE_PATH = os.path.join(self.File_path, FILE)
np.save(FILE_PATH,S)
elif self.enableCsv:
FILE = self.Filename + ("-signal-"+Label+"-%05d" % (int(n / self.Step)))+".csv"
FILE_PATH = os.path.join(self.File_path, FILE)
np.savetxt(FILE_PATH, S, delimiter=",")
else:
FILE = self.Filename + ("-signal-"+Label+"-%05d" % (int(n / self.Step))) + ".mat"
savemat(FILE, S)
@ -242,6 +248,17 @@ class SnapShots:
print("Saving numpy S-volume "+str(vol.shape)+" in "+FILE_PATH)
np.save(FILE_PATH,vol)
self.volS=[]
if self.enableCsv:
FILE = self.Filename + '-signal-'+self.Label+".csv"
FILE_PATH = os.path.join(self.File_path, FILE)
vol=np.array(self.volS)
if vol.shape[1]==1:
vol=vol.reshape(vol.shape[0],vol.shape[1])
elif vol.shape[2]==1:
vol=vol.reshape(vol.shape[0],vol.shape[2])
print("Saving reduced CSV S-volume "+str(vol.shape)+" in "+FILE_PATH)
np.savetxt(FILE_PATH, vol, delimiter=",")
self.volS=[]
if len(self.volP) > 0:
if self.enableNumPy: