WiscPlan_v2/WiscPlanPhotonkV125/matlab_frontend/merge_beamlets.m

function B = merge_beamlets(num_batches, pat_dir)

% num_batches = 4;


B = [];
for k = 1:num_batches
    batch_doses = read_ryan_beamlets([pat_dir, '\batch_dose' num2str(k-1) '.bin']);
    B = [B batch_doses];
end
write_ryan_beamlets([pat_dir '\batch_dose_unshifted.bin'],B);

% apply the shift (0.5 vox up and left) to make dose match geometry
f = waitbar(0, 'Correcting beamlet shift');
for beam_i = 1:numel(B)
    new_dose_list = B{beam_i};
    waitbar(beam_i/numel(B),f, ['Correcting beamlet shift: ' num2str(beam_i)])
    
    tabula = zeros(B{1, beam_i}.y_count, B{1, beam_i}.x_count, B{1, beam_i}.z_count);
    tabula(B{1, beam_i}.non_zero_indices) = B{1, beam_i}.non_zero_values;
    
    shifted_img = imtranslate(tabula, [-0.5,-0.5,-0.5]);
    nonzero_idx = find(shifted_img>0.005*max(shifted_img(:)));

    B{beam_i}.non_zero_indices = nonzero_idx;
    B{beam_i}.non_zero_values = shifted_img(nonzero_idx);
end
close(f)

for k = 1:numel(B)
    B{k}.num = k-1;
end
write_ryan_beamlets([pat_dir '\batch_dose.bin'],B);

write_ryan_beamlets([pat_dir '\batch_dose_backup.bin'],B);