Changeset 1744 for issm/trunk/src/m/classes/public/plot/plot_quiver.m

Timestamp:

08/18/09 09:50:07 (16 years ago)

Author:

Mathieu Morlighem

Message:

Added quiver plot

File:

• issm/trunk/src/m/classes/public/plot/plot_quiver.m (modified) (1 diff)

issm/trunk/src/m/classes/public/plot/plot_quiver.m

@@ -1 @@
-function plot_quivervel(md,options_structure,width,i);
-%PLOT_QUIVERVEL - plot arrow field of 2d velocities
+function plot_quiver(x,y,u,v,options_structure),
+%PLOT_QUIVER - quiver plot with colors
+%
+%   to be perfected tomorrow
 %
 %   Usage:
-%      plot_quivervel(md,options_structure,width,i);
+%      plot_quiver(x,y,u,v,options_structure)
 %
-%   See also: PLOTMODEL
+%   Example:
+%      plot_quiver(md.x,md.y,md.vx,md.vy,options_structure);
 
-%process data and model
-[x y z elements is2d]=processmesh(md,options_structure);
-[vx isongrid isquiver]=processdata(md,md.vx,options_structure);
-[vy isongrid isquiver]=processdata(md,md.vy,options_structure);
+%get norm Min and Max
+Norm=sqrt(u.^2+v.^2);
+Min=min(Norm);
+Max=max(Norm);
 
-%plot mesh quivervel
-subplot(width,width,i); 
-                                        
-%quiver 2d 
-if ~isnan(options_structure.density)
-        x=x(1:options_structure.density:end);
-        y=y(1:options_structure.density:end);
-        vx=vx(1:options_structure.density:end);
-        vy=vy(1:options_structure.density:end);
+%process options: scaling factor?
+if isnan(options_structure.scaling),
+        arrow=0.40;
+else
+        arrow=options_structure.scaling;
 end
-quiver(x,y,vx,vy);
 
-%apply options
-if isnan(options_structure.title)
-        options_structure.title='Velocity vectors';
-end 
-if isnan(options_structure.colorbar)
-        options_structure.colorbar=0;
+%number of colors?
+if isnumeric(options_structure.colorlevels),
+        if isnan(options_structure.colorlevels),
+                numcolors=30;
+        else
+                numcolors=options_structure.colorlevels;
+        end
+        levels=round_ice(linspace(Min,Max,numcolors+1),2);
+else
+        levels=zeros(1,length(options_structure.colorlevels)+2);
+        levels(1)=Min;
+        for i=1:length(options_structure.colorlevels)
+                levels(i+1)=options_structure.colorlevels{i};
+        end
+        levels(end)=Max;
+        levels=sort(unique(levels));
+        numcolors=length(levels)-1;
 end
-applyoptions(md,[],options_structure);
+
+%set the colormap 
+if numcolors==2;
+        %blue and red
+        c=[0 0 1;1 0 0];
+elseif numcolors==3,
+        %blue yellow and red
+        c=[0 0 1;1 1 0;1 0 0];
+else
+        %let jet choose
+        c=colormap(jet(numcolors));
+end
+
+%Scale data
+delta=((min(x)-max(x))^2+(min(y)-max(y))^2)/numel(x);
+u=arrow*sqrt(delta)*u./Norm;
+v=arrow*sqrt(delta)*v./Norm;
+
+%loop over the number of colors
+hold on
+h=[];
+for i=1:numcolors
+        pos=find( (Norm>=levels(i)) & (Norm<=levels(i+1)) );
+        hprime=quiver(x(pos),y(pos),u(pos),v(pos),'Color',c(i,:),'ShowArrowHead','on','AutoScale','off');
+        h=[h;hprime];
+end
+
+%take care of colorbar
+if  0 & ~strcmpi(options_structure.colorbar,'off'),
+        for i=1:length(levels),
+                scalevalues(i)=levels(i);
+                scalestring=[scalestring; sprintf('%8.4g',levels(i))];
+        end
+        set(colorbar,'YTickLabel',scalestring,'YTick',scalevalues);
+        error('debug')
+end