#ifndef _CUDA_SIMULATION_CUH_

#define _CUDA_SIMULATION_CUH_

#include <cuda_runtime.h>

#include "Particle.h"

#include "SimParams.h"

#include "SceneParams.h"

#include "NeighborSearch.h"

#define MAX_THREAD_NUM 512

void allocate_array(void** devPtr, size_t size);

void set_sim_params(SimParams& param_in);

void integrate_pbd(

ParticleSet* particles,

float deltaTime,

uint numParticles,

bool cd_on

);

void compute_grid_size(uint n, uint block_size, uint& num_blocks, uint& num_threads);

void calculate_hash(

CellData cell_data,

float3* pos,

uint num_particles

);

void sort_particles(

CellData cell_data,

uint numParticles

);

void reorder_data(

CellData cell_data,

float3* oldPos,

uint numParticles,

uint numCells

);

void compute_boundary_volume(

CellData data,

float* mass,

float* volume, // output: volume of particle

uint numParticles

);

void snow_simulation(

ParticleSet* sph_particles,

ParticleSet* dem_particles,

ParticleSet* boundary_particles,

ParticleDeviceData* phase_change_buffer,

CellData sph_cell_data,

CellData dem_cell_data,

CellData b_cell_data,

SceneParams& scene_params,

float dt,

int iterations,

bool sph_dem_correction,

bool sph_sph_correction,

bool dem_friction,

bool compute_temperature,

bool change_phase,

bool simulate_freezing,

bool simulate_melting,

bool dem_viscosity,

bool use_interlink,

bool dynamic_max_connections,

bool cd_on,

float temperature_variation

);

#endif