stv0g
/
pastie
1
0
Fork 0
Code Issues 2 Pull Requests Releases Wiki Activity

huge commit of refactoring work

This commit is contained in:
Steffen Vogel 2015-01-29 01:05:57 +01:00
parent 20891805e0
commit de953d0a6d
33 changed files with 321 additions and 209 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Pastie.pro.user
View File

@ -1,6 +1,6 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE QtCreatorProject>
<!-- Written by QtCreator 3.3.0, 2015-01-15T11:48:54. -->
<!-- Written by QtCreator 3.3.0, 2015-01-28T16:52:04. -->
<qtcreator>
<data>
<variable>EnvironmentId</variable>

7
camera.cpp
View File

@ -32,9 +32,10 @@ bool Camera::calibrate(QList<Image *> imgs, Pattern *pattern)
return false;
}
for (auto img : imgs) {
img->applyFilter(pattern);
PatternResult *result = (PatternResult *) img->getResult(pattern);
for (Image *img : imgs) {
PatternResult *result = (PatternResult*) pattern->apply(img);
img->results[pattern] = result;
if (result->isFound())
imagePoints.push_back(result->getPoints());

33
filterlist.cpp
View File

@ -58,7 +58,7 @@ bool FilterList::setData(const QModelIndex &index, const QVariant &value, int ro
case 1: filter->setShow(value == Qt::Checked); break;
}
emit filtersChanged();
emit filterChanged(filter);
break;
}
@ -101,36 +101,27 @@ void FilterList::add(Filter *filter)
push_back(filter);
endInsertRows();
connect(filter, SIGNAL(filterChanged()), this, SIGNAL(filtersChanged()));
connect(filter, SIGNAL(filterChanged(Filter *)), this, SIGNAL(filterChanged(Filter *)));
connect(filter, SIGNAL(filterApplied(Filter*)), this, SLOT(update(Filter *)));
emit filterAdded(filter);
}
void FilterList::execute(Image *img)
{
img->getMat().release();
try {
for (auto filter : *this)
img->applyFilter(filter);
} catch (Exception e) {
qCritical("%s", e.msg.c_str());
}
/* Update times and results */
emit dataChanged(
createIndex(0, 4),
createIndex(size(), 4)
);
}
void FilterList::reset()
{
qDebug() << "Reset filters";
for (Filter *filter : *this)
filter->reset();
emit filtersChanged();
emit filterChanged();
}
void FilterList::update(Filter *f)
{
emit dataChanged(
createIndex(indexOf(f), 0),
createIndex(indexOf(f)+1, 5)
);
}
Filter * FilterList::getCurrent()

6
filterlist.h
View File

@ -23,8 +23,7 @@ class FilterList :
Qt::ItemFlags flags(const QModelIndex & index) const;
bool setData(const QModelIndex &index, const QVariant &value, int role);
void add(Filter *filt);
void execute(Image *img);
void add(Filter *filter);
Filter * getCurrent();
@ -32,9 +31,10 @@ class FilterList :
public slots:
void reset();
void update(Filter *);
signals:
void filtersChanged();
void filterChanged(Filter *filter = NULL);
void filterSelected(Filter *filter);
void filterAdded(Filter *filter);
void filterRemoved(Filter *filter);

9
filters/filter.cpp
View File

@ -36,5 +36,14 @@ Result * Filter::apply(Image *img)
time = timer.nsecsElapsed();
}
img->results[this] = result;
emit filterApplied(this);
return result;
}
void Filter::settingChanged(Setting *)
{
emit filterChanged(this);
}

6
filters/filter.h
View File

@ -41,10 +41,12 @@ class Filter : public QObject
public slots:
virtual void reset() { }
virtual bool clicked(Point, QMouseEvent * = 0) { return false; }
virtual void clicked(Point, QMouseEvent *) { }
virtual void settingChanged(Setting *);
signals:
void filterChanged();
void filterChanged(Filter *);
void filterApplied(Filter *);
protected:
bool enabled = true;

5
filters/kmeans.cpp
View File

