fe_series.cc

Go to the documentation of this file.

// ------------------------------------------------------------------------
//
// SPDX-License-Identifier: LGPL-2.1-or-later
// Copyright (C) 2016 - 2021 by the deal.II authors
//
// This file is part of the deal.II library.
//
// Part of the source code is dual licensed under Apache-2.0 WITH
// LLVM-exception OR LGPL-2.1-or-later. Detailed license information
// governing the source code and code contributions can be found in
// LICENSE.md and CONTRIBUTING.md at the top level directory of deal.II.
//
// ------------------------------------------------------------------------
 
 
 
#include <deal.II/base/config.h>
 
#include <deal.II/fe/fe_series.h>
 
#include <iostream>
 
 
DEAL_II_NAMESPACE_OPEN
 
namespace FESeries
{
  std::pair<double, double>
  linear_regression(const std::vector<double> &x, const std::vector<double> &y)
  {
    FullMatrix<double> K(2, 2), invK(2, 2);
    Vector<double>     X(2), B(2);
 
    Assert(x.size() == y.size(),
           ExcMessage("x and y are expected to have the same size"));
 
    Assert(x.size() >= 2,
           ::ExcMessage(
             "at least two points are required for linear regression fit"));
 
    double sum_1 = 0.0, sum_x = 0.0, sum_x2 = 0.0, sum_y = 0.0, sum_xy = 0.0;
 
    for (unsigned int i = 0; i < x.size(); ++i)
      {
        sum_1 += 1.0;
        sum_x += x[i];
        sum_x2 += x[i] * x[i];
        sum_y += y[i];
        sum_xy += x[i] * y[i];
      }
 
    K(0, 0) = sum_1;
    K(0, 1) = sum_x;
    K(1, 0) = sum_x;
    K(1, 1) = sum_x2;
 
    B(0) = sum_y;
    B(1) = sum_xy;
 
    invK.invert(K);
    invK.vmult(X, B, false);
 
    return std::make_pair(X(1), X(0));
  }
} // namespace FESeries
 
DEAL_II_NAMESPACE_CLOSE