import type {Testbed} from "../Testbed";

type RAPIER_API = typeof import("@dimforge/rapier2d");

export function initWorld(RAPIER: RAPIER_API, testbed: Testbed) {
    let gravity = new RAPIER.Vector2(0.0, -9.81);
    let world = new RAPIER.World(gravity);
    let i, j;

    /*
     * Ground
     */
    let nsubdivs = 100;
    let points = [];
    let groundSize = 30.0;
    let stepSize = groundSize / nsubdivs;

    points.push(-groundSize / 2.0, 25.0);
    for (i = 1; i < nsubdivs; ++i) {
        let x = -groundSize / 2.0 + i * stepSize;
        let y = Math.cos(i * stepSize) * 2.0;
        points.push(x, y);
    }
    points.push(groundSize / 2.0, 25.0);
    let bodyDesc = RAPIER.RigidBodyDesc.fixed();
    let body = world.createRigidBody(bodyDesc);
    let colliderDesc = RAPIER.ColliderDesc.polyline(new Float32Array(points));
    world.createCollider(colliderDesc, body);

    /*
     * Create the cubes
     */
    let num = 10;
    let rad = 0.5;

    let shift = rad * 2.0;
    let centerx = shift * (num / 2);
    let centery = shift / 2.0;

    for (i = 0; i < num; ++i) {
        for (j = 0; j < num * 5; ++j) {
            let x = i * shift - centerx;
            let y = j * shift + centery + 3.0;

            // Build the rigid body.
            let bodyDesc = RAPIER.RigidBodyDesc.dynamic().setTranslation(x, y);
            let body = world.createRigidBody(bodyDesc);

            if (j % 2 == 0) {
                let colliderDesc = RAPIER.ColliderDesc.cuboid(rad, rad);
                world.createCollider(colliderDesc, body);
            } else {
                let colliderDesc = RAPIER.ColliderDesc.ball(rad);
                world.createCollider(colliderDesc, body);
            }
        }
    }

    testbed.setWorld(world);
    testbed.lookAt({
        target: {x: -10.0, y: -15.0},
        zoom: 10.0,
    });
}