Added the 3-D polar plot and documentation

ilent2 · Feb 19, 2019 · b349529 · b349529
1 parent 1c90ae8
commit b349529
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/+ott/Bsc.m b/+ott/Bsc.m
@@ -439,10 +439,18 @@ function visualiseFarfieldSphere(beam, varargin)
       % Generate a spherical surface visualisation of the far-field
       %
       % beam.visualiseFarfieldSphere(phi)
+      %
+      % Optional named arguments:
+      %   npts      num   Number of points to use for sphere surface
+      %   normalise bool  If intensity values should be normalised to 1
+      %   type      str   Type of visualisation to produce.
+      %       sphere    (default) draw a sphere with intensity as color
+      %       3dpolar   scale the radius by the intensity
 
       p = inputParser;
       p.addParameter('npts', 100);
       p.addParameter('normalise', false);
+      p.addParameter('type', 'sphere');
       p.parse(varargin{:});
 
       % build grid:
@@ -460,12 +468,21 @@ function visualiseFarfieldSphere(beam, varargin)
       if p.Results.normalise
         I = I ./ max(I(:));
       end
+
+      switch p.Results.type
+        case 'sphere'
+          surf(x,y,z,I,'facecolor','interp','edgecolor','none');
+        case '3dpolar'
+          surf(I.*x,I.*y,I.*z,I,'facecolor','interp','edgecolor','none');
+        otherwise
+          error('Unknown visualisation type');
+      end
 
-      surf(x,y,z,I,'facecolor','interp','edgecolor','none')
       zlabel('Z');
       xlabel('X');
       ylabel('Y');
       view(50, 20);
+      axis equal;
     end
 
     function [im, data] = visualiseFarfield(beam, varargin)