Files
ResInsight/ApplicationLibCode/UnitTests/RifFormationNamesReader-Test.cpp
T
Magne Sjaastad 601a3bbb78 Refactor RigFormationNames
- Rename RifColorLegendData to RifFormationNamesReader to better reflect the
class purpose of reading formation names from .lyr files.
- Remove inheritance of cvf::Object
- Improve const correctness
2026-02-27 14:17:21 +01:00

86 lines
2.8 KiB
C++

#include "gtest/gtest.h"
#include "RiaTestDataDirectory.h"
#include "RifFormationNamesReader.h"
#include "RigFormationNames.h"
#include "cvfColor3.h"
#include <QDir>
#include <QString>
#include <QStringList>
TEST( RifFormationNamesReader, ReadLYRFileWithoutColor )
{
QDir baseFolder( TEST_DATA_DIR );
const QString filename( "RifFormationNamesReader/Norne_ATW2013.lyr" );
const QString filePath = baseFolder.absoluteFilePath( filename );
EXPECT_TRUE( QFile::exists( filePath ) );
QString errormessage;
auto fm = RifFormationNamesReader::readFormationNamesFile( filePath, &errormessage );
EXPECT_TRUE( errormessage.isEmpty() );
QString formationName_K1 = fm->formationNameFromKLayerIdx( 0 );
int formationIndex = fm->formationIndexFromKLayerIdx( 1 );
EXPECT_TRUE( formationName_K1 == "Garn 3" );
EXPECT_EQ( 1, formationIndex );
}
TEST( RifFormationNamesReader, ReadLYRFileWithColorName )
{
QDir baseFolder( TEST_DATA_DIR );
const QString filename( "RifFormationNamesReader/Norne_ATW2013ColorName.lyr" );
const QString filePath = baseFolder.absoluteFilePath( filename );
EXPECT_TRUE( QFile::exists( filePath ) );
QString errormessage;
auto fm = RifFormationNamesReader::readFormationNamesFile( filePath, &errormessage );
EXPECT_TRUE( errormessage.isEmpty() );
QString formationName_K1 = fm->formationNameFromKLayerIdx( 1 );
int formationIndex = fm->formationIndexFromKLayerIdx( 1 );
cvf::Color3f formationColor;
bool colorPresent = fm->formationColorFromKLayerIdx( 1, &formationColor );
EXPECT_TRUE( colorPresent );
EXPECT_TRUE( formationName_K1 == "Garn 2" );
EXPECT_EQ( 1, formationIndex );
EXPECT_EQ( 1.0f, formationColor.r() );
EXPECT_EQ( 0.0f, formationColor.g() );
EXPECT_EQ( 0.0f, formationColor.b() );
}
TEST( RifFormationNamesReader, ReadLYRFileWithColorHTML )
{
QDir baseFolder( TEST_DATA_DIR );
const QString filename( "RifFormationNamesReader/Norne_ATW2013ColorHTML.lyr" );
const QString filePath = baseFolder.absoluteFilePath( filename );
EXPECT_TRUE( QFile::exists( filePath ) );
QString errormessage;
auto fm = RifFormationNamesReader::readFormationNamesFile( filePath, &errormessage );
EXPECT_TRUE( errormessage.isEmpty() );
QString formationName_K1 = fm->formationNameFromKLayerIdx( 1 );
int formationIndex = fm->formationIndexFromKLayerIdx( 1 );
cvf::Color3f formationColor;
bool colorPresent = fm->formationColorFromKLayerIdx( 1, &formationColor );
EXPECT_TRUE( colorPresent );
EXPECT_TRUE( formationName_K1 == "Garn 2" );
EXPECT_EQ( 1, formationIndex );
EXPECT_EQ( 1.0f, formationColor.r() );
EXPECT_EQ( 0.0f, formationColor.g() );
EXPECT_EQ( 0.0f, formationColor.b() );
}