WebSVN – seema-scanner – Diff – /src/SMScanner.cpp

+}
 void SMScanner::on_actionExport_Sets_triggered(){
-    cv::Mat frameBGR;
     QString dirName = QFileDialog::getExistingDirectory(this, "Export calibration sets", QString());
+    QProgressDialog progressDialog("Exporting Sets...", "Cancel", 0, 100, this);
+    progressDialog.setWindowModality(Qt::WindowModal);
+    progressDialog.setMinimumDuration(1000);
+    progressDialog.show();
+    cv::Mat frameBGR;
     for(unsigned int i=0; i<calibrationData.size(); i++){
         QString fileName = QString("%1/frame0_%2.png").arg(dirName).arg(i);
+        progressDialog.setValue(100.0*i/calibrationData.size());
         // Convert to BGR
         if(calibrationData[i].frame0.channels() == 1)
             cv::cvtColor(calibrationData[i].frame0, frameBGR, CV_BayerBG2BGR);
         else
             cv::cvtColor(calibrationData[i].frame1, frameBGR, CV_RGB2BGR);
         cv::imwrite(fileName.toStdString(), frameBGR);
         // Necessary to prevent pileup of video frame signals
         QCoreApplication::processEvents();
+        if(progressDialog.wasCanceled())
+            return;
+    }
+}
 void SMScanner::on_actionExport_Sequences_triggered(){
+    QString dirName = QFileDialog::getExistingDirectory(this, "Export frame sequences", QString());
-    QProgressDialog progressDialog("Exporting Sequences.", "Cancel", 0, 100, this);
+    QProgressDialog progressDialog("Exporting Sequences...", "Cancel", 0, 100, this);
     progressDialog.setWindowModality(Qt::WindowModal);
     progressDialog.setMinimumDuration(1000);
     progressDialog.show();
     cv::Mat frameBGR;
-    QString dirName = QFileDialog::getExistingDirectory(this, "Export frame sequences", QString());
     for(unsigned int i=0; i<captureData.size(); i++){
         QString seqDirName = QString("%1/sequence_%2").arg(dirName).arg(i);
         if(!QDir().mkdir(seqDirName))
             std::cerr << "Could not create directory " << seqDirName.toStdString() << std::endl;
         for(unsigned int j=0; j<captureData[i].frames0.size(); j++){
-            progressDialog.setValue(100.0*i/captureData.size() + 100.0*i/captureData.size()*j/captureData[i].frames0.size());
+            progressDialog.setValue(100.0*i/captureData.size() + 100.0/captureData.size()*j/captureData[i].frames0.size());
-            progressDialog.update();
             QString fileName0 = QString("%1/frames0_%2.png").arg(seqDirName).arg(j);
             // Convert Bayer to rgb (png needs BGR order)
             cv::cvtColor(captureData[i].frames0[j], frameBGR, CV_BayerBG2BGR);
             cv::imwrite(fileName0.toStdString(), frameBGR);
+    }
+}
 void SMScanner::on_actionExport_White_Frames_triggered(){
-    cv::Mat frameBGR;
     QString dirName = QFileDialog::getExistingDirectory(this, "Export frame sequences", QString());
+    QProgressDialog progressDialog("Exporting White Frames...", "Cancel", 0, 100, this);
+    progressDialog.setWindowModality(Qt::WindowModal);
+    progressDialog.setMinimumDuration(1000);
+    cv::Mat frameBGR;
     for(unsigned int i=0; i<captureData.size(); i++){
+        progressDialog.setValue(100.0*i/captureData.size());
         QString seqDirName = QString("%1/sequence_%2").arg(dirName).arg(i);
         if(!QDir().mkdir(seqDirName))
             std::cerr << "Could not create directory " << seqDirName.toStdString() << std::endl;
         QString fileName = QString("%1/frames0_0.png").arg(seqDirName);
         // Convert Bayer to rgb (png needs BGR order)
         cv::cvtColor(captureData[i].frames1[0], frameBGR, CV_BayerBG2BGR);
         cv::imwrite(fileName.toStdString(), frameBGR);
         // Necessary to prevent memory pileup
         QCoreApplication::processEvents();
+        if(progressDialog.wasCanceled())
+            return;
+    }
+}
 void SMScanner::on_reconstructButton_clicked(){
 //    saveDirectoryDialog.setFileMode(QFileDialog::Directory);
 //    saveDirectoryDialog.exec();
 //    QString directory = saveDirectoryDialog.directory().path();
 //    QString type = saveDirectoryDialog.selectedNameFilter();
+    QProgressDialog progressDialog("Exporting White Frames...", "Cancel", 0, 100, this);
+    progressDialog.setWindowModality(Qt::WindowModal);
+    progressDialog.setMinimumDuration(1000);
     // save point clouds in ply format
     for(unsigned int i=0; i<pointCloudData.size(); i++){
+        progressDialog.setValue(100.0*i/pointCloudData.size());
         QString fileName = QString("%1/pointcloud_%2.ply").arg(directory).arg(i);
         pcl::PointCloud<pcl::PointXYZRGBNormal>::Ptr pointCloudPCL(pointCloudData[i].pointCloud);
         //pcl::io::savePLYFileBinary(fileName.toStdString(), *pointCloudPCL);
         pcl::PLYWriter w;
         // Write to ply in binary without camera
         w.write<pcl::PointXYZRGBNormal> (fileName.toStdString(), *pointCloudPCL, true, false);
+        QCoreApplication::processEvents();
+        if(progressDialog.wasCanceled())
+            return;
+    }
     // save meshlab aln project file
     std::ofstream s(QString("%1/alignment.aln").arg(directory).toLocal8Bit());
     s << pointCloudData.size() << std::endl;
+    }
     calibrationData.clear();
     ui->calibrationListWidget->clear();
+    QProgressDialog progressDialog("Importing Sets...", "Cancel", 0, 100, this);
+    progressDialog.setWindowModality(Qt::WindowModal);
-    int nSets = fileNames0.size();
+    progressDialog.setMinimumDuration(1000);
+    int nSets = fileNames0.size();
     for(unsigned int i=0; i<nSets; i++){
+        progressDialog.setValue(100.0*i/nSets);
         SMCalibrationSet calibrationSet;
         QString fileName0 = QString("%1/frame0_%2.png").arg(dirName).arg(i);
         cv::Mat frame0BGR = cv::imread(fileName0.toStdString());
         cvtools::cvtColorBGRToBayerBG(frame0BGR, calibrationSet.frame0);
         item->setFlags(item->flags() | Qt::ItemIsUserCheckable);
         item->setCheckState(Qt::Unchecked);
         ui->calibrationListWidget->addItem(item);
         QCoreApplication::processEvents();
+        if(progressDialog.wasCanceled())
+            return;
+    }
     // Set enabled checkmark
     if(calibrationData.size() >= 2)
         ui->calibrateButton->setEnabled(true);
+}
 void SMScanner::on_actionImport_Sequences_triggered(){
     // NOTE: we do not know which algorithm was used!!!
     QString dirName = QFileDialog::getExistingDirectory(this, "Import captured sequences", QString());
     QDir dir(dirName);
     QStringList sequenceDirNames = dir.entryList(QStringList("sequence_*"));
+    }
     captureData.clear();
     ui->captureTreeWidget->clear();
