tahoma2d/toonz/sources/toonzqt/gutil.cpp
2016-04-20 10:24:52 +09:00

445 lines
12 KiB
C++

#include "toonzqt/gutil.h"
// TnzQt includes
#include "toonzqt/dvdialog.h"
// TnzCore includes
#include "traster.h"
#include "tpixelutils.h"
#include "tfilepath.h"
#include "tfiletype.h"
#include "tstroke.h"
#include "tcurves.h"
#include "trop.h"
#include "tmsgcore.h"
// Qt includes
#include <QPixmap>
#include <QImage>
#include <QPainter>
#include <QPainterPath>
#include <QIcon>
#include <QString>
#include <QApplication>
#include <QMouseEvent>
#include <QTabletEvent>
#include <QKeyEvent>
#include <QUrl>
#include <QFileInfo>
using namespace DVGui;
//-----------------------------------------------------------------------------
QString fileSizeString(qint64 size, int precision)
{
if (size < 1024)
return QString::number(size) + " Bytes";
else if (size < 1024 * 1024)
return QString::number(size / (1024.0), 'f', precision) + " KB";
else if (size < 1024 * 1024 * 1024)
return QString::number(size / (1024 * 1024.0), 'f', precision) + " MB";
else
return QString::number(size / (1024 * 1024 * 1024.0), 'f', precision) + " GB";
}
//----------------------------------------------------------------
QImage rasterToQImage(const TRasterP &ras, bool premultiplied, bool mirrored)
{
if (TRaster32P ras32 = ras) {
QImage image(ras->getRawData(), ras->getLx(), ras->getLy(),
premultiplied ? QImage::Format_ARGB32_Premultiplied : QImage::Format_ARGB32);
if (mirrored)
return image.mirrored();
return image;
} else if (TRasterGR8P ras8 = ras) {
QImage image(ras->getRawData(), ras->getLx(), ras->getLy(), ras->getWrap(), QImage::Format_Indexed8);
static QVector<QRgb> colorTable;
if (colorTable.size() == 0) {
int i;
for (i = 0; i < 256; i++)
colorTable.append(QColor(i, i, i).rgb());
}
image.setColorTable(colorTable);
if (mirrored)
return image.mirrored();
return image;
}
return QImage();
}
//-----------------------------------------------------------------------------
QPixmap rasterToQPixmap(const TRaster32P &ras, bool premultiplied)
{
QPixmap pixmap;
return pixmap.fromImage(rasterToQImage(ras, premultiplied));
}
//-----------------------------------------------------------------------------
TRaster32P rasterFromQImage(QImage image, bool premultiply, bool mirror) //no need of const& - Qt uses implicit sharing...
{
QImage copyImage = mirror ? image.mirrored() : image;
TRaster32P ras(image.width(), image.height(), image.width(), (TPixelRGBM32 *)copyImage.bits(), false);
if (premultiply)
TRop::premultiply(ras);
return ras->clone();
}
//-----------------------------------------------------------------------------
TRaster32P rasterFromQPixmap(QPixmap pixmap, bool premultiply, bool mirror) //no need of const& - Qt uses implicit sharing...
{
QImage image = pixmap.toImage();
return rasterFromQImage(image, premultiply, mirror);
}
//-----------------------------------------------------------------------------
void drawPolygon(QPainter &p,
const std::vector<QPointF> &points,
bool fill,
const QColor colorFill,
const QColor colorLine)
{
if (points.size() == 0)
return;
p.setPen(colorLine);
QPolygonF E0Polygon;
int i = 0;
for (i = 0; i < (int)points.size(); i++)
E0Polygon << QPointF(points[i]);
E0Polygon << QPointF(points[0]);
QPainterPath E0Path;
E0Path.addPolygon(E0Polygon);
if (fill)
p.fillPath(E0Path, QBrush(colorFill));
p.drawPath(E0Path);
}
//-----------------------------------------------------------------------------
void drawArrow(QPainter &p,
const QPointF a, const QPointF b, const QPointF c,
bool fill,
const QColor colorFill,
const QColor colorLine)
{
std::vector<QPointF> pts;
pts.push_back(a);
pts.push_back(b);
pts.push_back(c);
drawPolygon(p, pts, fill, colorFill, colorLine);
}
//-----------------------------------------------------------------------------
QPixmap scalePixmapKeepingAspectRatio(QPixmap pixmap, QSize size, QColor color)
{
if (pixmap.isNull() || pixmap.size() == size)
return pixmap;
QPixmap scaledPixmap = pixmap.scaled(size.width(), size.height(), Qt::KeepAspectRatio, Qt::SmoothTransformation);
QPixmap newPixmap(size);
newPixmap.fill(color);
QPainter painter(&newPixmap);
painter.drawPixmap(double(size.width() - scaledPixmap.width()) * 0.5,
double(size.height() - scaledPixmap.height()) * 0.5,
scaledPixmap);
return newPixmap;
}
//-----------------------------------------------------------------------------
QIcon createQIcon(const char *iconSVGName)
{
QString normal = QString(":Resources/") + iconSVGName + ".svg";
QString click = QString(":Resources/") + iconSVGName + "_click.svg";
QString over = QString(":Resources/") + iconSVGName + "_over.svg";
QIcon icon;
icon.addFile(normal, QSize(), QIcon::Normal, QIcon::Off);
icon.addFile(click, QSize(), QIcon::Normal, QIcon::On);
icon.addFile(over, QSize(), QIcon::Active);
return icon;
}
//-----------------------------------------------------------------------------
QIcon createQIconPNG(const char *iconPNGName)
{
QString normal = QString(":Resources/") + iconPNGName + ".png";
QString click = QString(":Resources/") + iconPNGName + "_click.png";
QString over = QString(":Resources/") + iconPNGName + "_over.png";
QIcon icon;
icon.addFile(normal, QSize(), QIcon::Normal, QIcon::Off);
icon.addFile(click, QSize(), QIcon::Normal, QIcon::On);
icon.addFile(over, QSize(), QIcon::Active);
return icon;
}
//-----------------------------------------------------------------------------
QIcon createQIconOnOff(const char *iconSVGName, bool withOver)
{
QString on = QString(":Resources/") + iconSVGName + "_on.svg";
QString off = QString(":Resources/") + iconSVGName + "_off.svg";
QString over = QString(":Resources/") + iconSVGName + "_over.svg";
QIcon icon;
icon.addFile(off, QSize(), QIcon::Normal, QIcon::Off);
icon.addFile(on, QSize(), QIcon::Normal, QIcon::On);
if (withOver)
icon.addFile(over, QSize(), QIcon::Active);
else
icon.addFile(on, QSize(), QIcon::Active);
return icon;
}
//-----------------------------------------------------------------------------
QIcon createQIconOnOffPNG(const char *iconPNGName, bool withOver)
{
QString on = QString(":Resources/") + iconPNGName + "_on.png";
QString off = QString(":Resources/") + iconPNGName + "_off.png";
QString over = QString(":Resources/") + iconPNGName + "_over.png";
QIcon icon;
icon.addFile(off, QSize(), QIcon::Normal, QIcon::Off);
icon.addFile(on, QSize(), QIcon::Normal, QIcon::On);
if (withOver)
icon.addFile(over, QSize(), QIcon::Active);
else
icon.addFile(on, QSize(), QIcon::Active);
return icon;
}
//-----------------------------------------------------------------------------
QString toQString(const TFilePath &path)
{
return QString::fromStdWString(path.getWideString());
}
//-----------------------------------------------------------------------------
bool isSpaceString(const QString &str)
{
int i;
QString space(" ");
for (i = 0; i < str.size(); i++)
if (str.at(i) != space.at(0))
return false;
return true;
}
//-----------------------------------------------------------------------------
bool isValidFileName(const QString &fileName)
{
if (fileName.isEmpty() ||
fileName.contains(":") ||
fileName.contains("\\") ||
fileName.contains("/") ||
fileName.contains(">") ||
fileName.contains("<") ||
fileName.contains("*") ||
fileName.contains("|") ||
fileName.contains("\"") ||
fileName.contains("?") ||
fileName.trimmed().isEmpty())
return false;
return true;
}
//-----------------------------------------------------------------------------
bool isValidFileName_message(const QString &fileName)
{
return isValidFileName(fileName) ? true : (DVGui::error(QObject::tr("The file name cannot be empty or contain any of the following "
"characters: (new line) \\ / : * ? \" |")),
false);
}
//-----------------------------------------------------------------------------
QString elideText(const QString &srcText, const QFont &font, int width)
{
QFontMetrics metrix(font);
int srcWidth = metrix.width(srcText);
if (srcWidth < width)
return srcText;
int tilde = metrix.width("~");
int block = (width - tilde) / 2;
QString text("");
int i;
for (i = 0; i < srcText.size(); i++) {
text += srcText.at(i);
if (metrix.width(text) > block)
break;
}
text[i] = '~';
QString endText("");
for (i = srcText.size() - 1; i >= 0; i--) {
endText.push_front(srcText.at(i));
if (metrix.width(endText) > block)
break;
}
endText.remove(0, 1);
text += endText;
return text;
}
//-----------------------------------------------------------------------------
QString elideText(const QString &srcText, const QFontMetrics &fm, int width, const QString &elideSymbol)
{
QString text(srcText);
for (int i = text.size(); i > 1 && fm.width(text) > width;)
text = srcText.left(--i).append(elideSymbol);
return text;
}
//-----------------------------------------------------------------------------
QUrl pathToUrl(const TFilePath &path)
{
return QUrl::fromLocalFile(QString::fromStdWString(path.getWideString()));
}
//-----------------------------------------------------------------------------
bool isResource(const QString &path)
{
const TFilePath fp(path.toStdWString());
TFileType::Type type = TFileType::getInfo(fp);
return (TFileType::isViewable(type) || type & TFileType::MESH_IMAGE || type == TFileType::AUDIO_LEVEL || type == TFileType::TABSCENE || type == TFileType::TOONZSCENE || fp.getType() == "tpl");
}
//-----------------------------------------------------------------------------
bool isResource(const QUrl &url)
{
return isResource(url.toLocalFile());
}
//-----------------------------------------------------------------------------
bool isResourceOrFolder(const QUrl &url)
{
struct locals {
static inline bool isDir(const QString &path)
{
return QFileInfo(path).isDir();
}
}; // locals
const QString &path = url.toLocalFile();
return (isResource(path) || locals::isDir(path));
}
//-----------------------------------------------------------------------------
bool acceptResourceDrop(const QList<QUrl> &urls)
{
int count = 0;
foreach (const QUrl &url, urls) {
if (isResource(url))
++count;
else
return false;
}
return (count > 0);
}
//-----------------------------------------------------------------------------
bool acceptResourceOrFolderDrop(const QList<QUrl> &urls)
{
int count = 0;
foreach (const QUrl &url, urls) {
if (isResourceOrFolder(url))
++count;
else
return false;
}
return (count > 0);
}
//-----------------------------------------------------------------------------
QPainterPath strokeToPainterPath(TStroke *stroke)
{
QPainterPath path;
int i, chunkSize = stroke->getChunkCount();
for (i = 0; i < chunkSize; i++) {
const TThickQuadratic *q = stroke->getChunk(i);
if (i == 0)
path.moveTo(toQPointF(q->getThickP0()));
path.quadTo(toQPointF(q->getThickP1()), toQPointF(q->getThickP2()));
}
return path;
}
//=============================================================================
// TabBarContainter
//-----------------------------------------------------------------------------
TabBarContainter::TabBarContainter(QWidget *parent)
: QFrame(parent)
{
setObjectName("TabBarContainer");
setFrameStyle(QFrame::StyledPanel);
}
//-----------------------------------------------------------------------------
void TabBarContainter::paintEvent(QPaintEvent *event)
{
QPainter p(this);
p.setPen(QColor(120, 120, 120));
p.drawLine(0, height() - 2, width(), height() - 2);
p.setPen(Qt::white);
p.drawLine(0, height() - 1, width(), height() - 1);
}
//=============================================================================
// ToolBarContainer
//-----------------------------------------------------------------------------
ToolBarContainer::ToolBarContainer(QWidget *parent)
: QFrame(parent)
{
setObjectName("ToolBarContainer");
setFrameStyle(QFrame::StyledPanel);
setSizePolicy(QSizePolicy::MinimumExpanding, QSizePolicy::Fixed);
}
//-----------------------------------------------------------------------------
void ToolBarContainer::paintEvent(QPaintEvent *event)
{
QPainter p(this);
p.setPen(QColor(120, 120, 120));
p.drawLine(0, 0, width(), 0);
}
//=============================================================================
QString operator+(const QString &a, const TFilePath &fp)
{
return a + QString::fromStdWString(fp.getWideString());
}