#include "gtest/gtest.h"

#include "RiaTestDataDirectory.h"

#include "RifStimPlanXmlReader.h"
#include "RigStimPlanFractureDefinition.h"

static const QString CASE_REAL_TEST_DATA_DIRECTORY_02 = QString( "%1/RifStimPlanXmlReader/" ).arg( TEST_DATA_DIR );

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
TEST( RifStimPlanXmlReaderTest, LoadFile )
{
    QString fileName = CASE_REAL_TEST_DATA_DIRECTORY_02 + "small_fracture.xml";

    double                           conductivityScaleFactor = 1.0;
    RiaDefines::EclipseUnitSystem    unit                    = RiaDefines::EclipseUnitSystem::UNITS_METRIC;
    QString                          errorMessage;
    RifStimPlanXmlReader::MirrorMode mode = RifStimPlanXmlReader::MirrorMode::MIRROR_AUTO;

    cvf::ref<RigStimPlanFractureDefinition> fractureData;

    fractureData = RifStimPlanXmlReader::readStimPlanXMLFile( fileName, conductivityScaleFactor, mode, unit, &errorMessage );

    EXPECT_TRUE( errorMessage.isEmpty() );
    EXPECT_TRUE( fractureData.notNull() );

    size_t xSamplesIncludingMirrorValues = 7;
    EXPECT_EQ( xSamplesIncludingMirrorValues, fractureData->xCount() );
    EXPECT_EQ( size_t( 5 ), fractureData->yCount() );
    EXPECT_EQ( size_t( 1 ), fractureData->timeSteps().size() );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
TEST( RifStimPlanXmlReaderTest, LoadFileNewFormat )
{
    QString fileName = CASE_REAL_TEST_DATA_DIRECTORY_02 + "contour_Metric.xml";

    double                           conductivityScaleFactor = 1.0;
    RiaDefines::EclipseUnitSystem    unit                    = RiaDefines::EclipseUnitSystem::UNITS_METRIC;
    QString                          errorMessage;
    RifStimPlanXmlReader::MirrorMode mode = RifStimPlanXmlReader::MirrorMode::MIRROR_AUTO;

    cvf::ref<RigStimPlanFractureDefinition> fractureData;

    fractureData = RifStimPlanXmlReader::readStimPlanXMLFile( fileName, conductivityScaleFactor, mode, unit, &errorMessage );

    EXPECT_TRUE( errorMessage.isEmpty() );
    EXPECT_TRUE( fractureData.notNull() );

    size_t xSamplesIncludingMirrorValues = 49;
    EXPECT_EQ( xSamplesIncludingMirrorValues, fractureData->xCount() );
    EXPECT_EQ( size_t( 23 ), fractureData->yCount() );
    EXPECT_EQ( size_t( 1 ), fractureData->timeSteps().size() );

    EXPECT_DOUBLE_EQ( 2773.680, fractureData->topPerfTvd() );
    EXPECT_DOUBLE_EQ( 2773.680, fractureData->bottomPerfTvd() );
    EXPECT_DOUBLE_EQ( 2804.160, fractureData->topPerfMd() );
    EXPECT_DOUBLE_EQ( 2804.770, fractureData->bottomPerfMd() );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
TEST( RifStimPlanXmlReaderTest, LoadFileNewFormatExtraParameters )
{
    QString fileName = CASE_REAL_TEST_DATA_DIRECTORY_02 + "contour_with_extra_parameters.xml";

    double                           conductivityScaleFactor = 1.0;
    RiaDefines::EclipseUnitSystem    unit                    = RiaDefines::EclipseUnitSystem::UNITS_METRIC;
    QString                          errorMessage;
    RifStimPlanXmlReader::MirrorMode mode = RifStimPlanXmlReader::MirrorMode::MIRROR_AUTO;

    cvf::ref<RigStimPlanFractureDefinition> fractureData;

    fractureData = RifStimPlanXmlReader::readStimPlanXMLFile( fileName, conductivityScaleFactor, mode, unit, &errorMessage );

    EXPECT_TRUE( errorMessage.isEmpty() );
    EXPECT_TRUE( fractureData.notNull() );

    EXPECT_DOUBLE_EQ( 12.34, fractureData->formationDip() );
    EXPECT_EQ( fractureData->orientation(), RigStimPlanFractureDefinition::Orientation::TRANSVERSE );
}