#include "gtest/gtest.h"

#include "RimSummaryCaseCollection.h"

#include <random>

#include <QDebug>

TEST( RimSummaryCaseCollection, EnsembleParameter )
{
    std::random_device                     rd;
    std::mt19937                           gen( rd() );
    std::uniform_real_distribution<double> meanDistribution( -10000.0, 10000.0 );
    std::uniform_real_distribution<double> variationDistribution( 0.0, 5000.0 );
    std::uniform_int_distribution<size_t>  countDistribution( 1u, 1000u );
    size_t                                 N = 1000;

    std::vector<EnsembleParameter::NameParameterPair> parameters;
    for ( size_t i = 0; i < N; ++i )
    {
        EnsembleParameter param;
        param.type = EnsembleParameter::TYPE_NUMERIC;

        size_t                                 valueCount = countDistribution( gen );
        double                                 meanValue  = meanDistribution( gen );
        double                                 range      = variationDistribution( gen );
        std::uniform_real_distribution<double> valueDistribution( meanValue - range, meanValue + range );
        double                                 maxValue = -std::numeric_limits<double>::max();
        double                                 minValue = std::numeric_limits<double>::max();
        for ( size_t j = 0; j < valueCount; ++j )
        {
            double value = valueDistribution( gen );
            maxValue     = std::max( maxValue, value );
            minValue     = std::min( minValue, value );
            param.values.push_back( QVariant( value ) );
        }

        param.minValue = minValue;
        param.maxValue = maxValue;

        double normStdDev = param.normalizedStdDeviation();
        EXPECT_GE( normStdDev, 0.0 );
        EXPECT_LE( normStdDev, std::sqrt( 2.0 ) );
        parameters.push_back( std::make_pair( QString( "%1" ).arg( i ), param ) );
    }
    size_t previousSize = parameters.size();
    EnsembleParameter::sortByBinnedVariation( parameters );
    size_t currentSize = parameters.size();
    EXPECT_EQ( previousSize, currentSize );

    int currentVariation = (int)EnsembleParameter::HIGH_VARIATION;
    for ( const EnsembleParameter::NameParameterPair& nameParamPair : parameters )
    {
        EXPECT_GE( nameParamPair.second.variationBin, (int)EnsembleParameter::LOW_VARIATION );
        EXPECT_LE( nameParamPair.second.variationBin, (int)EnsembleParameter::HIGH_VARIATION );
        EXPECT_LE( nameParamPair.second.variationBin, currentVariation );
        currentVariation = nameParamPair.second.variationBin;
    }
}