FullSubjectSim2.m

%% Monte Carlo Simulation of Ambiguous Decision Making
% Produces estimate of response distribution during stochastic choices in ambiguous decision making task

% Indirect Method
% Initialize variables

tic
h = waitbar(0,'Simulating...');

v = [5,8,20,50,125];
p = [.13,.25,.38,.5,.75];
a = [.25,.5,.75];

V = repmat(v,1,32)';
P_temp = ones(20,1)*p;
P = [P_temp(:);.5*ones(60,1)];
A_temp = ones(20,1)*a;
A = [zeros(100,1);A_temp(:)];

trialnumber = length(V);
simnumbs = 100000;

for sims = 1:simnumbs
    
    Cresults = zeros(8,5);
    
    while or(sum(sum(Cresults==1)) > 36, sum(sum(Cresults==0)) > 36)

        Choices = zeros(trialnumber,1);

        %alpha =  .473 + .212.*randn(1);        %Gains Normal distribution of measured Risk Preferences
        alpha = .97 + .75*randn(1);             %Loss
        alpha(alpha<0) = 0;
        %beta = .359 + .426.*randn(1);          %Gains Normal distribution of measured Ambiguity Preferences
        beta = -.48 + 1.01*randn(1);            %Loss

        noise = 0+.3*randn(1);
    
    
    
        for i = 1:trialnumber;
            %if (5^alpha)*(1+noise*randn(1)) > (P(i)*(1-beta*(A(i)/2)))*(V(i)^alpha)*(1+noise*randn(1));      % Subjective Value under ambiguity model Gain
            if (-((5)^alpha))*(1+noise*randn(1)) > (P(i)*(1-beta*(A(i)/2)))*(-(V(i)^alpha))*(1+noise*randn(1));      % Subjective Value under ambiguity model Loss
                Choices(i) = 0;
            else
                Choices(i) = 1; 
            end
        end


        Risk = Choices(1:100);
        Ambiguity = Choices(101:160);

        %Vresults = [mean(Risk(and(V(1:100)==5,P(1:100)==.13))),mean(Risk(V(1:100)==8)),mean(Risk(V(1:100)==20)),mean(Risk(V(1:100)==50)),mean(Risk(V(1:100)==125));...
            %mean(Ambiguity(V(101:160)==5)),mean(Ambiguity(V(101:160)==8)),mean(Ambiguity(V(101:160)==20)),mean(Ambiguity(V(101:160)==50)),mean(Ambiguity(V(101:160)==125))];
        %disp(Vresults)

        Cresults = zeros(8,5);

        for valindex = 1:5
            for riskindex = 1:5
                Cresults(riskindex,valindex) = mean(Choices(V==v(valindex) & P==p(riskindex) & A==0));
            end
            for ambindex = 6:8
                Cresults(ambindex,valindex) = mean(Choices(and(V==v(valindex),A==a(ambindex-5))));
            end
        end
    end
    
    
    MultiSubChoices(:,:,sims) = [V,P,A,Choices];
    MultiSubResults(:,:,sims) = Cresults;
    MultiParams(sims,:) = [alpha,beta];
    MultiNoise(sims) = noise;
    MultiFOSD(sims) = sum(Cresults(:,1))/.25;
    
    waitbar(sims/simnumbs,h)
end


MultiFOSD = (32-MultiFOSD)';
MultiNoise = MultiNoise';
Subindex = 1:simnumbs;

%close(h)
toc



%% Fit Model

tic
base = 0;
    
model = 'ambigNrisk'

fixed_valueP = 5;
fixed_valueN = -5;

fixed_prob = 1;
    
% Initialize parameters (this could be smarter)
if strcmp(model,'ambiguity')
   b0 = [-1 1]';
elseif strcmp(model,'power')
   b0 = [-1 0.8];
elseif strcmp(model, 'ambigNrisk')
   b0 = [-1 0.5 0.5]'; %slope, beta, alpha
elseif strcmp(model, 'ambigNriskFixSlope')
   b0 = [0.5 0.5]'; %beta, alpha
end


for s = 1:length(Subindex)
    
    
    %ACTUAL START
    
    %MultiSubChoices(:,:,sims) = [V,P,A,Choices];
    
    % get rid of no choice trials
    choice = MultiSubChoices(:,4,s);
    %vA = MultiSubChoices(:,1,s);            %Gains
    vA = -1*MultiSubChoices(:,1,s);        %Loss
    AL = MultiSubChoices(:,3,s);
    pA = MultiSubChoices(:,2,s);

    % separate fit for pos
    choiceP = choice(find(vA>0));
    vP = vA(find(vA>0));
    pP = pA(find(vA>0));
    AP = AL(find(vA>0));
    vFP = fixed_valueP * ones(length(choiceP),1);
    pFP = fixed_prob * ones(length(choiceP),1);
    
    [info,p] = fit_ambigNrisk_model_Constrained(choiceP,vFP,vP,pFP,pP,AP,model,b0,base);

    if strcmp(model,'ambigNrisk')
        slopeP = info.b(1);
        aP = info.b(3);
        bP = info.b(2);
    elseif strcmp(model,'ambigNriskFixSlope')
        slopeP = -1;
        aP = info.b(2);
        bP = info.b(1);
    end
    r2P = info.r2;
 
    % separate fit for neg
    choiceN = choice(find(vA<0));
    vN = vA(find(vA<0));
    pN = pA(find(vA<0));
    AN = AL(find(vA<0));
    vFN = fixed_valueN * ones(length(choiceN),1);
    pFN = fixed_prob * ones(length(choiceN),1);
    
    [info,p] = fit_ambigNrisk_model_Constrained(~choiceN,-vFN,-vN,pFN,pN,AN,model,b0,base);

    if strcmp(model,'ambigNrisk')
        slopeN = info.b(1);
        aN = info.b(3);
        bN = info.b(2);
    elseif strcmp(model,'ambigNriskFixSlope')
        slopeN = -1;
        aN = info.b(2);
        bN = info.b(1);
    end
    r2N = info.r2;
    
    %ParamsResults(s,:) = [aP,bP,slopeP,r2P];
    ParamsResults(s,:) = [aN,bN,slopeN,r2N];
    
    waitbar(s/length(Subindex),h,['Fitting... ' num2str(s)])

end

close(h)    
toc