cblock.cpp

Go to the documentation of this file.

// Copyright (c) 2024 anabrid GmbH
// Contact: https://www.anabrid.com/licensing/
// SPDX-License-Identifier: MIT OR GPL-2.0-or-later
 
#include <block/cblock.h>
#include <cstring>
#include <entity/visitor.h>
#include <utils/logging.h>
 
blocks::CBlock::CBlock(CBlockHAL *hardware) : FunctionBlock("C", hardware), hardware(hardware) {
  classifier.class_enum = CLASS_;
}
 
blocks::CBlock::CBlock() : CBlock(new CBlockHALDummy()) {}
 
float blocks::CBlock::get_factor(uint8_t idx) {
  if (idx >= NUM_COEFF)
    return 0.0f;
  return factors_[idx];
}
 
const std::array<float, blocks::CBlock::NUM_COEFF> &blocks::CBlock::get_factors() const {
  return factors_;
}
 
bool blocks::CBlock::set_factor(uint8_t idx, float factor) {
  if (idx >= NUM_COEFF)
    return false;
  if (factor > MAX_FACTOR or factor < MIN_FACTOR)
    return false;
  factors_[idx] = factor;
  return true;
}
 
void blocks::CBlock::set_factors(const std::array<float, NUM_COEFF> &factors) { factors_ = factors; }
 
utils::status blocks::CBlock::write_to_hardware() {
  if (!write_factors_to_hardware()) {
    LOG(ANABRID_PEDANTIC, __PRETTY_FUNCTION__);
    return utils::status::failure();
  }
  return utils::status::success();
}
 
bool blocks::CBlock::write_factors_to_hardware() {
  for (size_t i = 0; i < factors_.size(); i++) {
    auto factor = factors_[i] * gain_corrections_[i];
    if (!hardware->write_factor(i, factor))
      return false;
  }
  return true;
}
 
void blocks::CBlock::reset(entities::ResetAction action) {
  FunctionBlock::reset(action);
 
  if (action.has(entities::ResetAction::CIRCUIT_RESET))
    for (size_t i = 0; i < NUM_COEFF; i++)
      (void)set_factor(i, 1.0f);
 
  if (action.has(entities::ResetAction::CALIBRATION_RESET))
    reset_gain_corrections();
}
 
float blocks::CBlock::get_gain_correction(uint8_t idx) const {
  if (idx > NUM_COEFF)
    return -2.0f;
  return gain_corrections_[idx];
}
 
const std::array<float, blocks::CBlock::NUM_COEFF> &blocks::CBlock::get_gain_corrections() const {
  return gain_corrections_;
}
 
void blocks::CBlock::reset_gain_corrections() {
  std::fill(gain_corrections_.begin(), gain_corrections_.end(), 1.0f);
}
 
void blocks::CBlock::set_gain_corrections(const std::array<float, NUM_COEFF> &corrections) {
  gain_corrections_ = corrections;
};
 
bool blocks::CBlock::set_gain_correction(const uint8_t coeff_idx, const float correction) {
  if (coeff_idx > NUM_COEFF)
    return false;
  // Gain correction must be positive and close to 1
  if (correction < 0.0f || correction > MAX_GAIN_CORRECTION_ABS + 1.0f)
    return false;
 
  gain_corrections_[coeff_idx] = correction;
  return true;
};
 
ConfigResult blocks::CBlock::config(const pb_Config &cfg) {
#ifdef ANABRID_DEBUG_ENTITY_CONFIG
  Serial.println(__PRETTY_FUNCTION__);
#endif
  if (cfg.which_kind != pb_Config_coef_config_tag)
    return ConfigResult::err("expected coef block config");
 
  auto& coef_config = cfg.kind.coef_config;
  return _apply_elements(coef_config);
}
 
ConfigResult blocks::CBlock::_apply_elements(const pb_CoefConfig &cfg) {
  // Handle an array of factors
  for (size_t elem_idx = 0; elem_idx < cfg.elements_count; ++elem_idx) {
    auto& element = cfg.elements[elem_idx];
    if (!set_factor(element.idx, element.factor))
      return ConfigResult::err_fmt("CBlock factor %f is out of valid bounds", element.factor);
  }
 
  return ConfigResult::ok(true);
}
 
void blocks::CBlock::extract(entities::ExtractVisitor &collector) {
  auto& cfg = collector.create(pb_Config_coef_config_tag);
  auto& coef_config = cfg.kind.coef_config;
  coef_config.elements_count = NUM_COEFF;
  auto& idx = coef_config.elements_count = 0;
  for (;idx < NUM_COEFF; ++idx) {
    auto& elem = coef_config.elements[idx];
    elem.idx = idx;
    elem.factor = factors_[idx];
  }
}