BrushTool.java

/*
 * Copyright 2017 Laszlo Balazs-Csiki
 *
 * This file is part of Pixelitor. Pixelitor is free software: you
 * can redistribute it and/or modify it under the terms of the GNU
 * General Public License, version 3 as published by the Free
 * Software Foundation.
 *
 * Pixelitor is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Pixelitor. If not, see <http://www.gnu.org/licenses/>.
 */

package pixelitor.tools;

import pixelitor.colors.ColorUtils;
import pixelitor.colors.FgBgColors;
import pixelitor.gui.ImageComponent;
import pixelitor.layers.Drawable;

import javax.swing.*;
import java.awt.Color;
import java.awt.Cursor;
import java.awt.Graphics2D;
import java.awt.Point;
import java.awt.Shape;
import java.awt.event.MouseEvent;

/**
 * The brush tool
 */
public class BrushTool extends BlendingModeBrushTool {
    private Color drawingColor;

    public BrushTool() {
        super('b', "Brush", "brush_tool_icon.png",
                "click and drag to draw with the current brush, Shift-click to draw lines, right-click to draw with the background color",
                Cursor.getDefaultCursor()
        );
    }

    @Override
    public void initSettingsPanel() {
        addTypeSelector();
        addBrushSettingsButton();
        settingsPanel.addSeparator();
        addSizeSelector();
        addSymmetryCombo();

        settingsPanel.addSeparator();

        addBlendingModePanel();
    }

    @Override
    public void mousePressed(MouseEvent e, ImageComponent ic) {
        setupDrawingColor(e);
        super.mousePressed(e, ic);
    }

    @Override
    protected void initializeGraphics(Graphics2D g) {
        // reinitialize the color for each stroke
        g.setColor(drawingColor);
    }

    @Override
    protected void prepareProgrammaticBrushStroke(Drawable dr, Point start) {
        super.prepareProgrammaticBrushStroke(dr, start);
        graphics.setColor(FgBgColors.getFG());
    }

    private void setupDrawingColor(MouseEvent e) {
        if (SwingUtilities.isRightMouseButton(e)) {
            drawingColor = FgBgColors.getBG();
        } else if (SwingUtilities.isMiddleMouseButton(e)) {
            // TODO we never get here because isAltDown is always true for middle-button events, even if Alt is not pressed?
            // See source comment in java.awt.Event for ALT_MASK
            Color fg = FgBgColors.getFG();
            Color bg = FgBgColors.getBG();
            if (e.isControlDown()) {
                drawingColor = ColorUtils.calcHSBAverage(fg, bg);
            } else {
                drawingColor = ColorUtils.calcRGBAverage(fg, bg);
            }
        } else {
            drawingColor = FgBgColors.getFG();
        }
    }

    @Override
    public void trace(Drawable dr, Shape shape) {
        drawingColor = FgBgColors.getFG();
        super.trace(dr, shape);
    }
}