import simd

struct Box {
  var geometry: AABB
  var color: SIMD4<Float> = .one
}

struct Instance {
  let position: SIMD3<Float>
  let scale: SIMD3<Float>
  let rotation: simd_quatf
  let color: SIMD4<Float>

  init(
    position: SIMD3<Float> = .zero,
    scale: SIMD3<Float> = .one,
    rotation: simd_quatf = .identity,
    color: SIMD4<Float> = .one
  ) {
    self.position = position
    self.scale = scale
    self.rotation = rotation
    self.color = color
  }
}

let boxes: [Box] = [
  Box(geometry: .fromUnitCube(position: .init(0, -1, 0) * 2, scale: .init(10, 0.1, 10) * 2)),
  Box(geometry: .fromUnitCube(position: .init(-2.5, 0, -3) * 2, scale: .init(repeating: 2)), color: .init(1, 0.5, 0.75, 1)),
  Box(geometry: .fromUnitCube(position: .init(-2.5, -0.5, -5) * 2, scale: .init(repeating: 2)), color: .init(0.75, 1, 1, 1))
]

class Game: GameDelegate {

  private var fpsCalculator = FPSCalculator()
  var camera = Camera(fov: 60, size: .one, range: 0.06...50)
  var player = Player()
  var projection: matrix_float4x4 = .identity

  func fixedUpdate(_ time: GameTime) {
    
  }

  func update(_ time: GameTime) {
    fpsCalculator.frame(deltaTime: time.delta) { fps in
      print("FPS: \(fps)")
    }

    let deltaTime = min(Float(time.delta.asFloat), 1.0 / 15)

    player.update(deltaTime: deltaTime, boxes: boxes)
    camera.position = player.position
    camera.rotation =
      simd_quatf(angle: player.rotation.y, axis: .init(1, 0, 0)) *
      simd_quatf(angle: player.rotation.x, axis: .init(0, 1, 0))
  }

  func draw(_ renderer: Renderer, _ time: GameTime) {
    let totalTime = Float(time.total.asFloat)

    var instances: [Instance] = boxes.map {
      Instance(
        position: $0.geometry.center,
        scale: $0.geometry.size * 0.5,
        color: $0.color)
    }
    instances.append(
      Instance(
        position: .init(0, sin(totalTime * 1.5) * 0.5, -2) * 2,
        scale: .init(repeating: 0.5),
        rotation: .init(angle: totalTime * 3.0, axis: .init(0, 1, 0)),
        color: .init(0.5, 0.5, 1, 1)))
    renderer.batch(instances: instances, camera: self.camera)
  }

  func resize(_ size: Size<Int>) {
    self.camera.setSize(size)
  }
}