Subversion Repositories seema-scanner

Rev

Rev 103 | Rev 113 | Go to most recent revision | Only display areas with differences | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed

Rev 103 Rev 112
1
#include "SMScanner.h"
1
#include "SMScanner.h"
2
#include "ui_SMScanner.h"
2
#include "ui_SMScanner.h"
3
 
3
 
4
#include <QMetaObject>
4
#include <QMetaObject>
5
#include <QFileDialog>
5
#include <QFileDialog>
6
#include <QMessageBox>
6
#include <QMessageBox>
7
 
7
 
8
#include <pcl/io/pcd_io.h>
8
#include <pcl/io/pcd_io.h>
9
#include <pcl/io/ascii_io.h>
9
#include <pcl/io/ascii_io.h>
10
#include <pcl/io/ply_io.h>
10
#include <pcl/io/ply_io.h>
11
#include <pcl/io/png_io.h>
11
#include <pcl/io/png_io.h>
12
#include <pcl/io/vtk_io.h>
12
#include <pcl/io/vtk_io.h>
13
#include <vtkPolyDataWriter.h>
13
#include <vtkPolyDataWriter.h>
14
#include <pcl/conversions.h>
14
#include <pcl/conversions.h>
15
 
15
 
16
SMScanner::SMScanner(QWidget *parent) :QMainWindow(parent), ui(new Ui::SMScanner),
16
SMScanner::SMScanner(QWidget *parent) :QMainWindow(parent), ui(new Ui::SMScanner),
17
                                        calibrationReviewMode(false), captureReviewMode(false), lastCaptureId(-1){
17
                                        calibrationReviewMode(false), captureReviewMode(false), lastCaptureId(-1){
18
 
18
 
19
    // Register metatypes
19
    // Register metatypes
20
    qRegisterMetaType<cv::Mat>("cv::Mat");
20
    qRegisterMetaType<cv::Mat>("cv::Mat");
21
    qRegisterMetaType< std::vector<cv::Mat> >("std::vector<cv::Mat>");
21
    qRegisterMetaType< std::vector<cv::Mat> >("std::vector<cv::Mat>");
22
    qRegisterMetaType< std::vector<float> >("std::vector<float>");
22
    qRegisterMetaType< std::vector<float> >("std::vector<float>");
23
    qRegisterMetaType<SMCalibrationSet>("SMCalibrationSet");
23
    qRegisterMetaType<SMCalibrationSet>("SMCalibrationSet");
24
    qRegisterMetaType<SMCalibrationParameters>("SMCalibrationParameters");
24
    qRegisterMetaType<SMCalibrationParameters>("SMCalibrationParameters");
25
    qRegisterMetaTypeStreamOperators<SMCalibrationParameters>("SMCalibrationParameters");
25
    qRegisterMetaTypeStreamOperators<SMCalibrationParameters>("SMCalibrationParameters");
26
    qRegisterMetaType< std::vector<SMCalibrationSet> >("std::vector<SMCalibrationSet>");
26
    qRegisterMetaType< std::vector<SMCalibrationSet> >("std::vector<SMCalibrationSet>");
27
    qRegisterMetaType<SMFrameSequence>("SMFrameSequence");
27
    qRegisterMetaType<SMFrameSequence>("SMFrameSequence");
28
    qRegisterMetaType< std::vector<SMFrameSequence> >("std::vector<SMFrameSequence>");
28
    qRegisterMetaType< std::vector<SMFrameSequence> >("std::vector<SMFrameSequence>");
29
    qRegisterMetaType<pcl::PointCloud<pcl::PointXYZRGB>::Ptr>("pcl::PointCloud<pcl::PointXYZRGB>::Ptr");
29
    qRegisterMetaType<pcl::PointCloud<pcl::PointXYZRGB>::Ptr>("pcl::PointCloud<pcl::PointXYZRGB>::Ptr");
30
    qRegisterMetaType<SMPointCloud>("SMPointCloud");
30
    qRegisterMetaType<SMPointCloud>("SMPointCloud");
31
 
31
 
32
    // Setup ui (requires stream operators registered)
32
    // Setup ui (requires stream operators registered)
33
    ui->setupUi(this);
33
    ui->setupUi(this);
34
 
34
 
35
    // Restore geometry
35
    // Restore geometry
36
    this->restoreGeometry(settings.value("geometry/mainwindow").toByteArray());
36
    this->restoreGeometry(settings.value("geometry/mainwindow").toByteArray());
37
    this->restoreState(settings.value("state/mainwindow").toByteArray());
37
    this->restoreState(settings.value("state/mainwindow").toByteArray());
38
 
38
 
39
    // Set up threads
39
    // Set up threads
40
    captureWorker = new SMCaptureWorker;
40
    captureWorker = new SMCaptureWorker;
41
    captureWorkerThread = new QThread(this);
41
    captureWorkerThread = new QThread(this);
42
    captureWorkerThread->setObjectName("captureWorkerThread");
42
    captureWorkerThread->setObjectName("captureWorkerThread");
43
    captureWorker->moveToThread(captureWorkerThread);
43
    captureWorker->moveToThread(captureWorkerThread);
44
    captureWorkerThread->start();
44
    captureWorkerThread->start();
45
 
45
 
46
    // Connections
46
    // Connections
47
    connect(captureWorker, SIGNAL(newFrame(unsigned int, cv::Mat)), this, SLOT(onReceiveFrame(unsigned int, cv::Mat)));
47
    connect(captureWorker, SIGNAL(newFrame(unsigned int, cv::Mat)), this, SLOT(onReceiveFrame(unsigned int, cv::Mat)));
48
    connect(captureWorker, SIGNAL(newCalibrationSet(SMCalibrationSet)), this, SLOT(onReceiveCalibrationSet(SMCalibrationSet)));
48
    connect(captureWorker, SIGNAL(newCalibrationSet(SMCalibrationSet)), this, SLOT(onReceiveCalibrationSet(SMCalibrationSet)));
49
    connect(captureWorker, SIGNAL(newFrameSequence(SMFrameSequence)), this, SLOT(onReceiveFrameSequence(SMFrameSequence)));
49
    connect(captureWorker, SIGNAL(newFrameSequence(SMFrameSequence)), this, SLOT(onReceiveFrameSequence(SMFrameSequence)));
50
    connect(captureWorker, SIGNAL(rotatedTo(float)), this, SLOT(onReceiveRotatedTo(float)));
50
    connect(captureWorker, SIGNAL(rotatedTo(float)), this, SLOT(onReceiveRotatedTo(float)));
51
 
51
 
52
    // Start capturing
52
    // Start capturing
53
    QMetaObject::invokeMethod(captureWorker, "setup");
53
    QMetaObject::invokeMethod(captureWorker, "setup");
54
    QMetaObject::invokeMethod(captureWorker, "doWork");
54
    QMetaObject::invokeMethod(captureWorker, "doWork");
55
 
55
 
56
}
56
}
57
 
