AutomatedRIRGenerator.m

% This script generates AAES IRs for a set of rooms with varying
% dimensions, reflection coefficients and AAES channel counts.

if isempty(gcp('nocreate'))
    parpool('Processes');
end

% General audio parameters
sample_rate = 48000;
bit_depth = 24;

% Room dimensions [x y z; x y z; x y z]
rooms = [5.7 7.35 2.5; 8.74 17 5.5; 19.52 30.83 15];

% Wall absorption coefficients
% Read "Example Absorption Coefficients/Absorption Coefficients Info.txt" for
% more info
alphas = zeros(7, 6, 3);

for set = 1:3
    alphas(:, :, set) = readmatrix("Example Absorption Coefficients/alpha_set_" + set + ".dat");
end

% There is only ever one source and receiver per room
% These use the dimensions (room number, axis)
% [x y z; x y z; x y z]
% The sources are 2/3 along the x and y axes and 1.2m high
src_sets = [2*rooms(1,1)/3 2*rooms(1,2)/3 1.2
            2*rooms(2,1)/3 2*rooms(2,2)/3 1.2
            2*rooms(3,1)/3 2*rooms(3,2)/3 1.2];
% The receivers are 1/3 along the x and y axes and 1.2m high
rec_sets = [rooms(1,1)/3 rooms(1,2)/3 1.2
            rooms(2,1)/3 rooms(2,2)/3 1.2
            rooms(3,1)/3 rooms(3,2)/3 1.2];

% ls_sets_... are arranged with dimensions: (ls number, axis, room number)
% These aren't concatenated due to their dimensions
ls_sets_8ch = zeros(8,3,3);
ls_sets_12ch = zeros(12,3,3);
ls_sets_16ch = zeros(16,3,3);

% mic_sets_... are arranged with dimensions: (mic number, axis, room number)
mic_sets_8ch = zeros(8,3,3);
mic_sets_12ch = zeros(12,3,3);
mic_sets_16ch = zeros(16,3,3);

% Load coordinates
for room = 1:3
    ls_sets_8ch(:,:,room) = readmatrix("Example Transducer Coordinates/ls_positions_room_" + room + "_8ch.dat");
    ls_sets_12ch(:,:,room) = readmatrix("Example Transducer Coordinates/ls_positions_room_" + room + "_12ch.dat");
    ls_sets_16ch(:,:,room) = readmatrix("Example Transducer Coordinates/ls_positions_room_" + room + "_16ch.dat");
    mic_sets_8ch(:,:,room) = readmatrix("Example Transducer Coordinates/mic_positions_room_" + room + "_8ch.dat");
    mic_sets_12ch(:,:,room) = readmatrix("Example Transducer Coordinates/mic_positions_room_" + room + "_12ch.dat");
    mic_sets_16ch(:,:,room) = readmatrix("Example Transducer Coordinates/mic_positions_room_" + room + "_16ch.dat");
end

parfor channel_count_index = 1:3
    disp(4 + 4 * channel_count_index + "-channel AAES...");
    
    if channel_count_index == 1
        GenerateIRsForChannelCount(rooms, alphas, src_sets, rec_sets, ls_sets_8ch, mic_sets_8ch, sample_rate, bit_depth);
    elseif channel_count_index == 2
        GenerateIRsForChannelCount(rooms, alphas, src_sets, rec_sets, ls_sets_12ch, mic_sets_12ch, sample_rate, bit_depth);
    else
        GenerateIRsForChannelCount(rooms, alphas, src_sets, rec_sets, ls_sets_16ch, mic_sets_16ch, sample_rate, bit_depth);
    end
end

delete(gcp('nocreate'));

function GenerateIRsForChannelCount(rooms, alphas, src_sets, rec_sets, ls_sets, mic_sets, sample_rate, bit_depth)
    parfor room_num = 1:size(rooms, 1)
        disp("Room " + room_num + "...");
    
        current_config = RoomWithAAES(rooms(room_num,:), alphas, src_sets(room_num,:), rec_sets(room_num,:), ls_sets(:,:,room_num), mic_sets(:,:,room_num), sample_rate, bit_depth);
        current_config.GenerateIRs();
    end
end