cvSelectionExporter.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        CloudViewer: www.cloudViewer.org                  -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.cloudViewer.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#pragma once
 
#include "qPCL.h"
 
// LOCAL
#include "cvSelectionData.h"
 
// Qt
#include <QList>
#include <QString>
 
// STL
#include <functional>
 
// Forward declarations
class ccHObject;
class ccMesh;
class ccPointCloud;
class vtkPolyData;
class QWidget;
 
class QPCL_ENGINE_LIB_API cvSelectionExporter {
public:
    struct ExportOptions {
        QString name;           //< Name for the created object
        bool addToScene;        //< Whether to add to scene directly
        bool saveToFile;        //< Whether to save to file
        QString filename;       //< Output filename (if saveToFile=true)
        bool writeAscii;        //< Use ASCII format (if supported)
        bool compressed;        //< Use compression (if supported)
        QWidget* parentWidget;  //< Parent widget for dialogs
 
        // Constructor with default values
        ExportOptions()
            : addToScene(false),
              saveToFile(false),
              writeAscii(false),
              compressed(false),
              parentWidget(nullptr) {}
    };
 
    static ccMesh* exportToMesh(vtkPolyData* polyData,
                                const cvSelectionData& selectionData,
                                const ExportOptions& options = ExportOptions());
 
    static ccPointCloud* exportToPointCloud(
            vtkPolyData* polyData,
            const cvSelectionData& selectionData,
            const ExportOptions& options = ExportOptions());
 
    //=========================================================================
    // Direct Extraction from ccPointCloud/ccMesh (bypasses VTK conversion)
    //=========================================================================
 
    static ccPointCloud* exportFromSourceCloud(
            ccPointCloud* sourceCloud,
            const cvSelectionData& selectionData,
            const ExportOptions& options = ExportOptions());
 
    static ccMesh* exportFromSourceMesh(
            ccMesh* sourceMesh,
            const cvSelectionData& selectionData,
            const ExportOptions& options = ExportOptions());
 
    static bool exportToFile(vtkPolyData* polyData,
                             const cvSelectionData& selectionData,
                             const QString& filename,
                             bool writeAscii = false,
                             bool compressed = false);
 
    static vtkPolyData* extractSelection(vtkPolyData* polyData,
                                         const cvSelectionData& selectionData);
 
    //=========================================================================
    // Batch Export Functionality
    //=========================================================================
 
    static QList<ccMesh*> batchExportToMeshes(
            vtkPolyData* polyData,
            const QList<cvSelectionData>& selections,
            const QString& baseName = "Selection");
 
    static QList<ccPointCloud*> batchExportToPointClouds(
            vtkPolyData* polyData,
            const QList<cvSelectionData>& selections,
            const QString& baseName = "Selection");
 
    static int batchExportToFiles(
            vtkPolyData* polyData,
            const QList<cvSelectionData>& selections,
            const QString& outputDir,
            const QString& format,
            const QString& baseName = "selection",
            std::function<void(int)> progressCallback = nullptr);
 
    static bool exportNumbered(vtkPolyData* polyData,
                               const cvSelectionData& selection,
                               const QString& outputPath,
                               int number);
 
    static bool saveObjectToFileWithDialog(ccHObject* object,
                                           bool isMesh,
                                           QWidget* parent = nullptr);
 
private:
    static ccMesh* vtkPolyDataToCCMesh(vtkPolyData* polyData,
                                       const QString& name);
 
    static ccPointCloud* vtkPolyDataToCCPointCloud(vtkPolyData* polyData,
                                                   const QString& name);
 
    static bool saveObjectToFile(ccHObject* object,
                                 const QString& filename,
                                 bool writeAscii = false,
                                 bool compressed = false);
};