mirror of
https://github.com/c-sooyoung/fold_slice.git
synced 2026-09-17 23:59:11 +09:00
initial commit
This commit is contained in:
@@ -0,0 +1,301 @@
|
||||
%LOAD_PTYCHO_ML_RECONS Load ML reconstructions from h5 or mat file and return it as
|
||||
% structure, single dataset or directly into the workspace.
|
||||
% An additional argument can be passed to select subsections of the data.
|
||||
% Loading single datasets is only supported for at least 2 output
|
||||
% arguments.
|
||||
% Created by YJ based on PSI's function.
|
||||
%
|
||||
|
||||
function [object, probe, dx_spec] = load_aps_ML_recons( filename_with_path)
|
||||
|
||||
if ~ischar(filename_with_path)
|
||||
error('First argument has to be string')
|
||||
end
|
||||
|
||||
filename_with_path = utils.abspath(filename_with_path);
|
||||
%disp(filename_with_path)
|
||||
if ~exist(filename_with_path, 'file')
|
||||
error('Could not find reconstruction file %s', filename_with_path)
|
||||
end
|
||||
|
||||
object = load(filename_with_path,'object');
|
||||
object = object.object;
|
||||
|
||||
probe = load(filename_with_path,'probe');
|
||||
probe = probe.probe;
|
||||
parameter = load(filename_with_path,'p');
|
||||
dx_spec = parameter.p.dx_spec; %pixel size
|
||||
|
||||
%% legacy code: read recon from processed h5 files
|
||||
%{
|
||||
object_r = h5read(filename_with_path,'/object_r');
|
||||
object_i = h5read(filename_with_path,'/object_i');
|
||||
object = object_r + 1i*object_i;
|
||||
|
||||
probe_r = h5read(filename_with_path,'/probe_r');
|
||||
probe_i = h5read(filename_with_path,'/probe_i');
|
||||
probe = probe_r + 1i*probe_i;
|
||||
|
||||
% load pixel size
|
||||
dx_spec = h5read(filename_with_path,'/dx_spec');
|
||||
|
||||
% load projection angle
|
||||
%ang = h5read(filename_with_path,'/angle');
|
||||
%}
|
||||
|
||||
end
|
||||
|
||||
%% old PSI code, too complex for APS data
|
||||
%{
|
||||
function varargout = load_ptycho_ML_recons( filename_with_path, varargin )
|
||||
|
||||
import io.HDF.hdf5_load
|
||||
|
||||
varargout = {};
|
||||
|
||||
if ~ischar(filename_with_path)
|
||||
error('First argument has to be string')
|
||||
end
|
||||
|
||||
filename_with_path = utils.abspath(filename_with_path);
|
||||
|
||||
if ~exist(filename_with_path, 'file')
|
||||
error('Could not find reconstruction file %s', filename_with_path)
|
||||
end
|
||||
|
||||
if nargin > 1
|
||||
switch varargin{1}
|
||||
case {'pr'; 'probe'; 'probes'}
|
||||
section = 'probe';
|
||||
case {'ob'; 'obj'; 'objects'}
|
||||
section = 'object';
|
||||
otherwise
|
||||
section = varargin{1};
|
||||
end
|
||||
else
|
||||
section = 'full';
|
||||
end
|
||||
|
||||
if ~nargout
|
||||
output = 0;
|
||||
elseif nargout >=2
|
||||
output = 2;
|
||||
else
|
||||
output = 1;
|
||||
end
|
||||
|
||||
function assign_struct(val, val_name)
|
||||
switch output
|
||||
case 1
|
||||
varargout{1}.(val_name) = val;
|
||||
case 2
|
||||
varargout{end+1} = val;
|
||||
otherwise
|
||||
assignin('base', val_name, val);
|
||||
end
|
||||
end
|
||||
|
||||
function assign_val(struc)
|
||||
switch output
|
||||
case 1
|
||||
varargout{1} = struc;
|
||||
|
||||
case 2
|
||||
if isfield(struc, 'object')
|
||||
varargout{end+1} = struc.object;
|
||||
end
|
||||
if isfield(struc, 'probe')
|
||||
varargout{end+1} = struc.probe;
|
||||
end
|
||||
if isfield(struc, 'p')
|
||||
varargout{end+1} = struc.p;
|
||||
end
|
||||
|
||||
otherwise
|
||||
fn = fieldnames(struc);
|
||||
for ii=1:length(fn)
|
||||
assignin('base', fn{ii}, struc.(fn{ii}))
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
% check if it is a .mat file or a .cxs file
|
||||
[~, ~, ext] = fileparts(filename_with_path);
|
||||
switch ext
|
||||
%{
|
||||
case '.mat'
|
||||
switch section
|
||||
case 'recon'
|
||||
S = load(filename_with_path, 'object', 'probe');
|
||||
assign_val(S);
|
||||
|
||||
case 'full'
|
||||
S = load(filename_with_path);
|
||||
assign_val(S);
|
||||
|
||||
case 'object'
|
||||
S = load(filename_with_path, 'object');
|
||||
assign_val(S);
|
||||
|
||||
case 'probe'
|
||||
S = load(filename_with_path, 'probe');
|
||||
size(S)
|
||||
assign_val(S);
|
||||
|
||||
case 'p'
|
||||
S = load(filename_with_path, 'p');
|
||||
assign_val(S);
|
||||
|
||||
otherwise
|
||||
error('Unknown data section %s', section);
|
||||
end
|
||||
%}
|
||||
case {'.cxs','.h5'}
|
||||
%{
|
||||
if io.HDF.hdf5_dset_exists(filename_with_path, 'object', '/reconstruction', true)
|
||||
h5_path = '/reconstruction';
|
||||
else
|
||||
h5_path = '';
|
||||
end
|
||||
%}
|
||||
% reconstruction
|
||||
switch section
|
||||
%{
|
||||
case 'recon'
|
||||
% load object
|
||||
h = hdf5_load(filename_with_path, [h5_path '/object']);
|
||||
assign_struct(load_data_cell(h), 'object');
|
||||
% load probe
|
||||
h = hdf5_load(filename_with_path, [h5_path '/probes']);
|
||||
assign_struct(load_data_cell(h), 'probe');
|
||||
case 'full'
|
||||
% load object
|
||||
h = hdf5_load(filename_with_path, [h5_path '/object']);
|
||||
assign_struct(load_data_cell(h), 'object');
|
||||
% load probe
|
||||
h = hdf5_load(filename_with_path, [h5_path '/probes']);
|
||||
assign_struct(load_data_cell(h), 'probe');
|
||||
% load p
|
||||
p = convert2p(hdf5_load(filename_with_path, '/reconstruction/p', '-c'));
|
||||
if io.HDF.hdf5_dset_exists(filename_with_path, 'meta_all', '/measurement', true)
|
||||
p.meta = hdf5_load(filename_with_path, '/measurement/meta_all', '-c');
|
||||
elseif io.HDF.hdf5_dset_exists(filename_with_path, 'spec_all', '/measurement', true)
|
||||
p.meta = hdf5_load(filename_with_path, '/measurement/spec_all', '-c');
|
||||
end
|
||||
assign_struct(p, 'p');
|
||||
%}
|
||||
case 'object'
|
||||
% load object
|
||||
%h = hdf5_load(filename_with_path, [h5_path '/object']);
|
||||
%assign_struct(load_data_cell(h), 'object');
|
||||
object_r = h5read(filename_with_path,'/object_r');
|
||||
object_i = h5read(filename_with_path,'/object_i');
|
||||
object = object_r + 1i*object_i;
|
||||
assign_struct(object, 'object');
|
||||
|
||||
case 'probe'
|
||||
% load probe
|
||||
%h = hdf5_load(filename_with_path, [h5_path '/probes']);
|
||||
%assign_struct(load_data_cell(h), 'probe');
|
||||
probe_r = h5read(filename_with_path,'/probe_r');
|
||||
probe_i = h5read(filename_with_path,'/probe_i');
|
||||
probe = probe_r + 1i*probe_i;
|
||||
|
||||
assign_struct(probe, 'probe');
|
||||
|
||||
case 'p'
|
||||
% load p
|
||||
p = convert2p(hdf5_load(filename_with_path, '/reconstruction/p', '-c'));
|
||||
if io.HDF.hdf5_dset_exists(filename_with_path, 'meta_all', '/measurement', true)
|
||||
p.meta = hdf5_load(filename_with_path, '/measurement/meta_all', '-c');
|
||||
elseif io.HDF.hdf5_dset_exists(filename_with_path, 'spec_all', '/measurement', true)
|
||||
p.meta = hdf5_load(filename_with_path, '/measurement/spec_all', '-c');
|
||||
end
|
||||
assign_struct(p, 'p');
|
||||
otherwise
|
||||
error('Unknown data section %s', section);
|
||||
end
|
||||
|
||||
|
||||
|
||||
otherwise
|
||||
error('Unknown ptycho datatype %s.', ext)
|
||||
end
|
||||
|
||||
|
||||
|
||||
end
|
||||
|
||||
function tmp = load_data_cell(h)
|
||||
|
||||
fn = fieldnames(h);
|
||||
num_end = str2double(subsref(strsplit(fn{1}, '_'), struct('type', '{}', 'subs',{{length(strsplit(fn{1},'_'))}})));
|
||||
if length(fn)==2 && (strcmpi(fn{1}, 'i') || strcmpi(fn{1}, 'r'))
|
||||
tmp = permute(h.r + 1i*h.i, [2,1,3,4]);
|
||||
elseif isnumeric(num_end) && ~isnan(num_end)
|
||||
for ii=1:length(fn)
|
||||
if isstruct(h.(fn{ii}))
|
||||
tmp{ii} = load_data_cell(h.(fn{ii}));
|
||||
else
|
||||
if isnumeric(h.(fn{ii}))
|
||||
tmp{ii} = double(h.(fn{ii}));
|
||||
else
|
||||
tmp{ii} = h.(fn{ii});
|
||||
end
|
||||
end
|
||||
end
|
||||
% tmp = h;
|
||||
else
|
||||
for ii=1:length(fn)
|
||||
if isstruct(h.(fn{ii}))
|
||||
tmp.(fn{ii}) = load_data_cell(h.(fn{ii}));
|
||||
else
|
||||
if isnumeric(h.(fn{ii}))
|
||||
tmp.(fn{ii}) = double(h.(fn{ii}));
|
||||
else
|
||||
tmp.(fn{ii}) = h.(fn{ii});
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
end
|
||||
|
||||
function tmp = convert2p(h)
|
||||
|
||||
fn = fieldnames(h);
|
||||
for ii=1:length(fn)
|
||||
if isstruct(h.(fn{ii}))
|
||||
h.(fn{ii}) = load_data_cell(h.(fn{ii}));
|
||||
elseif isnumeric(h.(fn{ii}))
|
||||
h.(fn{ii}) = double(h.(fn{ii}));
|
||||
else
|
||||
continue;
|
||||
end
|
||||
end
|
||||
tmp = h;
|
||||
|
||||
% object
|
||||
for ii=1:length(h.objects)
|
||||
tmp.object{ii} = permute(h.objects{ii}, [2 1 3 4]);
|
||||
end
|
||||
tmp = rmfield(tmp, 'objects');
|
||||
|
||||
% probes
|
||||
pr = tmp.probes;
|
||||
tmp.probes = [];
|
||||
for ii=1:length(pr)
|
||||
tmp.probes(:,:,ii,:) = permute(pr{ii}, [2 1 3 4]);
|
||||
end
|
||||
|
||||
% positions
|
||||
tmp.positions = transpose(tmp.positions);
|
||||
tmp.positions_real = transpose(tmp.positions_real);
|
||||
tmp.positions_orig = transpose(tmp.positions_orig);
|
||||
|
||||
% ctr
|
||||
tmp.ctr = transpose(tmp.ctr);
|
||||
|
||||
end
|
||||
%}
|
||||
Reference in New Issue
Block a user