import org.frice.Game;
import org.frice.anim.move.AccelerateMove;
import org.frice.anim.move.SimpleMove;
import org.frice.obj.PhysicalObject;
import org.frice.obj.SideEffect;
import org.frice.obj.sub.ShapeObject;
import org.frice.resource.graphics.ColorResource;
import org.frice.utils.QuadTree;
import org.frice.utils.message.FLog;
import org.frice.utils.shape.FCircle;
import org.frice.utils.shape.FQuad;
import org.frice.utils.shape.FRectangle;
import org.frice.utils.time.FTimer;
import org.jetbrains.annotations.NotNull;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Random;

import static org.frice.Initializer.launch;

public class Demo7QuadTree extends Game {
	public static void main(String[] args) {
		launch(Demo7QuadTree.class);
	}

	private Random random = new Random();
	private FTimer timer = new FTimer(3000);
	private ShapeObject bird;
	private SideEffect gameOver;

	@Override
	public void onInit() {
		setSize(500, 800);
		setTitle("Flappy bird demo by ice1000");
		bird = new ShapeObject(ColorResource.宝强绿, new FCircle(20.0), 50.0, 200.0);
		bird.addAnim(AccelerateMove.getGravity());
		addObject(bird);
		gameOver = () -> {
			dialogShow("Game Over");
			System.exit(0);
		};
		addKeyListener(null, e -> {
			bird.stopAnims();
			bird.addAnim(AccelerateMove.getGravity());
			bird.addAnim(new SimpleMove(0, -400));
		}, null);
	}

	@Override
	public void onRefresh() {
		QuadTree quadTree = new QuadTree(0, new FQuad(0, 0, 500, 800));
		if (bird.getY() > getHeight() + 20) gameOver.invoke();
		if (timer.ended()) {
			quadTree.clear();
			ShapeObject[] newObjects = getObj(random.nextInt(400));
			instantAddObject(newObjects);
			Arrays.stream(newObjects).forEachOrdered(quadTree::insert);
		}

		ArrayList<List<PhysicalObject>> returnObjects = new ArrayList<>();
		quadTree.retrieve(returnObjects, bird);
		FLog.e(" size " + returnObjects.size());

		returnObjects.stream().filter(objs -> objs.contains(bird)).forEachOrdered(rets -> {
			rets.remove(bird);
			rets.forEach(wall -> {
				((ShapeObject) wall).setRes(ColorResource.基佬紫);
				if (bird.collides(wall)) gameOver.invoke();
			});
		});
	}

	@NotNull
	private ShapeObject[] getObj(int height) {
		return new ShapeObject[]{new ShapeObject(ColorResource.高坂穗乃果, new FRectangle(50, height), 550, 0) {{
			addAnim(new SimpleMove(-150, 0));
			addCollider(bird, gameOver);
		}}, new ShapeObject(ColorResource.高坂穗乃果, new FRectangle(50, getHeight() - height - 400), 550, height + 400) {{
			addAnim(new SimpleMove(-150, 0));
			addCollider(bird, gameOver);
		}}};
	}
}