scatter.cpp

#include "scatter.h"

bool TRTC_Scatter(const DVVectorLike& vec_in, const DVVectorLike& vec_map, DVVectorLike& vec_out)
{
	static TRTC_For s_for({ "view_vec_in", "view_vec_map", "view_vec_out"}, "idx",
		"    view_vec_out[view_vec_map[idx]] = (decltype(view_vec_out)::value_t)view_vec_in[idx];\n"
	);

	const DeviceViewable* args[] = { &vec_in, &vec_map, &vec_out };
	return s_for.launch_n(vec_in.size(), args);
}

bool TRTC_Scatter_If(const DVVectorLike& vec_in, const DVVectorLike& vec_map, const DVVectorLike& vec_stencil, DVVectorLike& vec_out)
{
	static TRTC_For s_for({ "view_vec_in", "view_vec_map", "view_vec_stencil", "view_vec_out"}, "idx",
		"    if(view_vec_stencil[idx])\n"
		"        view_vec_out[view_vec_map[idx]] = (decltype(view_vec_out)::value_t)view_vec_in[idx];\n"
	);

	const DeviceViewable* args[] = { &vec_in, &vec_map, &vec_stencil, &vec_out };
	return s_for.launch_n(vec_in.size(), args);
}

bool TRTC_Scatter_If(const DVVectorLike& vec_in, const DVVectorLike& vec_map, const DVVectorLike& vec_stencil, DVVectorLike& vec_out, const Functor& pred)
{
	static TRTC_For s_for({ "view_vec_in", "view_vec_map", "view_vec_stencil", "view_vec_out", "pred" }, "idx",
		"    if(pred(view_vec_stencil[idx]))\n"
		"        view_vec_out[view_vec_map[idx]] = (decltype(view_vec_out)::value_t)view_vec_in[idx];\n"
	);

	const DeviceViewable* args[] = { &vec_in, &vec_map, &vec_stencil, &vec_out, &pred};
	return s_for.launch_n(vec_in.size(), args);
}