57
 
58
void SMScanner::onReceiveFrame(unsigned int camId, cv::Mat frame){
58
void SMScanner::onReceiveFrame(unsigned int camId, cv::Mat frame){
59
 
59
 
60
    if(camId == 0){
60
    if(camId == 0){
61
        if(!calibrationReviewMode)
61
        if(!calibrationReviewMode)
62
            ui->calibrationCamera0Widget->showImageCV(frame);
62
            ui->calibrationCamera0Widget->showImageCV(frame);
63
        if(!captureReviewMode)
63
        if(!captureReviewMode)
64
            ui->captureCamera0Widget->showImageCV(frame);
64
            ui->captureCamera0Widget->showImageCV(frame);
65
    } else if(camId == 1){
65
    } else if(camId == 1){
66
        if(!calibrationReviewMode)
66
        if(!calibrationReviewMode)
67
            ui->calibrationCamera1Widget->showImageCV(frame);
67
            ui->calibrationCamera1Widget->showImageCV(frame);
68
        if(!captureReviewMode)
68
        if(!captureReviewMode)
69
            ui->captureCamera1Widget->showImageCV(frame);
69
            ui->captureCamera1Widget->showImageCV(frame);
70
    }
70
    }
71
}
71
}
72
 
72
 
73
void SMScanner::on_actionPreferences_triggered(){
73
void SMScanner::on_actionPreferences_triggered(){
74
 
74
 
75
    connect(&preferenceDialog, SIGNAL(accepted()), this, SLOT(onPreferencesChanged()));
75
    connect(&preferenceDialog, SIGNAL(accepted()), this, SLOT(onPreferencesChanged()));
76
 
76
 
77
    preferenceDialog.show();
77
    preferenceDialog.show();
78
}
78
}
79
 
79
 
80
void SMScanner::onPreferencesChanged(){
80
void SMScanner::onPreferencesChanged(){
81
 
81
 
82
//    // Stop capturing thread
82
//    // Stop capturing thread
83
//    connect(captureWorker, SIGNAL(finished()), captureWorker, SLOT(deleteLater()));
83
//    connect(captureWorker, SIGNAL(finished()), captureWorker, SLOT(deleteLater()));
84
//    connect(captureWorker, SIGNAL(finished()), captureWorkerThread, SLOT(quit()));
84
//    connect(captureWorker, SIGNAL(finished()), captureWorkerThread, SLOT(quit()));
85
//    QMetaObject::invokeMethod(captureWorker, "stopWork");
85
//    QMetaObject::invokeMethod(captureWorker, "stopWork");
86
//    captureWorkerThread->quit();
86
//    captureWorkerThread->quit();
87
//    captureWorkerThread->wait();
87
//    captureWorkerThread->wait();
88
 
88
 
89
//    // Restart capture worker
89
//    // Restart capture worker
90
//    captureWorker = new SMCaptureWorker;
90
//    captureWorker = new SMCaptureWorker;
91
//    captureWorkerThread = new QThread(this);
91
//    captureWorkerThread = new QThread(this);
92
//    captureWorkerThread->setObjectName("captureWorkerThread");
92
//    captureWorkerThread->setObjectName("captureWorkerThread");
93
//    captureWorker->moveToThread(captureWorkerThread);
93
//    captureWorker->moveToThread(captureWorkerThread);
94
//    captureWorkerThread->start();
94
//    captureWorkerThread->start();
95
 
95
 
96
//    connect(captureWorker, SIGNAL(newFrame(unsigned int, cv::Mat)), this, SLOT(onReceiveFrame(unsigned int, cv::Mat)));
96
//    connect(captureWorker, SIGNAL(newFrame(unsigned int, cv::Mat)), this, SLOT(onReceiveFrame(unsigned int, cv::Mat)));
97
//    connect(captureWorker, SIGNAL(newCalibrationSet(SMCalibrationSet)), this, SLOT(onReceiveCalibrationSet(SMCalibrationSet)));
97
//    connect(captureWorker, SIGNAL(newCalibrationSet(SMCalibrationSet)), this, SLOT(onReceiveCalibrationSet(SMCalibrationSet)));
98
//    connect(captureWorker, SIGNAL(newFrameSequence(SMFrameSequence)), this, SLOT(onReceiveFrameSequence(SMFrameSequence)));
98
//    connect(captureWorker, SIGNAL(newFrameSequence(SMFrameSequence)), this, SLOT(onReceiveFrameSequence(SMFrameSequence)));
99
//    connect(captureWorker, SIGNAL(rotatedTo(float)), this, SLOT(onReceiveRotatedTo(float)));
99
//    connect(captureWorker, SIGNAL(rotatedTo(float)), this, SLOT(onReceiveRotatedTo(float)));
100
 
100
 
101
//    QMetaObject::invokeMethod(captureWorker, "setup");
101
//    QMetaObject::invokeMethod(captureWorker, "setup");
102
//    QMetaObject::invokeMethod(captureWorker, "doWork");
102
//    QMetaObject::invokeMethod(captureWorker, "doWork");
103
 
103
 
104
    QMessageBox::warning(this, "Information", "You must quit and restart the program for new settings to take effect!", QMessageBox::Ok);
104
    QMessageBox::warning(this, "Information", "You must quit and restart the program for new settings to take effect!", QMessageBox::Ok);
105
 
105
 
106
}
106
}
107
 
107
 
108
void SMScanner::closeEvent(QCloseEvent *event){
108
void SMScanner::closeEvent(QCloseEvent *event){
109
 
109
 
110
    // Stop capturing thread
110
    // Stop capturing thread
111
    connect(captureWorker, SIGNAL(finished()), captureWorker, SLOT(deleteLater()));
111
    connect(captureWorker, SIGNAL(finished()), captureWorker, SLOT(deleteLater()));
112
    connect(captureWorker, SIGNAL(finished()), captureWorkerThread, SLOT(quit()));
112
    connect(captureWorker, SIGNAL(finished()), captureWorkerThread, SLOT(quit()));
113
    QMetaObject::invokeMethod(captureWorker, "stopWork");
113
    QMetaObject::invokeMethod(captureWorker, "stopWork");
114
    captureWorkerThread->quit();
114
    captureWorkerThread->quit();
115
    captureWorkerThread->wait();
115
    captureWorkerThread->wait();
116
 
116
 
117
    // Save window geometry
117
    // Save window geometry
118
    settings.setValue("geometry/mainwindow", this->saveGeometry());
118
    settings.setValue("geometry/mainwindow", this->saveGeometry());
119
    settings.setValue("state/mainwindow", this->saveState());
119
    settings.setValue("state/mainwindow", this->saveState());
120
 
120
 
121
    event->accept();
121
    event->accept();
122
 
122
 
123
}
123
}
124
 
124
 
125
SMScanner::~SMScanner(){
125
SMScanner::~SMScanner(){
126
    delete ui;
126
    delete ui;
127
}
127
}
128
 
