d-kitamura / auxiva-iss Goto Github PK

roomDimensions = [7 5 2.75]; 
microLocation = [3 1 1.5; 3.5 1 1.5; 4 1 1.5];
sourceLocation = [2 3.5 1.5; 4 3.5 1.5; 6 3.5 1.5];      

c = 343;                    % Sound velocity (m/s)
fs = 44100;                 % Sample frequency (samples/s)
beta = 0;                 % Reverberation time (s)
n = 4096;                   % Number of samples
mtype = 'omnidirectional';  % Type of microphone
order = -1;                 % -1 equals maximum reflection order!
dim = 3;                    % Room dimension
orientation = 0;            % Microphone orientation (rad)
hp_filter = 1;              % Enable high-pass filter
h = rir_generator(c, fs, microLocation, sourceLocation, roomDimensions, beta, n, mtype, order, dim, orientation, hp_filter);

mixSig(:,1) = conv(srcSig(:,1),h(1,:), 'same') + conv(srcSig(:,2),h(1,:), 'same') + conv(srcSig(:,3),h(1,:), 'same');
mixSig(:,2) = conv(srcSig(:,1),h(2,:), 'same') + conv(srcSig(:,2),h(2,:), 'same') + conv(srcSig(:,3),h(2,:), 'same');
mixSig(:,3) = conv(srcSig(:,1),h(3,:), 'same') + conv(srcSig(:,2),h(3,:), 'same') + conv(srcSig(:,3),h(3,:), 'same');

%% using AuxIVA
% Set parameters
seed = 1; % pseudo random seed
refMic = 1; % reference microphone for back projection
nSrc = 3; % number of sources
fftSize = 4096; % window length in STFT [points]
shiftSize = 2048; % shift length in STFT [points]
windowType = "hamming"; % window function used in STFT
nIter = 30; % number of iterations (define by checking convergence behavior with drawConv=true)
applyWhitening = false; % true or false (true: apply whitening to the observed multichannel spectrograms)
drawConv = true; % true or false (true: plot cost function values in each iteration and show convergence behavior, false: faster and do not plot cost function values)
% Fix random seed
RandStream.setGlobalStream(RandStream('mt19937ar','Seed',seed))
% Blind source separation based on AuxIVA-ISS
[estSig_auxiva, cost] = AuxIVAISS(mixSig, nSrc, targetFs, fftSize, shiftSize, windowType, nIter, refMic, applyWhitening, drawConv);