async function main() {
  const adapter = await navigator.gpu?.requestAdapter();
  const device = await adapter?.requestDevice();
  if (!device) {
    fail('need a browser that supports WebGPU');
    return;
  }
  const canvas = document.querySelector('canvas');
  const context = canvas.getContext('webgpu');
  const presentationFormat = navigator.gpu.getPreferredCanvasFormat();
  context.configure({device, format: presentationFormat});

  let Nparticles = 2**8;
  let dt = 0.04;
  let mass = 1.0/Nparticles;
  let workgroup_size = 256;
  let Nworkgroups = Nparticles/workgroup_size;

  const driftComputeModule = device.createShaderModule({
    label: 'nbody compute module',
    code: `
    	struct Particle {
        pos: vec3f, // x y z p vx vy vz p
        vel: vec3f,
      };
      @group(0) @binding(0) var<storage, read_write> particles: array<Particle>;

      @compute @workgroup_size(${workgroup_size}) fn drift( 
        @builtin(global_invocation_id) id: vec3u,
        @builtin(local_invocation_index) local_invocation_index: u32,
        @builtin(num_workgroups) num_workgroups: vec3<u32>,
        @builtin(workgroup_id) workgroup_id : vec3<u32>,
      ) {
        // Drift
        let pi = workgroup_id.x * ${workgroup_size} + local_invocation_index;
        particles[pi].pos += 0.5* ${dt} * particles[pi].vel;
      }
    `,
  });

  const kickComputeModule = device.createShaderModule({
    label: 'nbody compute module',
    code: `
    	struct Particle {
        pos: vec3f, // x y z p vx vy vz p
        vel: vec3f,
      };
      @group(0) @binding(0) var<storage, read_write> particles: array<Particle>;

      @compute @workgroup_size(${workgroup_size}) fn step( 
        @builtin(global_invocation_id) id: vec3u,
        @builtin(local_invocation_index) local_invocation_index: u32,
        @builtin(num_workgroups) num_workgroups: vec3<u32>,
        @builtin(workgroup_id) workgroup_id : vec3<u32>,
      ) {
        let pi = workgroup_id.x * ${workgroup_size} + local_invocation_index;

        // Kick
        for (var pj = 0u; pj<${Nparticles}; pj++){
          if (pi != pj){
            let rel_pos = particles[pj].pos - particles[pi].pos;
            let d = length(rel_pos) + 0.001;
            particles[pi].vel += ${dt} * ${mass}/(d*d*d) * rel_pos;
          }
        }
      }
    `,
  });

  

  const renderModule = device.createShaderModule({
    label: 'nbody render module',
    code: `
    	struct Particle {
        pos: vec3f,
        vel: vec3f,
      };
      @group(0) @binding(0) var<storage, read> particles: array<Particle>;
      @group(0) @binding(1) var<uniform> aspect: f32;

      struct InOut {
        @builtin(position) position: vec4f,
        @location(0) tex: vec2f,
      };

      @vertex fn vs( @builtin(vertex_index) vertexIndex : u32 ) -> InOut {
        let quad = array(  vec2f( -1, -1), vec2f( -1,  1), vec2f(  1, -1),
                           vec2f(  1,  1), vec2f(  1, -1), vec2f( -1,  1) );
        var pos = 0.05 * quad[vertexIndex%6].xy + 0.25 * particles[vertexIndex/6].pos.xy;
        pos.x *= aspect;

        return InOut(vec4f(pos,0,1), quad[vertexIndex%6]);
      }

      @fragment fn fs(inOut: InOut) -> @location(0) vec4f {
        let phi = atan2(inOut.tex.y,inOut.tex.x);
        var r = 1.0-length(inOut.tex);
        let a = pow(clamp(sin(phi*6.0),0.0,1.0),5.0/r);
        return vec4f(1,1,1,pow(1.1*r,8.0) + r*a);
      }
    `,
  });


  const computeBindGroupLayout = device.createBindGroupLayout({
    entries: [
      {
        binding: 0,
        visibility: GPUShaderStage.COMPUTE,
        buffer: {
          type: 'storage',
          minBindingSize: 0,
        },
      },
    ],
  });

  const computePipelineLayout = device.createPipelineLayout({
    bindGroupLayouts: [ computeBindGroupLayout ],
  });

  const driftComputePipeline = device.createComputePipeline({
    label: 'drift compute pipeline',
    layout: computePipelineLayout,
    compute: {
      module: driftComputeModule,
    },
  });

const kickComputePipeline = device.createComputePipeline({
    label: 'kick compute pipeline',
    layout: computePipelineLayout,
    compute: {
      module: kickComputeModule,
    },
  });

  const renderPipeline = device.createRenderPipeline({
    label: 'render pipeline',
    layout: "auto",
    vertex: {
      module: renderModule,
    },
    fragment: {
      module: renderModule,
      targets: [
        { format: presentationFormat,
          blend: {
            color: { srcFactor: 'src-alpha', dstFactor: 'one-minus-src-alpha' },
            alpha: { srcFactor: 'src-alpha', dstFactor: 'one-minus-src-alpha' },
          },
        }
      ],
    },
  });

  const initialConditions = new Float32Array(8*Nparticles);
  for (var i = 0; i<Nparticles*8; i++){
    initialConditions[i] = Math.random()-0.5;
  }

  const workBuffer = device.createBuffer({
    label: 'work buffer',
    size: initialConditions.byteLength,
    usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC | GPUBufferUsage.COPY_DST,
  });
  device.queue.writeBuffer(workBuffer, 0, initialConditions);


  const computeBindGroup = device.createBindGroup({
    label: 'bindGroup for compute',
    layout: computeBindGroupLayout,
    entries: [ { binding: 0, resource: { buffer: workBuffer } } ]
  });

  const uniformValues = new Float32Array([1.0]);
  const uniformBuffer = device.createBuffer({
    label: 'uniform buffer',
    size: uniformValues.byteLength,
    usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST,
  });

  const renderBindGroup = device.createBindGroup({
    label: 'bindGroup for render',
    layout: renderPipeline.getBindGroupLayout(0),
    entries: [ 
      { binding: 0, resource: { buffer: workBuffer } }, 
      { binding: 1, resource: { buffer: uniformBuffer } } ]
  });

  const renderPassDescriptor = {
    label: 'our basic canvas renderPass',
    colorAttachments: [
      { clearValue: [0.0, 0.0, 0.0, 1],
        loadOp: 'clear', storeOp: 'store', },
    ],
  };

  var width = 100, height=100;

  function render() {
    const computeEncoder = device.createCommandEncoder();    
    const computePass = computeEncoder.beginComputePass();
    computePass.setBindGroup(0, computeBindGroup);
    computePass.setPipeline(driftComputePipeline);
    computePass.dispatchWorkgroups(Nworkgroups);
    computePass.setPipeline(kickComputePipeline);
    computePass.dispatchWorkgroups(Nworkgroups);
    computePass.setPipeline(driftComputePipeline);
    computePass.dispatchWorkgroups(Nworkgroups);
    computePass.end();
    const computeCommandBuffer = computeEncoder.finish();
    device.queue.submit([computeCommandBuffer]);

    renderPassDescriptor.colorAttachments[0].view = context.getCurrentTexture().createView();   
    uniformValues.set([height/width]);
    device.queue.writeBuffer(uniformBuffer, 0, uniformValues);
    const renderEncoder = device.createCommandEncoder();
    const renderPass = renderEncoder.beginRenderPass(renderPassDescriptor);
    renderPass.setPipeline(renderPipeline); 
    renderPass.setBindGroup(0, renderBindGroup);
    renderPass.draw(6*Nparticles); 
    renderPass.end();

    const renderCommandBuffer = renderEncoder.finish();
    device.queue.submit([renderCommandBuffer]);

    requestAnimationFrame(render);
  }

  const observer = new ResizeObserver(entries => {
    for (const entry of entries) {
      const canvas = entry.target;
      width = entry.contentBoxSize[0].inlineSize;
      height = entry.contentBoxSize[0].blockSize;
      canvas.width = Math.max(1, Math.min(width, device.limits.maxTextureDimension2D));
      canvas.height = Math.max(1, Math.min(height, device.limits.maxTextureDimension2D));
      render();
    }
  });
  observer.observe(canvas);
}