+    QProgressDialog progressDialog("Importing Sequences...", "Cancel", 0, 100, this);
+    progressDialog.setWindowModality(Qt::WindowModal);
+    progressDialog.setMinimumDuration(1000);
     for(int s=0; s<sequenceDirNames.count(); s++){
         QDir sequenceDir(QString("%1/%2").arg(dirName).arg(sequenceDirNames.at(s)));
         QStringList fileNames0 = sequenceDir.entryList(QStringList("frames0_*.png"));
         SMFrameSequence sequence;
         for(int f=0; f<nFrames; f++){
+            progressDialog.setValue(100.0*s/sequenceDirNames.count() + 100.0/sequenceDirNames.count()*f/nFrames);
             cv::Mat frame0BGR, frame0BayerBG, frame1BGR, frame1BayerBG;
             QString fileName0 = QString("%1/%2/frames0_%3.png").arg(dirName).arg(sequenceDirNames.at(s)).arg(f);
             frame0BGR = cv::imread(fileName0.toStdString());
             cvtools::cvtColorBGRToBayerBG(frame0BGR, frame0BayerBG);
             sequence.frames0.push_back(frame0BayerBG);
             sequence.frames1.push_back(frame1BayerBG);
             QCoreApplication::processEvents();
+            if(progressDialog.wasCanceled())
+                return;
+        }
+        // Assign whatever algorithm is configured
         sequence.codec = settings.value("algorithm").toString();
         sequence.rotationAngle = 0;
         sequence.reconstructed = false;
         onReceiveFrameSequence(sequence);

Subversion Repositories seema-scanner

(root)/src/SMScanner.cpp @ 208 – Rev 142 → 143