Nlevels_no_dopler_with_z

%% field propagation
figure(1)
subplot(1,3,1); imagesc(z_1, t_1, I1_out_1); colorbar
xlabel('z')
ylabel('t')
zlabel('I_1')
title('I_1')
subplot(1,3,2); imagesc(z_1, t_1, I2_out_1); colorbar
xlabel('z')
ylabel('t')
zlabel('I_2')
title('I_2')
subplot(1,3,3); imagesc(z_1, t_1, I3_out_1); colorbar
xlabel('z')
ylabel('t')
zlabel('I_3')
title('I_3')




%% fields before and after the cell
figure(2)
subplot(1,3,1);
plot( ...
	t_1,I1_out_1(:,1),'-;before;',  ...
	t_1,I1_out_1(:,end), '-;after;' ...
	)
xlabel('t')
ylabel('I_1')
title('I_1 propagation')
subplot(1,3,2);
plot( ...
	t_1,I2_out_1(:,1),'-;before;',  ...
	t_1,I2_out_1(:,end), '-;after;' ...
	)
xlabel('t')
ylabel('I_2')
title('I_2 propagation')
subplot(1,3,3);
plot( ...
	t_1,I3_out_1(:,1),'-;before;',  ...
	t_1,I3_out_1(:,end), '-;after;' ...
	)
xlabel('t')
ylabel('I_3')
title('I_3 propagation')


%% all density matrix elements in one plot
% diagonal populations, 
% upper triangle real part of coherences, 
% lower diagonal imaginary part of coherences
figure(3)
subplot(4,4,1); imagesc (z_2, t_2, r11_out_2);  caxis([0,1]); colorbar
xlabel('z')
ylabel('t')
zlabel('rho_{11}')
title('rho_{11}')
subplot(4,4,6); imagesc (z_2, t_2, r22_out_2);  caxis([0,1]); colorbar
xlabel('z')
ylabel('t')
zlabel('rho_{22}')
title('rho_{22}')
subplot(4,4,11); imagesc (z_2, t_2, r33_out_2);  caxis([0,1]); colorbar
xlabel('z')
ylabel('t')
zlabel('rho_{33}')
title('rho_{33}')
subplot(4,4,16); imagesc (z_2, t_2, r44_out_2); caxis([0,1]);  colorbar
xlabel('z')
ylabel('t')
zlabel('rho_{44}')
title('rho_{44}')
% real parts of coherences
subplot(4,4,2); imagesc(z_2, t_2, r12_re_out_2); colorbar
xlabel('z')
ylabel('t')
zlabel('Real(rho_{12})')
title('Real(rho_{12})')
subplot(4,4,3); imagesc(z_2, t_2, r13_re_out_2); colorbar
xlabel('z')
ylabel('t')
zlabel('Real(rho_{13})')
title('Real(rho_{13})')
subplot(4,4,4); imagesc(z_2, t_2, r14_re_out_2); colorbar
xlabel('z')
ylabel('t')
zlabel('Real(rho_{14})')
title('Real(rho_{14})')
subplot(4,4,7); imagesc(z_2, t_2, r23_re_out_2); colorbar
xlabel('z')
ylabel('t')
zlabel('Real(rho_{23})')
title('Real(rho_{23})')
subplot(4,4,8); imagesc(z_2, t_2, r24_re_out_2); colorbar
xlabel('z')
ylabel('t')
zlabel('Real(rho_{24})')
title('Real(rho_{24})')
subplot(4,4,12); imagesc(z_2, t_2, r34_re_out_2); colorbar
xlabel('z')
ylabel('t')
zlabel('Real(rho_{34})')
title('Real(rho_{34})')
% imaginary parts of coherences
subplot(4,4,5); imagesc(z_2, t_2, r12_im_out_2); colorbar
xlabel('z')
ylabel('t')
zlabel('Imag(rho_{12})')
title('Imag(rho_{12})')
subplot(4,4,9); imagesc(z_2, t_2, r13_im_out_2); colorbar
xlabel('z')
ylabel('t')
zlabel('Imag(rho_{13})')
title('Imag(rho_{13})')
subplot(4,4,10); imagesc(z_2, t_2, r23_im_out_2); colorbar
xlabel('z')
ylabel('t')
zlabel('Imag(rho_{23})')
title('Imag(rho_{23})')
subplot(4,4,13); imagesc(z_2, t_2, r14_im_out_2); colorbar
xlabel('z')
ylabel('t')
zlabel('Imag(rho_{14})')
title('Imag(rho_{14})')
subplot(4,4,14); imagesc(z_2, t_2, r24_im_out_2); colorbar
xlabel('z')
ylabel('t')
zlabel('Imag(rho_{24})')
title('Imag(rho_{24})')
subplot(4,4,15); imagesc(z_2, t_2, r34_im_out_2); colorbar
xlabel('z')
ylabel('t')
zlabel('Imag(rho_{34})')
title('Imag(rho_{34})')