main.cpp

来自「利用C」· C++ 代码 · 共 51 行

// Copyright (C) 2008 Anders Logg.
// Licensed under the GNU LGPL Version 2.1.
//
// First added:  2008-03-11
// Last changed: 2007-03-11
//
// Testing evaluation at arbitrary points

#include <dolfin.h>
#include "Projection.h"

using namespace dolfin;

class F : public Function
{
public:
  
  F(Mesh& mesh) : Function(mesh) {}

  void eval(real* values, const real* x) const
  {
    values[0] = sin(3.0*x[0])*sin(3.0*x[1])*sin(3.0*x[2]);
  }
  
};

int main()
{
  UnitCube mesh(8, 8, 8);
  real x[3] = {0.3, 0.3, 0.3};
  real v[1] = {0.0};

  // A user-defined function
  F f(mesh);

  // Project to a discrete function
  ProjectionBilinearForm a;
  ProjectionLinearForm L(f);
  LinearPDE pde(a, L, mesh);
  Function g;
  pde.solve(g);

  // Evaluate user-defined function
  f.eval(v, x);
  message("f(x) = %g", v[0]);

  // Evaluate discrete function
  g.eval(v, x);
  message("g(x) = %g", v[0]);
}