package de.neemann.digital.draw.shapes;

import de.neemann.digital.core.element.ElementAttributes;
import de.neemann.digital.core.element.Keys;
import de.neemann.digital.core.element.PinDescriptions;
import de.neemann.digital.draw.elements.Pin;
import de.neemann.digital.draw.elements.Pins;
import de.neemann.digital.draw.graphics.Graphic;
import de.neemann.digital.draw.graphics.Polygon;
import de.neemann.digital.draw.graphics.Style;
import de.neemann.digital.draw.graphics.Vector;

import static de.neemann.digital.draw.shapes.FGFETShapeN.CHARGED_GATE;
import static de.neemann.digital.draw.shapes.GenericShape.SIZE;
import static de.neemann.digital.draw.shapes.GenericShape.SIZE2;

/**
 * The p-chan floating gate FET shape
 */
public class FGFETShapeP extends FETShape {

    private final boolean programmed;

    /**
     * Creates a new instance
     *
     * @param attributes the attributes
     * @param inputs     the inputs
     * @param outputs    the outputs
     */
    public FGFETShapeP(ElementAttributes attributes, PinDescriptions inputs, PinDescriptions outputs) {
        super(attributes, inputs, outputs);
        programmed = attributes.get(Keys.BLOWN);
        setXOffs(SIZE2 + 1);
    }

    @Override
    public Pins getPins() {
        return new Pins()
                .add(new Pin(new Vector(0, 0), getInputs().get(0)))
                .add(new Pin(new Vector(SIZE, 0), getOutputs().get(0)))
                .add(new Pin(new Vector(SIZE, SIZE * 2), getOutputs().get(1)));
    }


    @Override
    public void drawTo(Graphic graphic, boolean highLight) {
        super.drawTo(graphic, highLight);

        if (programmed)
            graphic.drawLine(new Vector(6, 2 * SIZE - 4), new Vector(6, 4), CHARGED_GATE);
        else
            graphic.drawLine(new Vector(6, 2 * SIZE - 4), new Vector(6, 4), Style.THIN);


        // the arrow
        graphic.drawLine(new Vector(SIZE2 + 1, SIZE), new Vector(SIZE2 + 7, SIZE), Style.THIN);
        graphic.drawPolygon(new Polygon(true)
                .add(SIZE - SIZE2 / 3 + 5, SIZE)
                .add(SIZE2 + 7, SIZE - SIZE2 / 4)
                .add(SIZE2 + 7, SIZE + SIZE2 / 4), Style.THIN_FILLED);
    }

}
