alexbrandmeyer@609: //
alexbrandmeyer@609: //  carfac.cc
alexbrandmeyer@609: //  CARFAC Open Source C++ Library
alexbrandmeyer@609: //
alexbrandmeyer@609: //  Created by Alex Brandmeyer on 5/10/13.
alexbrandmeyer@609: //
alexbrandmeyer@609: // This C++ file is part of an implementation of Lyon's cochlear model:
alexbrandmeyer@609: // "Cascade of Asymmetric Resonators with Fast-Acting Compression"
alexbrandmeyer@609: // to supplement Lyon's upcoming book "Human and Machine Hearing"
alexbrandmeyer@609: //
alexbrandmeyer@609: // Licensed under the Apache License, Version 2.0 (the "License");
alexbrandmeyer@609: // you may not use this file except in compliance with the License.
alexbrandmeyer@609: // You may obtain a copy of the License at
alexbrandmeyer@609: //
alexbrandmeyer@609: //     http://www.apache.org/licenses/LICENSE-2.0
alexbrandmeyer@609: //
alexbrandmeyer@609: // Unless required by applicable law or agreed to in writing, software
alexbrandmeyer@609: // distributed under the License is distributed on an "AS IS" BASIS,
alexbrandmeyer@609: // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
alexbrandmeyer@609: // See the License for the specific language governing permissions and
alexbrandmeyer@609: // limitations under the License.
alexbrandmeyer@609: 
alexbrandmeyer@609: #include "carfac.h"
alexbrandmeyer@609: void CARFAC::Design(int n_ears, long fs, CARParams car_params,
alexbrandmeyer@609:                     IHCParams ihc_params, AGCParams agc_params){
alexbrandmeyer@609:   n_ears_ = n_ears;
alexbrandmeyer@609:   fs_ = fs;
alexbrandmeyer@609:   ears_ = new Ear[n_ears_];
alexbrandmeyer@609:   for (int i = 0; i < n_ears_; i++){
alexbrandmeyer@609:     ears_[i].InitEar(fs_, car_params,ihc_params,agc_params);
alexbrandmeyer@609:   }
alexbrandmeyer@609:   n_ch_ = ears_[0].car_coeffs_.n_ch_;
alexbrandmeyer@609: }
alexbrandmeyer@609: 
alexbrandmeyer@609: CARFACOutput CARFAC::Run(FloatArray2d sound_data){
alexbrandmeyer@609:   //to store the final output
alexbrandmeyer@609:   CARFACOutput *output = new CARFACOutput();
alexbrandmeyer@609:   //to store the output of the individual segments
alexbrandmeyer@609:   CARFACOutput *seg_output = new CARFACOutput();
alexbrandmeyer@609:   
alexbrandmeyer@609:   int n_audio_channels = int(sound_data.cols());
alexbrandmeyer@609:   long seg_len = 441; //Fixed segment length for now
alexbrandmeyer@609:   long n_timepoints = sound_data.rows();
alexbrandmeyer@609:   double n_segs = ceil(double(n_timepoints)/double(seg_len));
alexbrandmeyer@609:   output->InitOutput(n_audio_channels, n_ch_, n_timepoints);
alexbrandmeyer@609:   seg_output->InitOutput(n_audio_channels, n_ch_, seg_len);
alexbrandmeyer@609:   //loop over individual audio segments
alexbrandmeyer@609:   long start, length; //to store the start and endpoints for each segment
alexbrandmeyer@609:   for (long i = 0; i < long(n_segs); i++){
alexbrandmeyer@609:     //determine start and end points
alexbrandmeyer@609:     start = (i * seg_len);
alexbrandmeyer@609:     if (i < n_segs - 1){
alexbrandmeyer@609:       length = seg_len;
alexbrandmeyer@609:     } else {
alexbrandmeyer@609:       length = n_timepoints - start; //the last segment can be shorter than the rest
alexbrandmeyer@609:     }
alexbrandmeyer@609:     RunSegment(sound_data.block(start,0,length,n_audio_channels),seg_output);
alexbrandmeyer@609:     output->MergeOutput(*seg_output, start, length);
alexbrandmeyer@609:   }
alexbrandmeyer@609:   
alexbrandmeyer@609:   return *output;
alexbrandmeyer@609: }
alexbrandmeyer@609: 
alexbrandmeyer@609: void CARFAC::RunSegment(FloatArray2d sound_data, CARFACOutput *seg_output){
alexbrandmeyer@609:   long n_timepoints = sound_data.rows();
alexbrandmeyer@609:   int n_ears = int(sound_data.cols());
alexbrandmeyer@609:   for (long i = 0; i < n_timepoints; i++){
alexbrandmeyer@609:     for (int j = 0; j < n_ears; j++){
alexbrandmeyer@609:       FPType input = sound_data(i,j);
alexbrandmeyer@609:       FloatArray car_out = ears_[j].CARStep(input);
alexbrandmeyer@609:       FloatArray ihc_out = ears_[j].IHCStep(car_out);
alexbrandmeyer@609:       bool updated = ears_[j].AGCStep(ihc_out);
alexbrandmeyer@609:       seg_output->ears_[j].nap_.block(0,i,n_ch_,1) = ihc_out;
alexbrandmeyer@609:       seg_output->ears_[j].bm_.block(0,i,n_ch_,1) = car_out;
alexbrandmeyer@609:       seg_output->ears_[j].ohc_.block(0,1,n_ch_,1) =
alexbrandmeyer@609:         ears_[j].car_state_.za_memory_;
alexbrandmeyer@609:       seg_output->ears_[j].agc_.block(0,i,n_ch_,1) =
alexbrandmeyer@609:         ears_[j].car_state_.zb_memory_;
alexbrandmeyer@609:     }
alexbrandmeyer@609:   }
alexbrandmeyer@609:   
alexbrandmeyer@609: }