atlas4py/generated_Diffusion.hpp at master · Stagno/atlas4py · GitHub

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
#define DAWN_GENERATED 1
#define DAWN_BACKEND_T CXXNAIVEICO
#include <driver-includes/unstructured_interface.hpp>
namespace dawn_generated {
namespace cxxnaiveico {
template <typename LibTag>
class generated {
private:
  struct stencil_156 {
    dawn::mesh_t<LibTag> const& m_mesh;
    int m_k_size;
    dawn::cell_field_t<LibTag, double>& m_in_field;
    dawn::cell_field_t<LibTag, double>& m_out_field;

  public:
    stencil_156(dawn::mesh_t<LibTag> const& mesh, int k_size,
                dawn::cell_field_t<LibTag, double>& in_field,
                dawn::cell_field_t<LibTag, double>& out_field)
        : m_mesh(mesh), m_k_size(k_size), m_in_field(in_field), m_out_field(out_field) {}

    ~stencil_156() {}

    void sync_storages() {}

    void run() {
      using dawn::deref;
      ;
      {
        for(int k = 0 + 0; k <= (m_k_size == 0 ? 0 : (m_k_size - 1)) + 0 + 0; ++k) {
          for(auto const& loc : getCells(LibTag{}, m_mesh)) {
            int cnt;
            cnt = reduceCellToCell(LibTag{}, m_mesh, loc, (int)0,
                                   [&](auto& lhs, auto const& red_loc) { return lhs += (int)1; });
            m_out_field(deref(LibTag{}, loc), k + 0) = reduceCellToCell(
                LibTag{}, m_mesh, loc, ((-cnt) * m_in_field(deref(LibTag{}, loc), k + 0)),
                [&](auto& lhs, auto const& red_loc) {
                  return lhs += m_in_field(deref(LibTag{}, red_loc), k + 0);
                });
            m_out_field(deref(LibTag{}, loc), k + 0) =
                (m_in_field(deref(LibTag{}, loc), k + 0) +
                 ((::dawn::float_type)0.100000 * m_out_field(deref(LibTag{}, loc), k + 0)));
          }
        }
      }
      sync_storages();
    }
  };
  static constexpr const char* s_name = "generated";
  stencil_156 m_stencil_156;

public:
  generated(const generated&) = delete;

  // Members

  generated(const dawn::mesh_t<LibTag>& mesh, int k_size,
            dawn::cell_field_t<LibTag, double>& in_field,
            dawn::cell_field_t<LibTag, double>& out_field)
      : m_stencil_156(mesh, k_size, in_field, out_field) {}

  void run() {
    m_stencil_156.run();
    ;
  }
};
} // namespace cxxnaiveico
} // namespace dawn_generated