#include "toonzqt/stylenameeditor.h"
// TnzQt includes
#include "toonzqt/gutil.h"
// TnzLib includes
#include "toonz/toonzfolders.h"
#include "toonz/palettecmd.h"
#include "toonz/tpalettehandle.h"
// TnzCore includes
#include "tsystem.h"
#include "tcommon.h"
#include "tcolorstyles.h"
#include "tpalette.h"
#include <QLineEdit>
#include <QPushButton>
#include <QHBoxLayout>
#include <QVBoxLayout>
#include <QGridLayout>
#include <QLabel>
#include <QScrollArea>
#include <QSettings>
#include <QMessageBox>
#include <QMenu>
#include <QContextMenuEvent>
#include <QFrame>
//------------------------------------------------------------
namespace {
const int areaColCount[WORD_COLUMN_AMOUNT] = {2, 2, 1};
const QString columnLabel[WORD_COLUMN_AMOUNT] = {AddWordButton::tr("Character"),
AddWordButton::tr("Part"),
AddWordButton::tr("Suffix")};
int indexToRow(int index, int columnId) {
return index / areaColCount[columnId];
}
int indexToCol(int index, int columnId) {
return index % areaColCount[columnId];
}
}; // namespace
//------------------------------------------------------------
NewWordDialog::NewWordDialog(QWidget* parent) {
setModal(true);
m_lineEdit = new QLineEdit(this);
QPushButton* okBtn = new QPushButton(tr("OK"), this);
QPushButton* cancelBtn = new QPushButton(tr("Cancel"), this);
// layout
QVBoxLayout* mainLay = new QVBoxLayout();
mainLay->setMargin(5);
mainLay->setSpacing(5);
{
mainLay->addWidget(new QLabel(tr("Enter new word"), this), 0,
Qt::AlignLeft | Qt::AlignVCenter);
mainLay->addWidget(m_lineEdit, 0);
QHBoxLayout* buttonsLay = new QHBoxLayout();
buttonsLay->setMargin(3);
buttonsLay->setSpacing(20);
{
buttonsLay->addSpacing(1);
buttonsLay->addWidget(okBtn, 0);
buttonsLay->addWidget(cancelBtn, 0);
buttonsLay->addSpacing(1);
}
mainLay->addLayout(buttonsLay);
}
setLayout(mainLay);
// signal-slot connections
bool ret = true;
ret = ret && connect(okBtn, SIGNAL(clicked(bool)), this, SLOT(accept()));
ret = ret && connect(cancelBtn, SIGNAL(clicked(bool)), this, SLOT(reject()));
}
//-------
QString NewWordDialog::getName() { return m_lineEdit->text(); }
//------------------------------------------------------------
WordButton::WordButton(const QString& text, QWidget* parent)
: QPushButton(text, parent) {
setFixedHeight(23);
setMinimumWidth(75);
setObjectName("WordButton");
setToolTip(text);
bool ret = connect(this, SIGNAL(clicked(bool)), this, SLOT(onClicked()));
assert(ret);
}
//-------
void WordButton::onClicked() { emit clicked(text()); }
//-------
void WordButton::onRemove() { emit removeWord(text()); }
//-------
void WordButton::contextMenuEvent(QContextMenuEvent* event) {
QMenu menu(this);
QAction* removeAct = new QAction(tr("Remove %1").arg(text()), &menu);
bool ret = connect(removeAct, SIGNAL(triggered()), this, SLOT(onRemove()));
menu.addAction(removeAct);
menu.exec(event->globalPos());
}
//------------------------------------------------------------
AddWordButton::AddWordButton(const int col, QWidget* parent)
: WordButton(columnLabel[col], parent), m_column(col) {
// setFixedSize(23, 23);
setIcon(QIcon(":Resources/plus.png"));
setIconSize(QSize(16, 16));
setToolTip(tr("Add New Word for %1").arg(columnLabel[col]));
}
//-------
void AddWordButton::onClicked() { emit clicked(m_column); }
//------------------------------------------------------------
//-------
// load word list from user's settings
void EasyInputArea::loadList() {
TFilePath fp(ToonzFolder::getMyModuleDir() +
TFilePath(styleNameEasyInputWordsFileName));
if (!TFileStatus(fp).doesExist()) return;
QSettings wordsSettings(toQString(fp), QSettings::IniFormat);
for (int a = 0; a < WORD_COLUMN_AMOUNT; a++) {
int size = wordsSettings.beginReadArray(QString::number(a));
if (size == 0) continue;
for (int i = 0; i < size; ++i) {
wordsSettings.setArrayIndex(i);
m_wordList[a].append(wordsSettings.value("word").toString());
}
wordsSettings.endArray();
}
}
//-------
// save word list to user's settings
void EasyInputArea::saveList() {
TFilePath fp(ToonzFolder::getMyModuleDir() +
TFilePath(styleNameEasyInputWordsFileName));
QSettings wordsSettings(toQString(fp), QSettings::IniFormat);
wordsSettings.clear();
for (int a = 0; a < WORD_COLUMN_AMOUNT; a++) {
wordsSettings.beginWriteArray(QString::number(a));
for (int i = 0; i < m_wordList[a].count(); ++i) {
wordsSettings.setArrayIndex(i);
wordsSettings.setValue("word", m_wordList[a].at(i));
}
wordsSettings.endArray();
}
}
//------
void EasyInputArea::updatePanelSize(int columnId) {
int itemCount = m_wordList[columnId].size() + 1;
int rowCount = tceil((double)itemCount / (double)areaColCount[columnId]);
QWidget* widget = m_scrollArea[columnId]->widget();
widget->setFixedSize(m_scrollArea[columnId]->width(), rowCount * 26 + 3);
}
//------
EasyInputArea::EasyInputArea(QWidget* parent) : QWidget(parent) {
loadList();
QHBoxLayout* mainLay = new QHBoxLayout();
mainLay->setMargin(0);
mainLay->setSpacing(3);
for (int a = 0; a < WORD_COLUMN_AMOUNT; a++) {
m_scrollArea[a] = new QScrollArea(this);
m_scrollArea[a]->setObjectName("SolidLineFrame");
QFrame* wordPanel = new QFrame(this);
QGridLayout* buttonsLay = new QGridLayout();
buttonsLay->setMargin(3);
buttonsLay->setSpacing(3);
{
int row = 0;
int col = 0;
// store word buttons
for (int s = 0; s < m_wordList[a].size(); s++) {
WordButton* button = new WordButton(m_wordList[a].at(s), this);
button->setFocusPolicy(Qt::NoFocus);
buttonsLay->addWidget(button, row, col);
connect(button, SIGNAL(clicked(const QString&)), this,
SIGNAL(wordClicked(const QString&)));
connect(button, SIGNAL(removeWord(const QString&)), this,
SLOT(onRemoveWord(const QString&)));
col++;
if (col == areaColCount[a]) {
col = 0;
row++;
}
}
// add button
AddWordButton* addWordButton = new AddWordButton(a, this);
addWordButton->setFocusPolicy(Qt::NoFocus);
buttonsLay->addWidget(addWordButton, row, col);
connect(addWordButton, SIGNAL(clicked(const int)), this,
SLOT(addWordButtonClicked(const int)));
}
for (int c = 0; c < areaColCount[a]; c++)
buttonsLay->setColumnStretch(c, 1);
wordPanel->setLayout(buttonsLay);
m_wordLayout[a] = buttonsLay;
m_scrollArea[a]->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
m_scrollArea[a]->setMinimumWidth(areaColCount[a] * 78 + 3);
m_scrollArea[a]->setWidget(wordPanel);
mainLay->addWidget(m_scrollArea[a], areaColCount[a]);
}
setLayout(mainLay);
}
//-------
void EasyInputArea::addWordButtonClicked(const int columnId) {
NewWordDialog dialog(this);
if (dialog.exec() == QDialog::Rejected) return;
QString word = dialog.getName();
if (word.isEmpty()) return;
bool found = false;
for (int i = 0; i < WORD_COLUMN_AMOUNT; i++) {
found = m_wordList[i].contains(word);
if (found) break;
}
if (found) {
QMessageBox::warning(this, tr("Warning"),
tr("%1 is already registered").arg(word));
return;
}
// Append word to the list
m_wordList[columnId].append(word);
// Append new WordButton
WordButton* button = new WordButton(word, this);
connect(button, SIGNAL(clicked(const QString&)), this,
SIGNAL(wordClicked(const QString&)));
connect(button, SIGNAL(removeWord(const QString&)), this,
SLOT(onRemoveWord(const QString&)));
button->setFocusPolicy(Qt::NoFocus);
int wordCount = m_wordList[columnId].count();
int row = indexToRow(wordCount - 1, columnId);
int col = indexToCol(wordCount - 1, columnId);
QWidget* addBtn = m_wordLayout[columnId]->itemAtPosition(row, col)->widget();
m_wordLayout[columnId]->addWidget(button, row, col);
// Move add button to the next index
col++;
if (col == areaColCount[columnId]) {
col = 0;
row++;
}
m_wordLayout[columnId]->addWidget(addBtn, row, col);
updatePanelSize(columnId);
saveList();
}
//------------------------------------------------------------
void EasyInputArea::onRemoveWord(const QString& word) {
int a = -1;
int index;
for (int i = 0; i < WORD_COLUMN_AMOUNT; i++) {
index = m_wordList[i].indexOf(word);
if (index >= 0) {
a = i;
break;
}
}
if (a < 0 || a >= WORD_COLUMN_AMOUNT) return;
// delete button
int row = indexToRow(index, a);
int col = indexToCol(index, a);
WordButton* wordButton = qobject_cast<WordButton*>(
m_wordLayout[a]->itemAtPosition(row, col)->widget());
if (!wordButton) return;
bool ret = true;
ret = ret && disconnect(wordButton, SIGNAL(clicked(const QString&)), this,
SIGNAL(wordClicked(const QString&)));
ret = ret && disconnect(wordButton, SIGNAL(removeWord(const QString&)), this,
SLOT(onRemoveWord(const QString&)));
assert(ret);
m_wordLayout[a]->removeWidget(wordButton);
wordButton->deleteLater();
// move the following buttons
for (int i = index + 1; i <= m_wordList[a].count(); i++) {
int row = indexToRow(i, a);
int col = indexToCol(i, a);
QWidget* button = m_wordLayout[a]->itemAtPosition(row, col)->widget();
col--;
if (col < 0) {
row--;
col = areaColCount[a] - 1;
}
m_wordLayout[a]->addWidget(button, row, col);
}
// remove word from the list
m_wordList[a].removeAt(index);
updatePanelSize(a);
}
//------------------------------------------------------------
void EasyInputArea::enterEvent(QEvent*) { emit mouseEnter(); }
//------------------------------------------------------------
void EasyInputArea::resizeEvent(QResizeEvent*) {
for (int i = 0; i < WORD_COLUMN_AMOUNT; i++) {
updatePanelSize(i);
}
}
//------------------------------------------------------------
StyleNameEditor::StyleNameEditor(QWidget* parent)
: Dialog(parent, false, false, "StyleNameEditor")
, m_paletteHandle(0)
, m_selectionStart(-1)
, m_selectionLength(0) {
setWindowTitle(tr("Name Editor"));
m_styleName = new QLineEdit(this);
m_okButton = new QPushButton(tr("OK"), this);
m_cancelButton = new QPushButton(tr("Cancel"), this);
m_applyButton = new QPushButton(tr("Apply and Next"), this);
EasyInputArea* easyInputArea = new EasyInputArea(this);
setFocusProxy(m_styleName);
m_styleName->setEnabled(false);
m_okButton->setEnabled(false);
m_okButton->setFocusPolicy(Qt::NoFocus);
m_applyButton->setEnabled(false);
m_applyButton->setFocusPolicy(Qt::NoFocus);
m_cancelButton->setFocusPolicy(Qt::NoFocus);
m_styleName->setObjectName("LargeSizedText");
easyInputArea->setFocusPolicy(Qt::NoFocus);
// QVBoxLayout* mainLayout = new QVBoxLayout();
m_topLayout->setMargin(10);
m_topLayout->setSpacing(5);
{
QHBoxLayout* inputLayout = new QHBoxLayout();
inputLayout->setMargin(0);
inputLayout->setSpacing(3);
{
inputLayout->addWidget(new QLabel(tr("Style Name"), this), 0);
inputLayout->addWidget(m_styleName, 1);
}
m_topLayout->addLayout(inputLayout, 0);
QHBoxLayout* buttonLayout = new QHBoxLayout();
buttonLayout->setMargin(0);
buttonLayout->setSpacing(3);
{
buttonLayout->addWidget(m_okButton);
buttonLayout->addWidget(m_applyButton);
buttonLayout->addWidget(m_cancelButton);
}
m_topLayout->addLayout(buttonLayout, 0);
m_topLayout->addSpacing(5);
m_topLayout->addWidget(new QLabel(tr("Easy Inputs"), this), 0,
Qt::AlignLeft);
m_topLayout->addWidget(easyInputArea, 1);
}
bool ret = true;
ret =
ret && connect(m_okButton, SIGNAL(pressed()), this, SLOT(onOkPressed()));
ret = ret && connect(m_cancelButton, SIGNAL(pressed()), this,
SLOT(onCancelPressed()));
ret = ret &&
connect(m_applyButton, SIGNAL(pressed()), this, SLOT(onApplyPressed()));
ret = ret && connect(easyInputArea, SIGNAL(wordClicked(const QString&)), this,
SLOT(onWordClicked(const QString&)));
ret = ret && connect(easyInputArea, SIGNAL(mouseEnter()), this,
SLOT(storeSelectionInfo()));
assert(ret);
}
//-------
void StyleNameEditor::setPaletteHandle(TPaletteHandle* ph) {
m_paletteHandle = ph;
connect(m_paletteHandle, SIGNAL(colorStyleSwitched()), this,
SLOT(onStyleSwitched()));
connect(m_paletteHandle, SIGNAL(paletteSwitched()), this,
SLOT(onStyleSwitched()));
m_styleName->setEnabled(true);
m_okButton->setEnabled(true);
m_applyButton->setEnabled(true);
}
//-------
void StyleNameEditor::showEvent(QShowEvent* e) {
if (m_paletteHandle) {
disconnect(m_paletteHandle, SIGNAL(colorStyleSwitched()), this,
SLOT(onStyleSwitched()));
disconnect(m_paletteHandle, SIGNAL(paletteSwitched()), this,
SLOT(onStyleSwitched()));
connect(m_paletteHandle, SIGNAL(colorStyleSwitched()), this,
SLOT(onStyleSwitched()));
connect(m_paletteHandle, SIGNAL(paletteSwitched()), this,
SLOT(onStyleSwitched()));
}
// update view
onStyleSwitched();
}
//-------disconnection
void StyleNameEditor::hideEvent(QHideEvent* e) {
disconnect(m_paletteHandle, SIGNAL(colorStyleSwitched()), this,
SLOT(onStyleSwitched()));
disconnect(m_paletteHandle, SIGNAL(paletteSwitched()), this,
SLOT(onStyleSwitched()));
}
//-----update display when the current style is switched
void StyleNameEditor::onStyleSwitched() {
if (!m_paletteHandle || !m_paletteHandle->getStyle()) return;
std::wstring styleName = m_paletteHandle->getStyle()->getName();
m_styleName->setText(QString::fromStdWString(styleName));
m_styleName->selectAll();
m_styleName->setFocus();
int styleId = m_paletteHandle->getStyleIndex();
setWindowTitle(tr("Name Editor: # %1").arg(styleId));
}
//-------
void StyleNameEditor::onOkPressed() {
onApplyPressed();
close();
}
//-------
void StyleNameEditor::onApplyPressed() {
if (!m_paletteHandle || !m_paletteHandle->getStyle()) return;
if (m_styleName->text() == "") return;
std::wstring newName = m_styleName->text().toStdWString();
PaletteCmd::renamePaletteStyle(m_paletteHandle, newName);
// move the current style to the next
TPalette* palette = m_paletteHandle->getPalette();
int currentIndex = m_paletteHandle->getStyleIndex();
TPalette::Page* page = palette->getStylePage(currentIndex);
int indexInPage = page->search(currentIndex);
// If indexInPage is at the end of the page, then move to the top of the next
// page
if (indexInPage == page->getStyleCount() - 1) {
int pageIndex = page->getIndex();
// if the current page is the last one, then move to the first page
while (1) {
pageIndex++;
if (pageIndex == palette->getPageCount()) pageIndex = 0;
page = palette->getPage(pageIndex);
if (page->getStyleCount() > 0) break;
}
currentIndex = page->getStyleId(0);
} else
currentIndex = page->getStyleId(indexInPage + 1);
// change the current index
m_paletteHandle->setStyleIndex(currentIndex);
}
//-------
void StyleNameEditor::onCancelPressed() { close(); }
//-------
// focus when the mouse enters
void StyleNameEditor::enterEvent(QEvent* e) {
activateWindow();
m_styleName->setFocus();
}
//-------
void StyleNameEditor::onWordClicked(const QString& word) {
if (m_selectionLength != 0)
m_styleName->setSelection(m_selectionStart, m_selectionLength);
else
m_styleName->setCursorPosition(m_selectionStart);
m_styleName->insert(word);
m_styleName->setFocus();
storeSelectionInfo();
}
//-------
// remember the selection of m_stylename when mouse entered in EasyInputArea
void StyleNameEditor::storeSelectionInfo() {
if (m_styleName->hasSelectedText()) {
m_selectionStart = m_styleName->selectionStart();
m_selectionLength = m_styleName->selectedText().length();
} else {
m_selectionStart = m_styleName->cursorPosition();
m_selectionLength = 0;
}
}