EnclaveDataRepository.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 "tarch/la/Vector.h"
 
#include <algorithm>
#include <iostream>
#include <map>
#include <string>
#include <vector>
 
namespace benchmarks::exahype2::kernelbenchmarks {
 
  class EnclaveDataRepository {
  public:
    static EnclaveDataRepository& getInstance();
 
    EnclaveDataRepository(const EnclaveDataRepository&)            = delete;
    EnclaveDataRepository& operator=(const EnclaveDataRepository&) = delete;
    EnclaveDataRepository(EnclaveDataRepository&&)                 = delete;
    EnclaveDataRepository& operator=(EnclaveDataRepository&&)      = delete;
 
    ~EnclaveDataRepository();
 
    void startCollection();
 
    void bookmarkCell(
      const tarch::la::Vector<DIMENSIONS, double>& cellCentre,
      const tarch::la::Vector<DIMENSIONS, double>& cellSize
    );
 
    void buildFromBookmarks();
 
    void clear();
 
    void syncCellsDeviceToHost();
    void syncFacesDeviceToHost();
 
    void syncCellsHostToDevice();
    void syncFacesHostToDevice();
 
    // --- GPU Data Accessors ---
    double* getQ();
    double* getQReconstructed();
 
    uint64_t* getFaceIndices();
    double*   getFacePool();
 
    uint64_t* getBoundaryFaceIndices();
    uint8_t*  getBoundaryFaceNumbers();
    uint64_t* getBoundaryCellIndices();
 
    tarch::la::Vector<DIMENSIONS, double>* getCellCentres();
    tarch::la::Vector<DIMENSIONS, double>* getCellSizes();
 
    int  getNumberOfEnclaveCells() const;
    int  getNumberOfBoundaryFaces() const;
    bool isReady() const;
 
    double getMinVolumeH() const;
    double getMaxVolumeH() const;
 
  private:
    EnclaveDataRepository();
 
    double* _combinedQBlockDevice = nullptr;
 
    using FaceKey = std::tuple<int, int, int>;
 
    double _minVolumeH;
    double _maxVolumeH;
 
    // --- State Management ---
    bool _isReady;
    int  _numberOfEnclaveCells;
    int  _numberOfBoundaryFaces;
 
    // --- GPU Device Pointers ---
    double* _QDevice;
    double* _QReconstructedDevice;
 
    tarch::la::Vector<DIMENSIONS, double>* _cellCentresDevice;
    tarch::la::Vector<DIMENSIONS, double>* _cellSizesDevice;
 
    // --- Face Data Management ---
    double*   _globalFacePoolDevice;
    uint64_t* _indicesOfFacesDevice;
 
    uint64_t* _indicesOfBoundaryFacesDevice;
    uint8_t*  _boundaryFaceNumbersDevice;
    uint64_t* _boundaryCellIndicesDevice;
 
    std::map<FaceKey, uint64_t> _uniqueFaceMap;
 
    // --- Host-side Bookmarking Data (temporary) ---
    std::vector<tarch::la::Vector<DIMENSIONS, double>> _bookmarkedCellCentres;
    std::vector<tarch::la::Vector<DIMENSIONS, double>> _bookmarkedCellSizes;
  };
} // namespace benchmarks::exahype2::kernelbenchmarks