128
 
129
void SMScanner::on_singleCalibrationButton_clicked(){
129
void SMScanner::on_singleCalibrationButton_clicked(){
130
 
130
 
131
    // If in review mode, go back to aquisition mode
131
    // If in review mode, go back to aquisition mode
132
    if(calibrationReviewMode){
132
    if(calibrationReviewMode){
133
        ui->calibrationListWidget->clearSelection();
133
        ui->calibrationListWidget->clearSelection();
134
        ui->singleCalibrationButton->setText("Single Aquisition");
134
        ui->singleCalibrationButton->setText("Single Aquisition");
135
        ui->batchCalibrationButton->setText("Batch Aquisition");
135
        ui->batchCalibrationButton->setText("Batch Aquisition");
136
        calibrationReviewMode = false;
136
        calibrationReviewMode = false;
137
        return;
137
        return;
138
    }
138
    }
139
 
139
 
140
    QMetaObject::invokeMethod(captureWorker, "acquireCalibrationSet", Q_ARG(float, -1.0));
140
    QMetaObject::invokeMethod(captureWorker, "acquireCalibrationSet", Q_ARG(float, -1.0));
141
 
141
 
142
}
142
}
143
 
143
 
144
 
144
 
145
void SMScanner::on_batchCalibrationButton_clicked(){
145
void SMScanner::on_batchCalibrationButton_clicked(){
146
 
146
 
147
    // If in review mode, go back to aquisition mode
147
    // If in review mode, go back to aquisition mode
148
    if(calibrationReviewMode){
148
    if(calibrationReviewMode){
149
        ui->calibrationListWidget->clearSelection();
149
        ui->calibrationListWidget->clearSelection();
150
        ui->singleCalibrationButton->setText("Single Aquisition");
150
        ui->singleCalibrationButton->setText("Single Aquisition");
151
        ui->batchCalibrationButton->setText("Batch Aquisition");
151
        ui->batchCalibrationButton->setText("Batch Aquisition");
152
        calibrationReviewMode = false;
152
        calibrationReviewMode = false;
153
        return;
153
        return;
154
    }
154
    }
155
 
155
 
156
    // Construct vector of angles
156
    // Construct vector of angles
157
    int angleStart = ui->calibrationBatchStartSpinBox->value();
157
    int angleStart = ui->calibrationBatchStartSpinBox->value();
158
    int angleEnd = ui->calibrationBatchEndSpinBox->value();
158
    int angleEnd = ui->calibrationBatchEndSpinBox->value();
159
    int angleStep = ui->calibrationBatchStepSpinBox->value();
159
    int angleStep = ui->calibrationBatchStepSpinBox->value();
160
 
160
 
161
    if(angleStart > angleEnd)
161
    if(angleStart > angleEnd)
162
        angleEnd += 360;
162
        angleEnd += 360;
163
 
163
 
164
    std::vector<float> angles;
164
    std::vector<float> angles;
165
    for(int i=angleStart; i<=angleEnd; i+=angleStep)
165
    for(int i=angleStart; i<=angleEnd; i+=angleStep)
166
        angles.push_back(i % 360);
166
        angles.push_back(i % 360);
167
 
167
 
168
    QMetaObject::invokeMethod(captureWorker, "acquireCalibrationSets", Q_ARG(std::vector<float>, angles));
168
    QMetaObject::invokeMethod(captureWorker, "acquireCalibrationSets", Q_ARG(std::vector<float>, angles));
169
 
169
 
170
    std::cout << "Aquiring sets at ";
170
    std::cout << "Aquiring sets at ";
171
    for(unsigned int i=0; i<angles.size(); i++)
171
    for(unsigned int i=0; i<angles.size(); i++)
172
        std::cout << angles[i] << " ";
172
        std::cout << angles[i] << " ";
173
    std::cout << " degrees" <<std::endl;
173
    std::cout << " degrees" <<std::endl;
174
}
174
}
175
 
175
 
176
 
176
 
177
void SMScanner::on_calibrationRotationDial_sliderReleased(){
177
void SMScanner::on_calibrationRotationDial_sliderReleased(){
178
 
178
 
179
    float angle = ui->calibrationRotationDial->value();
179
    float angle = ui->calibrationRotationDial->value();
180
    std::cout << "Rotation stage target: " << angle << std::endl;
180
    std::cout << "Rotation stage target: " << angle << std::endl;
181
    QMetaObject::invokeMethod(captureWorker, "rotateTo", Q_ARG(float, angle));
181
    QMetaObject::invokeMethod(captureWorker, "rotateTo", Q_ARG(float, angle));
182
 
182
 
183
    ui->captureRotationDial->setValue(ui->calibrationRotationDial->value());
183
    ui->captureRotationDial->setValue(ui->calibrationRotationDial->value());
184
}
184
}
185
 
185
 
186
void SMScanner::onReceiveCalibrationSet(SMCalibrationSet calibrationSet){
186
void SMScanner::onReceiveCalibrationSet(SMCalibrationSet calibrationSet){
187
    calibrationData.push_back(calibrationSet);
187
    calibrationData.push_back(calibrationSet);
188
 
188
 
189
    // Add identifier to list
189
    // Add identifier to list
190
    QListWidgetItem* item = new QListWidgetItem(QString("Calibration Set %1 -- %2 deg").arg(ui->calibrationListWidget->count()).arg(calibrationSet.rotationAngle), ui->calibrationListWidget);
190
    QListWidgetItem* item = new QListWidgetItem(QString("Calibration Set %1 -- %2 deg").arg(ui->calibrationListWidget->count()).arg(calibrationSet.rotationAngle), ui->calibrationListWidget);
191
    item->setFlags(item->flags() | Qt::ItemIsUserCheckable);
191
    item->setFlags(item->flags() | Qt::ItemIsUserCheckable);
192
    item->setCheckState(Qt::Checked);
192
    item->setCheckState(Qt::Checked);
193
    ui->calibrationListWidget->addItem(item);
193
    ui->calibrationListWidget->addItem(item);
194
 
194
 
195
    // Set enabled checkmark
195
    // Set enabled checkmark
196
    if(calibrationData.size() >= 2)
196
    if(calibrationData.size() >= 2)
197
        ui->calibrateButton->setEnabled(true);
197
        ui->calibrateButton->setEnabled(true);
198
}
198
}
199
 
199
 
