#include <freetype/freetype.h>
#include <assert.h>
#include <locale.h>
#include <ctype.h>

#include "CvxText.h"

//====================================================================
//====================================================================

CvxText::CvxText(const char *freeType)
{
	assert(freeType != NULL);

	if (FT_Init_FreeType(&m_library)) throw;
	if (FT_New_Face(m_library, freeType, 0, &m_face)) throw;

	restoreFont();

	setlocale(LC_ALL, "");

	m_backcolorEnable = true;
}


CvxText::~CvxText()
{
	FT_Done_Face(m_face);
	FT_Done_FreeType(m_library);
}


void CvxText::getFont(int *type, CvScalar *size, bool *underline, float *diaphaneity)
{
	if (type) *type = m_fontType;
	if (size) *size = m_fontSize;
	if (underline) *underline = m_fontUnderline;
	if (diaphaneity) *diaphaneity = m_fontDiaphaneity;
}

void CvxText::setFont(int *type, CvScalar *size, bool *underline, float *diaphaneity)
{
	if (type)
	{
		if (type >= 0) m_fontType = *type;
	}
	if (size)
	{
		m_fontSize.val[0] = fabs(size->val[0]);
		m_fontSize.val[1] = fabs(size->val[1]);
		m_fontSize.val[2] = fabs(size->val[2]);
		m_fontSize.val[3] = fabs(size->val[3]);
	}
	if (underline)
	{
		m_fontUnderline = *underline;
	}
	if (diaphaneity)
	{
		m_fontDiaphaneity = *diaphaneity;
	}
	FT_Set_Pixel_Sizes(m_face, size->val[0] , size->val[0]);
}

void CvxText::restoreFont()
{
	m_fontType = 0;

	m_fontSize.val[0] = 20;
	m_fontSize.val[1] = 0.5;
	m_fontSize.val[2] = 0.1;
	m_fontSize.val[3] = 0;

	m_fontUnderline = false;

	m_fontDiaphaneity = 1.0;

    m_scale = 1;

	FT_Set_Pixel_Sizes(m_face, (int)m_fontSize.val[0], (int)m_fontSize.val[0]);
}

int CvxText::putText(IplImage *img, const char    *text, CvPoint pos)
{
	return putText(img, text, pos, CV_RGB(255, 255, 255));
}
int CvxText::putText(IplImage *img, const wchar_t *text, CvPoint pos)
{
	return putText(img, text, pos, CV_RGB(255, 255, 255));
}

//

int CvxText::putText(IplImage *img, const char    *text, CvPoint pos, CvScalar color)
{
	if (img == NULL) return -1;
	if (text == NULL) return -1;
	int i;
	for (i = 0; text[i] != '\0'; ++i)
	{
		wchar_t wc = text[i];
        bool is_sascii = isascii(wc);
		if (is_sascii) mbtowc(&wc, &text[i++], 2);
		putWChar(img, wc, pos, color);
	}
	return i;
}
int CvxText::putText(IplImage *img, const wchar_t *text, CvPoint pos, CvScalar color)
{
	if (img == NULL) return -1;
	if (text == NULL) return -1;

	//

	int i;
	for (i = 0; text[i] != '\0'; ++i)
	{
		putWChar(img, text[i], pos, color);
	}
	return i;
}

void CvxText::drawBackGround(IplImage *img, CvPoint pos, int w, int h)
{
	cvRectangle(img, cvPoint(pos.x, pos.y + 3), cvPoint(pos.x + w, pos.y - h), m_backcolor, -1);
}


void CvxText::putWChar(IplImage *img, wchar_t wc, CvPoint &pos, CvScalar color)
{
	FT_UInt glyph_index = FT_Get_Char_Index(m_face, wc);
	FT_Load_Glyph(m_face, glyph_index, FT_LOAD_DEFAULT);
	FT_Render_Glyph(m_face->glyph, FT_RENDER_MODE_MONO);
	
	//

	FT_GlyphSlot slot = m_face->glyph;

	double space = m_fontSize.val[0] * m_fontSize.val[1] * m_scale;
	double sep = m_fontSize.val[0] * m_fontSize.val[2] * m_scale;


	int rows = slot->bitmap.rows;
	int cols = slot->bitmap.width;

	//
	if (m_backcolorEnable)
	drawBackGround(img, pos, (int)((cols ? cols : space) + sep), m_fontSize.val[0]);
	for (int i = 0; i < rows; ++i)
	{
		for (int j = 0; j < cols; ++j)
		{
			int off = ((img->origin == 0) ? i : (rows - 1 - i))
				* slot->bitmap.pitch + j / 8;

			if (slot->bitmap.buffer[off] & (0xC0 >> (j % 8)))
			{
				int r = (img->origin == 0) ? pos.y - (rows - 1 - i) : pos.y + i;
				int c = pos.x + j;
				if (r >= 0 && r < img->height
					&& c >= 0 && c < img->width)
				{
					CvScalar scalar = cvGet2D(img, r, c);

					float p = m_fontDiaphaneity;
					for (int k = 0; k < 4; ++k)
					{
						scalar.val[k] = scalar.val[k] * (1 - p) + color.val[k] * p;
					}
					cvSet2D(img, r, c, scalar);
				}
			}
		} // end for
	} // end for
	pos.x += (int)(cols ? cols : space) + sep;
}

int CvxText::putText(cv::Mat img, const wchar_t *text, CvPoint pos, CvScalar color) {
    IplImage iplImage(img);
    return putText(&iplImage, text,pos,color);
}

int CvxText::putText(cv::Mat img, const char *text, CvPoint pos, CvScalar color) {
    IplImage iplImage(img);
    return putText(&iplImage, text,pos,color);
}

int CvxText::putText(cv::Mat img, const char *text, CvPoint pos) {
     return putText(img, text, pos, CV_RGB(255,255,255));
}

int CvxText::putText(cv::Mat img, const wchar_t *text, CvPoint pos) {
    return putText(img, text, pos, CV_RGB(255,255,255));
}
//
//void CvxText::setScale(float scale) {
//    m_scale = scale;
//}
//
//void CvxText::getScale(float* scale) {
//    *scale = m_scale;
//}