Skip to content

Commit 0628a05

Browse files
author
Varvara Chiliaeva
committed
rename modules and classes for 2D version
1 parent d6ceffa commit 0628a05

File tree

4 files changed

+4124
-7
lines changed

4 files changed

+4124
-7
lines changed

scripts/main_SPC2D_2arms.py

+7-7
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,9 @@
55
performed after the acquisition and not in "real-time".
66
"""
77

8-
from spas.acquisition import init_2arms, setup_cam, AcquisitionParameters, setup_2arms, setup, acquire, acquire_2arms, snapshot, disconnect_2arms, captureVid, displaySpectro, setup_tuneSpectro, change_patterns
9-
from spas.metadata import MetaData, func_path, save_metadata_2arms
10-
from spas.reconstruction import reconstruction_hadamard
8+
from spas.acquisition_SPC2D import init_2arms, setup_cam_ids, AcquisitionParameters_SPC2D, setup_2arms, setup, acquire, acquire_2arms, snapshot, disconnect_2arms, captureVid, displaySpectro, setup_tuneSpectro, change_patterns
9+
from spas.metadata_SPC2D import MetaData_SPC2D, func_path, save_metadata_2arms
10+
from spas.reconstruction_SPC2D import reconstruction_hadamard_2D
1111
from spas.reconstruction_nn import ReconstructionParameters, setup_reconstruction, reorder_subsample
1212
from spas.noise import load_noise
1313
from spas.visualization import snapshotVisu, displayVid, plot_reco_without_NN, plot_reco_with_NN, extract_ROI_coord
@@ -29,7 +29,7 @@
2929
camPar = captureVid(camPar)
3030
#%% Set Camera Parameters
3131
# It is advice to execute this cell twice to take into account the parameter changement
32-
camPar = setup_cam(camPar,
32+
camPar = setup_cam_ids(camPar,
3333
pixelClock = 474, # Allowed values : [118, 237, 474] (MHz)
3434
fps = 220, # FrameRate boundary : [1 - No value(depend of the AOI size)]
3535
Gain = 0, # Gain boundary : [0 100]
@@ -73,7 +73,7 @@
7373
if 'mask_index' not in locals(): mask_index = []; x_mask_coord = []; y_mask_coord = [] # execute "mask_index = []" to not apply the mask
7474

7575
if all_path.aborted == False:
76-
metadata = MetaData(
76+
metadata = MetaData_SPC2D(
7777
output_directory = all_path.subfolder_path,
7878
pattern_order_source = 'C:/openspyrit/spas/stats/pattern_order_' + scan_mode + '_' + str(Np) + 'x' + str(Np) + '.npz',
7979
pattern_source = 'C:/openspyrit/spas/Patterns/' + scan_mode + '_' + str(Np) + 'x' + str(Np),
@@ -89,7 +89,7 @@
8989
force_change = False)
9090
except: pass
9191

92-
acquisition_parameters = AcquisitionParameters(pattern_compression = pattern_compression, pattern_dimension_x = Np, pattern_dimension_y = Np,
92+
acquisition_parameters = AcquisitionParameters_SPC2D(pattern_compression = pattern_compression, pattern_dimension_x = Np, pattern_dimension_y = Np,
9393
zoom = zoom, xw_offset = xw_offset, yh_offset = yh_offset, mask_index = mask_index,
9494
x_mask_coord = x_mask_coord, y_mask_coord = y_mask_coord)
9595

@@ -130,7 +130,7 @@
130130
save_metadata_2arms(metadata, DMD_params, spectrometer_params, camPar, acquisition_parameters)
131131
#%% Hadamard Reconstruction
132132
Q = wh.walsh2_matrix(Np)
133-
GT = reconstruction_hadamard(acquisition_parameters, 'walsh', Q, spectral_data, Np)
133+
GT = reconstruction_hadamard_2D(acquisition_parameters, 'walsh', Q, spectral_data, Np)
134134
plot_reco_without_NN(acquisition_parameters, GT, all_path)
135135
#%% Neural Network setup (executed it just one time)
136136
network_param = ReconstructionParameters(

0 commit comments

Comments
 (0)