package io.github.moulberry.notenoughupdates.mbgui;

import org.lwjgl.util.vector.Vector2f;

import java.util.Collection;
import java.util.List;

public abstract class MBGuiGroupAligned extends MBGuiGroup {

    //Serialized
    private List<MBGuiElement> children;
    private boolean vertical;

    public MBGuiGroupAligned(List<MBGuiElement> children, boolean vertical) {
        this.children = children;
        this.vertical = vertical;
        recalculate();
    }

    public abstract int getPadding();

    public Collection<MBGuiElement> getChildren() {
        return children;
    }

    public void recalculate() {
        for(MBGuiElement child : children) {
            child.recalculate();
        }

        if(vertical) {
            height = 0;
            for(int i=0; i<children.size(); i++) {
                MBGuiElement child = children.get(i);
                childrenPosition.put(child, new Vector2f(0, height));
                height += child.getHeight();
                if(i != children.size()-1) height += getPadding();
            }

            width = 0;
            for(MBGuiElement child : children) {
                int childWidth = child.getWidth();
                if(childWidth > width) {
                    width = childWidth;
                }
            }
        } else {
            width = 0;
            for(int i=0; i<children.size(); i++) {
                MBGuiElement child = children.get(i);
                childrenPosition.put(child, new Vector2f(width, 0));
                width += child.getWidth();
                if(i != children.size()-1) width += getPadding();
            }

            height = 0;
            for(MBGuiElement child : children) {
                int childHeight = child.getHeight();
                if(childHeight > height) {
                    height = childHeight;
                }
            }
        }

    }

}