loadvector.h

来自「Dolfin provide a high-performance linear」· C头文件 代码 · 共 74 行

// Copyright (C) 2006 Anders Logg.
// Licensed under the GNU LGPL Version 2.1.
//
// First added:  2006-08-21
// Last changed: 2006-08-21

#ifndef __LOAD_VECTOR_H
#define __LOAD_VECTOR_H

#include <dolfin/constants.h>
#include <dolfin/Vector.h>
#include <dolfin/PETScVector.h>
#include <dolfin/uBlasVector.h>
#include <dolfin/MatrixFactory.h>

namespace dolfin
{

  class Mesh;

  /// This class represents the standard load vector with constant
  /// load c on a given mesh, represented as a default DOLFIN vector.

  class LoadVector : public Vector
  {
  public:
  
    /// Construct load vector with constant load c on a given mesh
    LoadVector(Mesh& mesh, real c = 1.0) : Vector()
    {
      MatrixFactory::computeLoadVector(*this, mesh, c);
    }

  };

#ifdef HAVE_PETSC_H

  /// This class represents the standard load vector with constant
  /// load c on a given mesh, represented as a DOLFIN PETSc vector.

  class PETScLoadVector : public PETScVector
  {
  public:
  
    /// Construct load vector with constant load c on a given mesh
    PETScLoadVector(Mesh& mesh, real c = 1.0) : PETScVector()
    {
      MatrixFactory::computeLoadVector(*this, mesh, c);
    }

  };

#endif

  /// This class represents the standard load vector with constant
  /// load c on a given mesh, represented as a sparse DOLFIN uBlas
  /// vector.

  class uBlasLoadVector : public uBlasVector
  {
  public:
  
    /// Construct load vector with constant load c on a given mesh
    uBlasLoadVector(Mesh& mesh, real c = 1.0) : uBlasVector()
    {
      MatrixFactory::computeLoadVector(*this, mesh, c);
    }

  };

}

#endif