function fail(msg) {
  // eslint-disable-next-line no-alert
  alert(msg);
}

main();




/*
    /// Execute compute shaders
    const computeEncoder = device.createCommandEncoder();    
    const computePass = computeEncoder.beginComputePass();
    computePass.setBindGroup(0, computeBindGroup);
    computePass.setPipeline(computePipeline);
    computePass.dispatchWorkgroups(1);
    computePass.end();
    const computeCommandBuffer = computeEncoder.finish();
    device.queue.submit([computeCommandBuffer]);
*/


/*
  /// Shared bind group
  const computeBindGroupLayout = device.createBindGroupLayout({
    entries: [
      {
        binding: 0,
        visibility: GPUShaderStage.COMPUTE,
        buffer: {
          type: 'storage',
          minBindingSize: 0,
        },
      },
    ],
  });

  const computePipelineLayout = device.createPipelineLayout({
    bindGroupLayouts: [ computeBindGroupLayout ],
  });

  const computeBindGroup = device.createBindGroup({
    label: 'bindGroup for compute',
    layout: computeBindGroupLayout,
    entries: [ { binding: 0, resource: { buffer: workBuffer } } ]
  });

  const kickComputePipeline = device.createComputePipeline({
    label: 'kick compute pipeline',
    layout: computePipelineLayout,
    compute: {
      entryPoint: "kick",
      module: computeModule,
    },
  });

  const driftComputePipeline = device.createComputePipeline({
    label: 'drift compute pipeline',
    layout: computePipelineLayout,
    compute: {
      entryPoint: "drift",
      module: computeModule,
    },
  });
    
*/



/* 
  // Copy results to CPU

  const resultBuffer = device.createBuffer({
    label: 'result buffer',
    size: initialConditions.byteLength,
    usage: GPUBufferUsage.MAP_READ | GPUBufferUsage.COPY_DST,
  });

  const encoder = device.createCommandEncoder();
  encoder.copyBufferToBuffer(workBuffer, 0, resultBuffer, 0, resultBuffer.size);
  const commandBuffer = encoder.finish();
  device.queue.submit([commandBuffer]);

  await resultBuffer.mapAsync(GPUMapMode.READ);
  const result = new Float32Array(resultBuffer.getMappedRange().slice());
  console.log('result', result);
  resultBuffer.unmap();

*/