WiscPlan_v2/WiscPlanPhotonkV125/matlab_frontend/nrrd2geometry.m

function [Geometry patient_dir] = nrrd2geometry
    
    %% -----===== Get CT files =====-----
    [IMG_in, IMG_path, ~] = uigetfile(['C:\010-work\003_localGit\WiscPlan_v2\data\', '*.nrrd'], 'Select CT image');
    
%     IMG_in = 'test_ct.nrrd';
%     IMG_path = 'C:\010-work\003_localGit\WiscPlan_v2\PhantomData\';
    
    total_num_steps = 4;
    hWaitbar = waitbar(0);
    
    % geometry_file important parts:
    % Geometry.ROIS - actually only to get the binary mask of the target
    % Geometry.data - CT image in HU
    % Geometry.voxel_size - self explanatory
    % Geometry.start - self explanatory
    
    % these are derived from above:
    % +Geometry.rhomw - CT image with values in relative water density 
    % +Geometry.Smw - very similair to rhomw
    % +Geometry.Fmw2
    % +Geometry.BTV
    % +Geometry.ring
    
    if IMG_in==0 % no file selected, no change to CT data
        warning('No file selected!')
    else 
        [Geometry.data, meta]=nrrdread([IMG_path, IMG_in]);
    end
    
    % Geometry.rhomw
    Geometry.rhomw = CT2dens(Geometry.data, 'pinn');
    Geometry.rhomw(Geometry.rhomw < 0.0012) = 0.0012;
    [Geometry.Smw Geometry.Fmw2] = dens2mstp(Geometry.rhomw);
    
    % Geometry.voxel_size
%     VoxSizeString = meta.spacedirections;
%     C=textscan(VoxSizeString(2:end-1), '%f', 9,'Delimiter',{',',') ('});
%     C= C{1};
%     Geometry.voxel_size = [C(1), C(5), C(9)];
    Geometry.voxel_size = meta.spacedirections;
    
    % Geometry.start
%     StartString = meta.spaceorigin;
%     C=textscan(StartString(2:end-1), '%f', 3,'Delimiter',',');
%     Geometry.start=C{1};
%     Geometry.start=Geometry.start';
    Geometry.start=meta.spaceorigin;
    
    % Geometry.ROIS
    
    %% -----===== Get mask/contour files =====-----
    [TRGT_in, TRGT_path, ~] = uigetfile(['C:\010-work\003_localGit\WiscPlan_v2\CDP_data\','*.nrrd'], 'Select file with Target');
%     TRGT_in = 'test_seg.nrrd';
%     TRGT_path = 'C:\010-work\003_localGit\WiscPlan_v2\PhantomData\';
    
    if TRGT_in==0 % no file selected, no change to CT data
        warning('No file selected!')
    else 
        [TRGT_img, meta]=nrrdread([TRGT_path, TRGT_in]);
    end
    Geometry.ROIS{1}.ind = find(TRGT_img>0);
    Geometry.ROIS{1}.name= 'Target';

    
        %% -----===== Save geometry files =====-----
    patient_dir = uifile('getdir', 'Save the patient data to directory');
%     patient_dir = 'C:\010-work\003_localGit\WiscPlan_v2\data\PatientData\';
    
    % make directories
    mkdir(patient_dir);
    mkdir(fullfile(patient_dir, 'beamlet_batch_files'));
    mkdir(fullfile(patient_dir, 'geometry_files'));
    mkdir(fullfile(patient_dir, 'matlab_files'));
    mkdir(fullfile(patient_dir, 'opt_input'));
    mkdir(fullfile(patient_dir, 'opt_output'));

    %% -----===== BTV and Ring creation =====-----
    waitbar(1/total_num_steps, hWaitbar, 'Step 2: Assign target and BTV margin');
    ROI_names = cellfun(@(c)c.name, Geometry.ROIS, 'UniformOutput', false);

    [target_idx okay] = listdlg('ListString', ROI_names, ...
        'SelectionMode', 'single', 'Name', 'Target Selection', ...
        'PromptString', 'Please select the target ROI. ');
    if okay ~= 1
        msgbox('Plan creation aborted');
        delete(hWaitbar);
        return;
    end

    [BTV_margin_answer] = inputdlg({sprintf('Please enter the BTV margin (cm):\n(default 0.6 cm or 1 sigma, enter 0 to skip)')}, ...
        'BTV margin specification', 1, {'0.6'});
    if isempty(BTV_margin_answer)
        BTV_margin = 0.6;
    else
        BTV_margin = str2double(BTV_margin_answer{1});
    end

    % target_idx and BTV_margin are set. Expand PTV to BTV
    PTVmask = false(size(Geometry.rhomw));
    % for target_idx = target_indices
            PTVmask(Geometry.ROIS{target_idx}.ind) = 1;
    % end
    if BTV_margin > 0
        if exist('BTV_margin', 'var') && BTV_margin >= min(Geometry.voxel_size)
            bwD = bwdistsc(PTVmask, Geometry.voxel_size);
            Geometry.BTV = bwD <= BTV_margin;
        end
    end

    % Create btv
    Geometry.ROIS{end+1} = Geometry.ROIS{end};
    Geometry.ROIS{end}.name = 'BTV';
    Geometry.ROIS{end}.num_curves = 0;
    Geometry.ROIS{end}.curves = {};
    Geometry.ROIS{end}.ind = find(Geometry.BTV);
    Geometry.ROIS{end}.visible = false;

    % Create ring
    [ring_margin_answer] = inputdlg({sprintf('Please enter the ring margin (cm):')}, ...
        'Ring margin specification', 1, {'1'});
    if isempty(ring_margin_answer)
        ring_margin = 1;
    else
        ring_margin = str2double(ring_margin_answer{1});
    end
    bwD = bwdistsc(PTVmask, Geometry.voxel_size);
    Geometry.Ring = bwD <= ring_margin; % default ring radius 3 cm
    Geometry.Ring = xor(Geometry.Ring, PTVmask);
    Geometry.ROIS{end+1} = Geometry.ROIS{end};
    Geometry.ROIS{end}.name = 'Ring';
    Geometry.ROIS{end}.num_curves = 0;
    Geometry.ROIS{end}.curves = {};
    Geometry.ROIS{end}.ind = find(Geometry.Ring);
    Geometry.ROIS{end}.visible = false;

    
    
    waitbar(2.33/total_num_steps, hWaitbar, 'Step 3: Save matlab geometry files');
    % Save matlab geometry file
    save(fullfile(patient_dir, 'matlab_files', 'Geometry.mat'), 'Geometry');

    waitbar(2.66/total_num_steps, hWaitbar, 'Step 3: Save raw geometry files');
    % Write binary geometry files
    fid = fopen(fullfile(patient_dir, 'geometry_files', 'rhomw.bin'), 'w');
    fwrite(fid, Geometry.rhomw, 'single');
    fclose(fid);
    
    fid = fopen(fullfile(patient_dir, 'geometry_files', 'Smw.bin'), 'w');
    fwrite(fid, Geometry.Smw, 'single');
    fclose(fid);

    fid = fopen(fullfile(patient_dir, 'geometry_files', 'Fmw2.bin'), 'w');
    fwrite(fid, Geometry.Fmw2, 'single');
    fclose(fid);

    fid = fopen(fullfile(patient_dir, 'geometry_files', 'target_mask.bin'), 'w');
    fwrite(fid, Geometry.BTV, 'single');
    fclose(fid);


    % hWaitbar = waitbar(1/total_num_steps, 'Step 4, Create optimization geometry');

    delete(hWaitbar);
    waitfor(msgbox(['Plan geometry created successfully in ' '"' patient_dir '"']));
    
    
    

end