webgpu_manager.hpp

Go to the documentation of this file.

#pragma once
 
#include "utils_macros.hpp"
#include <webgpu/webgpu.hpp>
 
namespace netdem {
 
class Simulation;
struct ParticleDataGPU;
 
class WebGPUManager {
public:
  WebGPUManager();
 
  void Init(Simulation *const sim);
 
  void SendDomainData();
 
  void SendParticleData();
 
  VecXT<ParticleDataGPU> RecvParticleData();
 
  void MergeParticleData(VecXT<ParticleDataGPU> const &particle_data_list);
 
  wgpu::Device GetDevice();
 
  void Solve();
 
  ~WebGPUManager();
 
private:
  // A pointer to the simulation object that owns this scene.
  Simulation *sim{nullptr};
 
  wgpu::Buffer particle_buffer;
  size_t particle_size{0};
  size_t particle_data_size{0};
  size_t workgroup_size{32};
 
  // device
  wgpu::Device device;
};
 
} // namespace netdem