109 lines
3.1 KiB
C++
109 lines
3.1 KiB
C++
#include "gridpreview.h"
|
|
|
|
GridPreview::GridPreview(QWidget *parent)
|
|
: QWidget{parent}
|
|
{
|
|
colorMap.insert(QString("red"), Qt::red);
|
|
colorMap.insert(QString("green"), Qt::green);
|
|
colorMap.insert(QString("blue"), Qt::blue);
|
|
}
|
|
|
|
void GridPreview::setSize(int newWidth, int newHeight)
|
|
{
|
|
width = newWidth;
|
|
height = newHeight;
|
|
}
|
|
|
|
void GridPreview::addLine(Line &line)
|
|
{
|
|
lines.push_back(line);
|
|
}
|
|
|
|
void GridPreview::removeLine(int index)
|
|
{
|
|
lines.removeAt(index);
|
|
}
|
|
|
|
QString GridPreview::toLuaTable()
|
|
{
|
|
QString output(tableTemplate.arg(width).arg(height));
|
|
QString linesSubTable = "";
|
|
for(auto iter = lines.begin(); iter != lines.end(); iter++ )
|
|
{
|
|
linesSubTable += QString(" %1,\n").arg(iter->toLuaTable());
|
|
}
|
|
linesSubTable.removeLast(); // \n
|
|
linesSubTable.removeLast(); // ,
|
|
return output.arg(linesSubTable);
|
|
}
|
|
|
|
void GridPreview::paintEvent(QPaintEvent *event)
|
|
{
|
|
QPainter painter(this);
|
|
QPen pen(Qt::black, 2);
|
|
painter.setPen(pen);
|
|
QBrush brush(Qt::black);
|
|
bool widthBased = rect().width() > rect().height();
|
|
int cellSize = widthBased ?
|
|
rect().height() / height :
|
|
rect().width() / width;
|
|
|
|
int dotSize = cellSize * 0.5;
|
|
int filledDotSize = cellSize * 0.7;
|
|
|
|
int areaWidth = cellSize * width;
|
|
int areaHeight = cellSize * height;
|
|
|
|
int offsetX = (rect().width() - areaWidth) / 2;
|
|
int offsetY = (rect().height() - areaHeight) / 2;
|
|
|
|
QRect drawArea(offsetX, offsetY, areaWidth, areaHeight);
|
|
|
|
// Draw main grid
|
|
|
|
painter.drawRect(drawArea);
|
|
|
|
QRect cell(0, 0, cellSize, cellSize);
|
|
painter.setBrush(brush);
|
|
for (int cellX = 0; cellX < width; cellX++)
|
|
{
|
|
for (int cellY = 0; cellY < height; cellY++)
|
|
{
|
|
cell.setLeft(offsetX + (cellSize - dotSize) / 2 + cellX * cellSize);
|
|
cell.setWidth(dotSize);
|
|
cell.setTop(offsetY + (cellSize - dotSize) / 2 + cellY * cellSize);
|
|
cell.setHeight(dotSize);
|
|
painter.drawEllipse(cell);
|
|
}
|
|
}
|
|
|
|
// Draw line points
|
|
|
|
for (auto iter = lines.begin(); iter < lines.end(); iter++)
|
|
{
|
|
QColor color(colorFromString(iter->getColor()));
|
|
brush.setColor(color);
|
|
painter.setBrush(brush);
|
|
|
|
// Draw start
|
|
cell.setLeft(offsetX + (cellSize - filledDotSize) / 2 + (iter->getStartX() - 1) * cellSize);
|
|
cell.setWidth(filledDotSize);
|
|
cell.setTop(offsetY + (cellSize - filledDotSize) / 2 + (iter->getStartY() - 1) * cellSize);
|
|
cell.setHeight(filledDotSize);
|
|
painter.drawEllipse(cell);
|
|
|
|
// Draw end
|
|
cell.setLeft(offsetX + (cellSize - filledDotSize) / 2 + (iter->getEndX() - 1) * cellSize);
|
|
cell.setWidth(filledDotSize);
|
|
cell.setTop(offsetY + (cellSize - filledDotSize) / 2 + (iter->getEndY() - 1) * cellSize);
|
|
cell.setHeight(filledDotSize);
|
|
painter.drawEllipse(cell);
|
|
}
|
|
}
|
|
|
|
QColor GridPreview::colorFromString(QString color)
|
|
{
|
|
if (!colorMap.contains(color))
|
|
return colorMap[QString("red")];
|
|
return colorMap[color];
|
|
}
|