200
void SMScanner::on_calibrateButton_clicked(){
200
void SMScanner::on_calibrateButton_clicked(){
201
 
201
 
202
    // disable ui elements
202
    // disable ui elements
203
    ui->calibrateButton->setEnabled(false);
203
    ui->calibrateButton->setEnabled(false);
204
    ui->calibrationFrame->setEnabled(false);
204
    ui->calibrationFrame->setEnabled(false);
205
 
205
 
206
    // set checked flags
206
    // set checked flags
207
    for(unsigned int i=0; i<calibrationData.size(); i++){
207
    for(unsigned int i=0; i<calibrationData.size(); i++){
208
        calibrationData[i].checked = (ui->calibrationListWidget->item(i)->checkState() == Qt::Checked);
208
        calibrationData[i].checked = (ui->calibrationListWidget->item(i)->checkState() == Qt::Checked);
209
    }
209
    }
210
 
210
 
211
//    SMCalibrationWorker calibrationWorker;
211
//    SMCalibrationWorker calibrationWorker;
212
//    calibrationWorker.performCalibration(calibrationData);
212
//    calibrationWorker.performCalibration(calibrationData);
213
 
213
 
214
    // Set up calibration thread
214
    // Set up calibration thread
215
    calibrationWorker = new SMCalibrationWorker;
215
    calibrationWorker = new SMCalibrationWorker;
216
    calibrationWorkerThread = new QThread(this);
216
    calibrationWorkerThread = new QThread(this);
217
    calibrationWorkerThread->setObjectName("calibrationWorkerThread");
217
    calibrationWorkerThread->setObjectName("calibrationWorkerThread");
218
    calibrationWorker->moveToThread(captureWorkerThread);
218
    calibrationWorker->moveToThread(captureWorkerThread);
219
    calibrationWorkerThread->start();
219
    calibrationWorkerThread->start();
220
 
220
 
221
    // Connections
221
    // Connections
222
    connect(calibrationWorker, SIGNAL(newSetProcessed(int)), this, SLOT(onCalibrationSetProcessed(int)));
222
    connect(calibrationWorker, SIGNAL(newSetProcessed(int)), this, SLOT(onCalibrationSetProcessed(int)));
223
    connect(calibrationWorker, SIGNAL(newFrameResult(int,int,bool,cv::Mat)), this, SLOT(onCalibrationFrameResult(int,int,bool,cv::Mat)));
223
    connect(calibrationWorker, SIGNAL(newFrameResult(int,int,bool,cv::Mat)), this, SLOT(onCalibrationFrameResult(int,int,bool,cv::Mat)));
224
    connect(calibrationWorker, SIGNAL(done()), this, SLOT(onCalibrationDone()));
224
    connect(calibrationWorker, SIGNAL(done()), this, SLOT(onCalibrationDone()));
225
    connect(calibrationWorker, SIGNAL(done()), ui->pointCloudWidget, SLOT(updateCalibrationParameters()));
225
    connect(calibrationWorker, SIGNAL(done()), ui->pointCloudWidget, SLOT(updateCalibrationParameters()));
226
    connect(calibrationWorker, SIGNAL(done()), calibrationWorkerThread, SLOT(quit()));
226
    connect(calibrationWorker, SIGNAL(done()), calibrationWorkerThread, SLOT(quit()));
227
    connect(calibrationWorker, SIGNAL(done()), calibrationWorker, SLOT(deleteLater()));
227
    connect(calibrationWorker, SIGNAL(done()), calibrationWorker, SLOT(deleteLater()));
228
 
228
 
229
    // Start calibration
229
    // Start calibration
230
    QMetaObject::invokeMethod(calibrationWorker, "performCalibration", Q_ARG(std::vector<SMCalibrationSet>, calibrationData));
230
    QMetaObject::invokeMethod(calibrationWorker, "performCalibration", Q_ARG(std::vector<SMCalibrationSet>, calibrationData));
231
 
231
 
232
}
232
}
233
 
233
 
234
 
234
 
235
 
235
 
236
void SMScanner::onCalibrationSetProcessed(int idx){
236
void SMScanner::onCalibrationSetProcessed(int idx){
237
 
237
 
238
    ui->calibrationListWidget->setCurrentRow(idx);
238
    ui->calibrationListWidget->setCurrentRow(idx);
239
}
239
}
240
 
240
 
241
void SMScanner::onCalibrationDone(){
241
void SMScanner::onCalibrationDone(){
242
 
242
 
243
    std::cout << "Calibration done!" << std::endl;
243
    std::cout << "Calibration done!" << std::endl;
244
    ui->calibrateButton->setEnabled(true);
244
    ui->calibrateButton->setEnabled(true);
245
    ui->calibrationFrame->setEnabled(true);
245
    ui->calibrationFrame->setEnabled(true);
246
 
246
 
247
}
247
}
248
 
248
 
249
void SMScanner::on_calibrationListWidget_itemSelectionChanged(){
249
void SMScanner::on_calibrationListWidget_itemSelectionChanged(){
250
 
250
 
251
    // if selection is just cleared
251
    // if selection is just cleared
252
    if(ui->calibrationListWidget->selectedItems().empty())
252
    if(ui->calibrationListWidget->selectedItems().empty())
253
        return;
253
        return;
254
 
254
 
255
    int currentRow = ui->calibrationListWidget->currentRow();
255
    int currentRow = ui->calibrationListWidget->currentRow();
256
 
256
 
257
    calibrationReviewMode = true;
257
    calibrationReviewMode = true;
258
    ui->singleCalibrationButton->setText("Live View");
258
    ui->singleCalibrationButton->setText("Live View");
259
    ui->batchCalibrationButton->setText("Live View");
259
    ui->batchCalibrationButton->setText("Live View");
260
 
260
 
261
    if(!calibrationData[currentRow].frame0Result.empty())
261
    if(!calibrationData[currentRow].frame0Result.empty())
262
        ui->calibrationCamera0Widget->showImageCV(calibrationData[currentRow].frame0Result);
262
        ui->calibrationCamera0Widget->showImageCV(calibrationData[currentRow].frame0Result);
263
    else
263
    else
264
        ui->calibrationCamera0Widget->showImageCV(calibrationData[currentRow].frame0);
264
        ui->calibrationCamera0Widget->showImageCV(calibrationData[currentRow].frame0);
265
 
265
 
266
    if(!calibrationData[currentRow].frame1Result.empty())
266
    if(!calibrationData[currentRow].frame1Result.empty())
267
        ui->calibrationCamera1Widget->showImageCV(calibrationData[currentRow].frame1Result);
267
        ui->calibrationCamera1Widget->showImageCV(calibrationData[currentRow].frame1Result);
268
    else
268
    else
269
        ui->calibrationCamera1Widget->showImageCV(calibrationData[currentRow].frame1);
269
        ui->calibrationCamera1Widget->showImageCV(calibrationData[currentRow].frame1);
270
 
270
 
271
//     std::cout << "on_calibrationListWidget_itemSelectionChanged" << std::endl;
271
//     std::cout << "on_calibrationListWidget_itemSelectionChanged" << std::endl;
272
}
272
}
273
 
273
 
