From 53b7d03b39ae30bcca53dc294f213e6d455943f7 Mon Sep 17 00:00:00 2001 From: Tavian Barnes Date: Wed, 6 Oct 2010 01:05:17 -0400 Subject: Add base Integrator class, and simple compilation test. --- src/Makefile.am | 22 ++++++++++++++ src/vZ.hpp | 29 ++++++++++++++++++ src/vZ/integrator.hpp | 83 +++++++++++++++++++++++++++++++++++++++++++++++++++ src/vZ/simple.hpp | 56 ++++++++++++++++++++++++++++++++++ 4 files changed, 190 insertions(+) create mode 100644 src/Makefile.am create mode 100644 src/vZ.hpp create mode 100644 src/vZ/integrator.hpp create mode 100644 src/vZ/simple.hpp (limited to 'src') diff --git a/src/Makefile.am b/src/Makefile.am new file mode 100644 index 0000000..25f6803 --- /dev/null +++ b/src/Makefile.am @@ -0,0 +1,22 @@ +########################################################################### +## Copyright (C) 2009-2010 Tavian Barnes ## +## ## +## This file is part of The Dimension Build Suite. ## +## ## +## The Dimension Build Suite 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 3 of the ## +## License, or (at your option) any later version. ## +## ## +## The Dimension Build Suite 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. ## +## ## +## You should have received a copy of the GNU General Public License ## +## along with this program. If not, see . ## +########################################################################### + +nobase_include_HEADERS = vZ.hpp \ + vZ/integrator.hpp \ + vZ/simple.hpp diff --git a/src/vZ.hpp b/src/vZ.hpp new file mode 100644 index 0000000..1d7327b --- /dev/null +++ b/src/vZ.hpp @@ -0,0 +1,29 @@ +/************************************************************************* + * Copyright (C) 2009-2010 Tavian Barnes * + * * + * This file is part of The Dimension Library. * + * * + * The Dimension Library is free software; you can redistribute it and/ * + * or modify it under the terms of the GNU Lesser General Public License * + * as published by the Free Software Foundation; either version 3 of the * + * License, or (at your option) any later version. * + * * + * The Dimension Library 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 * + * Lesser General Public License for more details. * + * * + * You should have received a copy of the GNU Lesser General Public * + * License along with this program. If not, see * + * . * + *************************************************************************/ + +// vZ -- A Runge-Kutta type numerical integration library + +#ifndef VZ_HPP +#define VZ_HPP + +#include +#include + +#endif // VZ_HPP diff --git a/src/vZ/integrator.hpp b/src/vZ/integrator.hpp new file mode 100644 index 0000000..394f40f --- /dev/null +++ b/src/vZ/integrator.hpp @@ -0,0 +1,83 @@ +/************************************************************************* + * Copyright (C) 2009-2010 Tavian Barnes * + * * + * This file is part of The vZ Library. * + * * + * The vZ Library is free software; you can redistribute it and/or * + * modify it under the terms of the GNU Lesser General Public License as * + * published by the Free Software Foundation; either version 3 of the * + * License, or (at your option) any later version. * + * * + * The vZ Library 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 * + * Lesser General Public License for more details. * + * * + * You should have received a copy of the GNU Lesser General Public * + * License along with this program. If not, see * + * . * + *************************************************************************/ + +#ifndef VZ_INTEGRATOR_HPP +#define VZ_INTEGRATOR_HPP + +#include +#include + +namespace vZ +{ + // Base Integrator class + // + // All integration methods derrive from this class + // If the initial value problem is specified as + // y' = f(t, y); y(t0) = y0 + // then an Integrator could be constructed as Integrator(f, dt).y(y0).t(t0) + template + class GenericIntegrator + { + public: + typedef std::tr1::function Function; + + // By default, y and t start at zero + GenericIntegrator(Function f, Real dt) + : m_f(f), m_y(0), m_t(0), m_dt(dt) { } + virtual ~GenericIntegrator() { } + + GenericIntegrator& y(Real y) { m_y = y; return *this; } + GenericIntegrator& t(Real t) { m_t = t; return *this; } + GenericIntegrator& dt(Real dt) { m_dt = dt; return *this; } + + Real y() const { return m_y; } + Real t() const { return m_t; } + Real dt() const { return m_dt; } + + // Integrate until time t + void integrate(Real t_final); + + protected: + virtual void step(Real& t, Real& dt) = 0; + Function m_f; + + private: + Real m_y; + Real m_t, m_dt; + }; + + // Type alias + typedef GenericIntegrator Integrator; + + // Definitions + + template + void + GenericIntegrator::integrate(Real t_final) + { + while (m_t < t_final) { + m_dt = std::min(m_dt, t_final - m_t); + step(m_t, m_dt); + } + } +} + + +#endif // VZ_INTEGRATOR_HPP diff --git a/src/vZ/simple.hpp b/src/vZ/simple.hpp new file mode 100644 index 0000000..65b7560 --- /dev/null +++ b/src/vZ/simple.hpp @@ -0,0 +1,56 @@ +/************************************************************************* + * Copyright (C) 2009-2010 Tavian Barnes * + * * + * This file is part of The vZ Library. * + * * + * The vZ Library is free software; you can redistribute it and/or * + * modify it under the terms of the GNU Lesser General Public License as * + * published by the Free Software Foundation; either version 3 of the * + * License, or (at your option) any later version. * + * * + * The vZ Library 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 * + * Lesser General Public License for more details. * + * * + * You should have received a copy of the GNU Lesser General Public * + * License along with this program. If not, see * + * . * + *************************************************************************/ + +#ifndef VZ_SIMPLE_HPP +#define VZ_SIMPLE_HPP + +#include + +namespace vZ +{ + // Base class for non-adaptive RK-style algorithms + template + class GenericSimpleIntegrator : public GenericIntegrator + { + public: + typedef typename GenericIntegrator::Function Function; + + // Coefficients in the tableau representation of the RK algorithm + typedef std::vector > ACoefficients; + typedef std::vector BCoefficients; + + GenericSimpleIntegrator(Function f, Real dt, + ACoefficients a, BCoefficients b) + : GenericIntegrator(f, dt), m_a(a), m_b(b) { } + virtual ~GenericSimpleIntegrator() { } + + protected: + virtual void step(Real& t, Real& dt); + + private: + ACoefficients m_a; + BCoefficients m_b; + }; + + // Type alias + typedef GenericSimpleIntegrator SimpleIntegrator; +} + +#endif // VZ_SIMPLE_HPP -- cgit v1.2.3