@ -94,8 +94,9 @@ Result * KMeans::applyInternal(Image *img)
return new Result;
}
bool KMeans::clicked(Point p, QMouseEvent *)
void KMeans::clicked(Point p, QMouseEvent *)
{
colorFilterPoints.append(p);
return true;
emit filterChanged(this);
}

2
filters/kmeans.h
View File

@ -14,7 +14,7 @@ class KMeans : public Filter
QString getName() const { return "KMeans"; }
void reset();
bool clicked(Point, QMouseEvent *);
void clicked(Point, QMouseEvent *);
protected:

15
filters/paddetect.cpp
View File

@ -103,13 +103,18 @@ QString PadResult::getResult() const
return QString("count = %1").arg(size());
}
QList<Pad>::iterator PadResult::getNearest(const Point &p)
PadResult::iterator PadResult::getNearest(const Pad &p)
{
float minDist = FLT_MAX;
QList<Pad>::iterator nearest = end();
return getNearest(p.center);
}
for (auto it = begin(); it != end(); it++) {
float dist = norm(it->center - Point2f(p));
PadResult::iterator PadResult::getNearest(const Point2f &p)
{
double minDist = FLT_MAX;
iterator nearest = end();
for (iterator it = begin(); it != end(); it++) {
double dist = it->getDistance(p);
if (dist < minDist) {
nearest = it;
minDist = dist;

8
filters/paddetect.h
View File

@ -1,7 +1,7 @@
#ifndef PADDETECT_H
#define PADDETECT_H
#include <vector>
#include <QLinkedList>
#include "pad.h"
#include "filter.h"
@ -28,14 +28,16 @@ class PadDetect : public Filter
enum Mode mode;
};
class PadResult : public Result, public QList<Pad>
class PadResult : public Result, public QLinkedList<Pad>
{
public:
void drawResult(Painter *p) const;
QString getResult() const;
QList<Pad>::iterator getNearest(const Point &p);
iterator getNearest(const Point2f &p);
iterator getNearest(const Pad &p);
};
#endif // PADDETECT_H

12
filters/padfilter.cpp
View File

@ -24,10 +24,10 @@ Result * PadFilter::applyInternal(Image *img)
result->push_back(pad);
}
for (Point p : addPoints)
for (Point2f p : addPoints)
result->append(Pad(p));
for (Point p : delPoints)
for (Point2f p : delPoints)
result->erase(result->getNearest(p));
return result;
@ -44,12 +44,12 @@ void PadFilter::reset()
delPoints.clear();
}
bool PadFilter::clicked(Point p, QMouseEvent *me)
void PadFilter::clicked(Point p, QMouseEvent *me)
{
if (me->modifiers() & Qt::AltModifier)
delPoints.append(p);
delPoints.append(Point2f(p));
else
addPoints.append(p);
addPoints.append(Point2f(p));
return true;
emit filterChanged(this);
}

4
filters/padfilter.h
View File

@ -10,7 +10,7 @@ class PadFilter : public Filter
QString getName() const { return "PadFilter"; }
bool clicked(Point p, QMouseEvent *me);
void clicked(Point p, QMouseEvent *me);
void reset();
protected:
@ -20,7 +20,7 @@ class PadFilter : public Filter
Range<double> areaRange, ratioRange;
QList<Point> addPoints, delPoints;
QList<Point2f> addPoints, delPoints;
};
#endif // PADFILTER_H

117
filters/pathplanner.cpp
View File