274
void SMScanner::onCalibrationFrameResult(int idx, int camID, bool success, cv::Mat result){
274
void SMScanner::onCalibrationFrameResult(int idx, int camID, bool success, cv::Mat result){
275
 
275
 
276
//    std::cout << "onCalibrationFrameResult " << idx << camID << std::endl;
276
//    std::cout << "onCalibrationFrameResult " << idx << camID << std::endl;
277
 
277
 
278
    if(!success)
278
    if(!success)
279
        ui->calibrationListWidget->item(idx)->setCheckState(Qt::Unchecked);
279
        ui->calibrationListWidget->item(idx)->setCheckState(Qt::Unchecked);
280
    else {
280
    else {
281
        if(camID == 0)
281
        if(camID == 0)
282
            calibrationData[idx].frame0Result = result;
282
            calibrationData[idx].frame0Result = result;
283
        else if(camID == 1)
283
        else if(camID == 1)
284
            calibrationData[idx].frame1Result = result;
284
            calibrationData[idx].frame1Result = result;
285
    }
285
    }
286
 
286
 
287
}
287
}
288
 
288
 
289
void SMScanner::on_singleCaptureButton_clicked(){
289
void SMScanner::on_singleCaptureButton_clicked(){
290
 
290
 
291
    // If in review mode, go back to aquisition mode
291
    // If in review mode, go back to aquisition mode
292
    if(captureReviewMode){
292
    if(captureReviewMode){
293
        ui->captureTreeWidget->clearSelection();
293
        ui->captureTreeWidget->clearSelection();
294
        ui->singleCaptureButton->setText("Single Aquisition");
294
        ui->singleCaptureButton->setText("Single Aquisition");
295
        ui->batchCaptureButton->setText("Batch Aquisition");
295
        ui->batchCaptureButton->setText("Batch Aquisition");
296
        captureReviewMode = false;
296
        captureReviewMode = false;
297
        return;
297
        return;
298
    }
298
    }
299
 
299
 
300
    QMetaObject::invokeMethod(captureWorker, "acquireFrameSequence", Q_ARG(float, -1.0));
300
    QMetaObject::invokeMethod(captureWorker, "acquireFrameSequence", Q_ARG(float, -1.0));
301
 
301
 
302
}
302
}
303
 
303
 
304
 
304
 
305
void SMScanner::on_batchCaptureButton_clicked(){
305
void SMScanner::on_batchCaptureButton_clicked(){
306
 
306
 
307
    // If in review mode, go back to aquisition mode
307
    // If in review mode, go back to aquisition mode
308
    if(captureReviewMode){
308
    if(captureReviewMode){
309
        ui->captureTreeWidget->clearSelection();
309
        ui->captureTreeWidget->clearSelection();
310
        ui->singleCaptureButton->setText("Single Aquisition");
310
        ui->singleCaptureButton->setText("Single Aquisition");
311
        ui->batchCaptureButton->setText("Batch Aquisition");
311
        ui->batchCaptureButton->setText("Batch Aquisition");
312
        captureReviewMode = false;
312
        captureReviewMode = false;
313
        return;
313
        return;
314
    }
314
    }
315
 
315
 
316
    // Construct vector of angles
316
    // Construct vector of angles
317
    int angleStart = ui->captureBatchStartSpinBox->value();
317
    int angleStart = ui->captureBatchStartSpinBox->value();
318
    int angleEnd = ui->captureBatchEndSpinBox->value();
318
    int angleEnd = ui->captureBatchEndSpinBox->value();
319
    int angleStep = ui->captureBatchStepSpinBox->value();
319
    int angleStep = ui->captureBatchStepSpinBox->value();
320
 
320
 
321
    if(angleStart > angleEnd)
321
    if(angleStart > angleEnd)
322
        angleEnd += 360;
322
        angleEnd += 360;
323
 
323
 
324
    std::vector<float> angles;
324
    std::vector<float> angles;
325
    for(int i=angleStart; i<=angleEnd; i+=angleStep)
325
    for(int i=angleStart; i<=angleEnd; i+=angleStep)
326
        angles.push_back(i % 360);
326
        angles.push_back(i % 360);
327
 
327
 
328
    std::cout << "Aquiring sequences at ";
328
    std::cout << "Aquiring sequences at ";
329
    for(unsigned int i=0; i<angles.size(); i++)
329
    for(unsigned int i=0; i<angles.size(); i++)
330
        std::cout << angles[i] << " ";
330
        std::cout << angles[i] << " ";
331
    std::cout << " degrees" <<std::endl;
331
    std::cout << " degrees" <<std::endl;
332
 
332
 
333
    QMetaObject::invokeMethod(captureWorker, "acquireFrameSequences", Q_ARG(std::vector<float>, angles));
333
    QMetaObject::invokeMethod(captureWorker, "acquireFrameSequences", Q_ARG(std::vector<float>, angles));
334
}
334
}
335
 
335
 
336
void SMScanner::on_captureRotationDial_sliderReleased(){
336
void SMScanner::on_captureRotationDial_sliderReleased(){
337
 
337
 
338
    float angle = ui->captureRotationDial->value();
338
    float angle = ui->captureRotationDial->value();
339
    std::cout << "Rotation stage target: " << angle << std::endl;
339
    std::cout << "Rotation stage target: " << angle << std::endl;
340
    QMetaObject::invokeMethod(captureWorker, "rotateTo", Q_ARG(float, angle));
340
    QMetaObject::invokeMethod(captureWorker, "rotateTo", Q_ARG(float, angle));
341
 
341
 
342
    ui->calibrationRotationDial->setValue(ui->captureRotationDial->value());
342
    ui->calibrationRotationDial->setValue(ui->captureRotationDial->value());
343
}
343
}
344
 
344
 
345
void SMScanner::onReceiveFrameSequence(SMFrameSequence frameSequence){
345
void SMScanner::onReceiveFrameSequence(SMFrameSequence frameSequence){
346
 
346
 
347
    int id = lastCaptureId + 1;
347
    int id = lastCaptureId + 1;
348
    lastCaptureId += 1;
348
    lastCaptureId += 1;
349
 
349
 
350
    frameSequence.id = id;
350
    frameSequence.id = id;
351
 
351
 
352
    captureData.push_back(frameSequence);
352
    captureData.push_back(frameSequence);
353
 
353
 
354
    // Add identifier to list
354
    // Add identifier to list
355
    QTreeWidgetItem* item = new QTreeWidgetItem(ui->captureTreeWidget);
355
    QTreeWidgetItem* item = new QTreeWidgetItem(ui->captureTreeWidget);
356
    item->setText(0, QString("Frame Sequence %1 -- %2 deg").arg(id).arg(frameSequence.rotationAngle));
356
    item->setText(0, QString("Frame Sequence %1 -- %2 deg").arg(id).arg(frameSequence.rotationAngle));
357
    item->setFlags(item->flags() | Qt::ItemIsUserCheckable);
357
    item->setFlags(item->flags() | Qt::ItemIsUserCheckable);
358
    item->setData(0, Qt::UserRole, QPoint(captureData.size()-1, -1));
358
    item->setData(0, Qt::UserRole, QPoint(captureData.size()-1, -1));
359
    item->setCheckState(0, Qt::Checked);
359
    item->setCheckState(0, Qt::Checked);
360
    //ui->captureTreeWidget->addItem(item);
360
    //ui->captureTreeWidget->addItem(item);
361
 
361
 
362
    for(unsigned int i=0; i<frameSequence.frames0.size(); i++){
362
    for(unsigned int i=0; i<frameSequence.frames0.size(); i++){
363
        QTreeWidgetItem* subItem = new QTreeWidgetItem(item);
363
        QTreeWidgetItem* subItem = new QTreeWidgetItem(item);
364
        subItem->setText(0, QString("frames %1").arg(i));
364
        subItem->setText(0, QString("frames %1").arg(i));
365
        subItem->setData(0, Qt::UserRole, QPoint(id, i));
365
        subItem->setData(0, Qt::UserRole, QPoint(id, i));
366
    }
366
    }
367
 
367
 
368
    ui->reconstructButton->setEnabled(true);
368
    ui->reconstructButton->setEnabled(true);
369
}
369
}
370
 
