#include <gtest/gtest.h>

#include "ofxImageEffect.h"
#include "ofxhClip.h"
#include "pluginSupport/oliveclip.h"
#include "pluginSupport/image.h"

namespace
{
olive::VideoParams make_params(int width, int height,
							  olive::core::PixelFormat format, int channels,
							  bool premultiplied)
{
	olive::VideoParams params;
	params.set_width(width);
	params.set_height(height);
	params.set_format(format);
	params.set_channel_count(channels);
	params.set_premultiplied_alpha(premultiplied);
	return params;
}
}

TEST(PluginSupportImage, AllocateFromParamsSetsProperties)
{
	OFX::Host::ImageEffect::ClipDescriptor desc(kOfxImageEffectOutputClipName);
	olive::VideoParams params =
		make_params(640, 480, olive::core::PixelFormat::u8, 4, true);
	olive::plugin::OliveClipInstance clip(nullptr, desc, params);

	olive::plugin::Image image(clip);
	OfxRectI bounds = { 0, 0, 640, 480 };
	OfxRectI rod = bounds;
	image.allocate_from_params(params, bounds, rod, true);

	EXPECT_NE(image.data(), nullptr);
	EXPECT_EQ(image.width(), 640);
	EXPECT_EQ(image.height(), 480);
	EXPECT_EQ(image.row_bytes(), 640 * 4);
	EXPECT_EQ(image.pixel_format(), olive::core::PixelFormat::u8);
	EXPECT_EQ(image.channel_count(), 4);
	EXPECT_TRUE(image.premultiplied_alpha());
}

TEST(PluginSupportImage, EnsureAllocatedFromParamsClearsAndResizes)
{
	OFX::Host::ImageEffect::ClipDescriptor desc(kOfxImageEffectOutputClipName);
	olive::VideoParams params =
		make_params(64, 32, olive::core::PixelFormat::u8, 3, false);
	olive::plugin::OliveClipInstance clip(nullptr, desc, params);

	olive::plugin::Image image(clip);
	OfxRectI bounds = { 0, 0, 64, 32 };
	OfxRectI rod = bounds;
	image.allocate_from_params(params, bounds, rod, true);
	ASSERT_NE(image.data(), nullptr);
	image.data()[0] = 0xAB;

	image.ensure_allocated_from_params(params, bounds, rod, true);
	EXPECT_EQ(image.data()[0], 0);

	OfxRectI new_bounds = { 0, 0, 16, 16 };
	image.ensure_allocated_from_params(params, new_bounds, rod, false);
	EXPECT_EQ(image.width(), 16);
	EXPECT_EQ(image.height(), 16);
}

TEST(PluginSupportImage, PropertyFallbacks)
{
	OFX::Host::ImageEffect::ClipDescriptor desc(kOfxImageEffectOutputClipName);
	olive::VideoParams params =
		make_params(1, 1, olive::core::PixelFormat::invalid, 0, false);
	olive::plugin::OliveClipInstance clip(nullptr, desc, params);

	olive::plugin::Image image(clip);
	image.setStringProperty(kOfxImageEffectPropPixelDepth, kOfxBitDepthHalf);
	image.setStringProperty(kOfxImageEffectPropComponents,
							kOfxImageComponentRGB);
	image.setStringProperty(kOfxImageEffectPropPreMultiplication,
							kOfxImagePreMultiplied);
	image.setIntProperty(kOfxImagePropBounds, 10, 0);
	image.setIntProperty(kOfxImagePropBounds, 20, 1);
	image.setIntProperty(kOfxImagePropBounds, 42, 2);
	image.setIntProperty(kOfxImagePropBounds, 70, 3);

	EXPECT_EQ(image.pixel_format(), olive::core::PixelFormat::f16);
	EXPECT_EQ(image.channel_count(), 3);
	EXPECT_TRUE(image.premultiplied_alpha());
	EXPECT_EQ(image.width(), 32);
	EXPECT_EQ(image.height(), 50);
}

TEST(PluginSupportImage, AllocateSetsOfxProperties)
{
	OFX::Host::ImageEffect::ClipDescriptor desc(kOfxImageEffectOutputClipName);
	olive::VideoParams params =
		make_params(8, 6, olive::core::PixelFormat::f16, 4, false);
	olive::plugin::OliveClipInstance clip(nullptr, desc, params);

	olive::plugin::Image image(clip);
	OfxRectI bounds = { 2, 3, 10, 9 };
	OfxRectI rod = { 0, 0, 12, 12 };
	image.allocate(8, 6, olive::core::PixelFormat::f16, 4, false, bounds, rod,
				   true);

	EXPECT_NE(image.data(), nullptr);
	EXPECT_EQ(image.row_bytes(), 8 * 4 * 2);

	int bounds_props[4] = { 0 };
	image.getIntPropertyN(kOfxImagePropBounds, bounds_props, 4);
	EXPECT_EQ(bounds_props[0], bounds.x1);
	EXPECT_EQ(bounds_props[1], bounds.y1);
	EXPECT_EQ(bounds_props[2], bounds.x2);
	EXPECT_EQ(bounds_props[3], bounds.y2);

	int rod_props[4] = { 0 };
	image.getIntPropertyN(kOfxImagePropRegionOfDefinition, rod_props, 4);
	EXPECT_EQ(rod_props[0], rod.x1);
	EXPECT_EQ(rod_props[1], rod.y1);
	EXPECT_EQ(rod_props[2], rod.x2);
	EXPECT_EQ(rod_props[3], rod.y2);

	EXPECT_EQ(image.getStringProperty(kOfxImageEffectPropComponents),
			  std::string(kOfxImageComponentRGBA));
	EXPECT_EQ(image.getStringProperty(kOfxImageEffectPropPixelDepth),
			  std::string(kOfxBitDepthHalf));
	EXPECT_EQ(image.getStringProperty(kOfxImageEffectPropPreMultiplication),
			  std::string(kOfxImageUnPreMultiplied));
}

TEST(PluginSupportImage, EnsureAllocatedPreservesWithoutClear)
{
	OFX::Host::ImageEffect::ClipDescriptor desc(kOfxImageEffectOutputClipName);
	olive::VideoParams params =
		make_params(4, 4, olive::core::PixelFormat::u8, 3, false);
	olive::plugin::OliveClipInstance clip(nullptr, desc, params);

	olive::plugin::Image image(clip);
	OfxRectI bounds = { 0, 0, 4, 4 };
	OfxRectI rod = bounds;
	image.allocate_from_params(params, bounds, rod, true);
	ASSERT_NE(image.data(), nullptr);
	image.data()[0] = 0x5A;

	image.ensure_allocated_from_params(params, bounds, rod, false);
	EXPECT_EQ(image.data()[0], 0x5A);
}