create_dimensionality_reduction_colored_movie.m

function create_dimensionality_reduction_colored_movie(results_struct, varargin)
%CREATE_MOVIE_USING_PCA_COLORS Create movie where components are colored
%based on PCA components and intensity is based on DFF.
spatial = results_struct.spatial;
DFF = results_struct.DFF;

S=reshape(spatial,[size(spatial,1)*size(spatial,2),size(spatial,3)]);
S=S>0;

sz=[size(spatial,1),size(spatial,2)];
rescaled = mat2gray(DFF,[0 .25]);

if isempty(varargin) || varargin{1}=="pca"
    type = 'PCA';
    coeff = pca(rescaled');
elseif varargin{1}=="tsne"
    type = 'TSNE';
    coeff = tsne(rescaled, 'NumDimensions',3);
end
coeff = 255*mat2gray(coeff(:,1:3));
[~,filename,~] = fileparts(results_struct.filename);

output_name = [results_struct.dr filesep filename type 'Recoloredtest.tiff'];
for frame=1:size(DFF,2)
    S_scaled=S.* rescaled(:,frame)';
    %RGB = [mat2gray(coeff(:,1))'; mat2gray(coeff(:,2))'; mat2gray(coeff(:,3))'].^2;
    %RGB = RGB./repmat(sum(RGB,1), 3,1);
    RGB=[coeff(:,1)'; coeff(:,2)'; coeff(:,3)'];
    S_scaled_RGB = [S_scaled*RGB(1,:)' S_scaled*RGB(2,:)' S_scaled*RGB(3,:)']/255;%sqrt([S_scaled*RGB(1,:)' S_scaled*RGB(2,:)' S_scaled*RGB(3,:)']);
    S_scaled_RGB = reshape(full(S_scaled_RGB), [sz 3]);
    
    if frame==1
        imwrite(S_scaled_RGB, output_name)
    else
        imwrite(S_scaled_RGB, output_name, 'writemode', 'append')
    end
end