/****************************************************************************
 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>.
*****************************************************************************/

#include "onf_metricnbylasclass.h"
#include "ct_pointcloudindex/ct_pointcloudindexvector.h"
#include "ct_iterator/ct_pointiterator.h"

#include "ctliblas/tools/las/ct_lasdata.h"

ONF_MetricNbyLASClass::ONF_MetricNbyLASClass(QString pluginName) : CT_AbstractMetric_LAS(pluginName)
{
    declareAttributes();
}

ONF_MetricNbyLASClass::ONF_MetricNbyLASClass(const ONF_MetricNbyLASClass &other) : CT_AbstractMetric_LAS(other)
{
    declareAttributes();
    m_configAndResults = other.m_configAndResults;
}

QString ONF_MetricNbyLASClass::getShortDisplayableName() const
{
    return tr("Nombre de points par type/classe (LAS, Alti/Ht)");
}

QString ONF_MetricNbyLASClass::getShortDescription() const
{
    return tr("Calcul des nombres de points pour chaque type de retour et chaque type de classification LAS. "
              "Calcul de quelques indicateurs complémentaires (cf. détails). ");
}

QString ONF_MetricNbyLASClass::getDetailledDescription() const
{
    return tr("Les indicateurs suivants peuvent être calculés :"
              "<ul>"
              "<li><strong>N</strong> : Nombre total de points<br>"
              "<li><strong>N_first</strong> : Nombre de points First<br>"
              "<li><strong>N_last</strong> : Nombre de points Last<br>"
              "<li><strong>N_int</strong> : Nombre de points Intermediate<br>"
              "<li><strong>N_only</strong> : Nombre de points Only<br>"
              "<li><strong>N_error</strong> : Nombre de points en Erreur (nombre de retours ou numéro de retour aberrant)<br>"
              "<li><strong>N_Cla00_NeverClassified</strong> : Nombre de points classés 0<br>"
              "<li><strong>N_Cla01_Unclassified</strong> : Nombre de points classés 1<br>"
              "<li><strong>N_Cla02_Ground</strong> : Nombre de points classés 2<br>"
              "<li><strong>N_Cla03_LowVegetation</strong> : Nombre de points classés 3<br>"
              "<li><strong>N_Cla04_MediumVegetation</strong> : Nombre de points classés 4<br>"
              "<li><strong>N_Cla05_HighVegetation</strong> : Nombre de points classés 5<br>"
              "<li><strong>N_Cla06_Building</strong> : Nombre de points classés 6<br>"
              "<li><strong>N_Cla07_LowPointNoise</strong> : Nombre de points classés 7<br>"
              "<li><strong>N_Cla08_ModelKeyPoint</strong> : Nombre de points classés 8<br>"
              "<li><strong>N_Cla09_Water</strong> : Nombre de points classés 9<br>"
              "<li><strong>N_Cla10_Rail</strong> : Nombre de points classés 10<br>"
              "<li><strong>N_Cla11_RoadSurface</strong> : Nombre de points classés 11<br>"
              "<li><strong>N_Cla12_OverlapPoint</strong> : Nombre de points classés 12<br>"
              "<li><strong>N_Cla13_WireGuard</strong> : Nombre de points classés 13<br>"
              "<li><strong>N_Cla14_WireConductor</strong> : Nombre de points classés 14<br>"
              "<li><strong>N_Cla15_TransmissionTower</strong> : Nombre de points classés 15<br>"
              "<li><strong>N_Cla16_WireStructureConnector</strong> : Nombre de points classés 16<br>"
              "<li><strong>N_Cla17_BridgeDeck</strong> : Nombre de points classés 17<br>"
              "<li><strong>N_Cla18_HighNoise</strong> : Nombre de points classés 18<br>"
              "<li><strong>N_Cla19_63_Reserved</strong> : Nombre de points classés 19 à 63<br>"
              "<li><strong>N_Cla64_255_UserDefinable</strong> : Nombre de points classés 63 à 255<br>"
              "<li><strong>ZRange</strong> : Zmax - Zmin<br>"
              "<li><strong>NumberOfLines</strong> : Nombre de lignes vols couvrant la zone<br>"
              "<li><strong>N_bestLine</strong> : Nombre de points de la ligne de vols ayant le plus de points<br>"
              "<li><strong>N_secondLine</strong> : Nombre de points de la seconde ligne de vols ayant le plus de points<br>"
              "<li><strong>N_worstLine</strong> : Nombre de points de la ligne de vols ayant le moins de points"
              "</ul>");
}

