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

Fri 29 Nov 2024 04:30:50 PM CET

Browse Source
This commit is contained in:
sbosse 2024-11-29 16:32:25 +01:00
parent c39ce78b78
commit 2867e27291
1 changed files with 198 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

198
src/SimNDT/run_setup/snapshots.py Normal file
View File

@ -0,0 +1,198 @@
__author__ = 'Miguel Molero'
import os
import sys
import numpy as np
# import matplotlib.pyplot as plt
from scipy.misc import imsave
from scipy.io import savemat
import math
class SnapShots:
def __init__(self, Enable=False, Extension='.png', Step=200, Filename=None, File_path = None, dB=60, Color=0,
Field = 0,
enableFields=False,
enableSignals=False,
enableImages=False,
enableNumPy=False,
enableVolume=False,
enableView=False,
sensorShape = [],
sensorPlacement = [],
sensorSize = 0):
self.IsEnable = Enable
self.Extension = Extension
self.Step = Step
self.Filename = Filename
self.File_path = File_path
self.DB = dB
self.Color = Color
self.Field = Field
self.enableFields = enableFields
self.enableSignals = enableSignals
self.enableImages = enableImages
self.enableNumPy = enableNumPy
self.enableVolume = enableVolume
self.enableView = enableView
self.sensorShape = sensorShape
self.sensorPlacement = sensorPlacement
self.sensorSize = sensorSize
self.volF = []
self.volS = []
self.volP = []
print("SnapShots Field="+str(Field)+" enableNumPy="+str(enableNumPy)+" enableVolume="+str(enableVolume)+ " enableSignals="+str(enableSignals))
# if self.File_path does not exist, create it
if not os.path.exists(self.File_path):
os.makedirs(self.File_path)
def save_fields(self, Fx, Fy, n):
xn = Fx.shape[1]
yn = Fx.shape[0]
print('['+str(int(n / self.Step))+'] Saving fields x/y '+Label+' shape=('+str(xn)+','+str(yn)+')')
if self.enableVolume:
self.volF.append(np.array([Fx,Fy]))
elif self.enableNumPy:
FILE = self.Filename + ("-fields-%05d" % (int(n / self.Step)))
FILE_PATH = os.path.join(self.File_path, FILE)
np.save(FILE_PATH,np.array([Fx,Fy]))
print("Fields saved in numpy format at: ", FILE)
else:
dict = {}
dict['Vx'] = Fx
dict['Vy'] = Fy
FILE = self.Filename + ("-fields-%05d" % (int(n / self.Step))) + ".mat"
savemat(FILE, dict)
# @blab+
def save_signal(self, D , n, Label):
self.Label = Label
xn = D.shape[1]
yn = D.shape[0]
print('['+str(int(n / self.Step))+'] Saving signal '+Label+' shape=('+str(xn)+','+str(yn)+') sensorShape='+str(self.sensorShape))
dx = 1 # sensor delta
dy = 1
ox = 0 # offset
oy = 0
if len(self.sensorShape) > 0:
# size of sensor matrix
sx = self.sensorShape[0]
sy = self.sensorShape[1]
S = np.zeros([sy,sx])
if len(self.sensorPlacement):
p=self.sensorPlacement
ox=p[0]
dx=p[1]
oy=p[2]
dy=p[3]
else:
ox=dx=round(xn/(sx+1.0))
oy=dy=round(yn/(sy+1.0))
for x in range(0,sx):
for y in range(0,sy):
S[y,x]=D[oy+y*dy,ox+x*dx]
else:
S = D
if self.enableVolume:
self.volS.append(np.copy(S))
elif self.enableNumPy:
FILE = self.Filename + ("-signal-"+Label+"-%05d" % (int(n / self.Step)))
FILE_PATH = os.path.join(self.File_path, FILE)
np.save(FILE_PATH,S)
else:
FILE = self.Filename + ("-signal-"+Label+"-%05d" % (int(n / self.Step))) + ".mat"
savemat(FILE, S)
def save_power(self, Vx, Vy, n):
S = np.sqrt(Vx ** 2 + Vy ** 2)
xn = S.shape[1]
yn = S.shape[0]
print('['+str(int(n / self.Step))+'] Saving power '+Label+' shape=('+str(xn)+','+str(yn)+')')
if self.enableVolume:
self.volP.append(np.copy(S))
elif self.enableNumPy:
FILE = self.Filename + ("-power-%05d" % (int(n / self.Step)))
FILE_PATH = os.path.join(self.File_path, FILE)
np.save(FILE_PATH,S)
else:
FILE = self.Filename + ("-power-%05d" % (int(n / self.Step))) + ".mat"
FILE_PATH = os.path.join(self.File_path, FILE)
savemat(FILE_PATH, S)
def save_fig(self, SV, n, idx=None):
SV += self.DB
ind = np.nonzero(SV < 0)
SV[ind] = 0
SV /= np.max(SV)
if self.Color == 0:
cmap = plt.get_cmap('jet')
elif self.Color == 1:
cmap = plt.get_cmap('gray')
try:
_resize = False
M, N = np.shape(SV)
if M >= 2000:
_resize = True
if N >= 2000:
_resize = True
if _resize:
SVV = SV[::3, ::3]
rgba_img = cmap(SVV)
else:
SVV = SV[::2, ::2]
rgba_img = cmap(SVV)
rgb_img = np.delete(rgba_img, 3, 2)
if idx is not None:
FILE = self.Filename + "_insp%d_" % idx + str(int(n / self.Step)) + self.Extension
else:
FILE = self.Filename + ("%05d" % (int(n / self.Step))) + self.Extension
FILE_PATH = os.path.join(self.File_path, FILE)
imsave(FILE_PATH, rgb_img)
except:
raise ValueError("Too large Snapshots Size!!!!!!!!")
def save_vol(self):
if len(self.volF) > 0:
if self.enableNumPy:
FILE = self.Filename + '-fields'
FILE_PATH = os.path.join(self.File_path, FILE)
vol=np.array(self.volF)
print("Saving numpy S-volume "+str(vol.shape)+" in "+FILE_PATH)
np.save(FILE_PATH,vol)
self.volF=[]
if len(self.volS) > 0:
if self.enableNumPy:
FILE = self.Filename + '-signal-'+self.Label
FILE_PATH = os.path.join(self.File_path, FILE)
vol=np.array(self.volS)
print("Saving numpy S-volume "+str(vol.shape)+" in "+FILE_PATH)
np.save(FILE_PATH,vol)
self.volS=[]
if len(self.volP) > 0:
if self.enableNumPy:
FILE = self.Filename + '-power'
FILE_PATH = os.path.join(self.File_path, FILE)
vol=np.array(self.volP)
print("Saving numpy S-volume "+str(vol.shape)+" in "+FILE_PATH)
np.save(FILE_PATH,vol)
self.volP=[]