source: trunk/src/set_col_vec.m @ 767

%'set_col_vec': % sets the color code for vectors depending on a scalar and input parameters (used for plot_field)
%-----------------------------------------------------------------------
%function [colorlist,col_vec,minC,ColCode1,ColCode2,maxC]=colvec(colcode,vec_C)
%-----------------------------------------------------------------------
%OUTPUT
%colorlist(nb,3); %list of nb colors
%col_vec, size=[length(vec_C),3)];%list of color indices corresponding to vec_C
%minC, maxC: min and max of vec_C
%ColCode1, ColCode2: absolute threshold in vec_C corresponding to colcode.ColCode1 and colcode.ColCode2
%INPUT
% colcode: struture setting the colorcode for vectors
            % colcode.CName: 'ima_cor','black','white',...
            % colcode.ColorCode ='black', 'white', 'rgb','brg', '64 colors'
            % colcode.CheckFixVecColor =0; thresholds scaling relative to min and max, =1 fixed thresholds
            % colcode.MinVec; min 
            % colcode.MaxVec; max
            % colcode.ColCode1: first threshold for rgb, relative to min (0) and max (1)
            % colcode.ColCode2: second threshold for rgb, relative to min (0) and max (1), 
            % rmq: we need min <= ColCode1 <= ColCode2 <= max, otherwise
            % ColCode1 and ColCode2 are adjusted to the bounds
% vec_C: matlab vector representing the scalar setting the color
function [colorlist,col_vec,colcode_out]=set_col_vec(colcode,vec_C)
col_vec=ones(size(vec_C));%all vectors at color#1 by default

if ~isstruct(colcode),colcode=[];end;
colcode_out=colcode;%default
if isempty(vec_C) || ~isnumeric(vec_C)
    colorlist=[0 0 1]; %blue
    return
end

%% uniform color plot
check_multicolors=0;
%default input parameters
if ~isfield(colcode,'ColorCode') || isempty(colcode.ColorCode)
    colorlist=[0 0 1]; %blue
else
    if strcmp(colcode.ColorCode,'black')
        colorlist(1,:)=[0 0 0];%black
    elseif strcmp(colcode.ColorCode,'white')
        colorlist(1,:)=[1 1 1];%white
    else
        check_multicolors=1;
    end
end

if check_multicolors
    if (isfield(colcode,'CheckFixVecColor') && isequal(colcode.CheckFixVecColor,1))
        minC=colcode.MinVec;
        maxC=colcode.MaxVec;
    else
        minC=min(vec_C);
        maxC=max(vec_C);
    end
    colcode_out.MinVec=minC;
    colcode_out.MaxVec=maxC;
    if strcmp(colcode.ColorCode,'rgb')|| strcmp(colcode.ColorCode,'bgr')% 3 color representation
        if  isfield(colcode,'ColCode1')
            colcode_out.ColCode1=colcode.ColCode1;
        else
            colcode_out.ColCode1=minC+(maxC-minC)/3;%default
        end
        if  isfield(colcode,'ColCode2')
            colcode_out.ColCode2=colcode.ColCode2;
        else
            colcode_out.ColCode2=minC+2*(maxC-minC)/3;%default
        end
        colorlist(2,:)=[0 1 0];%green
        col_vec(vec_C < colcode_out.ColCode1)=1;% vectors with vec_C smaller than ColCode1 set to the first color (r or b)
        col_vec((vec_C >= colcode_out.ColCode1) & (vec_C < colcode_out.ColCode2))=2;% select green vectors
        col_vec(vec_C >= colcode_out.ColCode2)=3;
        if strcmp(colcode.ColorCode,'rgb')
            colorlist(1,:)=[1 0 0];%red
            colorlist(3,:)=[0 0 1];%blue
        else
            colorlist(1,:)=[0 0 1];%blue
            colorlist(3,:)=[1 0 0];%red
        end
    else
        colorjet=jet;% ususal colormap from blue to red
        sizlist=size(colorjet);
        indsel=ceil((sizlist(1)/64)*(1:64));
        colorlist(:,1)=colorjet(indsel,1);
        colorlist(:,2)=colorjet(indsel,2);
        colorlist(:,3)=colorjet(indsel,3);
        sizlist=size(colorlist);
        nblevel=sizlist(1);
        col2_1=maxC-minC;
        col_vec=1+floor(nblevel*(vec_C-minC)/col2_1);
        col_vec=col_vec.*(col_vec<= nblevel)+nblevel*(col_vec >nblevel);% take color #nblevel at saturation
        col_vec=col_vec.*(col_vec>= 1)+  (col_vec <1);% take color #1 for values below 1
    end
end