Subversion Repositories seema-scanner

Rev

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

Rev 74 Rev 75
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(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(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(this);
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(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(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
    QString dirName = QFileDialog::getExistingDirectory(this, "Export calibration sets", QString());
405
    QString dirName = QFileDialog::getExistingDirectory(this, "Export calibration sets", QString());
406
    for(int i=0; i<calibrationData.size(); i++){
406
    for(unsigned int i=0; i<calibrationData.size(); i++){
407
        QString fileName = QString("%1/frame0_%2.png").arg(dirName).arg(i);
407
        QString fileName = QString("%1/frame0_%2.png").arg(dirName).arg(i);
408
        cv::Mat frameBGR;
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
    }
414
    }
415
 
415
 
416
}
416
}
417
 
417
 
418
void SMScanner::on_actionExport_Sequences_triggered(){
418
void SMScanner::on_actionExport_Sequences_triggered(){
419
 
419
 
420
    QString dirName = QFileDialog::getExistingDirectory(this, "Export frame sequences", QString());
420
    QString dirName = QFileDialog::getExistingDirectory(this, "Export frame sequences", QString());
421
 
421
 
422
    for(int i=0; i<captureData.size(); i++){
422
    for(unsigned int i=0; i<captureData.size(); i++){
423
        QString seqDirName = QString("%1/sequence_%2").arg(dirName).arg(i);
423
        QString seqDirName = QString("%1/sequence_%2").arg(dirName).arg(i);
424
        if(!QDir().mkdir(seqDirName))
424
        if(!QDir().mkdir(seqDirName))
425
            std::cerr << "Could not create directory " << seqDirName.toStdString() << std::endl;
425
            std::cerr << "Could not create directory " << seqDirName.toStdString() << std::endl;
426
        for(int j=0; j<captureData[i].frames0.size(); j++){
426
        for(unsigned int j=0; j<captureData[i].frames0.size(); j++){
427
            QString fileName = QString("%1/frames0_%2.png").arg(seqDirName).arg(j);
427
            QString fileName = QString("%1/frames0_%2.png").arg(seqDirName).arg(j);
428
            cv::Mat frameBGR;
428
            cv::Mat frameBGR;
429
            cv::cvtColor(captureData[i].frames0[j], frameBGR, CV_RGB2BGR);
429
            cv::cvtColor(captureData[i].frames0[j], frameBGR, CV_RGB2BGR);
430
            cv::imwrite(fileName.toStdString(), frameBGR);
430
            cv::imwrite(fileName.toStdString(), frameBGR);
431
        }
431
        }
432
        for(int j=0; j<captureData[i].frames1.size(); j++){
432
        for(unsigned int j=0; j<captureData[i].frames1.size(); j++){
433
            QString fileName = QString("%1/frames1_%2.png").arg(seqDirName).arg(j);
433
            QString fileName = QString("%1/frames1_%2.png").arg(seqDirName).arg(j);
434
            cv::Mat frameBGR;
434
            cv::Mat frameBGR;
435
            cv::cvtColor(captureData[i].frames1[j], frameBGR, CV_RGB2BGR);
435
            cv::cvtColor(captureData[i].frames1[j], frameBGR, CV_RGB2BGR);
436
            cv::imwrite(fileName.toStdString(), frameBGR);
436
            cv::imwrite(fileName.toStdString(), frameBGR);
437
        }
437
        }
438
    }
438
    }
439
}
439
}
440
 
440
 
441
void SMScanner::on_reconstructButton_clicked(){
441
void SMScanner::on_reconstructButton_clicked(){
442
 
442
 
443
    // Set up reconstruction thread
443
    // Set up reconstruction thread
444
    reconstructionWorker = new SMReconstructionWorker;
444
    reconstructionWorker = new SMReconstructionWorker;
445
    reconstructionWorkerThread = new QThread(this);
445
    reconstructionWorkerThread = new QThread(this);
446
    reconstructionWorkerThread->setObjectName("reconstructionWorkerThread");
446
    reconstructionWorkerThread->setObjectName("reconstructionWorkerThread");
447
    reconstructionWorker->moveToThread(reconstructionWorkerThread);
447
    reconstructionWorker->moveToThread(reconstructionWorkerThread);
448
    reconstructionWorkerThread->start();
448
    reconstructionWorkerThread->start();
449
 
449
 
450
    // Connections
450
    // Connections
451
    connect(reconstructionWorker, SIGNAL(newPointCloud(SMPointCloud)), this, SLOT(onNewPointCloud(SMPointCloud)));
451
    connect(reconstructionWorker, SIGNAL(newPointCloud(SMPointCloud)), this, SLOT(onNewPointCloud(SMPointCloud)));
452
    connect(reconstructionWorker, SIGNAL(done()), reconstructionWorkerThread, SLOT(quit()));
452
    connect(reconstructionWorker, SIGNAL(done()), reconstructionWorkerThread, SLOT(quit()));
453
    connect(reconstructionWorker, SIGNAL(done()), reconstructionWorker, SLOT(deleteLater()));
453
    connect(reconstructionWorker, SIGNAL(done()), reconstructionWorker, SLOT(deleteLater()));
454
 
454
 
455
    // Start reconstructing
455
    // Start reconstructing
456
    QMetaObject::invokeMethod(reconstructionWorker, "setup");
456
    QMetaObject::invokeMethod(reconstructionWorker, "setup");
457
 
457
 
458
    for(int i=0; i<captureData.size(); i++){
458
    for(unsigned int i=0; i<captureData.size(); i++){
459
        if(!captureData[i].reconstructed & ui->captureTreeWidget->topLevelItem(i)->checkState(0) == Qt::Checked)
459
        if(!captureData[i].reconstructed & (ui->captureTreeWidget->topLevelItem(i)->checkState(0) == Qt::Checked))
460
            QMetaObject::invokeMethod(reconstructionWorker, "reconstructPointCloud", Q_ARG(SMFrameSequence, captureData[i]));
460
            QMetaObject::invokeMethod(reconstructionWorker, "reconstructPointCloud", Q_ARG(SMFrameSequence, captureData[i]));
461
    }
461
    }
462
 
462
 
463
}
463
}
464
 
464
 
465
void SMScanner::onNewPointCloud(SMPointCloud smCloud){
465
void SMScanner::onNewPointCloud(SMPointCloud smCloud){
466
 
466
 
467
    int id = smCloud.id;
467
    int id = smCloud.id;
468
    captureData[id].reconstructed = true;
468
    captureData[id].reconstructed = true;
469
 
469
 
470
    pointCloudData.push_back(smCloud);
470
    pointCloudData.push_back(smCloud);
471
 
471
 
472
    // Add identifier to list
472
    // Add identifier to list
473
    QListWidgetItem* item = new QListWidgetItem(QString("Point Cloud %1 -- %2 deg").arg(id).arg(smCloud.rotationAngle), ui->pointCloudsListWidget);
473
    QListWidgetItem* item = new QListWidgetItem(QString("Point Cloud %1 -- %2 deg").arg(id).arg(smCloud.rotationAngle), ui->pointCloudsListWidget);
474
    item->setFlags(item->flags() | Qt::ItemIsUserCheckable);
474
    item->setFlags(item->flags() | Qt::ItemIsUserCheckable);
475
    item->setData(Qt::UserRole, QVariant(id));
475
    item->setData(Qt::UserRole, QVariant(id));
476
    item->setCheckState(Qt::Checked);
476
    item->setCheckState(Qt::Checked);
477
 
477
 
478
    ui->pointCloudsListWidget->addItem(item);
478
    ui->pointCloudsListWidget->addItem(item);
479
 
479
 
480
//    ui->pointCloudWidget->addPointCloud(smCloud, id);
480
//    ui->pointCloudWidget->addPointCloud(smCloud, id);
481
}
481
}
482
 
482
 
483
 
483
 
484
 
484
 
485
void SMScanner::on_actionExport_Point_Clouds_triggered(){
485
void SMScanner::on_actionExport_Point_Clouds_triggered(){
486
 
486
 
487
    QString directory = QFileDialog::getExistingDirectory(this, "Export Point Clouds", QString());
487
    QString directory = QFileDialog::getExistingDirectory(this, "Export Point Clouds", QString());
488
 
488
 
489
//    //  Non native file dialog
489
//    //  Non native file dialog
490
//    QFileDialog saveDirectoryDialog(this, "Export Point Clouds", QString(), "*.pcd;;*.ply;;*.vtk;;*.png;;*.txt");
490
//    QFileDialog saveDirectoryDialog(this, "Export Point Clouds", QString(), "*.pcd;;*.ply;;*.vtk;;*.png;;*.txt");
491
//    saveDirectoryDialog.setDefaultSuffix("ply");
491
//    saveDirectoryDialog.setDefaultSuffix("ply");
492
//    saveDirectoryDialog.setFileMode(QFileDialog::Directory);
492
//    saveDirectoryDialog.setFileMode(QFileDialog::Directory);
493
//    saveDirectoryDialog.exec();
493
//    saveDirectoryDialog.exec();
494
//    QString directory = saveDirectoryDialog.directory().path();
494
//    QString directory = saveDirectoryDialog.directory().path();
495
//    QString type = saveDirectoryDialog.selectedNameFilter();
495
//    QString type = saveDirectoryDialog.selectedNameFilter();
496
 
496
 
497
    // save point clouds in ply format
497
    // save point clouds in ply format
498
    for(int i=0; i<pointCloudData.size(); i++){
498
    for(unsigned int i=0; i<pointCloudData.size(); i++){
499
        QString fileName = QString("%1/pointcloud_%2.ply").arg(directory).arg(i);
499
        QString fileName = QString("%1/pointcloud_%2.ply").arg(directory).arg(i);
500
        pcl::PointCloud<pcl::PointXYZRGB>::Ptr pointCloudPCL(pointCloudData[i].pointCloud);
500
        pcl::PointCloud<pcl::PointXYZRGB>::Ptr pointCloudPCL(pointCloudData[i].pointCloud);
501
        //pcl::io::savePLYFileBinary(fileName.toStdString(), *pointCloudPCL);
501
        //pcl::io::savePLYFileBinary(fileName.toStdString(), *pointCloudPCL);
502
        pcl::PLYWriter w;
502
        pcl::PLYWriter w;
503
        // Write to ply in binary without camera
503
        // Write to ply in binary without camera
504
        w.write<pcl::PointXYZRGB> (fileName.toStdString(), *pointCloudPCL, true, false);
504
        w.write<pcl::PointXYZRGB> (fileName.toStdString(), *pointCloudPCL, true, false);
505
    }
505
    }
506
 
506
 
507
    // save meshlab aln project file
507
    // save meshlab aln project file
508
    std::ofstream s(QString("%1/alignment.aln").arg(directory).toLocal8Bit());
508
    std::ofstream s(QString("%1/alignment.aln").arg(directory).toLocal8Bit());
509
    s << pointCloudData.size() << std::endl;
509
    s << pointCloudData.size() << std::endl;
510
    for(int i=0; i<pointCloudData.size(); i++){
510
    for(unsigned int i=0; i<pointCloudData.size(); i++){
511
        QString fileName = QString("pointcloud_%1.ply").arg(i);
511
        QString fileName = QString("pointcloud_%1.ply").arg(i);
512
        s << fileName.toStdString() << std::endl << "#" << std::endl;
512
        s << fileName.toStdString() << std::endl << "#" << std::endl;
513
        cv::Mat Tr = cv::Mat::eye(4, 4, CV_32F);
513
        cv::Mat Tr = cv::Mat::eye(4, 4, CV_32F);
514
        cv::Mat(pointCloudData[i].R.t()).copyTo(Tr.colRange(0, 3).rowRange(0, 3));
514
        cv::Mat(pointCloudData[i].R.t()).copyTo(Tr.colRange(0, 3).rowRange(0, 3));
515
        cv::Mat(-pointCloudData[i].R.t()*pointCloudData[i].T).copyTo(Tr.col(3).rowRange(0, 3));
515
        cv::Mat(-pointCloudData[i].R.t()*pointCloudData[i].T).copyTo(Tr.col(3).rowRange(0, 3));
516
        for(int j=0; j<Tr.rows; j++){
516
        for(int j=0; j<Tr.rows; j++){
517
            for(int k=0; k<Tr.cols; k++){
517
            for(int k=0; k<Tr.cols; k++){
518
                s << Tr.at<float>(j,k) << " ";
518
                s << Tr.at<float>(j,k) << " ";
519
            }
519
            }
520
            s << std::endl;
520
            s << std::endl;
521
        }
521
        }
522
    }
522
    }
523
    s << "0" << std::flush;
523
    s << "0" << std::flush;
524
    s.close();
524
    s.close();
525
}
525
}
526
 
526
 
527
void SMScanner::on_pointCloudsListWidget_itemChanged(QListWidgetItem *item){
527
void SMScanner::on_pointCloudsListWidget_itemChanged(QListWidgetItem *item){
528
 
528
 
529
    int id = item->data(Qt::UserRole).toInt();
529
    int id = item->data(Qt::UserRole).toInt();
530
 
530
 
531
    if(item->checkState() == Qt::Checked)
531
    if(item->checkState() == Qt::Checked)
532
        ui->pointCloudWidget->addPointCloud(pointCloudData[id], id);
532
        ui->pointCloudWidget->addPointCloud(pointCloudData[id], id);
533
    else
533
    else
534
        ui->pointCloudWidget->removePointCloud(id);
534
        ui->pointCloudWidget->removePointCloud(id);
535
 
535
 
536
}
536
}
537
 
537
 
538
void SMScanner::on_actionExport_Parameters_triggered(){
538
void SMScanner::on_actionExport_Parameters_triggered(){
539
 
539
 
540
    QString fileName = QFileDialog::getSaveFileName(this, "Export calibration parameters", QString(), "*.xml");
540
    QString fileName = QFileDialog::getSaveFileName(this, "Export calibration parameters", QString(), "*.xml");
541
    QFileInfo info(fileName);
541
    QFileInfo info(fileName);
542
    QString type = info.suffix();
542
    QString type = info.suffix();
543
    if(type == ""){
543
    if(type == ""){
544
        fileName.append(".xml");
544
        fileName.append(".xml");
545
    }
545
    }
546
 
546
 
547
    SMCalibrationParameters calibration = settings.value("calibration/parameters").value<SMCalibrationParameters>();
547
    SMCalibrationParameters calibration = settings.value("calibration/parameters").value<SMCalibrationParameters>();
548
    calibration.exportToXML(fileName);
548
    calibration.exportToXML(fileName);
549
}
549
}
550
 
550
 
551
void SMScanner::on_actionClear_Sequences_triggered(){
551
void SMScanner::on_actionClear_Sequences_triggered(){
552
 
552
 
553
    int res = QMessageBox::question(this, "Clear Captured Sequences", "Clear all captured data?", QMessageBox::Ok, QMessageBox::Cancel);
553
    int res = QMessageBox::question(this, "Clear Captured Sequences", "Clear all captured data?", QMessageBox::Ok, QMessageBox::Cancel);
554
 
554
 
555
    if(res == QMessageBox::Ok){
555
    if(res == QMessageBox::Ok){
556
        captureData.clear();
556
        captureData.clear();
557
        ui->captureTreeWidget->clear();
557
        ui->captureTreeWidget->clear();
558
    }
558
    }
559
}
559
}
560
 
560