package com.dfsek.terra.image;

import com.dfsek.terra.TerraWorld;
import com.dfsek.terra.biome.BiomeZone;
import com.dfsek.terra.biome.TerraBiomeGrid;
import com.dfsek.terra.config.base.ConfigUtil;
import com.dfsek.terra.debug.gui.DebugGUI;
import org.bukkit.World;
import org.polydev.gaea.biome.NormalizationUtil;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;

public class ImageLoader {
    private final BufferedImage image;
    private final Align align;
    private static final double INVERSE_ROOT_2 = 0.7071067811865475;

    public ImageLoader(File file, Align align) throws IOException {
        image = ImageIO.read(file);
        this.align = align;
    }

    public static void debugWorld(boolean genStep, World w) {
        if(!ConfigUtil.debug) return;
        BufferedImage newImg = new WorldImageGenerator(w, 1024, 1024).drawWorld(0, 0).getDraw();
        if(genStep) newImg = redrawStepped(newImg, w, Align.CENTER);
        DebugGUI debugGUI = new DebugGUI(newImg);
        debugGUI.start();
    }

    private static BufferedImage redrawStepped(BufferedImage original, World w, Align align) {
        BufferedImage newImg = copyImage(original);
        TerraBiomeGrid tb = TerraWorld.getWorld(w).getGrid();
        BiomeZone z = TerraWorld.getWorld(w).getZone();
        for(int x = 0; x < newImg.getWidth(); x++) {
            for(int y = 0; y < newImg.getHeight(); y++) {
                double[] noise;
                if(align.equals(Align.CENTER))
                    noise = tb.getGrid(x - original.getWidth() / 2, y - original.getHeight() / 2).getRawNoise(x - original.getWidth() / 2, y - original.getHeight() / 2);
                else noise = tb.getGrid(x, y).getRawNoise(x, y);
                newImg.setRGB(x, y, new Color((int) (NormalizationUtil.normalize(noise[0], tb.getGrid(x, y).getSizeX(), 4) * ((double) 255 / tb.getGrid(x, y).getSizeX())),
                        (int) (NormalizationUtil.normalize(noise[1], tb.getGrid(x, y).getSizeZ(), 4) * ((double) 255 / tb.getGrid(x, y).getSizeZ())),
                        (int) (NormalizationUtil.normalize(z.getNoise(x, y), z.getSize(), 4) * ((double) 255 / z.getSize())))
                        .getRGB());
            }
        }
        return newImg;
    }

    private static BufferedImage copyImage(BufferedImage source) {
        BufferedImage b = new BufferedImage(source.getWidth(), source.getHeight(), source.getType());
        Graphics g = b.getGraphics();
        g.drawImage(source, 0, 0, null);
        g.dispose();
        return b;
    }

    public void debug(boolean genStep, World w) {
        if (!ConfigUtil.debug) return;
        BufferedImage newImg = copyImage(image);
        if (genStep) {
            newImg = redrawStepped(image, w, align);
        }
        DebugGUI debugGUI = new DebugGUI(newImg);
        debugGUI.start();
    }

    public double getNoiseVal(int x, int y, Channel channel) {
        return ((double) (getChannel(x, y, channel) - 128) / 128) * INVERSE_ROOT_2;
    }

    public int getChannel(int x, int y, Channel channel) {
        int rgb;
        rgb = align.getRGB(image, x, y);
        switch(channel) {
            case RED:
                return rgb >> 16 & 0xff;
            case GREEN:
                return rgb >> 8 & 0xff;
            case BLUE:
                return rgb & 0xff;
            case ALPHA:
                return rgb >> 24 & 0xff;
            default:
                throw new IllegalArgumentException();
        }
    }

    public void debug(boolean genStep, World w) {
        if(!ConfigUtil.debug) return;
        BufferedImage newImg = copyImage(image);
        if(genStep) {
            newImg = redrawStepped(image, w, align);
        }
        DebugGUI debugGUI = new DebugGUI(newImg);
        debugGUI.start();
    }

    public double getNoiseVal(int x, int y, Channel channel) {
        return ((double) (getChannel(x, y, channel) - 128) / 128) * inverseRoot2;
    }

    public Align getAlign() {
        return align;
    }

    public enum Channel {
        RED, GREEN, BLUE, ALPHA
    }

    public enum Align {
        CENTER {
            @Override
            public int getRGB(BufferedImage image, int x, int y) {
                return Align.getRGBNoAlign(image, x - (image.getWidth() / 2), y - (image.getHeight() / 2));
            }
        },
        NONE {
            @Override
            public int getRGB(BufferedImage image, int x, int y) {
                return image.getRGB(Math.floorMod(x, image.getWidth()), Math.floorMod(y, image.getHeight()));
            }
        };

        private static int getRGBNoAlign(BufferedImage image, int x, int y) {
            return image.getRGB(Math.floorMod(x, image.getWidth()), Math.floorMod(y, image.getHeight()));
        }

        public abstract int getRGB(BufferedImage image, int x, int y);
    }
}