#include "TownImageWidget.h"

#include <QPainter>

TownImageWidget::TownImageWidget(QWidget *parent) : QWidget(parent) {
	setAttribute(Qt::WA_StyledBackground, true);
	//setFocusPolicy(Qt::WheelFocus);
}

void TownImageWidget::OnBeginDraw(const QPointF& start, const QPointF& end) {
	m_drawLine = true;

	m_start = start;
	m_end = end;

	update();
}

void TownImageWidget::OnEndDraw() {
	m_drawLine = false;
	update();
}

void TownImageWidget::paintEvent(QPaintEvent* event) {
	QWidget::paintEvent(event);

	if (!m_drawLine) {
		return;
	}

	QPainter painter(this);
	painter.setRenderHint(QPainter::Antialiasing);

	QPen pen;
	pen.setWidth(2);
	//pen.setColor(Qt::red);
	pen.setColor(QColor(92, 133, 252));

	painter.setPen(pen);

	painter.drawLine(m_start.x(), m_start.y(), m_end.x(), m_start.y());
	painter.drawLine(m_end.x(), m_start.y(), m_end.x(), m_end.y());
}