%Return subset of subjects based on the presence of curry file or EEGset

%type and absence of another EEGset type

%

%INPUTS

% present - string indicating a set type to draw subjects from; e.g.

% 'preart'; default: 'curry'

% missing - string indicating a set type that has not been created yet for

% the subject; e.g., 'postart'; [] will return all subjects

% specified by the present input

% main_dir - The main study directory

%

% Possible inputs for present and missing:

% - 'curry'

% - 'raw'

% - 'ICA'

% - 'preart'

% - 'postart'

% - 'erp'

% - 'bad_epochs'

%

% Default with no input is equivalent to get_subset('curry')

%

%Author: Eric Fields

%Version Date: 6 August 2023

%Copyright (c) 2023, Eric Fields

%All rights reserved.

%This code is free and open source software made available under the terms

%of the 3-clause BSD license:

%https://opensource.org/licenses/BSD-3-Clause

function subs_subset = get_subset(present, missing, main_dir)

%Set defaults for missing arguments

if nargin < 3

main_dir = pwd;

end

if nargin < 2

missing = [];

end

if ~nargin

present = 'curry';

end

%Check inputs

allowed_inputs = {'curry', 'raw', 'ICA', 'preart', 'postart', 'erp', 'bad_epochs'};

if ~any(strcmpi(present, allowed_inputs))

error('Input is incorrect. See >>help get_subset');

end

if ~any(strcmpi(missing, allowed_inputs)) && ~isempty(missing)

error('Input is not correct. See >>help get_subset');

end

%Get subject files

curry_files = get_files(fullfile(main_dir, 'curry'), '.cdt');

eeg_files = get_files(fullfile(main_dir, 'EEGsets'), '.set');

ica_files = get_files(fullfile(main_dir, 'ICA'), 'w.txt');

bad_epoch_files = get_files(fullfile(main_dir, 'ICA'), 'bad_epochs.csv');

erp_files = get_files(fullfile(main_dir, 'ERPsets'), '.erp');

erp_files = erp_files(~contains(erp_files, 'Hz'));

%Get subject IDs for present

if strcmpi(present, 'curry')

subs1 = cellfun(@(x) x(1:end-4), curry_files, 'UniformOutput', false);

elseif strcmpi(present, 'ICA')

subs1 = cellfun(@(x) x(1:end-9), ica_files, 'UniformOutput', false);

elseif strcmpi(present, 'ERP')

subs1 = cellfun(@(x) x(1:end-4), erp_files, 'UniformOutput', false);

elseif strcmpi(present, 'bad_epochs')

subs1 = cellfun(@(x) x(1:end-15), bad_epoch_files, 'UniformOutput', false);

else

subs1 = eeg_files(not(cellfun('isempty', strfind(eeg_files, present)))); %#ok<STRCL1>

subs1 = cellfun(@(x) x(1:end-(length(present)+5)), subs1, 'UniformOutput', false);

end

%Get subject IDs for missing

if ~isempty(missing)

if strcmpi(missing, 'curry')

subs2 = cellfun(@(x) x(1:end-4), curry_files, 'UniformOutput', false);

%Get rid of repeats (mutliple curry files per subject)

subs2 = cellfun(@(x) x(1:12), subs2, 'UniformOutput', false);

subs2 = unique(subs2);

elseif strcmpi(missing, 'ICA')

subs2 = cellfun(@(x) x(1:end-9), ica_files, 'UniformOutput', false);

elseif strcmpi(missing, 'ERP')

subs2 = cellfun(@(x) x(1:end-4), erp_files, 'UniformOutput', false);

elseif strcmpi(missing, 'bad_epochs')

subs2 = cellfun(@(x) x(1:end-15), bad_epoch_files, 'UniformOutput', false);

else

subs2 = eeg_files(not(cellfun('isempty', strfind(eeg_files, missing)))); %#ok<STRCL1>

subs2 = cellfun(@(x) x(1:end-(length(missing)+5)), subs2, 'UniformOutput', false);

end

end

%Get subset

if isempty(missing)

subs_subset = subs1;

else

subs_subset = subs1(~ismember(subs1, subs2));

end

end