#include "NameCreater.h"
#include <QtWidgets/QApplication>
#include <QTextCodec>
#include <QGraphicsScene>
#include <QGraphicsView>
#include <QGraphicsPixmapItem>
#include <QGraphicsTextItem>
#include <QFile>
#include <QDebug>
#include <QMessageBox>
#include <QDir>
#include <QDateTime>
#include <QFontDatabase>


#define BG_SAMPLE_PATH ":/NameCreater/resource/background_sample.jpg"
#define BG_PATH_FORMAT ":/NameCreater/resource/background_%1.jpg"
#define FONT_NAME "DFYuanStd-W8.otf"
#define FONT_SIZE 120
//#define FONT_SIZE 74

#define BG_MIX_TYPE_STR "mix"
#define BG_BOY_TYPE_STR "boy"
#define BG_GIRL_TYPE_STR "girl"

struct NameInfo_t {
	QString name1;
	QString name2;
	QString bg_path;
};

static QPointF BG_POS[] = {
	QPointF(159, 61), QPointF(482, 61), QPointF(802, 61),
	QPointF(1122, 61), QPointF(1449, 61), QPointF(1772, 61),
	QPointF(2095, 61), QPointF(2414, 61), QPointF(2738, 61),
	QPointF(3060, 61), QPointF(900, 2094), QPointF(900, 2420),
};

static int BG_POS_COUNT = sizeof(BG_POS) / sizeof(BG_POS[0]);


static QPointF NAME_POS[] = {
	QPointF(240, 430), QPointF(372, 1402),
	QPointF(564, 430), QPointF(694, 1402),
	QPointF(883, 430), QPointF(1014, 1402),
	QPointF(1205, 430), QPointF(1337, 1402),
	QPointF(1532, 430), QPointF(1663, 1402),
	QPointF(1854, 430), QPointF(1987, 1402),
	QPointF(2176, 430), QPointF(2309, 1402),
	QPointF(2498, 430), QPointF(2629, 1402),
	QPointF(2821, 430), QPointF(2952, 1402),
	QPointF(3145, 430), QPointF(3276, 1402),
	QPointF(1268, 2012), QPointF(2240, 1880),
	QPointF(1268, 2334), QPointF(2240, 2202),
};

static int NAME_POS_COUNT = sizeof(NAME_POS) / sizeof(NAME_POS[0]);





void saveToImage(const QString& filename, QGraphicsScene* scene) {

	scene->clearSelection();                                                  // Selections would also render to the file
	scene->setSceneRect(scene->itemsBoundingRect());                          // Re-shrink the scene to it's bounding contents
	QImage image(scene->sceneRect().size().toSize(), QImage::Format_ARGB32);  // Create the image with the exact size of the shrunk scene
	image.fill(Qt::transparent);                                              // Start all pixels transparent
	int dpm = 300 / 0.0254; // ~300 DPI
	image.setDotsPerMeterX(dpm);
	image.setDotsPerMeterY(dpm);


	QPainter painter(&image);
	scene->render(&painter);

#ifdef Q_OS_MAC
	QString folder_name = QCoreApplication::applicationDirPath() + "/../../../export_" + QDateTime::currentDateTime().toString("yyyyMMdd");
#else
	QString folder_name = QCoreApplication::applicationDirPath() + "/export_" + QDateTime::currentDateTime().toString("yyyyMMdd");
#endif

	qDebug() << "folder_name :" << folder_name;
	if (!QDir(folder_name).exists()) {
		QDir().mkdir(folder_name);
	}

	image.save(QString(folder_name).append("/").append(filename), "JPG", 100);

}

QList<NameInfo_t> getNameByFile(const QString& filename) {
	QList<NameInfo_t> nameList;

#ifdef Q_OS_MAC
	QString root_path = QCoreApplication::applicationDirPath() + "/../../../";
#else
	QString root_path = QCoreApplication::applicationDirPath() + "/";
#endif


	QFile file(root_path.append(filename));

	if (!file.open(QIODevice::ReadOnly)) {
		QMessageBox msgBox;
		msgBox.setText("Load file failed!");
		msgBox.exec();
		file.close();
		return nameList;
	}

	int row = 0;
	QTextCodec* tc = QTextCodec::codecForName("Big5");
	while (!file.atEnd())
	{
		NameInfo_t info;
		QByteArray line = file.readLine();
		QString name_tc = tc->toUnicode(line);
		QStringList csvList = name_tc.split(',');


		QString name1 = csvList.at(0);
		QString type_str = csvList.last();//QString::fromLocal8Bit(csvList.last());
		qDebug() << "row :" << row << " " << name1;
		type_str = type_str.remove(QRegExp("\r\n"));

		if (name1.compare("name_1") == 0) {
			continue;
		}

		info.name1 = name1;
		info.bg_path = QString(BG_PATH_FORMAT).arg(type_str);


		nameList.append(info);

		row++;
	}
	file.close();
	return nameList;
}

