#![allow(clippy::new_without_default)]
use core::iter::repeat;

use flax::*;
use glam::*;

component! {
    transform: Mat4,
    position: Vec3,
    rotation: Vec3,
    velocity: Vec3,
}

pub struct Benchmark(World, Query<(Component<Vec3>, Mutable<Vec3>)>);

impl Benchmark {
    pub fn new() -> Self {
        let mut world = World::new();

        let mut batch = BatchSpawn::new(10_000);
        batch
            .set(transform(), repeat(Mat4::from_scale(Vec3::ONE)))
            .unwrap();

        batch.set(position(), repeat(Vec3::X)).unwrap();
        batch.set(rotation(), repeat(Vec3::X)).unwrap();
        batch.set(velocity(), repeat(Vec3::X)).unwrap();

        batch.spawn(&mut world);
        let query = Query::new((velocity(), position().as_mut()));
        Self(world, query)
    }

    pub fn run(&mut self) {
        for (velocity, position) in
            &mut Query::new((velocity(), position().as_mut())).borrow(&self.0)
        {
            *position += *velocity
        }
    }

    pub fn run_manual_flatten(&mut self) {
        for (velocity, position) in &mut Query::new((velocity(), position().as_mut()))
            .borrow(&self.0)
            .iter_batched()
            .flatten()
        {
            *position += *velocity
        }
    }
}