370
 
371
void SMScanner::on_captureTreeWidget_itemSelectionChanged(){
371
void SMScanner::on_captureTreeWidget_itemSelectionChanged(){
372
 
372
 
373
    // if selection is just cleared
373
    // if selection is just cleared
374
    if(ui->captureTreeWidget->selectedItems().empty())
374
    if(ui->captureTreeWidget->selectedItems().empty())
375
        return;
375
        return;
376
 
376
 
377
    QTreeWidgetItem *item = ui->captureTreeWidget->currentItem();
377
    QTreeWidgetItem *item = ui->captureTreeWidget->currentItem();
378
 
378
 
379
    captureReviewMode = true;
379
    captureReviewMode = true;
380
 
380
 
381
    ui->singleCaptureButton->setText("Live View");
381
    ui->singleCaptureButton->setText("Live View");
382
    ui->batchCaptureButton->setText("Live View");
382
    ui->batchCaptureButton->setText("Live View");
383
 
383
 
384
    QPoint idx = item->data(0, Qt::UserRole).toPoint();
384
    QPoint idx = item->data(0, Qt::UserRole).toPoint();
385
 
385
 
386
    if(idx.y() != -1){
386
    if(idx.y() != -1){
387
        ui->captureCamera0Widget->showImageCV(captureData[idx.x()].frames0[idx.y()]);
387
        ui->captureCamera0Widget->showImageCV(captureData[idx.x()].frames0[idx.y()]);
388
        ui->captureCamera1Widget->showImageCV(captureData[idx.x()].frames1[idx.y()]);
388
        ui->captureCamera1Widget->showImageCV(captureData[idx.x()].frames1[idx.y()]);
389
    }
389
    }
390
 
390
 
391
//     std::cout << "on_captureTreeWidget_itemSelectionChanged" << std::endl;
391
//     std::cout << "on_captureTreeWidget_itemSelectionChanged" << std::endl;
392
}
392
}
393
 
393
 
394
 
394
 
395
void SMScanner::onReceiveRotatedTo(float angle){
395
void SMScanner::onReceiveRotatedTo(float angle){
396
 
396
 
397
    // update ui with new position
397
    // update ui with new position
398
    ui->calibrationRotationDial->setValue(angle);
398
    ui->calibrationRotationDial->setValue(angle);
399
    ui->captureRotationDial->setValue(angle);
399
    ui->captureRotationDial->setValue(angle);
400
 
400
 
401
}
401
}
402
 
402
 
403
void SMScanner::on_actionExport_Sets_triggered(){
403
void SMScanner::on_actionExport_Sets_triggered(){
404
 
404
 
-
 
405
    cv::Mat frameBGR;
405
    QString dirName = QFileDialog::getExistingDirectory(this, "Export calibration sets", QString());
406
    QString dirName = QFileDialog::getExistingDirectory(this, "Export calibration sets", QString());
406
    for(unsigned int i=0; i<calibrationData.size(); i++){
407
    for(unsigned int i=0; i<calibrationData.size(); i++){
407
        QString fileName = QString("%1/frame0_%2.png").arg(dirName).arg(i);
408
        QString fileName = QString("%1/frame0_%2.png").arg(dirName).arg(i);
408
        cv::Mat frameBGR;
-
 
409
        cv::cvtColor(calibrationData[i].frame0, frameBGR, CV_RGB2BGR);
409
        cv::cvtColor(calibrationData[i].frame0, frameBGR, CV_RGB2BGR);
410
        cv::imwrite(fileName.toStdString(), frameBGR);
410
        cv::imwrite(fileName.toStdString(), frameBGR);
411
        fileName = QString("%1/frame1_%2.png").arg(dirName).arg(i);
411
        fileName = QString("%1/frame1_%2.png").arg(dirName).arg(i);
412
        cv::cvtColor(calibrationData[i].frame1, frameBGR, CV_RGB2BGR);
412
        cv::cvtColor(calibrationData[i].frame1, frameBGR, CV_RGB2BGR);
413
        cv::imwrite(fileName.toStdString(), frameBGR);
413
        cv::imwrite(fileName.toStdString(), frameBGR);
-
 
414
        // Necessary to prevent memory pileup?
-
 
415
        QCoreApplication::processEvents();
414
    }
416
    }
415
 
417
 
416
}
418
}
417
 
419
 
418
void SMScanner::on_actionExport_Sequences_triggered(){
420
void SMScanner::on_actionExport_Sequences_triggered(){
419
 
421
 
-
 
422
    cv::Mat frameBGR;
420
    QString dirName = QFileDialog::getExistingDirectory(this, "Export frame sequences", QString());
423
    QString dirName = QFileDialog::getExistingDirectory(this, "Export frame sequences", QString());
421
 
-
 
422
    for(unsigned int i=0; i<captureData.size(); i++){
424
    for(unsigned int i=0; i<captureData.size(); i++){
423
        QString seqDirName = QString("%1/sequence_%2").arg(dirName).arg(i);
425
        QString seqDirName = QString("%1/sequence_%2").arg(dirName).arg(i);
424
        if(!QDir().mkdir(seqDirName))
426
        if(!QDir().mkdir(seqDirName))
425
            std::cerr << "Could not create directory " << seqDirName.toStdString() << std::endl;
427
            std::cerr << "Could not create directory " << seqDirName.toStdString() << std::endl;
426
        for(unsigned int j=0; j<captureData[i].frames0.size(); j++){
428
        for(unsigned int j=0; j<captureData[i].frames0.size(); j++){
427
            QString fileName = QString("%1/frames0_%2.png").arg(seqDirName).arg(j);
429
            QString fileName = QString("%1/frames0_%2.png").arg(seqDirName).arg(j);
428
            cv::Mat frameBGR;
-
 
429
            cv::cvtColor(captureData[i].frames0[j], frameBGR, CV_RGB2BGR);
430
            cv::cvtColor(captureData[i].frames0[j], frameBGR, CV_RGB2BGR);
430
            cv::imwrite(fileName.toStdString(), frameBGR);
431
            cv::imwrite(fileName.toStdString(), frameBGR);
-
 
432
            // Necessary to prevent memory pileup?
-
 
433
            QCoreApplication::processEvents();
431
        }
434
        }
432
        for(unsigned int j=0; j<captureData[i].frames1.size(); j++){
435
        for(unsigned int j=0; j<captureData[i].frames1.size(); j++){
433
            QString fileName = QString("%1/frames1_%2.png").arg(seqDirName).arg(j);
436
            QString fileName = QString("%1/frames1_%2.png").arg(seqDirName).arg(j);
434
            cv::Mat frameBGR;
-
 
435
            cv::cvtColor(captureData[i].frames1[j], frameBGR, CV_RGB2BGR);
437
            cv::cvtColor(captureData[i].frames1[j], frameBGR, CV_RGB2BGR);
436
            cv::imwrite(fileName.toStdString(), frameBGR);
438
            cv::imwrite(fileName.toStdString(), frameBGR);
-
 
439
            // Necessary to prevent memory pileup?
-
 
440
            QCoreApplication::processEvents();
437
        }
441
        }
438
    }
442
    }
439
}
443
}
440
 
