Automatically add zlabel for surface/3d plots (if present) about matlab2tikz HOT 16 CLOSED

% =========================================================================
% *** FUNCTION surfaceOpts
% =========================================================================
function [m2t,surfOpts,plotType] = surfaceOpts( m2t, handle )

  faceColor = get( handle, 'FaceColor');
  edgeColor = get( handle, 'EdgeColor');

  % Check for surf or mesh plot. Second argument in if-check corresponds to
  % default values for mesh plot in MATLAB.
  if strcmpi( faceColor, 'none') || ( strcmpi( edgeColor, 'flat' ) && isequal( faceColor, [1 1 1]))
      plotType = 'mesh';
  else
      plotType = 'surf';
  end

  surfOptions = cell(0);

  % Set opacity if FaceAlpha < 1 in MATLAB
  faceAlpha = get( handle, 'FaceAlpha');
  if faceAlpha ~= 1 && isnumeric( faceAlpha )
    surfOptions = appendOptions( surfOptions, sprintf( 'opacity=%g', faceAlpha ) );
  end

  if strcmpi( plotType, 'surf' )
    % Set shader for surface plot. 
    % TODO: find MATLAB equivalents for flat corner and flat mean  
    if strcmpi( edgeColor, 'none' ) && strcmpi( faceColor, 'flat' )
        surfOptions = appendOptions( surfOptions, sprintf( 'shader=flat' ) );
    elseif isnumeric( edgeColor) && strcmpi( faceColor, 'flat' )
        [ m2t, xEdgeColor ] = getColor( m2t, handle, edgeColor, 'patch' );
        % same as shader=flat,draw=\pgfkeysvalueof{/pgfplots/faceted color}
        surfOptions = appendOptions( surfOptions, sprintf( 'shader=faceted' ) );
        surfOptions = appendOptions( surfOptions, sprintf( 'draw=%s', xEdgeColor) );
    elseif strcmpi( edgeColor, 'none') && strcmpi( faceColor, 'interp' )
        surfOptions = appendOptions( surfOptions, sprintf( 'shader=interp') );
    else
        surfOptions = appendOptions( surfOptions, sprintf( 'shader=faceted' ) );
    end
    surfOpts = collapse( surfOptions , ',\n' );
  else % default for mesh plot is shader=flat
    surfOptions = appendOptions( surfOptions, sprintf( 'shader=flat' ) );
    surfOpts = collapse( surfOptions , ',\n' );
  end

end
% =========================================================================
% *** END FUNCTION surfaceOpts
% =========================================================================

% =========================================================================
% *** FUNCTION drawSurface
% =========================================================================
function [m2t,env] = drawSurface( m2t, handle )

    str = [];
    [m2t, opts, plotType] = surfaceOpts( m2t, handle );
    str = [ str, sprintf( [ '\n\\addplot3[%%\n%s,\n', opts ,']\ncoordinates{ \n' ], plotType ) ];

    dx = get(handle,'XData');
    dy = get(handle,'YData');
    dz = get(handle,'ZData');
    [col, row] = size(dz);

    % check, if surf plot is 'spectrogram' or 'surf' and run corresponding
    % algorithm.
    if isvector(dx)
        % - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        % plot is 'spectrogram'
        for i = 1:col
            for j = 1:row
                str = [ str, ...
                        sprintf('(%g,%g,%g)', dx(j), dy(i), dz(i,j) ) ];
            end
            % insert an empty line to tell Pgfplots about one row ending here
            str = [str, sprintf('\n\n')];
        end
        % - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    else
        % - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        % plot is 'surf'
        for i = 1:col
            for j = 1:row
                str = [ str, ...
                        sprintf('(%g,%g,%g)', dx(i,j), dy(i,j), dz(i,j) ) ];
            end
            % insert an empty line to tell Pgfplots about one row ending here
            str = [str, sprintf('\n\n')];
        end
        % - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    end %if-else

% TODO:
% - remove grids in spectrogram by either removing grid command
%   or adding: 'grid=none' from/in axis options
% - using a "real" colorbar instead of colorbar-png-flle
% - handling of huge data amounts in LaTeX.
% - correcting wrong colors

str = [str, sprintf('};\n\n')];
env = str;

end
% =========================================================================
% *** END FUNCTION drawSurface
% =========================================================================

% =========================================================================
% *** FUNCTION getColorMap
% =========================================================================
function [m2t,cmapOpts] = getColorMap( m2t, handle )

  cm = get( gcf, 'Colormap' );
  [M N] = size(cm);
  cmap = cell(0);
  for k = 1:M
      [m2t, xtempColor] = getColor(m2t, handle, cm(k,:), 'path');
      cmap = appendOptions( cmap, sprintf( 'color=(%s)', xtempColor ) );
  end
  cmapOpts = ['colormap={mymap}{' collapse( cmap , '; ') '}' ];

end
% =========================================================================
% *** END FUNCTION getColorMap
% =========================================================================

\begin{axis}[
    how to store options here??
]

[m2t,colorMap] = surfaceColorMap( m2t, handle );
env.options = appendOptions( env.options, colorMap );

function [ m2t, env ] = drawColorbar( m2t, handle, alignmentOptions)

  if ~isVisible( handle )
      return
  end

  % the actual contents of the TikZ file go here
  env = struct( 'name',     'axis', ...
                'comment',  'colorbar', ...
                'options',  [], ...
                'content',  [], ...
                'children', []  ...
              );
  % Setting the following to cell(0) straight away doesn't work unfortunately
  % as MATLAB(R) interprets structs with cell values as a cell array of structs.
  env.options  = cell(0);
  env.content  = cell(0);
  env.children = cell(0);

  if ~isfield( m2t, 'colorbarNo' )
      % Keep track of how many colorbars there are, to avoid
      % file name collision in case PNGs are used.
      m2t.colorbarNo = [];
  end

  % Assume that the parent axes pair is m2t.currentHandles.gca.
  try
      m2t.currentHandles.gca;
  catch
      error( [ 'm2t.currentHandles.gca not set although needed ', ...
               'by the color bar. The parent axes have not been printed yet.' ] ...
           )
  end

  % begin collecting axes options
  cbarOptions = cell( 0 );
  cbarOptions = [ cbarOptions, 'axis on top' ];

  % - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
  % set position, ticks etc. of the colorbar
  loc = get( handle, 'Location' );

  switch lower( loc )
      case { 'north', 'south', 'east', 'west' }
          userWarning( m2t, 'Don''t know how to deal with inner colorbars yet.' );
          return;

      case {'northoutside','southoutside'}
          if strcmpi(  loc , 'northoutside' )
              cbarOptions = [ cbarOptions 'colorbar horizontal' ...
                              'colorbar style={at={(0.5,1.1)}' ...     
                              'anchor=south' ...
                              'xticklabel pos=upper}' ];
          else
              cbarOptions = [ cbarOptions 'colorbar horizontal' ];
          end

      case {'eastoutside','westoutside'}
          if strcmpi(  loc , 'eastoutside' )
              cbarOptions = [ cbarOptions 'colorbar right' ];
           else
              cbarOptions = [ cbarOptions 'colorbar left' ];
           end

      otherwise
          error( 'drawColorbar: Unknown ''Location'' %s.', loc )
  end
  % set the options
  env.options = appendOptions( env.options, cbarOptions );

end
% =========================================================================
% *** END FUNCTION drawColorbar
% =========================================================================

env.options = appendOptions( env.options, sprintf( 'xlabel={%s}', xlabelText ) );

imagesc([1 10 100]);
colorbar('YScale','log')