SoiFilter.cpp

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        CloudViewer: www.cloudViewer.org                  -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.cloudViewer.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#include "SoiFilter.h"
 
// CV_DB_LIB
#include <CVLog.h>
#include <ecvPointCloud.h>
#include <ecvProgressDialog.h>
 
// Max number of characters per line in an ASCII file
// TODO: use QFile instead!
const int MAX_ASCII_FILE_LINE_LENGTH = 4096;
 
SoiFilter::SoiFilter()
    : FileIOFilter({"_Soisic Filter",
                    DEFAULT_PRIORITY,  // priority
                    QStringList{"soi"}, "soi",
                    QStringList{"Mensi Soisic cloud (*.soi)"}, QStringList(),
                    Import}) {}
 
CC_FILE_ERROR SoiFilter::loadFile(const QString& filename,
                                  ccHObject& container,
                                  LoadParameters& parameters) {
    // open the file
    FILE* fp = fopen(qPrintable(filename), "rt");
    if (!fp) return CC_FERR_READING;
 
    std::string line;
    line.resize(MAX_ASCII_FILE_LINE_LENGTH);
    unsigned nbScansTotal = 0;
    unsigned nbPointsTotal = 0;
 
    // we read the first line
    char* eof = fgets((char*)line.c_str(), MAX_ASCII_FILE_LINE_LENGTH, fp);
    char* pEnd;
 
    // header
    while ((strcmp((char*)line.substr(0, 4).c_str(), "#CC#") != 0) &&
           (eof != nullptr)) {
        if (strcmp(line.substr(0, 4).c_str(), "#NP#") == 0) {
            std::string numPoints(line, 4, line.size() - 4);
            nbPointsTotal = strtol(numPoints.c_str(), &pEnd, 0);
            // CVLog::Print("[SoiFilter::loadFile] Total number of points:
            // %i",nbPointsTotal);
        } else if (strcmp(line.substr(0, 4).c_str(), "#NS#") == 0) {
            std::string numScans(line, 4, line.size() - 4);
            nbScansTotal = strtol(numScans.c_str(), &pEnd, 0);
            // CVLog::Print("[SoiFilter::loadFile] Total number of scans:
            // %i",nbScansTotal);
        }
 
        eof = fgets((char*)line.c_str(), MAX_ASCII_FILE_LINE_LENGTH, fp);
    }
 
    if ((nbScansTotal == 0) || (nbPointsTotal == 0)) {
        CVLog::Warning(
                "[SoiFilter::loadFile] No points or scans defined in this "
                "file!");
        fclose(fp);
        return CC_FERR_NO_LOAD;
    }
 
    // Progress dialog
    QScopedPointer<ecvProgressDialog> pDlg(0);
    if (parameters.parentWidget) {
        pDlg.reset(new ecvProgressDialog(
                false, parameters.parentWidget));  // cancel is not supported
        pDlg->setMethodTitle(QObject::tr("Open SOI file"));
        pDlg->setInfo(QObject::tr("%1 scans / %2 points")
                              .arg(nbScansTotal)
                              .arg(nbPointsTotal));
        pDlg->start();
    }
    cloudViewer::NormalizedProgress nprogress(pDlg.data(), nbPointsTotal);
 
    // Scan by scan
    for (unsigned k = 0; k < nbScansTotal; k++) {
        char* eof = fgets((char*)line.c_str(), MAX_ASCII_FILE_LINE_LENGTH, fp);
 
        // we only look for points (we ignore the rest)
        while ((strcmp(line.substr(0, 4).c_str(), "#pt#") != 0) &&
               (eof != nullptr))
            eof = fgets((char*)line.c_str(), MAX_ASCII_FILE_LINE_LENGTH, fp);
 
        unsigned nbOfPoints = 0;
 
        if (strcmp(line.substr(0, 4).c_str(), "#pt#") == 0) {
            std::string numPoints(line, 4, line.size() - 4);
            nbOfPoints = strtol(numPoints.c_str(), &pEnd, 0);
            // CVLog::Print("[SoiFilter::loadFile] Scan %i - points:
            // %i",k+1,nbOfPoints);
        } else {
            CVLog::Warning("[SoiFilter::loadFile] Can't find marker '#pt#'!");
            fclose(fp);
            return CC_FERR_WRONG_FILE_TYPE;
        }
 
        if (nbOfPoints == 0) {
            CVLog::Warning("[SoiFilter::loadFile] Scan #%i is empty!", k);
            continue;
        }
 
        // Creation de la liste de points
        QString name = QString("unnamed - Scan #%1").arg(k);
 
        ccPointCloud* loadedCloud = new ccPointCloud(name);
        if (!loadedCloud->reserveThePointsTable(nbOfPoints) ||
            !loadedCloud->reserveTheRGBTable()) {
            fclose(fp);
            delete loadedCloud;
            return CC_FERR_NOT_ENOUGH_MEMORY;
        }
        loadedCloud->showColors(true);
 
        // we can read points now
        for (unsigned i = 0; i < nbOfPoints; i++) {
            float P[3];
            int c = 0;
            if (fscanf(fp, "%f %f %f %i", P, P + 1, P + 2, &c) < 4) {
                fclose(fp);
                delete loadedCloud;
                return CC_FERR_MALFORMED_FILE;
            }
 
            loadedCloud->addPoint(CCVector3::fromArray(P));
            loadedCloud->addGreyColor(static_cast<ColorCompType>(
                    c << 3));  //<<2 ? <<3 ? we lack some info. here ...
 
            if (pDlg) {
                nprogress.oneStep();
            }
        }
 
        container.addChild(loadedCloud);
    }
 
    fclose(fp);
 
    return CC_FERR_NO_ERROR;
}