/*
  Copyright (C) 2013 Birunthan Mohanathas

  This program is free software; you can redistribute it and/or
  modify it under the terms of the GNU General Public License
  as published by the Free Software Foundation; either version 2
  of the License, or (at your option) any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
*/

#ifndef RM_GFX_CANVASD2D_H_
#define RM_GFX_CANVASD2D_H_

#include "Canvas.h"
#include "TextFormatD2D.h"
#include <string>
#include <GdiPlus.h>
#include <d2d1.h>
#include <d2d1helper.h>
#include <dwrite.h>
#include <wincodec.h>

namespace Gfx {

// Provides a Direct2D/DirectWrite implementation of Canvas.
class CanvasD2D : public Canvas
{
public:
	CanvasD2D();
	~CanvasD2D();

	virtual void Resize(int w, int h);

	virtual bool BeginDraw();
	virtual void EndDraw();

	virtual Gdiplus::Graphics& BeginGdiplusContext() override;
	virtual void EndGdiplusContext() override;

	virtual HDC GetDC() override;
	virtual void ReleaseDC(HDC dc) override;

	virtual TextFormat* CreateTextFormat() override { return new TextFormatD2D(); }

	virtual bool IsTransparentPixel(int x, int y) override;

	virtual void SetAntiAliasing(bool enable) override;
	virtual void SetTextAntiAliasing(bool enable) override;

	virtual void Clear(const Gdiplus::Color& color) override;

	virtual void DrawTextW(const WCHAR* str, UINT strLen, const TextFormat& format, Gdiplus::RectF& rect, const Gdiplus::SolidBrush& brush) override;
	virtual bool MeasureTextW(const WCHAR* str, UINT strLen, const TextFormat& format, Gdiplus::RectF& rect) override;
	virtual bool MeasureTextLinesW(const WCHAR* str, UINT strLen, const TextFormat& format, Gdiplus::RectF& rect, UINT& lines) override;

	virtual void DrawBitmap(Gdiplus::Bitmap* bitmap, const Gdiplus::Rect& dstRect, const Gdiplus::Rect& srcRect) override;

	virtual void FillRectangle(Gdiplus::Rect& rect, const Gdiplus::SolidBrush& brush) override;

private:
	friend class TextFormatD2D;

	CanvasD2D(const CanvasD2D& other) {}

	static bool Initialize();
	static void Finalize();

	void DiscardDeviceResources();

	void BeginTargetDraw();
	void EndTargetDraw();

	ID2D1RenderTarget* m_Target;
	ID2D1GdiInteropRenderTarget* m_InteropTarget;
	IWICBitmap* m_Bitmap;

	// GDI+ objects that share the pixel data of m_Bitmap.
	Gdiplus::Graphics* m_GdipGraphics;
	Gdiplus::Bitmap* m_GdipBitmap;

	// If true, the BeginDraw() function of this class has been called and the matching EndDraw()
	// has not been called yet.
	bool m_BeginDrawCalled;

	// If true, the BeginDraw() function of the render target has been called and the matching
	// EndDraw() has not been called yet.
	bool m_TargetBeginDrawCalled;

	static UINT c_Instances;
	static ID2D1Factory* c_D2D;
	static IDWriteFactory* c_DW;
	static IWICImagingFactory* c_WIC;
};

}  // namespace Gfx

#endif