removing_nan_imp.hpp

Go to the documentation of this file.

// Copyright (c) 2024 Andrea Enrico Franzoni (andreaenrico.franzoni@gmail.com)
//
// This file is part of PPCKO
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of PPCKO and associated documentation files (the PPCKO software), to deal
// PPCKO without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of PPCKO, and to permit persons to whom PPCKO is
// furnished to do so, subject to the following conditions:
//
// PPCKO IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH PPCKO OR THE USE OR OTHER DEALINGS IN
// PPCKO.
 
#include "removing_nan.hpp"
 

 

template<typename T,REM_NAN MA_t>
void
removing_nan<T,MA_t>::row_removal(Eigen::Block<Eigen::Matrix<T,-1,-1>,1>& row, MAT<REM_NAN::MR>)
{
  //counting how many non-NaNs
  const auto n_not_nan = std::count_if(row.begin(),row.end(),[](T el){return !std::isnan(el);});
  std::vector<T> el_not_nan;
  el_not_nan.reserve(n_not_nan);
  //saving non-NaNs values
  std::copy_if(row.begin(),row.end(),std::back_inserter(el_not_nan),[](T el){return !std::isnan(el);});
  //evaluating their mean
  const T mean = std::accumulate(el_not_nan.begin(),el_not_nan.end(),static_cast<T>(0),std::plus{})/static_cast<T>(n_not_nan);
  el_not_nan.clear();
  //replacing
  std::replace_if(row.begin(),row.end(),[](T el){return std::isnan(el);},mean);
}  
 

template<typename T,REM_NAN MA_t>
void
removing_nan<T,MA_t>::row_removal(Eigen::Block<Eigen::Matrix<T,-1,-1>,1>& row, MAT<REM_NAN::ZR>)
{
  std::replace_if(row.begin(),row.end(),[](T el){return std::isnan(el);},static_cast<T>(0)); 
}