PolyBase.h

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
//   NScD Oak Ridge National Laboratory, European Spallation Source,
//   Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
#pragma once
 
#include "MantidGeometry/DllConfig.h"
 
namespace Mantid {
 
namespace mathLevel {
 
class MANTID_GEOMETRY_DLL PolyBase {
private:
  int iDegree;                 
  std::vector<double> afCoeff; 
  double Eaccuracy;            
 
  int solveQuadratic(std::complex<double> &, std::complex<double> &) const;
  int solveCubic(std::complex<double> &, std::complex<double> &, std::complex<double> &) const;
 
public:
  explicit PolyBase(int const);
  PolyBase(int const, double const);
 
  // member access
  void setDegree(int const);
  int getDegree() const;
  operator const std::vector<double> &() const;
  operator std::vector<double> &();
  double operator[](int const) const;
  double &operator[](int const);
 
  // evaluation
  double operator()(double const) const;
 
  // arithmetic updates
  PolyBase &operator+=(const PolyBase &);
  PolyBase &operator-=(const PolyBase &);
  PolyBase &operator*=(const PolyBase &);
 
  // arithmetic operations
  PolyBase operator+(const PolyBase &) const;
  PolyBase operator-(const PolyBase &) const;
  PolyBase operator*(const PolyBase &) const;
  //  PolyBase operator/(const PolyBase&) const;
 
  // input is degree 0 poly
  PolyBase operator+(double const) const; // input is degree 0 poly
  PolyBase operator-(double const) const;
  PolyBase operator*(double const) const;
  PolyBase operator/(double const) const;
 
  PolyBase operator-() const;
 
  // input is degree 0 poly
  PolyBase &operator+=(double const);
  PolyBase &operator-=(double const);
  PolyBase &operator*=(double const);
  PolyBase &operator/=(double const);
 
  // derivation
  PolyBase getDerivative() const;
  PolyBase &derivative();
 
  // inversion ( invpoly[i] = poly[degree-i] for 0 <= i <= degree )
  PolyBase GetInversion() const;
 
  void compress(double const);
 
  void divide(const PolyBase &, PolyBase &, PolyBase &, double const = -1.0) const;
 
  std::vector<double> realRoots(double const = -1.0);
  std::vector<std::complex<double>> calcRoots(double const = -1.0);
 
  void write(std::ostream &) const;
};
 
PolyBase operator*(double const, const PolyBase &);
std::ostream &operator<<(std::ostream &, const PolyBase &);
 
} // namespace mathLevel
 
} // NAMESPACE Mantid