clear;clc


% The data in medsy01 medication matrix are not real, but for illustration purposes only.
% If you need to rerun this part you will need to request permission to
% access the data directly from the ABCD, and redownload the medsy01 matrix
% with your own subject list.
load('C:\Users\Konstantinos Tsirlis\Dropbox\My Desktop\Mourao - Miranda\ABCD results\ABCD Github Repo\medsy01_reduced.mat')
load('C:\Users\Konstantinos Tsirlis\Dropbox\My Desktop\Mourao - Miranda\ABCD results\ABCD Github Repo\patientOrderFinal.mat')



dummy_var = medsy01(medsy01.eventname=="baseline_year_1_arm_1",:);


dummy = table2array(dummy_var(:,5:end));

adhd_subj_ids = [];

% for j = 1:size(dummy,2)
% j
% for i = 1:size(dummy,1)
% 
% if contains(dummy(i,j),'dderall') || contains(dummy(i,j),'oncerta') || contains(dummy(i,j),'vekeo') || contains(dummy(i,j),'ocalin')...
%          || contains(dummy(i,j),'uillivant') || contains(dummy(i,j),'italin') || contains(dummy(i,j),'trattera') || contains(dummy(i,j),'yvanse')...
%          || contains(dummy(i,j),'ethylphenidate') || contains(dummy(i,j),'omoxetine') || contains(dummy(i,j),'isdexamfetamine') || contains(dummy(i,j),'imesylate')
% adhd_subj_ids = [adhd_subj_ids; i];
% end
% end
% end


% The identifiers in here are not real.
load('C:\Users\Konstantinos Tsirlis\Dropbox\My Desktop\Mourao - Miranda\ABCD results\ABCD Github Repo\adhd_subj_ids.mat')

adhd_subj_ids = unique(adhd_subj_ids);
adhd_subj = dummy_var(adhd_subj_ids,:);


[~,ia,ib] = intersect(patientOrderFinal,adhd_subj.subjectkey);
adhd_patientOrderFinal = patientOrderFinal(ia,:);
adhd_patientOrderFinal_idx = ia;





% In both cases below change the directory to where the main model is,
% abcdAll_noA\framework\spls_fair20-5_fair_modeA_pub_noAnom_new_common_bmi\.
load('C:\Users\Konstantinos Tsirlis\Dropbox\My Desktop\Mourao - Miranda\ABCD results\abcdAll_noA\framework\spls_fair20-5_fair_modeA_pub_noAnom_new_common_bmi\outmat_1.mat','oteid')
cd('C:\Users\Konstantinos Tsirlis\Dropbox\My Desktop\Mourao - Miranda\ABCD results\abcdAll_noA\framework\spls_fair20-5_fair_modeA_pub_noAnom_new_common_bmi\res\level4')
load('P.mat');
P = reshape(P,[11288,2]);

PP = P;
PP(adhd_patientOrderFinal_idx,:) = [];
PPP = P(adhd_patientOrderFinal_idx,:);
corr(P(oteid,1),P(oteid,2))
oteid_PP = oteid;
oteid_PP(adhd_patientOrderFinal_idx,:) = [];
corr(PP(oteid_PP,1),PP(oteid_PP,2))
oteid_PPP = oteid(adhd_patientOrderFinal_idx);
corr(PPP(oteid_PPP,1),PPP(oteid_PPP,2))


NoMedSubjects = PP;
MedSubjects = PPP;
NoMedSubjects_psychosocial = [mean(PP(:,2)) std(PP(:,2))]
MedSubjects_psychosocial = [mean(PPP(:,2)) std(PPP(:,2))]

[~,p_psychosocial,~,tstat_psychosocial] = ttest2(PP(:,2),PPP(:,2),'Vartype','unequal')
[~,p_brain,~,tstat_brain] = ttest2(PP(:,1),PPP(:,1),'Vartype','unequal')