%% Plotting function for visualizing kymograph for all roaches within a single trial

function out=roachKymograph(kym,thetaData)

data=thetaData;
data(:,sum(data)==0)=[];

figure;
hold on;

kymIm=zeros(size(kym,1),size(kym,2),3);
kymIm(:,:,3)=~kym; kymIm(:,:,2)=~kym; kymIm(:,:,1)=~kym; kymIm=kymIm*0.5;

image(kymIm);

data(data==0)=NaN;
data=(data+pi())*360/(2*pi())/2;  % rescale angles to match values of angles

out=zeros(size(data))+NaN;

for i=1:size(data,2)
   
    colorTemp=hsv2rgb([i/size(data,2) .8 .9]);
    
    scatter(data(:,i),1:size(data,1),8,'MarkerFaceColor',colorTemp,'MarkerEdgeColor','none');
    
    
    
%     
%     for j=1:(size(data,1)-1)
%         if ~isnan(data(j,i))
%             
%        out(j,i)= kym(j,ceil(data(j,i)));
%        if ~isnan(data(j+1,i))
%            if data(j+1,i)-data(j,i)>0
%                scatter(data(j,i),j,22,'MarkerFaceColor','none','MarkerEdgeColor',[1 0 0]);
%            elseif data(j+1,i)-data(j,i)<0
%                scatter(data(j,i),j,22,'MarkerFaceColor','none','MarkerEdgeColor',[0 1 0]);
%            end
%        end
%        
%         end
%     end
    
end

set(gca,'YDir','Reverse');

set(gca,'XTickLabel',[0:40:360]);
ylim([0 size(data,1)])
xlim([0 size(kym,2)])