[ReA ImA]=meshgrid(-10:0.02:10,10:-0.02:-10);

A=ReA+i*ImA;
magA=abs(A);

%%%%CHOOSE WHICH PART OF THE QUESTION YOU WANT TO DO, THEN RUN SCRIPT
part=4;
%%%%%%%%%%%
 colormap('default');
if (part==1)

    % (i)
    theta=3*pi/4;%3*pi/4;

    phi=7*pi/4;
    N=12;
    Q=(abs(cos(theta./2).*exp(-abs(A).^2/2)+exp(i.*phi)*sin(theta./2)*exp(-abs(A).^2./2).*A.^N./sqrt(factorial(N)))).^2;

   

    imagesc(Q);
   
elseif (part==2)
    %(ii)
    B=3;
    Q=abs(0.5*(exp(-abs(A-B).^2)+exp(-abs(A+B).^2)+exp(-1*(abs(A).^2+abs(B).^2-conj(A).*B+conj(B).*A))+exp(-1*(abs(A).^2+abs(B).^2+conj(A).*B-conj(B).*A))));
  
    colormap('grays')
    imagesc(log(Q));

elseif (part==3)
    %(iii)
    phi=0;
    N=100;
    Q=zeros(size(ReA));
    for k=1:N
        Q=Q+exp(i*k*phi).*A.^k/sqrt(factorial(k));
        k
    end
    
    Q=abs(Q).^2;
    Q=Q.*exp(-abs(A).^2)./N;
    
  

    imagesc(Q);
    
elseif(part==4)
    %(iv)
    E=4;
    figure(1)
    Q=exp(-abs(A).^2)./(cosh(E)).*exp(tanh(E)*0.5*(A.^2+conj(A).^2));
    

    colormap('default');
    imshow(Q);
    
    %imshow(max(max(Q))-Q);
elseif(part==5)
    %(iv)
    N=40;
    B=4;
    E=pi/128;
    Q=zeros(size(ReA));
    t=124;
    
    for n=0:N
        Q=Q+(conj(A).*B).^n.*exp(-0.5*(abs(A).^2+abs(B).^2)).*exp(i*E*n*(n-1)*t)/factorial(n);
        n
    end
    
    Q=abs(Q).^2;
    
    figure(1)
    colormap('default');
 
    imagesc(Q);
    str1(1) = {['t=' int2str(t)]};
    text(40,70,str1,'Color','w','FontSize',24);
end