DataRepository.h

Go to the documentation of this file.

// This file is part of the ExaHyPE2 project. For conditions of distribution and
// use, please see the copyright notice at www.peano-framework.org
#pragma once
 
#include "Constants.h"
#include "celldata/FVSolverQ.h"
#include "celldata/FVSolverQReconstructed.h"
#include "facedata/FVSolverQNew.h"
#include "peano4/datamanagement/CellMarker.h"
#include "tarch/la/Vector.h"
 
#include <array>
#include <string>
#include <utility>
#include <vector>
 
namespace benchmarks::exahype2::kernelbenchmarks {
  class DataRepository {
  public:
    struct Indexing {
      static int linearise(
        const tarch::la::Vector<DIMENSIONS, int>&    cellCentre,
        const tarch::la::Vector<DIMENSIONS, double>& cellSize
      );

      static tarch::la::Vector<DIMENSIONS, int> getFaceIndices(
        const tarch::la::Vector<DIMENSIONS, double>& cellCentre,
        const tarch::la::Vector<DIMENSIONS, double>& cellSize
      );
 
      static int getFaceIndex(
        const tarch::la::Vector<DIMENSIONS, double>& cellCentre,
        const tarch::la::Vector<DIMENSIONS, double>& cellSize,
        int                                          axis
      );
    };
 
    static DataRepository& getInstance();
 
    DataRepository(const DataRepository&)            = delete;
    DataRepository& operator=(const DataRepository&) = delete;
    DataRepository(DataRepository&&)                 = delete;
    DataRepository& operator=(DataRepository&&)      = delete;
 
    ~DataRepository();
 
    double* getCellQ(int index);
    double* getCellQ(
      const tarch::la::Vector<DIMENSIONS, double>& cellCentre,
      const tarch::la::Vector<DIMENSIONS, double>& cellSize
    );
 
    double* getFaceQNew(int index, int axis);
    double* getFaceQNew(
      const tarch::la::Vector<DIMENSIONS, double>& cellCentre,
      const tarch::la::Vector<DIMENSIONS, double>& cellSize,
      int                                          axis
    );
    double* getFaceQNewDevice(
      const tarch::la::Vector<DIMENSIONS, double>& cellCentre,
      const tarch::la::Vector<DIMENSIONS, double>& cellSize,
      int                                          axis
    );
 
    void copyFacesToDevice();
    void copyFacesFromDevice();
 
    int getFaceCardinality();
    // QReconstructed is not stored explicitly anymore, as it is just used to store the patch+halo temporarily before
    // the Riemann step but we still need its size, therefore the getter.
    int getCellQReconstructedCardinality();
    int getCellQCardinality();
 
  private:
    friend struct celldata::FVSolverQ;
    friend struct celldata::FVSolverQReconstructed;
 
    using CellQ    = std::array<double, celldata::FVSolverQ::Cardinality>;
    using FaceQNew = std::array<double, facedata::FVSolverQNew::Cardinality>;
 
    struct DepthLevel {
      // cellData
      std::vector<CellQ> Q;
      // Other cellData would go here also as a std::vector<CellQ...>
 
      // Stores faces based on linearised order (face GridSize = 3^Depth * (3^Depth)+1) (not based on cellKeys)
      std::array<std::vector<FaceQNew>, DIMENSIONS> faces;
      std::array<double*, DIMENSIONS>               facesDevice;
    };
 
    DepthLevel _depthLevel;
 
    DataRepository();
  };
} // namespace benchmarks::exahype2::kernelbenchmarks