{%- include "structs.wgsl" -%}

@group(0) @binding(0)
var<storage, read> input_0: Array;

@group(0) @binding(1)
var<storage, read_write> output_0: Array;

@compute @workgroup_size(256, 1, 1)
fn main(@builtin(global_invocation_id) global_id: vec3<u32>) {
	let gidx = global_id.x;

	if (gidx < {{ o_lens[0] }}u) {
		var rest = gidx;
		{%- for chunk in o_chunks[0] -%}
			{% if loop.last %}
				let d_{{ loop.index0 }} = rest; 
			{% else %}
				let d_{{ loop.index0 }} = rest / {{ chunk }}u; 
				rest = gidx % {{ chunk }}u; 
			{% endif %}
		{%- endfor %}

		let index = 
			{%- for chunks in i_chunks[0] -%}
				{% set scale = scales | nth(n=loop.index0) %}
				{%- if not loop.first %}
					+ 
				{%- endif -%}
				u32(floor(
					(f32(d_{{ loop.index0 }}) + 0.5) / {{ scale }}
				)) * {{ chunks  }}u 
			{%- endfor -%}
		;

		output_0.data[gidx] = input_0.data[index];
	}
}