% Copyright 2012, Google, Inc.
% Author: Richard F. Lyon
%
% This Matlab file is part of an implementation of Lyon's cochlear model:
% "Cascade of Asymmetric Resonators with Fast-Acting Compression"
% to supplement Lyon's upcoming book "Human and Machine Hearing"
%
% Licensed under the Apache License, Version 2.0 (the "License");
% you may not use this file except in compliance with the License.
% You may obtain a copy of the License at
%
%     http://www.apache.org/licenses/LICENSE-2.0
%
% Unless required by applicable law or agreed to in writing, software
% distributed under the License is distributed on an "AS IS" BASIS,
% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
% See the License for the specific language governing permissions and
% limitations under the License.

%% Test/demo hacking for CARFAC Matlab stuff:

clear variables

%%
use_plan_file = 1;
if use_plan_file
  
  file_signal = wavread('plan.wav');
  file_signal = file_signal(8100+(1:20000));  % trim for a faster test
  
else
    flist = [1000];
    alist = [1];
    flist = 1000;
    alist = 1;
    sine_signal = 0;
    times = (0:19999)' / 22050;
    for fno = 1:length(flist)
      sine_signal = sine_signal + alist(fno)*sin(flist(fno)*2*pi*times);
    end
    growth_power = 0;  % use 0 for flat, 4 or more for near exponential
    file_signal = 1.0 * (sine_signal .* (times/max(times)).^growth_power);
end

% repeat with negated signal to compare responses:
% file_signal = [file_signal; -file_signal];

% make a long test signal by repeating at different levels:
dB = -80;
test_signal =  10^(dB/20)* file_signal(1:4000) % lead-in [];
for dB =  -80:20:60
  test_signal = [test_signal; file_signal * 10^(dB/20)];
end

%%
CF_struct = CARFAC_Design;  % default design

%% Run mono, then stereo test:

agc_plot_fig_num = 6;

for n_ears = 1:2
  if n_ears == 2
    % For the 2-channel pass, add a silent second channel:
    test_signal = [test_signal, zeros(size(test_signal))];
  end
  
  CF_struct = CARFAC_Init(CF_struct, n_ears);

  [CF_struct, nap_decim, nap, BM] = CARFAC_Run(CF_struct, test_signal, ...
    agc_plot_fig_num);

%   nap = deskew(nap);  % deskew doesn't make much difference

%   dB_BM = 10/log(10) * log(filter(1, [1, -0.995], BM(:, 38:40, :).^2));
  dB_BM = 10/log(10) * log(filter(1, [1, -0.995], BM(:, 20:50, :).^2));

  % only ear 1:
  MultiScaleSmooth(dB_BM(5000:200:end, :, 1), 1);

  % Display results for 1 or 2 ears:
  for ear = 1:n_ears
    smooth_nap = nap_decim(:, :, ear);
    if n_ears == 1
      mono_max = max(smooth_nap(:));
    end
    figure(3 + ear + n_ears)  % Makes figures 5, ...
    image(63 * ((max(0, smooth_nap)/mono_max)' .^ 0.5))
    title('smooth nap from nap decim')
    colormap(1 - gray);
  end

  % Show resulting data, even though M-Lint complains:
  CF_struct
  CF_struct.CAR_state
  CF_struct.AGC_state
  min_max_decim = [min(nap_decim(:)), max(nap_decim(:))]

end

% Expected result:  Figure 3 looks like figure 2, a tiny bit darker.
% and figure 4 is empty (all zero)