ONF_MetricNbyLASClass::Config ONF_MetricNbyLASClass::metricConfiguration() const
{
    return m_configAndResults;
}

void ONF_MetricNbyLASClass::setMetricConfiguration(const ONF_MetricNbyLASClass::Config &conf)
{
    m_configAndResults = conf;
}

CT_AbstractConfigurableElement *ONF_MetricNbyLASClass::copy() const
{
    return new ONF_MetricNbyLASClass(*this);
}

void ONF_MetricNbyLASClass::computeMetric()
{    
    m_configAndResults.n.value = 0;
    m_configAndResults.i_mean_1st.value = 0;
    m_configAndResults.n_last.value = 0;
    m_configAndResults.n_intermediate.value = 0;
    m_configAndResults.n_only.value = 0;
    m_configAndResults.n_error.value = 0;

    m_configAndResults.n_cla00.value = 0;
    m_configAndResults.n_cla01.value = 0;
    m_configAndResults.n_cla02.value = 0;
    m_configAndResults.n_cla03.value = 0;
    m_configAndResults.n_cla04.value = 0;
    m_configAndResults.n_cla05.value = 0;
    m_configAndResults.n_cla06.value = 0;
    m_configAndResults.n_cla07.value = 0;
    m_configAndResults.n_cla08.value = 0;
    m_configAndResults.n_cla09.value = 0;
    m_configAndResults.n_cla10.value = 0;
    m_configAndResults.n_cla11.value = 0;
    m_configAndResults.n_cla12.value = 0;
    m_configAndResults.n_cla13.value = 0;
    m_configAndResults.n_cla14.value = 0;
    m_configAndResults.n_cla15.value = 0;
    m_configAndResults.n_cla16.value = 0;
    m_configAndResults.n_cla17.value = 0;
    m_configAndResults.n_cla18.value = 0;
    m_configAndResults.n_cla19_63.value = 0;
    m_configAndResults.n_cla64_255.value = 0;

    m_configAndResults.max_m_min.value = std::numeric_limits<double>::quiet_NaN();
    m_configAndResults.numberOfLines.value = 0;
    m_configAndResults.nBestLine.value = 0;
    m_configAndResults.nSecondLine.value = 0;
    m_configAndResults.nWorstLine.value = 0;

    double max = -std::numeric_limits<double>::max();
    double min = std::numeric_limits<double>::max();

    QMap<quint16, size_t> linesCounts;

    CT_PointIterator itP(pointCloud());
    while(itP.hasNext())
    {
        const CT_Point& point = itP.next().currentPoint();

        if ((plotArea() == nullptr) || plotArea()->contains(point(0), point(1)))
        {
            ++(m_configAndResults.n.value);

            const size_t index = itP.currentGlobalIndex();

            CT_LASData lasData;
            lasAttributes()->getLASDataAt(index, lasData);

            if (lasData._Return_Number == 1 && lasData._Number_of_Returns >= 1) {++(m_configAndResults.i_mean_1st.value);}
            if (lasData._Return_Number == 1 && lasData._Number_of_Returns == 1) {++(m_configAndResults.n_only.value);}
            if (lasData._Return_Number != 1 && lasData._Return_Number == lasData._Number_of_Returns && lasData._Number_of_Returns > 1) {++(m_configAndResults.n_last.value);}
            if (lasData._Return_Number > 1 && lasData._Return_Number != lasData._Number_of_Returns && lasData._Number_of_Returns > 1) {++(m_configAndResults.n_intermediate.value);}
            if (lasData._Return_Number < 1 || lasData._Number_of_Returns < 1 || lasData._Return_Number > lasData._Number_of_Returns) {++(m_configAndResults.n_error.value);}

            if (lasData._Classification == 0) {++(m_configAndResults.n_cla00.value);}
            else if (lasData._Classification == 1) {++(m_configAndResults.n_cla01.value);}
            else if (lasData._Classification == 2) {++(m_configAndResults.n_cla02.value);}
            else if (lasData._Classification == 3) {++(m_configAndResults.n_cla03.value);}
            else if (lasData._Classification == 4) {++(m_configAndResults.n_cla04.value);}
            else if (lasData._Classification == 5) {++(m_configAndResults.n_cla05.value);}
            else if (lasData._Classification == 6) {++(m_configAndResults.n_cla06.value);}
            else if (lasData._Classification == 7) {++(m_configAndResults.n_cla07.value);}
            else if (lasData._Classification == 8) {++(m_configAndResults.n_cla08.value);}
            else if (lasData._Classification == 9) {++(m_configAndResults.n_cla09.value);}
            else if (lasData._Classification == 10) {++(m_configAndResults.n_cla10.value);}
            else if (lasData._Classification == 11) {++(m_configAndResults.n_cla11.value);}
            else if (lasData._Classification == 12) {++(m_configAndResults.n_cla12.value);}
            else if (lasData._Classification == 13) {++(m_configAndResults.n_cla13.value);}
            else if (lasData._Classification == 14) {++(m_configAndResults.n_cla14.value);}
            else if (lasData._Classification == 15) {++(m_configAndResults.n_cla15.value);}
            else if (lasData._Classification == 16) {++(m_configAndResults.n_cla16.value);}
            else if (lasData._Classification == 17) {++(m_configAndResults.n_cla17.value);}
            else if (lasData._Classification == 18) {++(m_configAndResults.n_cla18.value);}
            else if (lasData._Classification >= 19 && lasData._Classification <= 63) {++(m_configAndResults.n_cla19_63.value);}
            else {++(m_configAndResults.n_cla64_255.value);}

            if (point(2) > max) {max = point(2);}
            if (point(2) < min) {min = point(2);}

            linesCounts[lasData._Point_Source_ID] = linesCounts.value(lasData._Point_Source_ID, 0) + 1;
        }
    }

    QList<size_t> counts = linesCounts.values();
    std::sort(counts.begin(), counts.end());

    m_configAndResults.numberOfLines.value = linesCounts.size();

    if (counts.size() > 1)
    {
        m_configAndResults.nBestLine.value = counts.last();
    } else {
        m_configAndResults.nBestLine.value = 0;
    }

    if (counts.size() > 1)
    {
        m_configAndResults.nSecondLine.value = counts.at(counts.size() - 2);
        m_configAndResults.nWorstLine.value  = counts.first();
    } else {
        m_configAndResults.nSecondLine.value = 0;
        m_configAndResults.nWorstLine.value  = 0;
    }

    if (m_configAndResults.n.value > 0)
    {
        m_configAndResults.max_m_min.value = max - min;
    }

    setAttributeValueVaB(m_configAndResults.n);

    setAttributeValueVaB(m_configAndResults.i_mean_1st);
    setAttributeValueVaB(m_configAndResults.n_last);
    setAttributeValueVaB(m_configAndResults.n_intermediate);
    setAttributeValueVaB(m_configAndResults.n_only);
    setAttributeValueVaB(m_configAndResults.n_error);

    setAttributeValueVaB(m_configAndResults.n_cla00);
    setAttributeValueVaB(m_configAndResults.n_cla01);
    setAttributeValueVaB(m_configAndResults.n_cla02);
    setAttributeValueVaB(m_configAndResults.n_cla03);
    setAttributeValueVaB(m_configAndResults.n_cla04);
    setAttributeValueVaB(m_configAndResults.n_cla05);
    setAttributeValueVaB(m_configAndResults.n_cla06);
    setAttributeValueVaB(m_configAndResults.n_cla07);
    setAttributeValueVaB(m_configAndResults.n_cla08);
    setAttributeValueVaB(m_configAndResults.n_cla09);
    setAttributeValueVaB(m_configAndResults.n_cla10);
    setAttributeValueVaB(m_configAndResults.n_cla11);
    setAttributeValueVaB(m_configAndResults.n_cla12);
    setAttributeValueVaB(m_configAndResults.n_cla13);
    setAttributeValueVaB(m_configAndResults.n_cla14);
    setAttributeValueVaB(m_configAndResults.n_cla15);
    setAttributeValueVaB(m_configAndResults.n_cla16);
    setAttributeValueVaB(m_configAndResults.n_cla17);
    setAttributeValueVaB(m_configAndResults.n_cla18);
    setAttributeValueVaB(m_configAndResults.n_cla19_63);
    setAttributeValueVaB(m_configAndResults.n_cla64_255);

    setAttributeValueVaB(m_configAndResults.max_m_min);
    setAttributeValueVaB(m_configAndResults.numberOfLines);
    setAttributeValueVaB(m_configAndResults.nBestLine);
    setAttributeValueVaB(m_configAndResults.nSecondLine);
    setAttributeValueVaB(m_configAndResults.nWorstLine);
}