int main(int argc, char* argv[])
{
	int export_times = 0;



	QApplication a(argc, argv);
	//    NameCreater w;
	//    w.show();
#ifdef Q_OS_MAC
	QString root_path = QCoreApplication::applicationDirPath() + "/../../../";
#else
	QString root_path = QCoreApplication::applicationDirPath() + "/";
#endif

	QString font_path = root_path.append(FONT_NAME);
	QFile font_res(font_path);
	if (!font_res.open(QIODevice::ReadOnly)) {
		QMessageBox msgBox;
		msgBox.setText("Can not load font file!" + font_path);
		msgBox.exec();
		font_res.close();
		return 0;
	}
	int id = QFontDatabase::addApplicationFontFromData(font_res.readAll());
	QStringList family_list = QFontDatabase::applicationFontFamilies(id);
	QFont font = QFont(family_list.at(0));
	font.setPixelSize(FONT_SIZE);
	QColor font_color(0, 0, 0);


	QList<NameInfo_t> nameList = getNameByFile("name.csv");
	if (nameList.isEmpty()) {
		QMessageBox msgBox;
		msgBox.setText(("Name is empty !"));
		msgBox.exec();
	}

	QGraphicsScene scene(0, 0, 3508, 2482);
	scene.setBackgroundBrush(QBrush(QColor(255, 255, 255)));
	QGraphicsView view(&scene);



	//QImage image(BG_SAMPLE_PATH);
	//QGraphicsPixmapItem* bgItem = new QGraphicsPixmapItem(QPixmap::fromImage(image));
	//scene.addItem(bgItem);


	for (int n = 0; n < nameList.length(); n++) {
		if (n >= BG_POS_COUNT)break;
		NameInfo_t name_info = nameList.at(n);
		QImage name_image(name_info.bg_path);
		QGraphicsPixmapItem* name_bgItem = new QGraphicsPixmapItem(QPixmap::fromImage(name_image));
		bool rot = (n == 10 || n == 11);
		name_bgItem->setRotation(rot ? -90 : 0);
		name_bgItem->setPos(BG_POS[n]);
		scene.addItem(name_bgItem);


		if (name_info.name1.isEmpty()) {
			QMessageBox msgBox;
			msgBox.setText(QString().asprintf("Name is Empty at %d", export_times + 1));
			msgBox.exec();
		}
		QString name = name_info.name1;
		for (int j = 0; j < 2; j++) {
			int pos_idx = n * 2 + j;
			int xscale, yscale;

			xscale = yscale = (j == 1) ? -1 : 1;
			QPointF pos = NAME_POS[pos_idx];



			QGraphicsTextItem* textItem1 = new QGraphicsTextItem(name.mid(0, 1));
			textItem1->setFont(font);
			textItem1->setDefaultTextColor(font_color);
			if (!rot) {
				textItem1->setPos(pos);
				textItem1->setTransform(QTransform::fromScale(xscale, yscale));
			}
			else {
				textItem1->setPos(pos);
				textItem1->setRotation(-90);
				textItem1->setTransform(QTransform::fromScale(xscale, yscale));
			}



			scene.addItem(textItem1);

			QGraphicsTextItem* textItem2 = new QGraphicsTextItem(name.mid(1, 1));
			textItem2->setFont(font);
			textItem2->setDefaultTextColor(font_color);
			if (!rot) {

				textItem2->setPos(pos.x(), pos.y() + 145 * xscale);
				textItem2->setTransform(QTransform::fromScale(xscale, yscale));
			}
			else {
				textItem2->setPos(pos.x() + 145 * xscale, pos.y());
				textItem2->setRotation(-90);
				textItem2->setTransform(QTransform::fromScale(xscale, yscale));
			}

			scene.addItem(textItem2);

			QGraphicsTextItem* textItem3 = new QGraphicsTextItem(name.mid(2, 1));
			textItem3->setFont(font);
			textItem3->setDefaultTextColor(font_color);
			if (!rot) {
				textItem3->setPos(pos.x(), pos.y() + 290 * xscale);
				textItem3->setTransform(QTransform::fromScale(xscale, yscale));
			}
			else {
				textItem3->setPos(pos.x() + 290 * xscale, pos.y());
				textItem3->setRotation(-90);
				textItem3->setTransform(QTransform::fromScale(xscale, yscale));
			}

			scene.addItem(textItem3);

		}


	
	}
	QString file_name = QString("%1_.jpg").arg(QString::number(export_times + 1));
	saveToImage(file_name, &scene);
	export_times++;

	/*QMessageBox msgBox;
	msgBox.setText(QString().asprintf("Export Finshed. Count:%d", export_times));
	msgBox.exec();*/



	font_res.close();
	return a.exec();
}