r4fterman/pdf.forms

package org.pdf.forms.widgets.components;

import java.awt.*;
import java.awt.geom.Rectangle2D;

import javax.swing.*;

import org.pdf.forms.fonts.FontHandler;
import org.pdf.forms.widgets.IWidget;

public class PdfButton extends JButton implements IPdfComponent {

    private static final float FONT_SIZE = 11f;

    public PdfButton(
            final String buttonText,
            final FontHandler fontHandler) {
        super(buttonText);
        setFont(fontHandler.getDefaultFont().deriveFont(FONT_SIZE));

        /* needed so the background color of the button can be set */
        setContentAreaFilled(false);
        setOpaque(true);
    }

    private boolean standardUnderline;
    private boolean isStrikethrough;
    private boolean doubleUnderline;
    private boolean wordUnderline;

    @Override
    public void paintComponent(final Graphics graphics) {
        super.paintComponent(graphics);

        final String text = getText();
        final FontMetrics fontMetrics = getFontMetrics(getFont());
        final Rectangle2D textBounds = fontMetrics.getStringBounds(text, graphics);

        final int x = getInsets().left;
        final int y = getHeight() / 2 + (int) (textBounds.getHeight() / 2);
        final int width = (int) textBounds.getWidth() + x;

        if (standardUnderline) {
            graphics.drawLine(x, y, width, y);

            if (doubleUnderline) {
                graphics.drawLine(x, y + 2, width, y + 2);
            }
        } else if (wordUnderline) {
            underlineWord(graphics, text, fontMetrics, x, y);
        }

        if (isStrikethrough) {
            graphics.drawLine(x, getHeight() / 2, width, getHeight() / 2);
        }
    }

    private void underlineWord(
            final Graphics graphics,
            final String text,
            final FontMetrics fontMetrics,
            final int x,
            final int y) {
        int startX = x;
        for (int i = 0; i < text.length(); i++) {
            final char currentChar = text.charAt(i);
            final int charWidth = fontMetrics.charWidth(currentChar);
            if (currentChar != ' ') {
                graphics.drawLine(startX, y, startX + charWidth, y);
                if (doubleUnderline) {
                    graphics.drawLine(startX, y + 2, startX + charWidth, y + 2);
                }
            }
            startX += charWidth;
        }
    }

    @Override
    public void setUnderlineType(final int type) {
        if (type == IWidget.UNDERLINE_SINGLE) {
            standardUnderline = true;
            doubleUnderline = false;
            wordUnderline = false;
        } else if (type == IWidget.UNDERLINE_DOUBLE) {
            standardUnderline = true;
            doubleUnderline = true;
            wordUnderline = false;
        } else if (type == IWidget.UNDERLINE_WORD_SINGLE) {
            standardUnderline = false;
            doubleUnderline = false;
            wordUnderline = true;
        } else if (type == IWidget.UNDERLINE_WORD_DOUBLE) {
            standardUnderline = false;
            doubleUnderline = true;
            wordUnderline = true;
        }
    }

    @Override
    public void setStrikethrough(final boolean isStrikethrough) {
        this.isStrikethrough = isStrikethrough;
    }
}