void ONF_MetricNbyLASClass::declareAttributes()
{
    registerAttributeVaB(m_configAndResults.n, CT_AbstractCategory::DATA_NUMBER, "N");
    registerAttributeVaB(m_configAndResults.i_mean_1st, CT_AbstractCategory::DATA_NUMBER, "N_first");
    registerAttributeVaB(m_configAndResults.n_last, CT_AbstractCategory::DATA_NUMBER, "N_last");
    registerAttributeVaB(m_configAndResults.n_intermediate, CT_AbstractCategory::DATA_NUMBER, "N_int");
    registerAttributeVaB(m_configAndResults.n_only, CT_AbstractCategory::DATA_NUMBER, "N_only");
    registerAttributeVaB(m_configAndResults.n_error, CT_AbstractCategory::DATA_NUMBER, "N_error");

    registerAttributeVaB(m_configAndResults.n_cla00, CT_AbstractCategory::DATA_NUMBER, "N_Cla00_NeverClassified");
    registerAttributeVaB(m_configAndResults.n_cla01, CT_AbstractCategory::DATA_NUMBER, "N_Cla01_Unclassified");
    registerAttributeVaB(m_configAndResults.n_cla02, CT_AbstractCategory::DATA_NUMBER, "N_Cla02_Ground");
    registerAttributeVaB(m_configAndResults.n_cla03, CT_AbstractCategory::DATA_NUMBER, "N_Cla03_LowVegetation");
    registerAttributeVaB(m_configAndResults.n_cla04, CT_AbstractCategory::DATA_NUMBER, "N_Cla04_MediumVegetation");
    registerAttributeVaB(m_configAndResults.n_cla05, CT_AbstractCategory::DATA_NUMBER, "N_Cla05_HighVegetation");
    registerAttributeVaB(m_configAndResults.n_cla06, CT_AbstractCategory::DATA_NUMBER, "N_Cla06_Building");
    registerAttributeVaB(m_configAndResults.n_cla07, CT_AbstractCategory::DATA_NUMBER, "N_Cla07_LowPointNoise");
    registerAttributeVaB(m_configAndResults.n_cla08, CT_AbstractCategory::DATA_NUMBER, "N_Cla08_ModelKeyPoint");
    registerAttributeVaB(m_configAndResults.n_cla09, CT_AbstractCategory::DATA_NUMBER, "N_Cla09_Water");
    registerAttributeVaB(m_configAndResults.n_cla10, CT_AbstractCategory::DATA_NUMBER, "N_Cla10_Rail");
    registerAttributeVaB(m_configAndResults.n_cla11, CT_AbstractCategory::DATA_NUMBER, "N_Cla11_RoadSurface");
    registerAttributeVaB(m_configAndResults.n_cla12, CT_AbstractCategory::DATA_NUMBER, "N_Cla12_OverlapPoint");
    registerAttributeVaB(m_configAndResults.n_cla13, CT_AbstractCategory::DATA_NUMBER, "N_Cla13_WireGuard");
    registerAttributeVaB(m_configAndResults.n_cla14, CT_AbstractCategory::DATA_NUMBER, "N_Cla14_WireConductor");
    registerAttributeVaB(m_configAndResults.n_cla15, CT_AbstractCategory::DATA_NUMBER, "N_Cla15_TransmissionTower");
    registerAttributeVaB(m_configAndResults.n_cla16, CT_AbstractCategory::DATA_NUMBER, "N_Cla16_WireStructureConnector");
    registerAttributeVaB(m_configAndResults.n_cla17, CT_AbstractCategory::DATA_NUMBER, "N_Cla17_BridgeDeck");
    registerAttributeVaB(m_configAndResults.n_cla18, CT_AbstractCategory::DATA_NUMBER, "N_Cla18_HighNoise");
    registerAttributeVaB(m_configAndResults.n_cla19_63, CT_AbstractCategory::DATA_NUMBER, "N_Cla19_63_Reserved");
    registerAttributeVaB(m_configAndResults.n_cla64_255, CT_AbstractCategory::DATA_NUMBER, "N_Cla64_255_UserDefinable");

    registerAttributeVaB(m_configAndResults.max_m_min, CT_AbstractCategory::DATA_NUMBER, "ZRange");

    registerAttributeVaB(m_configAndResults.numberOfLines, CT_AbstractCategory::DATA_NUMBER, "NumberOfLines");
    registerAttributeVaB(m_configAndResults.nBestLine, CT_AbstractCategory::DATA_NUMBER, "N_bestLine");
    registerAttributeVaB(m_configAndResults.nSecondLine, CT_AbstractCategory::DATA_NUMBER, "N_secondLine");
    registerAttributeVaB(m_configAndResults.nWorstLine, CT_AbstractCategory::DATA_NUMBER, "N_worstLine");
}