DataLayoutConverter.h

Go to the documentation of this file.

 
#pragma once
 
#include "autopas/options/DataLayoutOption.h"
 
namespace autopas::utils {
 
template <class FunctorSoaWrapper>
class DataLayoutConverter {
 public:
  explicit DataLayoutConverter(FunctorSoaWrapper *functor, DataLayoutOption dataLayout)
      : _functor(functor), _dataLayout(dataLayout) {}
 
  template <class ParticleCell>
  void loadDataLayout(ParticleCell &cell) {
    // (Explicit) static cast required for Apple Clang (last tested version: 15.0.0)
    switch (static_cast<DataLayoutOption::Value>(_dataLayout)) {
      case DataLayoutOption::aos: {
        return;
      }
      case DataLayoutOption::soa: {
        _functor->SoALoader(cell, cell._particleSoABuffer, 0, /*skipSoAResize*/ false);
        return;
      }
    }
  }
 
  template <class ParticleCell>
  void storeDataLayout(ParticleCell &cell) {
    // (Explicit) static cast required for Apple Clang (last tested version: 15.0.0)
    switch (static_cast<DataLayoutOption::Value>(_dataLayout)) {
      case DataLayoutOption::aos: {
        return;
      }
      case DataLayoutOption::soa: {
        _functor->SoAExtractor(cell, cell._particleSoABuffer, 0);
        return;
      }
    }
  }
 
 private:
  FunctorSoaWrapper *_functor;
 
  DataLayoutOption _dataLayout;
};
 
}  // namespace autopas::utils