create_localization_matrix.m

function loc = create_localization_matrix(spatial_locations, Ny, Nx, loc_fun_name, loc_params)
% Forms a localization matrix
%
% INPUTS:
% spatial_locations is an array 
% Ny is the number of Y locations
% Nx is the number of X locations
% loc_fun_name is a string: 'gaspari_cohn', 'bolin_wallin', 'askey', 'wendland'
% params is a struct which holds localization parameters
%
% OUTPUT:
% loc = localization matrix
%
% Author: Zofia Stanley

% Check lengths
loc_params.Ny = Ny;
loc_params.Nx = Nx;
N = Ny + Nx;
assert(length(spatial_locations)==N, 'Number of spatial locations does not match expectations.')

% Create distance matrix
dis = create_distance_matrix(spatial_locations, Ny);

% Create localization matrix
switch loc_fun_name
    case 'gaspari_cohn' 
        loc = gaspari_cohn(dis, loc_params);
    case 'bolin_wallin'
        loc = bolin_wallin(dis, loc_params);
    case 'askey'
        loc = askey(dis, loc_params);
    case 'wendland'
        loc = wendland(dis, loc_params);
    case 'no_loc'
        loc = ones(N);
    otherwise
        warning('Unexpected localization function name. No localization matrix created.')
end

end