OptimizationTechniques/Work2/scripts/plot3dFun.m

function plot3dFun(fun, x_lim, y_lim, size, plot_title, filename)
    % 3D plots a function
    % fun:          The function to plot
    % x_lim:        The range for x axis. ex: [-2, 2]
    % y_lim:        The range for y axis. ex: [0, 2]
    % size:         The number of points for each axis
    % plot_title:   The latex title for the plot
    %

    global image_width, 
    global image_height;

    % Generate a grid for x and y
    x_space = linspace(x_lim(1), x_lim(2), size);
    y_space = linspace(y_lim(1), y_lim(2), size);
    [X, Y] = meshgrid(x_space, y_space);
    
    % Evaluate the function on the grid
    Z = fun(X, Y);
    
    % 3D plot
    figure('Name', 'f(x,y)', 'NumberTitle', 'off');
    set(gcf, 'Position', [100, 100, image_width, image_height]); % Set the figure size
    surf(X, Y, Z);
    
    % Customize the plot
    xlabel('x'); % Label for x-axis
    ylabel('y'); % Label for y-axis
    zlabel('f(x, y)'); % Label for z-axis
    title(plot_title, 'Interpreter', 'latex', 'FontSize', 16); % Title of the plot
    colorbar;

    % save the figure
    if strcmp(filename, '') == 0
        print(gcf, filename, '-dpng', '-r300');
    end
end