444
 
441
void SMScanner::on_reconstructButton_clicked(){
445
void SMScanner::on_reconstructButton_clicked(){
442
 
446
 
443
    // Set up reconstruction thread
447
    // Set up reconstruction thread
444
    reconstructionWorker = new SMReconstructionWorker;
448
    reconstructionWorker = new SMReconstructionWorker;
445
    reconstructionWorkerThread = new QThread(this);
449
    reconstructionWorkerThread = new QThread(this);
446
    reconstructionWorkerThread->setObjectName("reconstructionWorkerThread");
450
    reconstructionWorkerThread->setObjectName("reconstructionWorkerThread");
447
    reconstructionWorker->moveToThread(reconstructionWorkerThread);
451
    reconstructionWorker->moveToThread(reconstructionWorkerThread);
448
    reconstructionWorkerThread->start();
452
    reconstructionWorkerThread->start();
449
 
453
 
450
    // Connections
454
    // Connections
451
    connect(reconstructionWorker, SIGNAL(newPointCloud(SMPointCloud)), this, SLOT(onNewPointCloud(SMPointCloud)));
455
    connect(reconstructionWorker, SIGNAL(newPointCloud(SMPointCloud)), this, SLOT(onNewPointCloud(SMPointCloud)));
452
    connect(reconstructionWorker, SIGNAL(done()), reconstructionWorkerThread, SLOT(quit()));
456
    connect(reconstructionWorker, SIGNAL(done()), reconstructionWorkerThread, SLOT(quit()));
453
    connect(reconstructionWorker, SIGNAL(done()), reconstructionWorker, SLOT(deleteLater()));
457
    connect(reconstructionWorker, SIGNAL(done()), reconstructionWorker, SLOT(deleteLater()));
454
 
458
 
455
    // Start reconstructing
459
    // Start reconstructing
456
    QMetaObject::invokeMethod(reconstructionWorker, "setup");
460
    QMetaObject::invokeMethod(reconstructionWorker, "setup");
457
 
461
 
458
    for(unsigned int i=0; i<captureData.size(); i++){
462
    for(unsigned int i=0; i<captureData.size(); i++){
459
        if(!captureData[i].reconstructed & (ui->captureTreeWidget->topLevelItem(i)->checkState(0) == Qt::Checked))
463
        if(!captureData[i].reconstructed & (ui->captureTreeWidget->topLevelItem(i)->checkState(0) == Qt::Checked))
460
            QMetaObject::invokeMethod(reconstructionWorker, "reconstructPointCloud", Q_ARG(SMFrameSequence, captureData[i]));
464
            QMetaObject::invokeMethod(reconstructionWorker, "reconstructPointCloud", Q_ARG(SMFrameSequence, captureData[i]));
461
    }
465
    }
462
 
466
 
463
}
467
}
464
 
468
 
465
void SMScanner::onNewPointCloud(SMPointCloud smCloud){
469
void SMScanner::onNewPointCloud(SMPointCloud smCloud){
466
 
470
 
467
    int id = smCloud.id;
471
    int id = smCloud.id;
468
 
472
 
469
    for(int i=0; i<captureData.size(); i++){
473
    for(int i=0; i<captureData.size(); i++){
470
        if(captureData[i].id == id)
474
        if(captureData[i].id == id)
471
            captureData[i].reconstructed = true;
475
            captureData[i].reconstructed = true;
472
    }
476
    }
473
 
477
 
474
    pointCloudData.push_back(smCloud);
478
    pointCloudData.push_back(smCloud);
475
 
479
 
476
    // Add identifier to list
480
    // Add identifier to list
477
    QListWidgetItem* item = new QListWidgetItem(QString("Point Cloud %1 -- %2 deg").arg(id).arg(smCloud.rotationAngle), ui->pointCloudsListWidget);
481
    QListWidgetItem* item = new QListWidgetItem(QString("Point Cloud %1 -- %2 deg").arg(id).arg(smCloud.rotationAngle), ui->pointCloudsListWidget);
478
    item->setFlags(item->flags() | Qt::ItemIsUserCheckable);
482
    item->setFlags(item->flags() | Qt::ItemIsUserCheckable);
479
    item->setData(Qt::UserRole, QVariant(id));
483
    item->setData(Qt::UserRole, QVariant(id));
480
    item->setCheckState(Qt::Checked);
484
    item->setCheckState(Qt::Checked);
481
 
485
 
482
    ui->pointCloudsListWidget->addItem(item);
486
    ui->pointCloudsListWidget->addItem(item);
483
 
487
 
484
//    ui->pointCloudWidget->addPointCloud(smCloud, id);
488
//    ui->pointCloudWidget->addPointCloud(smCloud, id);
485
}
489
}
486
 
490
 
487
 
491
 
488
 
492
 
