LRFaceData.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 "peano4/utils/Globals.h"
#include "tarch/accelerator/accelerator.h"
#include "tarch/accelerator/MemoryManager.h"
#include "tarch/la/Vector.h"
 
#include <functional>
#include <string>
 
namespace exahype2 {
  template <typename inType = double, typename outType = double> struct FaceData {
    inType* (*QIn)[2];
    tarch::la::Vector<DIMENSIONS, double>* faceCentre;
    tarch::la::Vector<DIMENSIONS, double>* faceSize;
 
    double* t;
    double* dt;

    int* id;

    const int numberOfFaces;

    const tarch::MemoryLocation memoryLocation;

    const int targetDevice;

    outType* (*QOut)[2];

    FaceData(
      inType*                                      QIn_[2],
      const tarch::la::Vector<DIMENSIONS, double>& faceCentre_,
      const tarch::la::Vector<DIMENSIONS, double>& faceSize_,
      double                                       t_,
      double                                       dt_,
      outType*                                     QOut_[2],
      tarch::MemoryLocation                        memoryLocation_ = tarch::MemoryLocation::Heap,
      int                                          targetDevice_   = tarch::accelerator::Device::DefaultDevice
    );
 
    FaceData(
      int                   numberOfFaces_,
      tarch::MemoryLocation memoryLocation_ = tarch::MemoryLocation::Heap,
      int                   targetDevice_   = tarch::accelerator::Device::DefaultDevice
    );
 
    FaceData(const FaceData& copy)       = delete;
    FaceData& operator=(const FaceData&) = delete;
 
    ~FaceData();
 
    std::string toString() const;
  };
 
} // namespace exahype2
 
template <typename inType, typename outType>
exahype2::FaceData<inType, outType>::FaceData(
  inType*                                      QIn_[2],
  const tarch::la::Vector<DIMENSIONS, double>& faceCentre_,
  const tarch::la::Vector<DIMENSIONS, double>& faceSize_,
  double                                       t_,
  double                                       dt_,
  outType*                                     QOut_[2],
  tarch::MemoryLocation                        memoryLocation_,
  int                                          targetDevice_
):
  FaceData(1, memoryLocation_, targetDevice_) {
  QIn[0]        = QIn_;
  faceCentre[0] = faceCentre_;
  faceSize[0]   = faceSize_;
  t[0]          = t_;
  dt[0]         = dt_;
  QOut[0]       = QOut_;
  id[0]         = -1;
}
 
template <typename inType, typename outType>
exahype2::FaceData<inType, outType>::
  FaceData(int numberOfFaces_, tarch::MemoryLocation memoryLocation_, int targetDevice_):
  numberOfFaces(numberOfFaces_),
  memoryLocation(memoryLocation_),
  targetDevice(targetDevice_) {
 
  QIn = tarch::accelerator::MemoryManager::getInstance().allocate<inType* [2]>(
    numberOfFaces_,
    memoryLocation_,
    targetDevice_
  );
  faceCentre = tarch::accelerator::MemoryManager::getInstance().allocate<tarch::la::Vector<DIMENSIONS, double>>(
    numberOfFaces_,
    memoryLocation_,
    targetDevice_
  );
  faceSize = tarch::accelerator::MemoryManager::getInstance().allocate<tarch::la::Vector<DIMENSIONS, double>>(
    numberOfFaces_,
    memoryLocation_,
    targetDevice_
  );
  t = tarch::accelerator::MemoryManager::getInstance().allocate<double>(numberOfFaces_, memoryLocation_, targetDevice_);
  dt = tarch::accelerator::MemoryManager::getInstance().allocate<double>(
    numberOfFaces_,
    memoryLocation_,
    targetDevice_
  );
  id   = tarch::accelerator::MemoryManager::getInstance().allocate<int>(numberOfFaces_, memoryLocation_, targetDevice_);
  QOut = tarch::accelerator::MemoryManager::getInstance().allocate<outType* [2]>(
    numberOfFaces_,
    memoryLocation_,
    targetDevice_
  );
}
 
template <typename inType, typename outType> exahype2::FaceData<inType, outType>::~FaceData() {
  // tarch::freeMemory(QIn, memoryLocation, targetDevice);
  // tarch::freeMemory(faceCentre, memoryLocation, targetDevice);
  // tarch::freeMemory(faceSize, memoryLocation, targetDevice);
  // tarch::freeMemory(t, memoryLocation, targetDevice);
  // tarch::freeMemory(dt, memoryLocation, targetDevice);
  // tarch::freeMemory(QOut, memoryLocation, targetDevice);
  // tarch::freeMemory(id, memoryLocation, targetDevice);
 
  tarch::accelerator::MemoryManager::getInstance().free(QIn, targetDevice);
  tarch::accelerator::MemoryManager::getInstance().free(faceCentre, targetDevice);
  tarch::accelerator::MemoryManager::getInstance().free(faceSize, targetDevice);
  tarch::accelerator::MemoryManager::getInstance().free(t, targetDevice);
  tarch::accelerator::MemoryManager::getInstance().free(dt, targetDevice);
  tarch::accelerator::MemoryManager::getInstance().free(QOut, targetDevice);
  tarch::accelerator::MemoryManager::getInstance().free(id, targetDevice);
}
 
template <typename inType, typename outType> std::string exahype2::FaceData<inType, outType>::toString() const {
  std::ostringstream msg;
  msg << "[";
  for (int i = 0; i < numberOfFaces; i++) {
    msg << "(x=" << faceCentre[i] << ",h=" << faceSize[i] << ",t=" << t[i] << ",dt=" << dt[i] << ",id=" << id[i] << ")";
  }
  msg << "]";
  return msg.str();
}