qColorimetricSegmenter.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
 
// ##########################################################################
// #                                                                        #
// #            CLOUDVIEWER PLUGIN: ColorimetricSegmenter                   #
// #                                                                        #
// #  This program is free software; you can redistribute it and/or modify  #
// #  it under the terms of the GNU General Public License as published by  #
// #  the Free Software Foundation; version 2 of the License.               #
// #                                                                        #
// #  This program is distributed in the hope that it will be useful,       #
// #  but WITHOUT ANY WARRANTY; without even the implied warranty of        #
// #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         #
// #  GNU General Public License for more details.                          #
// #                                                                        #
// #    COPYRIGHT:  Tri-Thien TRUONG, Ronan COLLIER, Mathieu LETRONE       #
// #                                                                        #
// ##########################################################################
 
#include "ecvStdPluginInterface.h"
 
// cloudViewer
#include <ReferenceCloud.h>
 
// Qt
#include <QObject>
#include <QtGui>
 
class ccPointCloud;
 
class ColorimetricSegmenter : public QObject, public ccStdPluginInterface {
    Q_OBJECT
    Q_INTERFACES(ccPluginInterface ccStdPluginInterface)
    Q_PLUGIN_METADATA(IID
                      "ecvcorp.cloudviewer.plugin.ColorimetricSegmenter" FILE
                      "info.json")
 
public:
    explicit ColorimetricSegmenter(QObject* parent = nullptr);
    ~ColorimetricSegmenter() override = default;
 
    // inherited from ccStdPluginInterface
    void onNewSelection(const ccHObject::Container& selectedEntities) override;
    QList<QAction*> getActions() override;
 
private:
    std::vector<ccPointCloud*> getSelectedPointClouds();
 
    void filterRgb();
 
    void filterHSV();
 
    void filterScalar();
 
    void HistogramClustering();
 
    void KmeansClustering();
 
    bool addPoint(cloudViewer::ReferenceCloud& filteredCloud, unsigned int j);
 
    template <typename T>
    void createClouds(T& dlg,
                      ccPointCloud* cloud,
                      const cloudViewer::ReferenceCloud& filteredCloudInside,
                      const cloudViewer::ReferenceCloud& filteredCloudOutside,
                      QString name);
 
    void createCloud(ccPointCloud* cloud,
                     const cloudViewer::ReferenceCloud& referenceCloud,
                     QString name);
 
    void filterRgbWithSegmentation();
 
    typedef QSharedPointer<cloudViewer::ReferenceCloud> Region;
    typedef std::vector<Region> RegionSet;
 
    static bool RegionGrowing(RegionSet& regions,
                              ccPointCloud* pointCloud,
                              const unsigned TNN,
                              const double TPP,
                              const double TD);
 
    static bool RegionMergingAndRefinement(RegionSet& mergedRegions,
                                           ccPointCloud* basePointCloud,
                                           const RegionSet& regions,
                                           const unsigned TNN,
                                           const double TRR,
                                           const double TD,
                                           const unsigned Min);
 
private:  // members
    QAction* m_action_filterRgb;
    // QAction* m_action_filterRgbWithSegmentation;
    QAction* m_action_filterHSV;
    QAction* m_action_filterScalar;
    QAction* m_action_histogramClustering;
    QAction* m_action_kMeansClustering;
 
    bool m_addPointError;
};