X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=synfig-core%2Fsrc%2Fsynfig%2Fpolynomial_root.h;fp=synfig-core%2Fsrc%2Fsynfig%2Fpolynomial_root.h;h=09e31e4b2e0116a71516e1fb857ccfcc1fee8256;hb=a095981e18cc37a8ecc7cd237cc22b9c10329264;hp=0000000000000000000000000000000000000000;hpb=9459638ad6797b8139f1e9f0715c96076dbf0890;p=synfig.git diff --git a/synfig-core/src/synfig/polynomial_root.h b/synfig-core/src/synfig/polynomial_root.h new file mode 100644 index 0000000..09e31e4 --- /dev/null +++ b/synfig-core/src/synfig/polynomial_root.h @@ -0,0 +1,121 @@ +/* === S Y N F I G ========================================================= */ +/*! \file polynomial_root.h +** \brief Polynomial Root Finder Header +** +** $Id$ +** +** \legal +** Copyright (c) 2002-2005 Robert B. Quattlebaum Jr., Adrian Bentley +** Copyright (c) 2007 Chris Moore +** +** This package is free software; you can redistribute it and/or +** modify it under the terms of the GNU General Public License as +** published by the Free Software Foundation; either version 2 of +** the License, or (at your option) any later version. +** +** This package is distributed in the hope that it will be useful, +** but WITHOUT ANY WARRANTY; without even the implied warranty of +** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +** General Public License for more details. +** \endlegal +*/ +/* ========================================================================= */ + +/* === S T A R T =========================================================== */ + +#ifndef __SYNFIG_POLYNOMIAL_ROOT_H +#define __SYNFIG_POLYNOMIAL_ROOT_H + +/* === H E A D E R S ======================================================= */ + +#include +#include + +/* === M A C R O S ========================================================= */ + +/* === T Y P E D E F S ===================================================== */ + +/* === C L A S S E S & S T R U C T S ======================================= */ +template < typename T = float, typename F = float > +class Polynomial : public std::vector //a0 + a1x + a2x^2 + ... + anx^n +{ +public: + + //Will maintain all lower constants + void degree(unsigned int d, const T & def = (T)0) { resize(d+1,def); } + unsigned int degree()const { return this->size() - 1; } + + const Polynomial & operator+=(const Polynomial &p) + { + if(p.size() > this->size()) + resize(p.size(), (T)0); + + for(int i = 0; i < p.size(); ++i) + { + (*this)[i] += p[i]; + } + return *this; + } + + const Polynomial & operator-=(const Polynomial &p) + { + if(p.size() > this->size()) + resize(p.size(), (T)0); + + for(int i = 0; i < p.size(); ++i) + { + (*this)[i] -= p[i]; + } + return *this; + } + + const Polynomial & operator*=(const Polynomial &p) + { + if(p.size() < 1) + { + this->resize(0); + return *this; + } + + unsigned int i,j; + std::vector nc(*this); + + //in place for constant stuff + for(i = 0; i < nc.size(); ++i) + { + (*this)[i] *= p[0]; + } + + if(p.size() < 2) return *this; + + this->resize(this->size() + p.degree()); + for(int i = 0; i < nc.size(); ++i) + { + for(int j = 1; j < p.size(); ++j) + { + nc[i+j] += nc[i]*p[j]; + } + } + + return *this; + } +}; + +class RootFinder +{ + std::vector< std::complex > workcoefs; + int its; + +public: + std::vector< std::complex > coefs; //the number of coefficients determines the degree of polynomial + + std::vector< std::complex > roots; + + void find_all_roots(bool polish); +}; + + + +/* === E N D =============================================================== */ + +#endif