%% Main code
% Prepare work space
clc

% Check for miscelaneous transcriptions in the work space
miscLS = who('-regexp', '^ls(([0-9]$)|($))');
newLS = CheckmiscLS(miscLS);
if ~isempty(newLS); ls = eval(newLS); end

% Search the workspace for ls and lss and check for the tracker
if ~ValidateWorkspace(ls, lss); return; end

% Update progress
if ~exist('remainingFiles', 'var'); remainingFiles = []; end
remainingFiles = FindRemainingFiles(ls, lss, remainingFiles);

% Transfer labels
lss = TransferLabels(ls, lss);

% Prompt user for Inits
reviewerInit = input('Please enter your initials: ', 's');

% Save latest transcription and progress
uisave({'lss', 'remainingFiles', 'reviewerInit'}, '*.mat')


%% Functions
function ls = CheckmiscLS(miscLS)

if isempty(miscLS); error('No valid ls transcription file was found in the workspace. Please check that the file has been exported from the application.'); end
if length(miscLS) == 1
    ls = string(miscLS); 
    fprintf('Variable %s will be used as the user transcription variable.\n', ls)
    return
end

fprintf(2, 'Warning: multiple potential ''ls'' files were found in the work space. \n\n')

nLS = length(miscLS);
lsNames = string(miscLS);
formatPat = '\t%i. %s\n';
formattedstring = compose(formatPat, (1:nLS)', lsNames);
listSet = join(formattedstring' , '');

validResp = 0:nLS;
while true
    fprintf('Select the most up to date ls file by inputing a number from 0 to %i\n', nLS)
    fprintf(listSet{1});
    userResp = input('Resp:');

    if any(ismember(validResp, userResp))
        fprintf('%s has been selected.\n', lsNames(userResp))
        break
    else
        fprintf('%i is an invalid response.\n', userResp)
    end
end

ls = lsNames(userResp);

end

function extStatus = ValidateWorkspace(ls, lss)

if ~exist("ls", "var")
    fprintf(2, 'No exported transcript found. Please export your transciription to the work space as ''ls''\n')
    extStatus = false;
    return
end

if ~exist("lss", 'var')
    fprintf(2, 'The original transcription file is missing, please ensure the work you have done is saved to ''ls'' and load in the original transcription\n');
    extStatus = false;
    return
end

extStatus = true;

end

function remainingFiles = FindRemainingFiles(ls, lss, remainingFiles)

% Find identical labels
nFiles = lss.NumMembers;
isUntouched = false(nFiles, 1);
for i = 1:nFiles
    isUntouched(i) = isequal(ls.Labels.Words{i}.ROILimits, lss.Labels.Words{i}.ROILimits);
end

% Account for previosuly edited files
if ~isempty(remainingFiles)
    wasPreviouslyUntouched = ~ismember(1:nFiles, remainingFiles.File_Pos);
    isUntouched(wasPreviouslyUntouched) = false;
end

% Prepare output
File_Pos = find(isUntouched);
memberNames = ls.getMemberNames;
File_Name = memberNames(isUntouched);
remainingFiles = table(File_Pos, File_Name);

end

function lss = TransferLabels(ls, lss)

% Clear lss labels
resetLabelValues(lss)

% Move ls labels to lss
nTrials = height(ls.Labels);
for iTrial = 1:nTrials
    curROI = ls.Labels.Words{iTrial}.ROILimits;
    curVal = ls.Labels.Words{iTrial}.Value;
    if isempty(curROI)
        continue
    end
    setLabelValue(lss, iTrial, 'Words', curROI, curVal)
end

end

%% For testing
% load('C:\Users\chris\Box\Nelson Lab Data\Other Data\NTAud Transcription\Transcript_OrigAudio_AudSentStim.mat', 'lss');
% test = load('C:\Users\chris\Box\Nelson Lab Data\Other Data\NTAud Transcription\Transcript_Version2_AudSent.mat');
% ls = test.lss;

% testpos = 6;
% resetLabelValues(lss, testpos)
% setLabelValue(lss, testpos, 'Words', ls.Labels.Words{testpos}.ROILimits, ls.Labels.Words{testpos}.Value)
