173 lines
7 KiB
C++
173 lines
7 KiB
C++
#include "BaseTestCase.h"
|
|
|
|
#include <cmath>
|
|
#include "NoiseGenerator.h"
|
|
#include "TerrainDefinition.h"
|
|
#include "TerrainHeightMap.h"
|
|
#include "PaintedGridBrush.h"
|
|
#include "FloatNode.h"
|
|
|
|
/* Noise sin period is defined at 20.0 */
|
|
#define X_FACTOR (M_PI / 10.0)
|
|
|
|
static double _noise1dMock(double x)
|
|
{
|
|
return sin(x * X_FACTOR) * 0.5 + 0.5;
|
|
}
|
|
|
|
static double _noise2dMock(double x, double)
|
|
{
|
|
return sin(x * X_FACTOR) * 0.5 + 0.5;
|
|
}
|
|
|
|
static double _noise3dMock(double x, double, double)
|
|
{
|
|
return sin(x * X_FACTOR) * 0.5 + 0.5;
|
|
}
|
|
|
|
class TerrainPainting_Test : public BaseTestCase {
|
|
protected:
|
|
virtual void SetUp() {
|
|
terrain = new TerrainDefinition(NULL);
|
|
terrain->height = 3.0;
|
|
terrain->_height_noise->clearLevels();
|
|
terrain->propWaterHeight()->setValue(0.0);
|
|
NoiseGenerator::NoiseLevel level = {1.0, 2.0, -1.0};
|
|
terrain->_height_noise->addLevel(level);
|
|
noise_state.resetOffsets();
|
|
terrain->_height_noise->setState(noise_state);
|
|
terrain->_height_noise->setCustomFunction(_noise1dMock, _noise2dMock, _noise3dMock);
|
|
}
|
|
|
|
virtual void TearDown()
|
|
{
|
|
delete terrain;
|
|
}
|
|
|
|
TerrainDefinition* terrain;
|
|
NoiseState noise_state;
|
|
};
|
|
|
|
TEST_F(TerrainPainting_Test, grid)
|
|
{
|
|
/* Test base grid */
|
|
EXPECT_DOUBLE_EQ(0.0, terrain->getGridHeight(0, 0, 0));
|
|
EXPECT_DOUBLE_EQ(0.0, terrain->getGridHeight(0, 1, 0));
|
|
EXPECT_DOUBLE_EQ(0.0, terrain->getGridHeight(0, 0, 1));
|
|
EXPECT_DOUBLE_EQ(sin(1.0 * X_FACTOR), terrain->getGridHeight(1, 0, 0));
|
|
EXPECT_DOUBLE_EQ(sin(2.0 * X_FACTOR), terrain->getGridHeight(2, 0, 0));
|
|
EXPECT_DOUBLE_EQ(sin(3.0 * X_FACTOR), terrain->getGridHeight(3, 0, 0));
|
|
EXPECT_DOUBLE_EQ(sin(4.0 * X_FACTOR), terrain->getGridHeight(4, 0, 0));
|
|
EXPECT_DOUBLE_EQ(1.0, terrain->getGridHeight(5, 0, 0));
|
|
EXPECT_DOUBLE_EQ(sin(4.0 * X_FACTOR), terrain->getGridHeight(6, 0, 0));
|
|
EXPECT_DOUBLE_EQ(-sin(1.0 * X_FACTOR), terrain->getGridHeight(-1, 0, 0));
|
|
EXPECT_DOUBLE_EQ(0.0, terrain->getGridHeight(10, 0, 0));
|
|
EXPECT_DOUBLE_EQ(-1.0, terrain->getGridHeight(15, 0, 0));
|
|
EXPECT_DOUBLE_EQ(0.0, terrain->getGridHeight(20, 0, 0));
|
|
EXPECT_DOUBLE_EQ(-1.0, terrain->getGridHeight(-5, 0, 0));
|
|
|
|
/* Test interpolated result */
|
|
EXPECT_DOUBLE_EQ(terrain->getInterpolatedHeight(0.0, 0.0, 0, 0), 0.0);
|
|
EXPECT_DOUBLE_IN_RANGE(terrain->getInterpolatedHeight(0.5, 0.0, 0, 0), 0.1564, 0.1566);
|
|
EXPECT_DOUBLE_EQ(terrain->getInterpolatedHeight(1.0, 0.0, 0, 0), sin(1.0 * X_FACTOR));
|
|
EXPECT_DOUBLE_EQ(terrain->getInterpolatedHeight(0.0, 0.0, 1, 0), 0.0);
|
|
EXPECT_DOUBLE_IN_RANGE(terrain->getInterpolatedHeight(0.5, 0.0, 1, 0), 3.0 * 0.1564, 3.0 * 0.1566);
|
|
EXPECT_DOUBLE_EQ(terrain->getInterpolatedHeight(1.0, 0.0, 1, 0), 3.0 * sin(1.0 * X_FACTOR));
|
|
}
|
|
|
|
static void _checkBrushResultSides(TerrainDefinition* terrain, PaintedGridBrush*, double center, double midhard, double hard, double midsoft, double soft, double exter, double neg_midhard, double neg_hard, double neg_midsoft, double neg_soft, double neg_exter)
|
|
{
|
|
EXPECT_DOUBLE_EQ(terrain->getGridHeight(0, 0, 1), center);
|
|
|
|
EXPECT_DOUBLE_EQ(terrain->getGridHeight(1, 0, 1), midhard);
|
|
EXPECT_DOUBLE_EQ(terrain->getGridHeight(2, 0, 1), hard);
|
|
EXPECT_DOUBLE_EQ(terrain->getGridHeight(3, 0, 1), midsoft);
|
|
EXPECT_DOUBLE_EQ(terrain->getGridHeight(4, 0, 1), soft);
|
|
EXPECT_DOUBLE_EQ(terrain->getGridHeight(5, 0, 1), exter);
|
|
|
|
EXPECT_DOUBLE_EQ(terrain->getGridHeight(-1, 0, 1), neg_midhard);
|
|
EXPECT_DOUBLE_EQ(terrain->getGridHeight(-2, 0, 1), neg_hard);
|
|
EXPECT_DOUBLE_EQ(terrain->getGridHeight(-3, 0, 1), neg_midsoft);
|
|
EXPECT_DOUBLE_EQ(terrain->getGridHeight(-4, 0, 1), neg_soft);
|
|
EXPECT_DOUBLE_EQ(terrain->getGridHeight(-5, 0, 1), neg_exter);
|
|
}
|
|
|
|
static void _checkBrushResult(TerrainDefinition* terrain, PaintedGridBrush* brush, double center, double midhard, double hard, double midsoft, double soft, double exter, int mirror)
|
|
{
|
|
if (mirror)
|
|
{
|
|
_checkBrushResultSides(terrain, brush, center, midhard, hard, midsoft, soft, exter, -midhard, -hard, -midsoft, -soft, -exter);
|
|
}
|
|
else
|
|
{
|
|
_checkBrushResultSides(terrain, brush, center, midhard, hard, midsoft, soft, exter, midhard, hard, midsoft, soft, exter);
|
|
}
|
|
}
|
|
|
|
TEST_F(TerrainPainting_Test, brush_flatten)
|
|
{
|
|
/* Set up */
|
|
PaintedGridBrush brush(2.0, 2.0, 4.0);
|
|
terrain->height = 1.0;
|
|
terrain->_height_noise->forceValue(0.0);
|
|
|
|
/* Test flattening center at 0.5 */
|
|
_checkBrushResult(terrain, &brush, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0);
|
|
terrain->height_map->brushFlatten(brush, 0.0, 0.0, 0.5, 1.0);
|
|
_checkBrushResult(terrain, &brush, 0.5, 0.5, 0.5, 0.25, 0.0, 0.0, 0);
|
|
|
|
/* Test brush strength */
|
|
terrain->height_map->clearPainting();
|
|
_checkBrushResult(terrain, &brush, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0);
|
|
terrain->height_map->brushFlatten(brush, 0.0, 0.0, 0.5, 0.01);
|
|
_checkBrushResult(terrain, &brush, 0.005, 0.005, 0.005, 0.0025, 0.0, 0.0, 0);
|
|
|
|
/* Test cumulative effect */
|
|
terrain->height_map->brushFlatten(brush, 0.0, 0.0, 0.5, 0.01);
|
|
_checkBrushResult(terrain, &brush, 0.00995, 0.00995, 0.00995, 0.0049875, 0.0, 0.0, 0);
|
|
|
|
/* Test with height modifier */
|
|
terrain->height = 10.0;
|
|
terrain->height_map->clearPainting();
|
|
_checkBrushResult(terrain, &brush, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0);
|
|
terrain->height_map->brushFlatten(brush, 0.0, 0.0, 0.5, 1.0);
|
|
_checkBrushResult(terrain, &brush, 0.05, 0.05, 0.05, 0.025, 0.0, 0.0, 0);
|
|
}
|
|
|
|
TEST_F(TerrainPainting_Test, brush_reset)
|
|
{
|
|
/* Set up */
|
|
PaintedGridBrush brush(2.0, 2.0, 4.0);
|
|
PaintedGridBrush brush_full(4.0, 0.0, 4.0);
|
|
terrain->height = 1.0;
|
|
terrain->_height_noise->forceValue(1.0);
|
|
|
|
/* Test resetting at center */
|
|
_checkBrushResult(terrain, &brush, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 0);
|
|
terrain->height_map->brushFlatten(brush_full, 0.0, 0.0, 2.0, 1.0);
|
|
_checkBrushResult(terrain, &brush, 2.0, 2.0, 2.0, 2.0, 2.0, 1.0, 0);
|
|
terrain->height_map->brushReset(brush, 0.0, 0.0, 1.0);
|
|
_checkBrushResult(terrain, &brush, 1.0, 1.0, 1.0, 1.5, 2.0, 1.0, 0);
|
|
|
|
/* Test brush strength */
|
|
terrain->height_map->clearPainting();
|
|
_checkBrushResult(terrain, &brush, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 0);
|
|
terrain->height_map->brushFlatten(brush_full, 0.0, 0.0, 2.0, 1.0);
|
|
_checkBrushResult(terrain, &brush, 2.0, 2.0, 2.0, 2.0, 2.0, 1.0, 0);
|
|
terrain->height_map->brushReset(brush, 0.0, 0.0, 0.1);
|
|
_checkBrushResult(terrain, &brush, 1.9, 1.9, 1.9, 1.95, 2.0, 1.0, 0);
|
|
|
|
/* Test cumulative effect */
|
|
terrain->height_map->brushReset(brush, 0.0, 0.0, 0.1);
|
|
_checkBrushResult(terrain, &brush, 1.81, 1.81, 1.81, 1.9025, 2.0, 1.0, 0);
|
|
|
|
/* Test with height modifier */
|
|
terrain->height = 10.0;
|
|
terrain->height_map->clearPainting();
|
|
_checkBrushResult(terrain, &brush, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 0);
|
|
terrain->height_map->brushFlatten(brush_full, 0.0, 0.0, 2.0, 1.0);
|
|
_checkBrushResult(terrain, &brush, 1.1, 1.1, 1.1, 1.1, 1.1, 1.0, 0);
|
|
terrain->height_map->brushReset(brush, 0.0, 0.0, 0.1);
|
|
_checkBrushResult(terrain, &brush, 1.099, 1.099, 1.099, 1.0995, 1.1, 1.0, 0);
|
|
}
|