@ -9,35 +9,78 @@ PathPlanner::PathPlanner(Filter *s, enum Algorithm a) :
algoMap[NEAREST_NEIGHBOUR] = "Nearest Neighbour";
algoMap[REPETETIVE_NEAREST_NEIGHBOUR] = "Repetitive Nearest Neighbour";
//settings["Algorithm"] = new EnumSetting(this, (int &) algo, algoMap);
settings["Algorithm"] = new EnumSetting(this, (int &) algo, algoMap);
}
Result * PathPlanner::applyInternal(Image *img)
{
PadResult *list = dynamic_cast<PadResult*>(img->getResult(source));
if (list) {
PathResult *path = new PathResult;
PadResult vertices = *list; /* create a copy */
PathResult *path = NULL;
PadResult *padResult = dynamic_cast<PadResult*>(img->getResult(source));
if (vertices.size() < 1 || vertices.size() > 1000) {
qWarning() << "Found too much/less pads (" << vertices.size() << ")! Skipping PathPlanner";
if (padResult) {
if (padResult->size() < 1 || padResult->size() > 1000) {
qWarning() << "Found too much/less pads (" << padResult->size() << ")! Skipping PathPlanner";
return path;
}
switch (algo) {
case NEAREST_NEIGHBOUR:
path->append(vertices.takeFirst());
case NEAREST_NEIGHBOUR: {
path = new PathResult;
PadResult pads = *padResult; /* create a copy */
PadResult::iterator start = pads.begin(); /* fixed */
Pad last = *start;
while (!vertices.isEmpty()) {
auto nearest = vertices.getNearest(path->last().center);
path->append(*start);
pads.erase(start);
while (!pads.isEmpty()) {
PadResult::iterator nearest = pads.getNearest(last);
path->append(*nearest);
vertices.erase(nearest);
}
break;
pads.erase(nearest);
case REPETETIVE_NEAREST_NEIGHBOUR:
// FIXME: implement
break;
last = *nearest;
}
} break;
case REPETETIVE_NEAREST_NEIGHBOUR: {
double minLength = FLT_MAX;
for (int i = 0; i < padResult->size(); i++) {
double currentLength = 0;
PathResult *currentPath = new PathResult;
PadResult pads = *padResult;
PadResult::iterator start = pads.begin() + i;
Pad last = *start;
currentPath->append(*start);
pads.erase(start);
while (!pads.isEmpty()) {
PadResult::iterator nearest = pads.getNearest(last);
CV_Assert(nearest != pads.end());
currentLength += last.getDistance(*nearest);
if (currentLength > minLength)
break;
currentPath->append(*nearest);
pads.erase(nearest);
last = *nearest;
}
if (currentLength > minLength)
delete currentPath;
else {
if (path)
delete path;
minLength = currentLength;
path = currentPath;
}
}
} break;
}
return path;
@ -48,39 +91,25 @@ Result * PathPlanner::applyInternal(Image *img)
return new Result;
}
#if 0
Mat PathPlanner::calcDistanceMat(QList<Point2f> vertices)
{
int size = vertices.size();
Mat distance = Mat::zeros(size, size, CV_32F);
for (int i = 0; i < size; i++) {
for (int j = i+1; j < size; j++) {
distance.at<float>(i, j) = Pad::distance(pads[i], pads[j]);
}
}
return distance;
}
#endif
void PathResult::drawResult(Painter *p) const
{
if (size()) {
if (size() >= 2) {
QPen pe = p->pen();
QBrush br(Qt::SolidPattern);
pe.setWidth(2);
for (int i = 1; i < size(); i++) {
const_iterator prev = begin();
const_iterator curr = begin() + 1;
for (int i = 0; curr != end(); i++, curr++, prev++) {
QColor c = QColor::fromHsl(360.0 / size() * i, 255, 128);
pe.setColor(c);
br.setColor(c);
p->setPen(pe);
p->setBrush(br);
p->drawLine(toQt(at(i-1)), toQt(at(i)));
p->drawEllipse(toQt(at(i)), 1, 1);
p->drawLine(toQt(*prev), toQt(*curr));
p->drawEllipse(toQt(*curr), 1, 1);
}
p->drawMarker(toQt((Point2i) (Point2f) first()));
@ -88,11 +117,17 @@ void PathResult::drawResult(Painter *p) const
}
}
float PathResult::getLength() const
double PathResult::getLength() const
{
float length = 0;
for (int i = 1; i < size(); i++)
length += norm(at(i-1).center - at(i).center);
double length = 0;
if (size() >= 2) {
const_iterator curr = begin() + 1;
const_iterator prev = begin();
for (; curr != end(); prev++, curr++)
length += prev->getDistance(*curr);
}
return length;
}

2
filters/pathplanner.h
View File

@ -30,7 +30,7 @@ class PathResult : public PadResult
public:
void drawResult(Painter *p) const;
float getLength() const;
double getLength() const;
QString getResult() const;
};

10
filters/setting.cpp
View File

@ -6,11 +6,13 @@
#include "viewer.h"
#include "filter.h"
Setting::Setting(Filter *f, QString tip)
: toolTip(tip)
Setting::Setting(Filter *f, QString tip) :
filter(f),
toolTip(tip)
{
connect(this, SIGNAL(valueChanged()), f, SIGNAL(filterChanged()));
connect(this, SIGNAL(valueChanged()), f, SLOT(reset()));
connect(this, &Setting::valueChanged, [&]() {
filter->settingChanged(this);
});
}
QWidget * Setting::getWidget(QWidget *parent)

4
filters/watershed.cpp
View File

@ -1,6 +1,6 @@
#include "watershed.h"
bool Watershed::clicked(Point pos, QMouseEvent *me)
void Watershed::clicked(Point pos, QMouseEvent *me)
{
qDebug() << getName() << ": Marker " << ((me->modifiers() & Qt::AltModifier) ? "deleted" : "added") << " at:" << toQt(pos);
@ -9,7 +9,7 @@ bool Watershed::clicked(Point pos, QMouseEvent *me)
else
addMarkers.push_back(pos);
return true;
emit filterChanged(this);
}
void Watershed::reset()

2
filters/watershed.h
View File

@ -8,7 +8,7 @@ class Watershed : public Filter
public:
QString getName() const { return "Watershed"; }
bool clicked(Point pos, QMouseEvent *me = 0);
void clicked(Point pos, QMouseEvent *me = 0);
public slots:
void reset();

17
image.cpp
View File

@ -26,7 +26,7 @@ Image::~Image()
void Image::load(QString p)
{
if (p != "")
if (!p.isNull())
path = p;
source = cv::imread(path.toStdString());
@ -35,19 +35,12 @@ void Image::load(QString p)
void Image::save(QString p)
{
if (p != "")
if (!p.isNull())
path = p;
saved = imwrite(path.toStdString(), getMat());
}
void Image::applyFilter(Filter *filter)
{
Result *result = filter->apply(this);
if (result)
results[filter] = result;
}
Mat & Image::getMat()
{
if (!filtered.data)
@ -56,6 +49,12 @@ Mat & Image::getMat()
return filtered;
}
QImage Image::getQImage()
{
return toQImage(getMat());
}
const Mat & Image::getSourceMat()
{
if (!loaded)

6
image.h
View File

@ -3,6 +3,7 @@
#include <QString>
#include <QMap>
#include <QImage>
#include <opencv2/core.hpp>
#include <opencv2/imgproc.hpp>
@ -23,8 +24,6 @@ class Image
void save(QString path = QString());
void load(QString path = QString());
void applyFilter(Filter *filter);
/* Getter */
bool isLoaded() const { return loaded; }
bool isSaved() const { return saved; }
@ -32,13 +31,14 @@ class Image
Result * getResult(Filter *f) { return results[f]; }
QString getPath() { return path; }
Mat & getMat();
QImage getQImage();
const Mat & getSourceMat();
void setSourceMat(const Mat &m);
protected:
QMap<Filter*, Result*> results;
protected:
QString path;
Mat source;
Mat filtered;

10
imagelist.cpp
View File

@ -1,6 +1,7 @@
#include <QFileDialog>
#include <QDir>
#include "painter.h"
#include "imagelist.h"
#include "mainwindow.h"
@ -132,7 +133,14 @@ void ImageList::load(QStringList files)
void ImageList::save(QString path)
{
getCurrent()->save(path);
Image *img = getCurrent();
QImage qimg = img->getQImage();
Painter p(&qimg);
p.setRatio(1e-3 * qimg.width());
p.drawOverlay(img);
qimg.save(path);
}
void ImageList::loadFilePicker()

2
images.qrc
View File

@ -31,6 +31,8 @@
<file>resources/go-top.png</file>
<file>resources/process-stop.png</file>
<file>resources/start-here.png</file>
<file>resources/bed_circ.png</file>
<file>resources/bed_rect.png</file>
</qresource>
<qresource prefix="/math">
<file>resources/matbr_left.png</file>

25
main.cpp
View File

@ -2,15 +2,16 @@
#include <QApplication>
#include <QCameraInfo>
/* Filters */
#include "filters.h"
#include "imagelist.h"
#include "filterlist.h"
#include "source.h"
#include "camera.h"
#include "robot.h"
Source *source;
Camera *cam;
Robot *robot;
FilterList *filters;
ImageList *images;
MainWindow *mwindow;
@ -27,7 +28,6 @@ int main(int argc, char *argv[])
cam = new Camera(source);
filters = new FilterList;
images = new ImageList;
mwindow = new MainWindow;
QStringList imgs = QCoreApplication::arguments();
imgs.removeFirst();
@ -35,22 +35,25 @@ int main(int argc, char *argv[])
images->load(imgs);
/* Setup pipeline */
Pattern *pat = new Pattern(Size(2, 2), Size(60, 60), Pattern::QUADRILINEAR_MARKERS);
filters->add(pat);
filters->add(new Perspective(cam, pat));
filters->add(new Resize(Range<int>(400, 1000)));
Pattern *pattern = new Pattern(Size(2, 2), Size(60, 60), Pattern::QUADRILINEAR_MARKERS);
PadDetect *pads = new PadDetect();
PadFilter *filter = new PadFilter(pads);
PathPlanner *planner = new PathPlanner(filter, PathPlanner::REPETETIVE_NEAREST_NEIGHBOUR);
filters->add(pattern);
filters->add(new Perspective(cam, pattern));
filters->add(new Resize(Range<int>(400, 1000)));
filters->add(new Blur(Blur::GAUSSIAN, Size(3, 3)));
filters->add(new KMeans(4));
filters->add(new Convert(COLOR_BGR2GRAY));
filters->add(new Threshold(Threshold::OTSU));
PadDetect *pads = new PadDetect();
PadFilter *filter = new PadFilter(pads);
filters->add(new Morph(MORPH_CLOSE, MORPH_RECT));
filters->add(pads);
filters->add(filter);
filters->add(new PathPlanner(filter, PathPlanner::NEAREST_NEIGHBOUR));
filters->add(planner);
robot = new Robot(pattern, planner);
mwindow = new MainWindow;
mwindow->show();

48
mainwindow.cpp
View File

@ -30,15 +30,15 @@ MainWindow::MainWindow(QWidget *parent) :
});
/* Connect actions */
connect(images, &ImageList::newImage, ui->viewer, &Viewer::showImage);
connect(source, &Source::newImage, ui->viewer, &Viewer::showImage);
connect(filters, &FilterList::filtersChanged,ui->viewer, &Viewer::updateImage);
connect(ui->actionRedraw, &QAction::triggered, ui->viewer, &Viewer::updateImage);
connect(images, &ImageList::newImage, this, &MainWindow::showImage);
connect(source, &Source::newFrame, this, &MainWindow::showFrame);
connect(filters, &FilterList::filterChanged, this, &MainWindow::render);
connect(ui->actionRedraw, &QAction::triggered, this, &MainWindow::render);
connect(ui->actionAbout, &QAction::triggered, this, &MainWindow::showAbout);
connect(ui->actionExit, &QAction::triggered, this, &MainWindow::close);
connect(ui->actionReset, &QAction::triggered, ui->viewer, &Viewer::reset);
connect(ui->actionCalibrate, &QAction::triggered, ui->tabCalibration, &TabCalibration::doCalibration);
connect(ui->actionSnapshot, &QAction::triggered, ui->tabCamera, &TabCamera::doSnapshot);
connect(ui->actionAbout, &QAction::triggered, this, &MainWindow::showAbout);
connect(ui->actionExit, &QAction::triggered, this, &MainWindow::close);
connect(ui->actionClear, &QAction::triggered, images, &ImageList::clear);
connect(ui->actionLoad, &QAction::triggered, images, &ImageList::loadFilePicker);
connect(ui->actionSave, &QAction::triggered, images, &ImageList::saveFilePicker);
@ -46,6 +46,7 @@ MainWindow::MainWindow(QWidget *parent) :
connect(ui->actionPrevImage, &QAction::triggered, images, &ImageList::prevImage);
connect(ui->actionReset, &QAction::triggered, filters, &FilterList::reset);
connect(ui->actionPlay, &QAction::triggered, source, &Source::play);
connect(ui->actionTabImages, &QAction::triggered, [=]() { ui->tabWidget->setCurrentWidget(ui->tabImages); });
connect(ui->actionTabFilters, &QAction::triggered, [=]() { ui->tabWidget->setCurrentWidget(ui->tabFilters); });
connect(ui->actionTabCamera, &QAction::triggered, [=]() { ui->tabWidget->setCurrentWidget(ui->tabCamera); });
@ -55,11 +56,6 @@ MainWindow::MainWindow(QWidget *parent) :
showMaximized();
}
void MainWindow::showEvent(QShowEvent *)
{
ui->viewer->showImage(images->getCurrent());
}
MainWindow::~MainWindow()
{
delete ui;
@ -73,3 +69,33 @@ void MainWindow::showAbout()
about.setupUi(&dialog);
dialog.exec();
}
void MainWindow::showImage(Image *i)
{
source->play(false);
showFrame(i);
}
void MainWindow::showFrame(Image *i)
{
currentImage = i;
render();
}
void MainWindow::render()
{
currentImage->getMat().release();
try {
for (Filter *filter : *filters)
filter->apply(currentImage);
} catch (Exception e) {
qCritical("%s", e.what());
}
qDebug() << "Rendering completed";
ui->viewer->showImage(currentImage);
}

11
mainwindow.h
View File

@ -29,16 +29,19 @@ class MainWindow : public QMainWindow
explicit MainWindow(QWidget *parent = 0);
~MainWindow();
/* Getter */
Image * getCurrentImage() { return currentImage; }
public slots:
void showAbout();
void showImage(Image *img);
void showFrame(Image *img);
void render();
protected:
Ui::MainWindow *ui;
void showEvent(QShowEvent *se);
signals:
void newImage(Image *img);
Image *currentImage;
};
#endif // MAINWINDOW_H

3
pad.h
View File

@ -13,8 +13,11 @@ class Pad : public RotatedRect
Pad(const Point &p);
Pad(const RotatedRect &p);
/* Getter */
double getArea() const { return size.area(); }
double getRatio() const { return qMax(size.width, size.height) / qMin(size.width, size.height); }
double getDistance(const Pad &b) const { return norm(center - b.center); }
double getDistance(const Point2f &b) const { return norm(center - b); }
Point2f getCenter() const { return center; }
Point2f const * getVertexes() const { return vertexes; }

37
painter.cpp
View File

@ -1,11 +1,39 @@
#include "painter.h"
#include "filters/filter.h"
Painter::Painter(QPaintDevice *dev) :
QPainter(dev)
{
setRenderHint(Antialiasing);
}
void Painter::drawOverlay(Image *img)
{
for (Filter *filter : img->results.keys()) {
Result *result = img->getResult(filter);
if (result && filter->isShown()) {
save();
/*QTransform t;
for (auto it2 = it+1; it2 != filters->end(); it2++) {
Result *result2 = img->getResult(*it2);
if (result2 && (*it2)->isShown())
t *= result2->getTransform();
}
p.setWorldTransform(t * transform);*/
result->drawResult(this);
restore();
}
}
}
void Painter::drawMarker(const QPoint &center, int r)
{
QBrush br(Qt::NoBrush);
setBrush(br);
r /= getRatio();
r *= ratio;
drawEllipse(center, r, r);
drawLine(center + QPoint(+r, +r), center - QPoint(+r, +r));
@ -27,12 +55,7 @@ void Painter::setPen(const QPen &pen)
{
QPen p(pen);
p.setWidthF((double) p.widthF() / getRatio());
p.setWidthF((double) p.widthF() * ratio);
QPainter::setPen(p);
}
double Painter::getRatio()
{
return (transform().m11() + transform().m22()) / 2;
}

13
painter.h
View File

@ -3,15 +3,16 @@
#include <QPainter>
#include "image.h"
#include "cast.h"
#include "pad.h"
class Painter : public QPainter
{
public:
Painter(QPaintDevice *dev) :
QPainter(dev)
{ }
Painter(QPaintDevice *dev);
void drawOverlay(Image *p);
void drawMarker(const QPoint &center, int radius = 8);
void drawPad(const Pad &pad);
@ -19,7 +20,11 @@ class Painter : public QPainter
/* Proxy to adjust width */
void setPen(const QPen &pen);
double getRatio();
double getRatio() const { return ratio; }
void setRatio(double r) { ratio = r; }
protected:
double ratio;
};
#endif // PAINTER_H

2
source.cpp
View File

@ -75,7 +75,7 @@ void Source::tick()
if (read(m)) {
last.setSourceMat(m);
emit newImage(&last);
emit newFrame(&last);
}
}

2
source.h
View File

@ -31,7 +31,7 @@ class Source :
void play(bool);
signals:
void newImage(Image *img);
void newFrame(Image *img);
protected slots:
void tick();

6
tabfilters.cpp
View File

@ -13,6 +13,11 @@ TabFilters::TabFilters(QWidget *parent) :
ui->tblFilters->setModel(filters);
ui->tblFilters->setSelectionModel(&filters->selection);
ui->tblFilters->resizeColumnsToContents();
ui->tblFilters->resizeRowsToContents();
for (Filter *filter : *filters)
ui->wdgSettings->addWidget(filter->getWidget(this));
connect(ui->tblFilters->verticalHeader(), &QHeaderView::sectionCountChanged, [&]() {
ui->tblFilters->resizeColumnsToContents();
@ -25,7 +30,6 @@ TabFilters::TabFilters(QWidget *parent) :
connect(filters, &FilterList::filterAdded, [&] (Filter *filter) {
ui->wdgSettings->addWidget(filter->getWidget(this));
});
}
TabFilters::~TabFilters()

3
tabimages.cpp
View File

@ -14,11 +14,14 @@ TabImages::TabImages(QWidget *parent) :
ui->tblImages->setModel(images);
ui->tblImages->setSelectionModel(&images->selection);
ui->tblImages->setColumnWidth(0, 200);
ui->tblImages->resizeColumnsToContents();
ui->tblImages->resizeRowsToContents();
connect(ui->tblImages->verticalHeader(), &QHeaderView::sectionCountChanged, [&]() {
ui->tblImages->resizeColumnsToContents();
ui->tblImages->resizeRowsToContents();
});
connect(ui->btnClear, &QPushButton::clicked, images, &ImageList::clear);
connect(ui->btnLoad, &QPushButton::clicked, images, &ImageList::loadFilePicker);
connect(ui->btnSave, &QPushButton::clicked, images, &ImageList::saveFilePicker);

94
viewer.cpp
View File

@ -17,6 +17,7 @@ extern FilterList *filters;
Viewer::Viewer(QWidget *parent) :
QGLWidget(parent),
mouseMode(MOUSE_NONE),
img(NULL)
{ }
@ -36,10 +37,10 @@ void Viewer::paintEvent(QPaintEvent *)
Painter p(this);
QPen pe;
p.setRenderHint(QPainter::Antialiasing);
p.fillRect(rect(), Qt::black);
p.setClipRect(viewport);
p.setTransform(transform);
p.setRatio(1 / transform.m11());
pe.setStyle(Qt::SolidLine);
pe.setWidth(2);
@ -48,26 +49,10 @@ void Viewer::paintEvent(QPaintEvent *)
if (img) {
p.drawImage(window, qimg, window);
for (auto it = filters->begin(); it != filters->end(); it++) {
Result *result = img->getResult(*it);
if (result && (*it)->isShown()) {
p.save();
/*QTransform t;
for (auto it2 = it+1; it2 != filters->end(); it2++) {
Result *result2 = img->getResult(*it2);
if (result2 && (*it2)->isShown())
t *= result2->getTransform();
}
p.setWorldTransform(t * transform);*/
result->drawResult(&p);
p.restore();
}
}
p.drawOverlay(img);
}
if (!first.isNull() && !last.isNull()) {
if (mouseMode == MOUSE_ZOOMING) {
QRect sel(unmap(first), unmap(last));
pe.setWidth(2);
@ -85,6 +70,11 @@ void Viewer::paintEvent(QPaintEvent *)
void Viewer::wheelEvent(QWheelEvent * we)
{
if (mouseMode != MOUSE_NONE) {
we->ignore();
return;
}
QPoint numPixels = we->pixelDelta();
QPoint numDegrees = we->angleDelta() / 8;
@ -113,80 +103,70 @@ void Viewer::wheelEvent(QWheelEvent * we)
void Viewer::mousePressEvent(QMouseEvent *me)
{
if (viewport.contains(me->pos())) {
first = me->pos();
last = me->pos();
if (me->modifiers() & Qt::AltModifier)
first = me->pos();
mouseMode = MOUSE_ZOOMING;
else
last = me->pos();
mouseMode = MOUSE_DRAGGING;
}
}
void Viewer::mouseMoveEvent(QMouseEvent *me)
{
if (viewport.contains(me->pos())) {
if (me->modifiers() & Qt::AltModifier) {
update();
}
else {
if (mouseMode == MOUSE_ZOOMING)
{ } /* draw zoom rectangle */
else if (mouseMode == MOUSE_DRAGGING) {
QPoint delta = (me->pos() - last) / transform.m11();
window.moveCenter(window.center() - delta);
updateWindow(window);
update();
}
last = me->pos();
}
last = me->pos();
update();
}
void Viewer::mouseReleaseEvent(QMouseEvent *me)
{
if (viewport.contains(me->pos())) {
if (me->modifiers() & Qt::AltModifier) {
if (!first.isNull() && !last.isNull())
if (mouseMode == MOUSE_ZOOMING) {
if (!first.isNull() && !last.isNull()) {
updateWindow(QRect(unmap(first), unmap(last)).normalized());
first = QPoint();
last = QPoint();
}
}
else if (last.isNull()) {
else if (mouseMode == MOUSE_DRAGGING) {
QPoint pos = unmap(me->pos());
Filter *f = filters->getCurrent();
qDebug() << "Clicked at: " << pos << "(" << me->pos() << ")";
if (f) {
if (f->clicked(toCv(pos), me))
updateImage();
qDebug() << "Clicked at: " << pos << "(" << me->pos() << ") << for filer " << f->getName();
f->clicked(toCv(pos), me);
}
}
}
first = last = QPoint();
mouseMode = MOUSE_NONE;
first = last = QPoint();
}
update();
}
void Viewer::showImage(Image *next)
{
if (next) {
if (next != img)
filters->reset();
img = next;
qimg = img->getQImage();
img = next;
updateImage();
}
}
if (window.size() != qimg.size())
updateWindow(QRect(QPoint(), qimg.size()));
void Viewer::updateImage()
{
if (img) {
filters->execute(img);
Mat &m = img->getMat();
qimg = toQImage(m);
if (window.size() != toQt(m.size()))
updateWindow(QRect(0, 0, m.cols, m.rows));
update();
}
update();
}
void Viewer::updateWindow(const QRect &win)

7
viewer.h
View File

@ -18,6 +18,12 @@ class Viewer : public QGLWidget
explicit Viewer(QWidget *parent = 0);
protected:
enum MouseMode {
MOUSE_NONE,
MOUSE_ZOOMING,
MOUSE_DRAGGING
} mouseMode;
void paintEvent(QPaintEvent *pe);
void resizeEvent(QResizeEvent *re);
@ -41,7 +47,6 @@ class Viewer : public QGLWidget
public slots:
void showImage(Image *img);
void updateImage();
void reset();
};