/****************************************************************************
 Copyright (C) 2010-2012 the Office National des Forêts (ONF), France
                         All rights reserved.

 Contact : alexandre.piboule@onf.fr

 Developers : Alexandre PIBOULE (ONF)

 This file is part of PluginONF library.

 PluginONF is free library: you can redistribute it and/or modify
 it under the terms of the GNU Lesser General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 PluginONF 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.

 You should have received a copy of the GNU Lesser General Public License
 along with PluginONF.  If not, see <http://www.gnu.org/licenses/lgpl.html>.
*****************************************************************************/

#ifndef ONF_STEPSEGMENTCROWNS_H
#define ONF_STEPSEGMENTCROWNS_H

#include "ct_step/abstract/ct_abstractstep.h"
#include "ct_tools/model/ct_autorenamemodels.h"

#include "ct_itemdrawable/model/outModel/ct_outstdsingularitemmodel.h"
#include "ct_itemdrawable/model/outModel/ct_outstdgroupmodel.h"
#include "ct_pointcloudindex/ct_pointcloudindexvector.h"
#include "ct_triangulation/ct_delaunayt.h"

#include "actions/onf_actiondefineheightlayer.h"
#include "actions/onf_actionsegmentcrowns.h"

#include "ct_itemdrawable/abstract/ct_abstractgrid3d.h"
#include "ct_itemdrawable/ct_scene.h"
#include "ct_itemdrawable/ct_grid2dxy.h"
#include "ct_itemdrawable/ct_polygon2d.h"


class CT_AbstractSingularItemDrawable;
class CT_AbstractItemGroup;


class ONF_StepSegmentCrowns: public CT_AbstractStep
{
    Q_OBJECT

public:

    /*! \brief Step constructor
     *
     * Create a new instance of the step
     *
     * \param dataInit Step parameters object
     */
    ONF_StepSegmentCrowns(CT_StepInitializeData &dataInit);

    /*! \brief Step description
     *
     * Return a description of the step function
     */
    QString getStepDescription() const;

    /**
     * @brief Inherit this method if you want to return your own detailled description.
     *
     *        By default return a empty string
     */
    virtual QString getStepDetailledDescription() const;

    /*! \brief Step copy
     *
     * Step copy, used when a step is added by step contextual menu
     */
    CT_VirtualAbstractStep* createNewInstance(CT_StepInitializeData &dataInit);

protected:

    /*! \brief Input results specification
     *
     * Specification of input results models needed by the step (IN)
     */
    void createInResultModelListProtected();

    /*! \brief Parameters DialogBox
     *
     * DialogBox asking for step parameters
     */
    void createPostConfigurationDialog();

    /*! \brief Output results specification
     *
     * Specification of output results models created by the step (OUT)
     */
    void createOutResultModelListProtected();

    /*! \brief Algorithm of the step
     *
     * Step computation, using input results, and creating output results
     */
    void compute();

    void initManualMode();

    void useManualMode(bool quit = false);

private:

    // Step parameters
    DocumentInterface                           *m_doc;
    int                                         m_status;
    CT_AbstractActionForGraphicsView            *_action;
    QList<CT_Scene*>                            m_itemDrawableToAdd;

    CT_OutStdSingularItemModel *_outDensityGridModel;
    CT_OutStdSingularItemModel *_outMNSGridModel;
    CT_ResultGroup                  *_outResult;


    ONF_ActionDefineHeightLayer_gridContainer *_gridContainer;
    CT_Grid2DXY<int>                         *_clustersGrid;

    QList<CT_Scene*>            _sceneList;
    double                       _xmin;
    double                       _ymin;
    double                       _zmin;
    double                       _xmax;
    double                       _ymax;
    double                       _zmax;



    int dropEmptyClusters();

    void addPointsToExtractedScenes(CT_ResultGroup *resultIn,
                                    const QMap<int, CT_PointCloudIndexVector *> &indexVectorMap);

    void registerScenes(const QMap<int, CT_PointCloudIndexVector *> &indexVectorMap,
                        const QMap<int, CT_StandardItemGroup *> &sceneGroupMap);

    void registerClusterCells(QMap<int, QList<Eigen::Vector2d*> *> &cellsMapByCluster,
                              QMap<int, size_t> &clusterCounts,
                              QMap<int, double> &clusterZMax);

    void createConvexHulls(QMap<int, QList<Eigen::Vector2d *> *> &cellsMapByCluster,
                           const QMap<int, CT_StandardItemGroup *> &sceneGroupMap,
                           QMap<int, CT_Polygon2DData *> &convexHullsMap);

    void computeMetrics(const QMap<int, CT_StandardItemGroup*> &sceneGroupMap,
                        const QMap<int, CT_Polygon2DData *> &convexHullsMap,
                        const QMap<int, size_t> &clusterCounts,
                        const QMap<int, double> &clusterZMax);

};

#endif // ONF_STEPSEGMENTCROWNS_H