File: Expression.cpp

package info (click to toggle)
dolfin 2018.1.0.post1-16
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 28,764 kB
  • sloc: xml: 104,040; cpp: 98,856; python: 22,511; makefile: 204; sh: 182
file content (85 lines) | stat: -rw-r--r-- 2,377 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
// Copyright (C) 2007 Anders Logg
//
// This file is part of DOLFIN.
//
// DOLFIN 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.
//
// DOLFIN 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 DOLFIN. If not, see <http://www.gnu.org/licenses/>.
//
// Unit tests for the function library

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

using namespace dolfin;

namespace
{
  void arbitrary_eval()
  {
    class F0 : public Expression
    {
    public:
      F0() {}
      void eval(Array<double>& values, const Array<double>& x) const
      { values[0] = sin(3.0*x[0])*sin(3.0*x[1])*sin(3.0*x[2]); }
    };

    class F1 : public Expression
    {
    public:
      F1() {}
      void eval(Array<double>& values, const Array<double>& x) const
      { values[0] = 1.0 + 3.0*x[0] + 4.0*x[1] + 0.5*x[2]; }
    };

    auto mesh = std::make_shared<UnitCubeMesh>(8, 8, 8);

    Array<double> x(3);
    x[0] = 0.31; x[1] = 0.32; x[2] = 0.33;

    Array<double> u0(1);
    Array<double> u1(1);

    // User-defined functions (one from finite element space, one not)
    F0 f0;
    auto f1 = std::make_shared<F1>();

    // Test evaluation of a user-defined function
    f0.eval(u0, x);
    CHECK(u0[0] == Approx(sin(3.0*x[0])*sin(3.0*x[1])*sin(3.0*x[2])));

    // Test for single core only
    if (dolfin::MPI::size(mesh->mpi_comm()) == 1)
    {
    // Test evaluation of a discrete function
      auto V = std::make_shared<Projection::FunctionSpace>(mesh);
      Projection::BilinearForm a(V, V);
      Projection::LinearForm L(V);
      L.f = f1;
      Function g(V);
      solve(a == L, g);

      const double tol = 1.0e-6;
      f1->eval(u0, x);
      g.eval(u1, x);
      CHECK(std::abs(u0[0] - u1[0]) < tol);
    }
  }
}

//-----------------------------------------------------------------------------
TEST_CASE( "Test arbitray eval", "[eval]" )
{
  CHECK_NOTHROW(arbitrary_eval());
}