PDESystem.hpp

Go to the documentation of this file.

//  This file is part of ff3d - http://www.freefem.org/ff3d
//  Copyright (C) 2001, 2002, 2003 Stéphane Del Pino

//  This program 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, or (at your option)
//  any later version.

//  This program 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, write to the Free Software Foundation,
//  Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  

//  $Id: PDESystem.hpp,v 1.6 2007/11/02 20:03:18 delpinux Exp $

#ifndef PDE_SYSTEM_HPP
#define PDE_SYSTEM_HPP

#include <vector>

#include <PDEProblem.hpp>
#include <ReferenceCounting.hpp>

#include <Problem.hpp>

class PDESystem
  : public Problem
{
private:
  std::vector<ConstReferenceCounting<PDEProblem> >
  __pdeProblems;                
public:
  void add(ReferenceCounting<PDEProblem> pdeProblem)
  {
    __pdeProblems.push_back(pdeProblem);
  }

  const PDEProblem&
  operator[](const size_t& i) const
  {
    ASSERT(i<__pdeProblems.size());
    return *(__pdeProblems[i]);
  }

  const ScalarFunctionBase&
  secondMember(const size_t& i) const
  {
    return *__pdeProblems[i]->pde().secondMember();
  }

  const BoundaryConditionSet&
  boundaryConditionSet(const size_t& i) const
  {
    return __pdeProblems[i]->boundaryConditionSet();
  }

  ReferenceCounting<Problem>
  operator*(const ConstReferenceCounting<ScalarFunctionBase>& c) const
  {
    PDESystem* newPDESystem = new PDESystem(*this);

    for (size_t i=0; i<this->numberOfEquations(); ++i) {
      (*newPDESystem).__pdeProblems[i] = (*__pdeProblems[i]) * c;
    }
    return newPDESystem;
  }

  size_t numberOfUnknown() const
  {
    return __pdeProblems.size();
  }

  size_t numberOfEquations() const
  {
    return __pdeProblems.size();
  }

  friend std::ostream&
  operator << (std::ostream& os, const PDESystem& pdeSystem)
  {
    for(size_t i=0; i<pdeSystem.numberOfEquations(); ++i) {
      os << *(pdeSystem.__pdeProblems[i]) << '\n';
    }
    return os;
  }

  PDESystem ()
    : Problem(Problem::pde, 0)
  {
    ;
  }

  PDESystem (const size_t& dimension)
    : Problem(Problem::pde, 0)
  {
    __pdeProblems.reserve(dimension);
  }

  PDESystem(const PDESystem& p)
    : Problem(p),
      __pdeProblems(p.__pdeProblems)
  {
  }

  ~PDESystem ()
  {
    ;
  }
};

#endif // PDE_SYSTEM_HPP

---