view trunk/carfac/carfac_output.cc @ 679:594b410c2aed

Sixth revision of Alex Brandmeyer's C++ implementation. Only small changes in response to Lyon's comments from r285. Note: I tried to use a consistent indentation with two spaces, but also preserving parenthetical structure to make reading the longer equations easier. Please advise if this is OK. Additional documentation and tests with non-standard parameter sets will be added in a subsequent revision.

//
//  carfac_output.cc
//  CARFAC Open Source C++ Library
//
//  Created by Alex Brandmeyer on 5/10/13.
//
// This C++ 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.

#include "carfac_output.h"

using std::vector;

void CARFACOutput::Init(const int n_ears, const bool store_nap,
                        const bool store_nap_decim, const bool store_bm,
                        const bool store_ohc, const bool store_agc) {
  n_ears_ = n_ears;
  store_nap_ = store_nap;
  store_nap_decim_ = store_nap_decim;
  store_bm_ = store_bm;
  store_ohc_ = store_ohc;
  store_agc_ = store_agc;

}

void CARFACOutput::StoreOutput(const vector<Ear>& ears) {
  if (store_nap_) {
    nap_.push_back(vector<FloatArray>());
    for (auto ear : ears) {
      nap_.back().push_back(ear.ihc_out());
    }
  }
  if (store_ohc_) {
    ohc_.push_back(vector<FloatArray>());
    for (auto ear : ears) {
      ohc_.back().push_back(ear.za_memory());
    }
  }
  if (store_agc_) {
    agc_.push_back(vector<FloatArray>());
    for (auto ear : ears) {
      agc_.back().push_back(ear.zb_memory());
    }
  }
  if (store_bm_) {
    bm_.push_back(vector<FloatArray>());
    for (auto ear : ears) {
      bm_.back().push_back(ear.zy_memory());
    }
  }
}