tahoma2d/toonz/sources/tnztools/fullcolorbrushtool.cpp
2022-04-10 12:21:41 -04:00

1210 lines
43 KiB
C++

#include "fullcolorbrushtool.h"
// TnzTools includes
#include "tools/tool.h"
#include "tools/cursors.h"
#include "tools/toolutils.h"
#include "tools/toolhandle.h"
#include "tools/tooloptions.h"
#include "mypainttoonzbrush.h"
// TnzQt includes
#include "toonzqt/dvdialog.h"
// TnzLib includes
#include "toonz/tpalettehandle.h"
#include "toonz/txsheethandle.h"
#include "toonz/txshlevelhandle.h"
#include "toonz/tobjecthandle.h"
#include "toonz/tframehandle.h"
#include "toonz/ttileset.h"
#include "toonz/ttilesaver.h"
#include "toonz/strokegenerator.h"
#include "toonz/tstageobject.h"
#include "toonz/palettecontroller.h"
#include "toonz/mypaintbrushstyle.h"
#include "toonz/preferences.h"
#include "toonz/toonzfolders.h"
#include "toonz/tcolumnhandle.h"
#include "toonz/tstageobjectcmd.h"
// TnzCore includes
#include "tgl.h"
#include "tproperty.h"
#include "trasterimage.h"
#include "tenv.h"
#include "tpalette.h"
#include "trop.h"
#include "tstream.h"
#include "tstroke.h"
#include "timagecache.h"
#include "tpixelutils.h"
#include "tsystem.h"
#include "perspectivetool.h"
// Qt includes
#include <QCoreApplication> // Qt translation support
//----------------------------------------------------------------------------------
TEnv::IntVar FullcolorBrushMinSize("FullcolorBrushMinSize", 1);
TEnv::IntVar FullcolorBrushMaxSize("FullcolorBrushMaxSize", 5);
TEnv::IntVar FullcolorPressureSensitivity("FullcolorPressureSensitivity", 1);
TEnv::DoubleVar FullcolorBrushHardness("FullcolorBrushHardness", 100);
TEnv::DoubleVar FullcolorMinOpacity("FullcolorMinOpacity", 100);
TEnv::DoubleVar FullcolorMaxOpacity("FullcolorMaxOpacity", 100);
TEnv::DoubleVar FullcolorModifierSize("FullcolorModifierSize", 0);
TEnv::DoubleVar FullcolorModifierOpacity("FullcolorModifierOpacity", 100);
TEnv::IntVar FullcolorModifierEraser("FullcolorModifierEraser", 0);
TEnv::IntVar FullcolorModifierLockAlpha("FullcolorModifierLockAlpha", 0);
TEnv::StringVar FullcolorBrushPreset("FullcolorBrushPreset", "<custom>");
TEnv::IntVar FullcolorBrushSnapGrid("FullcolorBrushSnapGrid", 0);
//----------------------------------------------------------------------------------
#define CUSTOM_WSTR L"<custom>"
//----------------------------------------------------------------------------------
namespace {
class FullColorBrushUndo final : public ToolUtils::TFullColorRasterUndo {
TPoint m_offset;
QString m_id;
public:
FullColorBrushUndo(TTileSetFullColor *tileSet, TXshSimpleLevel *level,
const TFrameId &frameId, bool isFrameCreated,
const TRasterP &ras, const TPoint &offset)
: ToolUtils::TFullColorRasterUndo(tileSet, level, frameId, isFrameCreated,
false, 0)
, m_offset(offset) {
static int counter = 0;
m_id = QString("FullColorBrushUndo") + QString::number(counter++);
TImageCache::instance()->add(m_id.toStdString(), TRasterImageP(ras));
}
~FullColorBrushUndo() { TImageCache::instance()->remove(m_id); }
void redo() const override {
insertLevelAndFrameIfNeeded();
TRasterImageP image = getImage();
TRasterP ras = image->getRaster();
TRasterImageP srcImg =
TImageCache::instance()->get(m_id.toStdString(), false);
ras->copy(srcImg->getRaster(), m_offset);
TTool::getApplication()->getCurrentXsheet()->notifyXsheetChanged();
notifyImageChanged();
}
int getSize() const override {
return sizeof(*this) + ToolUtils::TFullColorRasterUndo::getSize();
}
QString getToolName() override { return QString("Raster Brush Tool"); }
int getHistoryType() override { return HistoryType::BrushTool; }
};
} // namespace
//************************************************************************
// FullColor Brush Tool implementation
//************************************************************************
FullColorBrushTool::FullColorBrushTool(std::string name)
: TTool(name)
, m_thickness("Size", 1, 1000, 1, 5, false)
, m_pressure("Pressure", true)
, m_opacity("Opacity", 0, 100, 100, 100, true)
, m_hardness("Hardness:", 0, 100, 100)
, m_modifierSize("ModifierSize", -3, 3, 0, true)
, m_modifierOpacity("ModifierOpacity", 0, 100, 100, true)
, m_modifierEraser("ModifierEraser", false)
, m_modifierLockAlpha("Lock Alpha", false)
, m_preset("Preset:")
, m_minCursorThick(0)
, m_maxCursorThick(0)
, m_enabledPressure(false)
, m_toonz_brush(0)
, m_tileSet(0)
, m_tileSaver(0)
, m_notifier(0)
, m_presetsLoaded(false)
, m_firstTime(true)
, m_snapGrid("Grid", false) {
bind(TTool::RasterImage | TTool::EmptyTarget);
m_thickness.setNonLinearSlider();
m_prop.bind(m_thickness);
m_prop.bind(m_hardness);
m_prop.bind(m_opacity);
m_prop.bind(m_modifierSize);
m_prop.bind(m_modifierOpacity);
m_prop.bind(m_modifierEraser);
m_prop.bind(m_modifierLockAlpha);
m_prop.bind(m_pressure);
m_prop.bind(m_snapGrid);
m_prop.bind(m_preset);
m_preset.setId("BrushPreset");
m_modifierEraser.setId("RasterEraser");
m_modifierLockAlpha.setId("LockAlpha");
m_pressure.setId("PressureSensitivity");
m_snapGrid.setId("SnapGrid");
m_brushTimer.start();
}
//---------------------------------------------------------------------------------------------------
ToolOptionsBox *FullColorBrushTool::createOptionsBox() {
TPaletteHandle *currPalette =
TTool::getApplication()->getPaletteController()->getCurrentLevelPalette();
ToolHandle *currTool = TTool::getApplication()->getCurrentTool();
return new BrushToolOptionsBox(0, this, currPalette, currTool);
}
//---------------------------------------------------------------------------------------------------
void FullColorBrushTool::onCanvasSizeChanged() {
onDeactivate();
setWorkAndBackupImages();
}
//---------------------------------------------------------------------------------------------------
void FullColorBrushTool::onColorStyleChanged() {
getApplication()->getCurrentTool()->notifyToolChanged();
}
//---------------------------------------------------------------------------------------------------
void FullColorBrushTool::updateTranslation() {
m_thickness.setQStringName(tr("Size"));
m_pressure.setQStringName(tr("Pressure"));
m_opacity.setQStringName(tr("Opacity"));
m_hardness.setQStringName(tr("Hardness:"));
m_preset.setQStringName(tr("Preset:"));
m_modifierSize.setQStringName(tr("Size"));
m_modifierOpacity.setQStringName(tr("Opacity"));
m_modifierEraser.setQStringName(tr("Eraser"));
m_modifierLockAlpha.setQStringName(tr("Lock Alpha"));
m_snapGrid.setQStringName(tr("Grid"));
}
//---------------------------------------------------------------------------------------------------
void FullColorBrushTool::onActivate() {
if (!m_notifier) m_notifier = new FullColorBrushToolNotifier(this);
updateCurrentStyle();
if (m_firstTime) {
m_firstTime = false;
std::wstring wpreset =
QString::fromStdString(FullcolorBrushPreset.getValue()).toStdWString();
if (wpreset != CUSTOM_WSTR) {
initPresets();
if (!m_preset.isValue(wpreset)) wpreset = CUSTOM_WSTR;
m_preset.setValue(wpreset);
FullcolorBrushPreset = m_preset.getValueAsString();
loadPreset();
} else
loadLastBrush();
}
setWorkAndBackupImages();
onColorStyleChanged();
}
//--------------------------------------------------------------------------------------------------
void FullColorBrushTool::onDeactivate() {
if (m_mousePressed) leftButtonUp(m_mousePos, m_mouseEvent);
m_workRaster = TRaster32P();
m_backUpRas = TRasterP();
}
//--------------------------------------------------------------------------------------------------
void FullColorBrushTool::updateWorkAndBackupRasters(const TRect &rect) {
if (rect.isEmpty()) return;
TRasterImageP ri = TImageP(getImage(false, 1));
if (!ri) return;
TRasterP ras = ri->getRaster();
const int denominator = 8;
TRect enlargedRect = rect + m_lastRect;
int dx = (enlargedRect.getLx() - 1) / denominator + 1;
int dy = (enlargedRect.getLy() - 1) / denominator + 1;
if (m_lastRect.isEmpty()) {
enlargedRect.x0 -= dx;
enlargedRect.y0 -= dy;
enlargedRect.x1 += dx;
enlargedRect.y1 += dy;
TRect _rect = enlargedRect * ras->getBounds();
if (_rect.isEmpty()) return;
m_workRaster->extract(_rect)->copy(ras->extract(_rect));
m_backUpRas->extract(_rect)->copy(ras->extract(_rect));
} else {
if (enlargedRect.x0 < m_lastRect.x0) enlargedRect.x0 -= dx;
if (enlargedRect.y0 < m_lastRect.y0) enlargedRect.y0 -= dy;
if (enlargedRect.x1 > m_lastRect.x1) enlargedRect.x1 += dx;
if (enlargedRect.y1 > m_lastRect.y1) enlargedRect.y1 += dy;
TRect _rect = enlargedRect * ras->getBounds();
if (_rect.isEmpty()) return;
TRect _lastRect = m_lastRect * ras->getBounds();
QList<TRect> rects = ToolUtils::splitRect(_rect, _lastRect);
for (int i = 0; i < rects.size(); i++) {
m_workRaster->extract(rects[i])->copy(ras->extract(rects[i]));
m_backUpRas->extract(rects[i])->copy(ras->extract(rects[i]));
}
}
m_lastRect = enlargedRect;
}
//--------------------------------------------------------------------------------------------------
bool FullColorBrushTool::askRead(const TRect &rect) { return askWrite(rect); }
//--------------------------------------------------------------------------------------------------
bool FullColorBrushTool::askWrite(const TRect &rect) {
if (rect.isEmpty()) return true;
m_strokeRect += rect;
m_strokeSegmentRect += rect;
updateWorkAndBackupRasters(rect);
m_tileSaver->save(rect);
return true;
}
//--------------------------------------------------------------------------------------------------
bool FullColorBrushTool::preLeftButtonDown() {
touchImage();
if (m_isFrameCreated) {
setWorkAndBackupImages();
// When the xsheet frame is selected, whole viewer will be updated from
// SceneViewer::onXsheetChanged() on adding a new frame.
// We need to take care of a case when the level frame is selected.
if (m_application->getCurrentFrame()->isEditingLevel()) invalidate();
}
return true;
}
//---------------------------------------------------------------------------------------------------
void FullColorBrushTool::leftButtonDown(const TPointD &pos,
const TMouseEvent &e) {
TPointD previousBrushPos = m_brushPos;
m_brushPos = m_mousePos = pos;
m_mousePressed = true;
m_mouseEvent = e;
Viewer *viewer = getViewer();
if (!viewer) return;
TRasterImageP ri = (TRasterImageP)getImage(true);
if (!ri) ri = (TRasterImageP)touchImage();
if (!ri) return;
TTool::Application *app = TTool::getApplication();
if (!app) return;
TXshLevel *level = app->getCurrentLevel()->getLevel();
if (level == NULL) {
m_active = false;
return;
}
if ((e.isShiftPressed() || e.isCtrlPressed()) && !e.isAltPressed()) {
m_isStraight = true;
m_firstPoint = pos;
m_lastPoint = pos;
}
if (e.isAltPressed()) {
m_isStraight = true;
m_firstPoint = pos;
m_lastPoint = pos;
}
if (m_snapGrid.getValue()) {
m_firstPoint = pos;
m_lastPoint = pos;
}
/* update color here since the current style might be switched with numpad
* shortcut keys */
updateCurrentStyle();
TRasterP ras = ri->getRaster();
if (!(m_workRaster && m_backUpRas)) setWorkAndBackupImages();
m_workRaster->lock();
TPointD rasCenter = ras->getCenterD();
TPointD point(pos + rasCenter);
double pressure;
if (getApplication()->getCurrentLevelStyle()->getTagId() ==
4001) // mypaint brush case
pressure = m_enabledPressure && e.isTablet() ? e.m_pressure : 0.5;
else
pressure = m_enabledPressure ? e.m_pressure : 1.0;
m_oldPressure = pressure;
m_tileSet = new TTileSetFullColor(ras->getSize());
m_tileSaver = new TTileSaverFullColor(ras, m_tileSet);
mypaint::Brush mypaintBrush;
applyToonzBrushSettings(mypaintBrush);
m_toonz_brush = new MyPaintToonzBrush(m_workRaster, *this, mypaintBrush);
m_strokeRect.empty();
m_strokeSegmentRect.empty();
m_toonz_brush->beginStroke();
m_toonz_brush->strokeTo(point, pressure, restartBrushTimer());
TRect updateRect = m_strokeSegmentRect * ras->getBounds();
if (!updateRect.isEmpty())
ras->extract(updateRect)->copy(m_workRaster->extract(updateRect));
TPointD thickOffset(m_maxCursorThick * 0.5, m_maxCursorThick * 0.5);
TRectD invalidateRect = convert(m_strokeSegmentRect) - rasCenter;
invalidateRect += TRectD(m_brushPos - thickOffset, m_brushPos + thickOffset);
invalidateRect +=
TRectD(previousBrushPos - thickOffset, previousBrushPos + thickOffset);
invalidate(invalidateRect.enlarge(2.0));
m_perspectiveIndex = -1;
}
//-------------------------------------------------------------------------------------------------------------
void FullColorBrushTool::leftButtonDrag(const TPointD &pos,
const TMouseEvent &e) {
TRectD invalidateRect;
invalidateRect = TRectD(m_firstPoint, m_lastPoint).enlarge(2);
m_lastPoint = pos;
TPointD usePos = pos;
if (e.isAltPressed() || m_snapGrid.getValue()) {
double distance = (m_brushPos.x - m_maxCursorThick + 1) * 0.5;
TRectD brushRect =
TRectD(TPointD(m_brushPos.x - distance, m_brushPos.y - distance),
TPointD(m_brushPos.x + distance, m_brushPos.y + distance));
invalidateRect += (brushRect);
PerspectiveTool *perspectiveTool = dynamic_cast<PerspectiveTool *>(
TTool::getTool("T_PerspectiveGrid", TTool::RasterImage));
std::vector<PerspectiveObject *> perspectiveObjs =
perspectiveTool->getPerspectiveObjects();
TPointD pointToUse = TPointD(0.0, 0.0);
TPointD dpiScale = getViewer()->getDpiScale();
TPointD refPoint = m_firstPoint;
refPoint.x *= dpiScale.x;
refPoint.y *= dpiScale.y;
if (e.isAltPressed() || m_perspectiveIndex < 0) {
// let's get info about our current location
double denominator = m_lastPoint.x - m_firstPoint.x;
double numerator = m_lastPoint.y - m_firstPoint.y;
if (areAlmostEqual(denominator, 0.0, 0.0001)) {
denominator = denominator < 0 ? -0.0001 : 0.0001;
}
if (areAlmostEqual(numerator, 0.0, 0.0001)) {
numerator = numerator < 0 ? -0.0001 : 0.0001;
}
double slope = (numerator / denominator);
double angle = std::atan(slope) * (180 / 3.14159);
// now let's get the angle of each of the assistant points
std::vector<double> anglesToAssistants;
for (auto data : perspectiveObjs) {
TPointD point = data->getReferencePoint(refPoint);
point.x /= dpiScale.x;
point.y /= dpiScale.y;
double newDenominator = point.x - m_firstPoint.x;
double newNumerator = point.y - m_firstPoint.y;
if (areAlmostEqual(newDenominator, 0.0, 0.0001)) {
newDenominator = newDenominator < 0 ? -0.0001 : 0.0001;
}
if (areAlmostEqual(newNumerator, 0.0, 0.0001)) {
newNumerator = newNumerator < 0 ? -0.0001 : 0.0001;
}
double newSlope = (newNumerator / newDenominator);
double newAngle = std::atan(newSlope) * (180 / 3.14159);
anglesToAssistants.push_back(newAngle);
}
// figure out which angle is closer
double difference = 360;
for (int i = 0; i < anglesToAssistants.size(); i++) {
double newDifference = abs(angle - anglesToAssistants.at(i));
if (newDifference < difference || (180 - newDifference) < difference) {
difference = std::min(newDifference, (180 - newDifference));
pointToUse = perspectiveObjs.at(i)->getReferencePoint(refPoint);
pointToUse.x /= dpiScale.x;
pointToUse.y /= dpiScale.y;
m_perspectiveIndex = i;
}
}
} else {
pointToUse =
perspectiveObjs.at(m_perspectiveIndex)->getReferencePoint(refPoint);
pointToUse.x /= dpiScale.x;
pointToUse.y /= dpiScale.y;
}
double distanceFirstToLast =
std::sqrt(std::pow((m_lastPoint.x - m_firstPoint.x), 2) +
std::pow((m_lastPoint.y - m_firstPoint.y), 2));
double distanceLastToAssistant =
std::sqrt(std::pow((pointToUse.x - m_lastPoint.x), 2) +
std::pow((pointToUse.y - m_lastPoint.y), 2));
double distanceFirstToAssistant =
std::sqrt(std::pow((pointToUse.x - m_firstPoint.x), 2) +
std::pow((pointToUse.y - m_firstPoint.y), 2));
if (distanceFirstToAssistant == 0.0) distanceFirstToAssistant = 0.001;
double ratio = distanceFirstToLast / distanceFirstToAssistant;
double newX;
double newY;
// flip the direction if the last point is farther than the first point
if (distanceFirstToAssistant < distanceLastToAssistant &&
distanceFirstToLast < distanceLastToAssistant) {
newX = ((1 + ratio) * m_firstPoint.x) - (ratio * pointToUse.x);
newY = ((1 + ratio) * m_firstPoint.y) - (ratio * pointToUse.y);
} else {
newX = ((1 - ratio) * m_firstPoint.x) + (ratio * pointToUse.x);
newY = ((1 - ratio) * m_firstPoint.y) + (ratio * pointToUse.y);
}
usePos = m_lastPoint = TPointD(newX, newY);
invalidateRect += TRectD(m_firstPoint, m_lastPoint).enlarge(2);
} else if (e.isCtrlPressed()) {
double distance = (m_brushPos.x - m_maxCursorThick + 1) * 0.5;
TRectD brushRect =
TRectD(TPointD(m_brushPos.x - distance, m_brushPos.y - distance),
TPointD(m_brushPos.x + distance, m_brushPos.y + distance));
invalidateRect += (brushRect);
double denominator = m_lastPoint.x - m_firstPoint.x;
if (denominator == 0) denominator = 0.001;
double slope = ((m_lastPoint.y - m_firstPoint.y) / denominator);
double radAngle = std::atan(abs(slope));
double angle = radAngle * (180 / 3.14159);
if (abs(angle) >= 82.5)
m_lastPoint.x = m_firstPoint.x;
else if (abs(angle) < 7.5)
m_lastPoint.y = m_firstPoint.y;
else {
double xDistance = m_lastPoint.x - m_firstPoint.x;
double yDistance = m_lastPoint.y - m_firstPoint.y;
double totalDistance =
std::sqrt(std::pow(xDistance, 2) + std::pow(yDistance, 2));
double xLength = 0.0;
double yLength = 0.0;
if (angle >= 7.5 && angle < 22.5) {
yLength = std::sin(15 * (3.14159 / 180)) * totalDistance;
xLength = std::cos(15 * (3.14159 / 180)) * totalDistance;
} else if (angle >= 22.5 && angle < 37.5) {
yLength = std::sin(30 * (3.14159 / 180)) * totalDistance;
xLength = std::cos(30 * (3.14159 / 180)) * totalDistance;
} else if (angle >= 37.5 && angle < 52.5) {
yLength = std::sin(45 * (3.14159 / 180)) * totalDistance;
xLength = std::cos(45 * (3.14159 / 180)) * totalDistance;
} else if (angle >= 52.5 && angle < 67.5) {
yLength = std::sin(60 * (3.14159 / 180)) * totalDistance;
xLength = std::cos(60 * (3.14159 / 180)) * totalDistance;
} else if (angle >= 67.5 && angle < 82.5) {
yLength = std::sin(75 * (3.14159 / 180)) * totalDistance;
xLength = std::cos(75 * (3.14159 / 180)) * totalDistance;
}
if (yDistance == abs(yDistance)) {
m_lastPoint.y = m_firstPoint.y + yLength;
} else {
m_lastPoint.y = m_firstPoint.y - yLength;
}
if (xDistance == abs(xDistance)) {
m_lastPoint.x = m_firstPoint.x + xLength;
} else {
m_lastPoint.x = m_firstPoint.x - xLength;
}
}
}
// keep this here for possible eventual variable width
// if (getApplication()->getCurrentLevelStyle()->getTagId() ==
// 4001) // mypaint brush case
// m_oldPressure = m_enabledPressure && e.isTablet() ? e.m_pressure : 0.5;
// else
// m_oldPressure = m_enabledPressure ? e.m_pressure : 1.0;
if (m_isStraight) {
m_mousePos = pos;
m_brushPos = pos;
invalidate(invalidateRect);
return;
}
TPointD previousBrushPos = m_brushPos;
m_brushPos = m_mousePos = pos;
m_mouseEvent = e;
TRasterImageP ri = (TRasterImageP)getImage(true);
if (!ri) return;
if (!m_toonz_brush) return;
TRasterP ras = ri->getRaster();
TPointD rasCenter = ras->getCenterD();
TPointD point(usePos + rasCenter);
double pressure;
if (getApplication()->getCurrentLevelStyle()->getTagId() ==
4001) // mypaint brush case
pressure = m_enabledPressure && e.isTablet() ? e.m_pressure : 0.5;
else
pressure = m_enabledPressure ? e.m_pressure : 1.0;
m_strokeSegmentRect.empty();
m_toonz_brush->strokeTo(point, pressure, restartBrushTimer());
TRect updateRect = m_strokeSegmentRect * ras->getBounds();
if (!updateRect.isEmpty())
ras->extract(updateRect)->copy(m_workRaster->extract(updateRect));
TPointD thickOffset(m_maxCursorThick * 0.5, m_maxCursorThick * 0.5);
invalidateRect = convert(m_strokeSegmentRect) - rasCenter;
invalidateRect += TRectD(m_brushPos - thickOffset, m_brushPos + thickOffset);
invalidateRect +=
TRectD(previousBrushPos - thickOffset, previousBrushPos + thickOffset);
invalidate(invalidateRect.enlarge(2.0));
}
//---------------------------------------------------------------------------------------------------------------
void FullColorBrushTool::leftButtonUp(const TPointD &pos,
const TMouseEvent &e) {
TPointD previousBrushPos = m_brushPos;
m_brushPos = m_mousePos = pos;
TRasterImageP ri = (TRasterImageP)getImage(true);
if (!ri) return;
if (!m_toonz_brush) return;
TRasterP ras = ri->getRaster();
TPointD rasCenter = ras->getCenterD();
TPointD point;
if (e.isCtrlPressed() || m_snapGrid.getValue() || e.isAltPressed())
point = TPointD(m_lastPoint + rasCenter);
else
point = TPointD(pos + rasCenter);
double pressure;
if (getApplication()->getCurrentLevelStyle()->getTagId() ==
4001) // mypaint brush case
pressure = m_enabledPressure && e.isTablet() ? e.m_pressure : 0.5;
else
pressure = m_enabledPressure ? e.m_pressure : 1.0;
if (m_isStraight) {
pressure = m_oldPressure;
}
m_strokeSegmentRect.empty();
m_toonz_brush->strokeTo(point, pressure, restartBrushTimer());
m_toonz_brush->endStroke();
TRect updateRect = m_strokeSegmentRect * ras->getBounds();
if (!updateRect.isEmpty())
ras->extract(updateRect)->copy(m_workRaster->extract(updateRect));
TPointD thickOffset(m_maxCursorThick * 0.5, m_maxCursorThick * 0.5);
TRectD invalidateRect = convert(m_strokeSegmentRect) - rasCenter;
invalidateRect += TRectD(m_brushPos - thickOffset, m_brushPos + thickOffset);
invalidateRect +=
TRectD(previousBrushPos - thickOffset, previousBrushPos + thickOffset);
invalidate(invalidateRect.enlarge(2.0));
if (m_toonz_brush) {
delete m_toonz_brush;
m_toonz_brush = 0;
}
m_lastRect.empty();
m_workRaster->unlock();
if (m_tileSet->getTileCount() > 0) {
delete m_tileSaver;
bool isEditingLevel = m_application->getCurrentFrame()->isEditingLevel();
bool renameColumn = m_isFrameCreated;
if (!isEditingLevel && renameColumn) TUndoManager::manager()->beginBlock();
TTool::Application *app = TTool::getApplication();
TXshLevel *level = app->getCurrentLevel()->getLevel();
TXshSimpleLevelP simLevel = level->getSimpleLevel();
TFrameId frameId = getCurrentFid();
TRasterP subras = ras->extract(m_strokeRect)->clone();
TUndoManager::manager()->add(new FullColorBrushUndo(
m_tileSet, simLevel.getPointer(), frameId, m_isFrameCreated, subras,
m_strokeRect.getP00()));
// Column name renamed to level name only if was originally empty
if (!isEditingLevel && renameColumn) {
int col = app->getCurrentColumn()->getColumnIndex();
TXshColumn *column = app->getCurrentXsheet()->getXsheet()->getColumn(col);
int r0, r1;
column->getRange(r0, r1);
if (r0 == r1) {
TStageObjectId columnId = TStageObjectId::ColumnId(col);
std::string columnName =
QString::fromStdWString(simLevel->getName()).toStdString();
TStageObjectCmd::rename(columnId, columnName, app->getCurrentXsheet());
}
TUndoManager::manager()->endBlock();
}
}
notifyImageChanged();
m_strokeRect.empty();
m_mousePressed = false;
m_isStraight = false;
m_oldPressure = -1.0;
m_perspectiveIndex = -1;
}
//---------------------------------------------------------------------------------------------------------------
void FullColorBrushTool::mouseMove(const TPointD &pos, const TMouseEvent &e) {
struct Locals {
FullColorBrushTool *m_this;
void setValue(TIntPairProperty &prop,
const TIntPairProperty::Value &value) {
prop.setValue(value);
m_this->onPropertyChanged(prop.getName());
TTool::getApplication()->getCurrentTool()->notifyToolChanged();
}
void addMinMax(TIntPairProperty &prop, double add) {
const TIntPairProperty::Range &range = prop.getRange();
TIntPairProperty::Value value = prop.getValue();
value.second =
tcrop<double>(value.second + add, range.first, range.second);
value.first = tcrop<double>(value.first + add, range.first, range.second);
setValue(prop, value);
}
void addMinMaxSeparate(TIntPairProperty &prop, double min, double max) {
if (min == 0.0 && max == 0.0) return;
const TIntPairProperty::Range &range = prop.getRange();
TIntPairProperty::Value value = prop.getValue();
value.first += min;
value.second += max;
if (value.first > value.second) value.first = value.second;
value.first = tcrop<double>(value.first, range.first, range.second);
value.second = tcrop<double>(value.second, range.first, range.second);
setValue(prop, value);
}
} locals = {this};
// if (e.isAltPressed() && !e.isCtrlPressed()) {
// const TPointD &diff = pos - m_mousePos;
// double add = (fabs(diff.x) > fabs(diff.y)) ? diff.x : diff.y;
// locals.addMinMax(m_thickness, int(add));
//} else
// if (e.isCtrlPressed() && e.isAltPressed()) {
// const TPointD &diff = pos - m_mousePos;
// double max = diff.x / 2;
// double min = diff.y / 2;
// locals.addMinMaxSeparate(m_thickness, int(min), int(max));
//} else {
m_brushPos = pos;
//}
m_mousePos = pos;
invalidate();
}
//-------------------------------------------------------------------------------------------------------------
void FullColorBrushTool::draw() {
if (TRasterImageP ri = TRasterImageP(getImage(false))) {
if (m_isStraight) {
tglDrawSegment(m_firstPoint, m_lastPoint);
}
// If toggled off, don't draw brush outline
if (!Preferences::instance()->isCursorOutlineEnabled()) return;
TRasterP ras = ri->getRaster();
double alpha = 1.0;
double alphaRadius = 3.0;
double pixelSize = sqrt(tglGetPixelSize2());
// circles with lesser radius looks more bold
// to avoid these effect we'll reduce alpha for small radiuses
double minX = m_minCursorThick / (alphaRadius * pixelSize);
double maxX = m_maxCursorThick / (alphaRadius * pixelSize);
double minAlpha = alpha * (1.0 - 1.0 / (1.0 + minX));
double maxAlpha = alpha * (1.0 - 1.0 / (1.0 + maxX));
glPushAttrib(GL_ALL_ATTRIB_BITS);
tglEnableBlending();
tglEnableLineSmooth(true, 0.5);
if (m_minCursorThick < m_maxCursorThick - pixelSize) {
glColor4d(1.0, 1.0, 1.0, minAlpha);
tglDrawCircle(m_brushPos, (m_minCursorThick + 1) * 0.5 - pixelSize);
glColor4d(0.0, 0.0, 0.0, minAlpha);
tglDrawCircle(m_brushPos, (m_minCursorThick + 1) * 0.5);
}
glColor4d(1.0, 1.0, 1.0, maxAlpha);
tglDrawCircle(m_brushPos, (m_maxCursorThick + 1) * 0.5 - pixelSize);
glColor4d(0.0, 0.0, 0.0, maxAlpha);
tglDrawCircle(m_brushPos, (m_maxCursorThick + 1) * 0.5);
glPopAttrib();
}
}
//--------------------------------------------------------------------------------------------------------------
void FullColorBrushTool::onEnter() { updateCurrentStyle(); }
//----------------------------------------------------------------------------------------------------------
void FullColorBrushTool::onLeave() {
m_minCursorThick = 0;
m_maxCursorThick = 0;
}
//----------------------------------------------------------------------------------------------------------
TPropertyGroup *FullColorBrushTool::getProperties(int targetType) {
if (!m_presetsLoaded) initPresets();
return &m_prop;
}
//----------------------------------------------------------------------------------------------------------
void FullColorBrushTool::onImageChanged() { setWorkAndBackupImages(); }
//----------------------------------------------------------------------------------------------------------
void FullColorBrushTool::setWorkAndBackupImages() {
TRasterImageP ri = (TRasterImageP)getImage(false, 1);
if (!ri) return;
TRasterP ras = ri->getRaster();
TDimension dim = ras->getSize();
if (!m_workRaster || m_workRaster->getLx() != dim.lx ||
m_workRaster->getLy() != dim.ly)
m_workRaster = TRaster32P(dim);
if (!m_backUpRas || m_backUpRas->getLx() != dim.lx ||
m_backUpRas->getLy() != dim.ly ||
m_backUpRas->getPixelSize() != ras->getPixelSize())
m_backUpRas = ras->create(dim.lx, dim.ly);
m_strokeRect.empty();
m_lastRect.empty();
}
//------------------------------------------------------------------
bool FullColorBrushTool::onPropertyChanged(std::string propertyName) {
if (m_propertyUpdating) return true;
updateCurrentStyle();
if (propertyName == "Preset:") {
if (m_preset.getValue() != CUSTOM_WSTR)
loadPreset();
else // Chose <custom>, go back to last saved brush settings
loadLastBrush();
FullcolorBrushPreset = m_preset.getValueAsString();
m_propertyUpdating = true;
getApplication()->getCurrentTool()->notifyToolChanged();
m_propertyUpdating = false;
return true;
}
FullcolorBrushMinSize = m_thickness.getValue().first;
FullcolorBrushMaxSize = m_thickness.getValue().second;
FullcolorPressureSensitivity = m_pressure.getValue();
FullcolorBrushHardness = m_hardness.getValue();
FullcolorMinOpacity = m_opacity.getValue().first;
FullcolorMaxOpacity = m_opacity.getValue().second;
FullcolorModifierSize = m_modifierSize.getValue();
FullcolorModifierOpacity = m_modifierOpacity.getValue();
FullcolorModifierEraser = m_modifierEraser.getValue() ? 1 : 0;
FullcolorModifierLockAlpha = m_modifierLockAlpha.getValue() ? 1 : 0;
FullcolorBrushSnapGrid = m_snapGrid.getValue() ? 1 : 0;
if (m_preset.getValue() != CUSTOM_WSTR) {
m_preset.setValue(CUSTOM_WSTR);
FullcolorBrushPreset = m_preset.getValueAsString();
m_propertyUpdating = true;
getApplication()->getCurrentTool()->notifyToolChanged();
m_propertyUpdating = false;
}
return true;
}
//------------------------------------------------------------------
void FullColorBrushTool::initPresets() {
if (!m_presetsLoaded) {
// If necessary, load the presets from file
m_presetsLoaded = true;
m_presetsManager.load(ToonzFolder::getMyModuleDir() + "brush_raster.txt");
}
// Rebuild the presets property entries
const std::set<BrushData> &presets = m_presetsManager.presets();
m_preset.deleteAllValues();
m_preset.addValue(CUSTOM_WSTR);
m_preset.setItemUIName(CUSTOM_WSTR, tr("<custom>"));
std::set<BrushData>::const_iterator it, end = presets.end();
for (it = presets.begin(); it != end; ++it) m_preset.addValue(it->m_name);
}
//----------------------------------------------------------------------------------------------------------
void FullColorBrushTool::loadPreset() {
const std::set<BrushData> &presets = m_presetsManager.presets();
std::set<BrushData>::const_iterator it;
it = presets.find(BrushData(m_preset.getValue()));
if (it == presets.end()) return;
const BrushData &preset = *it;
try // Don't bother with RangeErrors
{
m_thickness.setValue(
TIntPairProperty::Value(std::max((int)preset.m_min, 1), preset.m_max));
m_hardness.setValue(preset.m_hardness, true);
m_opacity.setValue(
TDoublePairProperty::Value(preset.m_opacityMin, preset.m_opacityMax));
m_pressure.setValue(preset.m_pressure);
m_modifierSize.setValue(preset.m_modifierSize);
m_modifierOpacity.setValue(preset.m_modifierOpacity);
m_modifierEraser.setValue(preset.m_modifierEraser);
m_modifierLockAlpha.setValue(preset.m_modifierLockAlpha);
} catch (...) {
}
}
//------------------------------------------------------------------
void FullColorBrushTool::addPreset(QString name) {
// Build the preset
BrushData preset(name.toStdWString());
preset.m_min = m_thickness.getValue().first;
preset.m_max = m_thickness.getValue().second;
preset.m_hardness = m_hardness.getValue();
preset.m_opacityMin = m_opacity.getValue().first;
preset.m_opacityMax = m_opacity.getValue().second;
preset.m_pressure = m_pressure.getValue();
preset.m_modifierSize = m_modifierSize.getValue();
preset.m_modifierOpacity = m_modifierOpacity.getValue();
preset.m_modifierEraser = m_modifierEraser.getValue();
preset.m_modifierLockAlpha = m_modifierLockAlpha.getValue();
// Pass the preset to the manager
m_presetsManager.addPreset(preset);
// Reinitialize the associated preset enum
initPresets();
// Set the value to the specified one
m_preset.setValue(preset.m_name);
FullcolorBrushPreset = m_preset.getValueAsString();
}
//------------------------------------------------------------------
void FullColorBrushTool::removePreset() {
std::wstring name(m_preset.getValue());
if (name == CUSTOM_WSTR) return;
m_presetsManager.removePreset(name);
initPresets();
// No parameter change, and set the preset value to custom
m_preset.setValue(CUSTOM_WSTR);
FullcolorBrushPreset = m_preset.getValueAsString();
}
//------------------------------------------------------------------
void FullColorBrushTool::loadLastBrush() {
m_thickness.setValue(
TIntPairProperty::Value(FullcolorBrushMinSize, FullcolorBrushMaxSize));
m_pressure.setValue(FullcolorPressureSensitivity ? 1 : 0);
m_opacity.setValue(
TDoublePairProperty::Value(FullcolorMinOpacity, FullcolorMaxOpacity));
m_hardness.setValue(FullcolorBrushHardness);
m_modifierSize.setValue(FullcolorModifierSize);
m_modifierOpacity.setValue(FullcolorModifierOpacity);
m_modifierEraser.setValue(FullcolorModifierEraser ? true : false);
m_modifierLockAlpha.setValue(FullcolorModifierLockAlpha ? true : false);
m_snapGrid.setValue(FullcolorBrushSnapGrid ? true : false);
}
//------------------------------------------------------------------
void FullColorBrushTool::updateCurrentStyle() {
m_currentColor = TPixel32::Black;
if (TTool::Application *app = getApplication()) {
if (app->getCurrentObject()->isSpline()) {
m_currentColor = TPixel32::Red;
} else if (TPalette *plt = app->getCurrentPalette()->getPalette()) {
int style = app->getCurrentLevelStyleIndex();
TColorStyle *colorStyle = plt->getStyle(style);
m_currentColor = colorStyle->getMainColor();
}
}
int prevMinCursorThick = m_minCursorThick;
int prevMaxCursorThick = m_maxCursorThick;
m_enabledPressure = m_pressure.getValue();
if (TMyPaintBrushStyle *brushStyle = getBrushStyle()) {
double radiusLog = brushStyle->getBrush().getBaseValue(
MYPAINT_BRUSH_SETTING_RADIUS_LOGARITHMIC) +
m_modifierSize.getValue() * log(2.0);
double radius = exp(radiusLog);
m_minCursorThick = m_maxCursorThick = (int)round(2.0 * radius);
} else {
m_minCursorThick = std::max(m_thickness.getValue().first, 1);
m_maxCursorThick =
std::max(m_thickness.getValue().second, m_minCursorThick);
if (!m_enabledPressure) m_minCursorThick = m_maxCursorThick;
}
// if this function is called from onEnter(), the clipping rect will not be
// set in order to update whole viewer.
if (prevMinCursorThick == 0 && prevMaxCursorThick == 0) return;
if (m_minCursorThick != prevMinCursorThick ||
m_maxCursorThick != prevMaxCursorThick) {
TRectD rect(
m_brushPos - TPointD(m_maxCursorThick + 2, m_maxCursorThick + 2),
m_brushPos + TPointD(m_maxCursorThick + 2, m_maxCursorThick + 2));
invalidate(rect);
}
}
//------------------------------------------------------------------
double FullColorBrushTool::restartBrushTimer() {
double dtime = m_brushTimer.nsecsElapsed() * 1e-9;
m_brushTimer.restart();
return dtime;
}
//------------------------------------------------------------------
TMyPaintBrushStyle *FullColorBrushTool::getBrushStyle() {
if (TTool::Application *app = getApplication())
return dynamic_cast<TMyPaintBrushStyle *>(app->getCurrentLevelStyle());
return 0;
}
//------------------------------------------------------------------
void FullColorBrushTool::applyClassicToonzBrushSettings(
mypaint::Brush &mypaintBrush) {
const double precision = 1e-5;
const double hardnessOpacity = 0.1;
double minThickness = 0.5 * m_thickness.getValue().first;
double maxThickness = 0.5 * m_thickness.getValue().second;
double minOpacity = 0.01 * m_opacity.getValue().first;
double maxOpacity = 0.01 * m_opacity.getValue().second;
double hardness = 0.01 * m_hardness.getValue();
TPixelD color = PixelConverter<TPixelD>::from(m_currentColor);
double colorH = 0.0;
double colorS = 0.0;
double colorV = 0.0;
RGB2HSV(color.r, color.g, color.b, &colorH, &colorS, &colorV);
// avoid log(0)
if (minThickness < precision) minThickness = precision;
if (maxThickness < precision) maxThickness = precision;
// tune hardness opacity for better visual softness
hardness *= hardness;
double opacityAmplifier = 1.0 - hardnessOpacity + hardness * hardnessOpacity;
minOpacity *= opacityAmplifier;
maxOpacity *= opacityAmplifier;
// reset
mypaintBrush.fromDefaults();
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_OPAQUE_MULTIPLY, 1.0);
mypaintBrush.setMappingN(MYPAINT_BRUSH_SETTING_OPAQUE_MULTIPLY,
MYPAINT_BRUSH_INPUT_PRESSURE, 0);
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_HARDNESS,
0.5 * hardness + 0.5);
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_COLOR_H, colorH / 360.0);
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_COLOR_S, colorS);
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_COLOR_V, colorV);
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_DABS_PER_ACTUAL_RADIUS,
5.0 + hardness * 10.0);
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_DABS_PER_BASIC_RADIUS, 0.0);
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_DABS_PER_SECOND, 0.0);
// thickness may be dynamic
if (minThickness + precision >= maxThickness) {
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_RADIUS_LOGARITHMIC,
log(maxThickness));
mypaintBrush.setMappingN(MYPAINT_BRUSH_SETTING_RADIUS_LOGARITHMIC,
MYPAINT_BRUSH_INPUT_PRESSURE, 0);
} else {
double minThicknessLog = log(minThickness);
double maxThicknessLog = log(maxThickness);
double baseThicknessLog = 0.5 * (minThicknessLog + maxThicknessLog);
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_RADIUS_LOGARITHMIC,
baseThicknessLog);
mypaintBrush.setMappingN(MYPAINT_BRUSH_SETTING_RADIUS_LOGARITHMIC,
MYPAINT_BRUSH_INPUT_PRESSURE, 2);
mypaintBrush.setMappingPoint(MYPAINT_BRUSH_SETTING_RADIUS_LOGARITHMIC,
MYPAINT_BRUSH_INPUT_PRESSURE, 0, 0.0,
minThicknessLog - baseThicknessLog);
mypaintBrush.setMappingPoint(MYPAINT_BRUSH_SETTING_RADIUS_LOGARITHMIC,
MYPAINT_BRUSH_INPUT_PRESSURE, 1, 1.0,
maxThicknessLog - baseThicknessLog);
}
// opacity may be dynamic
if (minOpacity + precision >= maxOpacity) {
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_OPAQUE, maxOpacity);
mypaintBrush.setMappingN(MYPAINT_BRUSH_SETTING_OPAQUE,
MYPAINT_BRUSH_INPUT_PRESSURE, 0);
} else {
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_OPAQUE, minOpacity);
mypaintBrush.setMappingN(MYPAINT_BRUSH_SETTING_OPAQUE,
MYPAINT_BRUSH_INPUT_PRESSURE, 2);
mypaintBrush.setMappingPoint(MYPAINT_BRUSH_SETTING_OPAQUE,
MYPAINT_BRUSH_INPUT_PRESSURE, 0, 0.0, 0.0);
mypaintBrush.setMappingPoint(MYPAINT_BRUSH_SETTING_OPAQUE,
MYPAINT_BRUSH_INPUT_PRESSURE, 1, 1.0,
maxOpacity - minOpacity);
}
if (m_modifierLockAlpha.getValue()) {
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_LOCK_ALPHA, 1.0);
}
}
void FullColorBrushTool::applyToonzBrushSettings(mypaint::Brush &mypaintBrush) {
TMyPaintBrushStyle *mypaintStyle = getBrushStyle();
if (mypaintStyle) {
const double precision = 1e-5;
double modifierSize = m_modifierSize.getValue() * log(2.0);
double modifierOpacity = 0.01 * m_modifierOpacity.getValue();
bool modifierEraser = m_modifierEraser.getValue();
bool modifierLockAlpha = m_modifierLockAlpha.getValue();
TPixelD color = PixelConverter<TPixelD>::from(m_currentColor);
double colorH = 0.0;
double colorS = 0.0;
double colorV = 0.0;
RGB2HSV(color.r, color.g, color.b, &colorH, &colorS, &colorV);
mypaintBrush.fromBrush(mypaintStyle->getBrush());
float baseSize =
mypaintBrush.getBaseValue(MYPAINT_BRUSH_SETTING_RADIUS_LOGARITHMIC);
float baseOpacity = mypaintBrush.getBaseValue(MYPAINT_BRUSH_SETTING_OPAQUE);
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_RADIUS_LOGARITHMIC,
baseSize + modifierSize);
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_OPAQUE,
baseOpacity * modifierOpacity);
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_COLOR_H, colorH / 360.0);
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_COLOR_S, colorS);
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_COLOR_V, colorV);
if (modifierEraser) {
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_ERASER, 1.0);
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_LOCK_ALPHA, 0.0);
} else if (modifierLockAlpha) {
// lock-alpha already disables eraser
mypaintBrush.setBaseValue(MYPAINT_BRUSH_SETTING_LOCK_ALPHA, 1.0);
}
} else {
applyClassicToonzBrushSettings(mypaintBrush);
}
}
//==========================================================================================================
FullColorBrushToolNotifier::FullColorBrushToolNotifier(FullColorBrushTool *tool)
: m_tool(tool) {
if (TTool::Application *app = m_tool->getApplication()) {
if (TXshLevelHandle *levelHandle = app->getCurrentLevel()) {
bool ret = connect(levelHandle, SIGNAL(xshCanvasSizeChanged()), this,
SLOT(onCanvasSizeChanged()));
assert(ret);
}
if (TPaletteHandle *paletteHandle = app->getCurrentPalette()) {
bool ret;
ret = connect(paletteHandle, SIGNAL(colorStyleChanged(bool)), this,
SLOT(onColorStyleChanged()));
assert(ret);
ret = connect(paletteHandle, SIGNAL(colorStyleSwitched()), this,
SLOT(onColorStyleChanged()));
assert(ret);
}
}
}
//==========================================================================================================
FullColorBrushTool fullColorPencil("T_Brush");