489
void SMScanner::on_actionExport_Point_Clouds_triggered(){
493
void SMScanner::on_actionExport_Point_Clouds_triggered(){
490
 
494
 
491
    QString directory = QFileDialog::getExistingDirectory(this, "Export Point Clouds", QString());
495
    QString directory = QFileDialog::getExistingDirectory(this, "Export Point Clouds", QString());
492
 
496
 
493
//    //  Non native file dialog
497
//    //  Non native file dialog
494
//    QFileDialog saveDirectoryDialog(this, "Export Point Clouds", QString(), "*.pcd;;*.ply;;*.vtk;;*.png;;*.txt");
498
//    QFileDialog saveDirectoryDialog(this, "Export Point Clouds", QString(), "*.pcd;;*.ply;;*.vtk;;*.png;;*.txt");
495
//    saveDirectoryDialog.setDefaultSuffix("ply");
499
//    saveDirectoryDialog.setDefaultSuffix("ply");
496
//    saveDirectoryDialog.setFileMode(QFileDialog::Directory);
500
//    saveDirectoryDialog.setFileMode(QFileDialog::Directory);
497
//    saveDirectoryDialog.exec();
501
//    saveDirectoryDialog.exec();
498
//    QString directory = saveDirectoryDialog.directory().path();
502
//    QString directory = saveDirectoryDialog.directory().path();
499
//    QString type = saveDirectoryDialog.selectedNameFilter();
503
//    QString type = saveDirectoryDialog.selectedNameFilter();
500
 
504
 
501
    // save point clouds in ply format
505
    // save point clouds in ply format
502
    for(unsigned int i=0; i<pointCloudData.size(); i++){
506
    for(unsigned int i=0; i<pointCloudData.size(); i++){
503
        QString fileName = QString("%1/pointcloud_%2.ply").arg(directory).arg(i);
507
        QString fileName = QString("%1/pointcloud_%2.ply").arg(directory).arg(i);
504
        pcl::PointCloud<pcl::PointXYZRGB>::Ptr pointCloudPCL(pointCloudData[i].pointCloud);
508
        pcl::PointCloud<pcl::PointXYZRGB>::Ptr pointCloudPCL(pointCloudData[i].pointCloud);
505
        //pcl::io::savePLYFileBinary(fileName.toStdString(), *pointCloudPCL);
509
        //pcl::io::savePLYFileBinary(fileName.toStdString(), *pointCloudPCL);
506
        pcl::PLYWriter w;
510
        pcl::PLYWriter w;
507
        // Write to ply in binary without camera
511
        // Write to ply in binary without camera
508
        w.write<pcl::PointXYZRGB> (fileName.toStdString(), *pointCloudPCL, true, false);
512
        w.write<pcl::PointXYZRGB> (fileName.toStdString(), *pointCloudPCL, true, false);
509
    }
513
    }
510
 
514
 
511
    // save meshlab aln project file
515
    // save meshlab aln project file
512
    std::ofstream s(QString("%1/alignment.aln").arg(directory).toLocal8Bit());
516
    std::ofstream s(QString("%1/alignment.aln").arg(directory).toLocal8Bit());
513
    s << pointCloudData.size() << std::endl;
517
    s << pointCloudData.size() << std::endl;
514
    for(unsigned int i=0; i<pointCloudData.size(); i++){
518
    for(unsigned int i=0; i<pointCloudData.size(); i++){
515
        QString fileName = QString("pointcloud_%1.ply").arg(i);
519
        QString fileName = QString("pointcloud_%1.ply").arg(i);
516
        s << fileName.toStdString() << std::endl << "#" << std::endl;
520
        s << fileName.toStdString() << std::endl << "#" << std::endl;
517
        cv::Mat Tr = cv::Mat::eye(4, 4, CV_32F);
521
        cv::Mat Tr = cv::Mat::eye(4, 4, CV_32F);
518
        cv::Mat(pointCloudData[i].R.t()).copyTo(Tr.colRange(0, 3).rowRange(0, 3));
522
        cv::Mat(pointCloudData[i].R.t()).copyTo(Tr.colRange(0, 3).rowRange(0, 3));
519
        cv::Mat(-pointCloudData[i].R.t()*pointCloudData[i].T).copyTo(Tr.col(3).rowRange(0, 3));
523
        cv::Mat(-pointCloudData[i].R.t()*pointCloudData[i].T).copyTo(Tr.col(3).rowRange(0, 3));
520
        for(int j=0; j<Tr.rows; j++){
524
        for(int j=0; j<Tr.rows; j++){
521
            for(int k=0; k<Tr.cols; k++){
525
            for(int k=0; k<Tr.cols; k++){
522
                s << Tr.at<float>(j,k) << " ";
526
                s << Tr.at<float>(j,k) << " ";
523
            }
527
            }
524
            s << std::endl;
528
            s << std::endl;
525
        }
529
        }
526
    }
530
    }
527
    s << "0" << std::flush;
531
    s << "0" << std::flush;
528
    s.close();
532
    s.close();
529
}
533
}
530
 
534
 
531
void SMScanner::on_pointCloudsListWidget_itemChanged(QListWidgetItem *item){
535
void SMScanner::on_pointCloudsListWidget_itemChanged(QListWidgetItem *item){
532
 
536
 
533
    int id = item->data(Qt::UserRole).toInt();
537
    int id = item->data(Qt::UserRole).toInt();
534
 
538
 
535
    if(item->checkState() == Qt::Checked)
539
    if(item->checkState() == Qt::Checked)
536
        ui->pointCloudWidget->addPointCloud(pointCloudData[id], id);
540
        ui->pointCloudWidget->addPointCloud(pointCloudData[id], id);
537
    else
541
    else
538
        ui->pointCloudWidget->removePointCloud(id);
542
        ui->pointCloudWidget->removePointCloud(id);
539
 
543
 
540
}
544
}
541
 
545
 
542
void SMScanner::on_actionExport_Parameters_triggered(){
546
void SMScanner::on_actionExport_Parameters_triggered(){
543
 
547
 
544
    QString fileName = QFileDialog::getSaveFileName(this, "Export calibration parameters", QString(), "*.xml");
548
    QString fileName = QFileDialog::getSaveFileName(this, "Export calibration parameters", QString(), "*.xml");
545
    QFileInfo info(fileName);
549
    QFileInfo info(fileName);
546
    QString type = info.suffix();
550
    QString type = info.suffix();
547
    if(type == ""){
551
    if(type == ""){
548
        fileName.append(".xml");
552
        fileName.append(".xml");
549
    }
553
    }
550
 
554
 
551
    SMCalibrationParameters calibration = settings.value("calibration/parameters").value<SMCalibrationParameters>();
555
    SMCalibrationParameters calibration = settings.value("calibration/parameters").value<SMCalibrationParameters>();
552
    calibration.exportToXML(fileName);
556
    calibration.exportToXML(fileName);
553
}
557
}
554
 
558
 
555
void SMScanner::on_actionClear_Sequences_triggered(){
559
void SMScanner::on_actionClear_Sequences_triggered(){
556
 
560
 
557
    int res = QMessageBox::question(this, "Clear Captured Sequences", "Clear all captured data?", QMessageBox::Ok, QMessageBox::Cancel);
561
    int res = QMessageBox::question(this, "Clear Captured Sequences", "Clear all captured data?", QMessageBox::Ok, QMessageBox::Cancel);
558
 
562
 
559
    if(res == QMessageBox::Ok){
563
    if(res == QMessageBox::Ok){
560
        captureData.clear();
564
        captureData.clear();
561
        ui->captureTreeWidget->clear();
565
        ui->captureTreeWidget->clear();
562
    }
566
    }
563
}
567
}
564
 
568