#include "frmToolsTree.h"
#include <QHeaderView>
#include <QStyleFactory>
#include <QDir>
#include <QPluginLoader>
#include "GlobalVariable.h"

QString ThemeColor = "#20B2AA";

frmToolsTree::frmToolsTree(QWidget* parent)
	: QWidget(parent)
{
	ui.setupUi(this);
	setWindowFlags(Qt::FramelessWindowHint);
	ToolsTreeWidgetInit();
}

//工具列表初始化
void frmToolsTree::ToolsTreeWidgetInit()
{
	ToolTreeWidget = new QtTreeWidget(this);
	QGridLayout* ToolGLayout = new QGridLayout(ui.ToolTreeFrame);
	ToolGLayout->setContentsMargins(0, 0, 0, 0);
	ToolGLayout->setSpacing(5);
	ToolGLayout->addWidget(ToolTreeWidget, 0, 0);
	ToolTreeWidget->setIconSize(QSize(24, 24));
	//显示虚线
	ToolTreeWidget->setStyle(QStyleFactory::create("windows"));
	//更换系统默认图标
	ToolTreeWidget->setStyleSheet("QTreeWidget::branch:has-children:!has-siblings:closed,\
	QWidget{border: 1px;border-style: solid;border-color: #d9d9d9}\
	QTreeWidget::item{color: #363636}\
	QTreeWidget::item{height: 30px}\
    QTreeView::item::selected{background-color:#96B5A8}\
	QTreeWidget::branch:closed:has-children:!has-siblings{border-style: none; border-image: none; image: url(:/res/ico/unfold.png);}\
	QTreeWidget::branch:closed:has-children:has-siblings{border-image: none; image: url(:/res/ico/unfold.png);}\
	QTreeWidget::branch:open:has-children:!has-siblings{border-image: none; image: url(:/res/ico/fold.png);}\
	QTreeWidget::branch:open:has-children:has-siblings{border-image: none; image: url(:/res/ico/fold.png);}"
	"QScrollBar:vertical{" //设置滚动条样式
	"background:#FFFFF2;"  //背景色  
	"padding-top:20px;"    //上预留位置(放置向上箭头）  
	"padding-bottom:20px;" //下预留位置(放置向下箭头）  
	"padding-left:2px;"    //左预留位置 
	"padding-right:2px;"   //右预留位置 
	"border-left:1px solid #d7d7d7;}" //左分割线  
	"QScrollBar::handle:vertical{" //滑块样式  
	"background:#dbdbdb;"  //滑块颜色  
	"border-radius:6px;"   //边角圆
	"min-height:20px;}"    //滑块最小高度
	"QScrollBar::handle:vertical:hover{" //鼠标触及滑块样式  
	"background:#d0d0d0;}" //滑块颜色  
	"QScrollBar::add-line:vertical{" //向下箭头样式  
	"background:url(:/res/ico/down.png) center no-repeat;}"
	"QScrollBar::sub-line:vertical{" //向上箭头样式
	"background:url(:/res/ico/up.png) center no-repeat;}");
	//设置字体及字体大小
	QFont font("Microsoft YaHei");
	font.setPixelSize(16.5);
	ToolTreeWidget->setFont(font);
	ToolTreeWidget->header()->setVisible(false);
	ToolTreeWidget->clear();
	this->setAcceptDrops(false);
	//工具列表
	ToolsPair toolPair;	
	//图像处理
	LoadPlugins("图像处理");  //加载图像处理	
	toolPair.first = "图像处理";
	toolPair.second = ToolNameList;
	ToolNamesVec.push_back(toolPair);
	ToolNameList.clear();
	//检测识别
	LoadPlugins("检测识别");  //加载检测识别		
	toolPair.first = "检测识别";
	toolPair.second = ToolNameList;
	ToolNamesVec.push_back(toolPair);
	ToolNameList.clear();
	//标定工具
	LoadPlugins("标定工具");  //加载标定工具	
	toolPair.first = "标定工具";
	toolPair.second = ToolNameList;
	ToolNamesVec.push_back(toolPair);
	ToolNameList.clear();
	//对位工具
	LoadPlugins("对位工具");  //加载对位工具		
	toolPair.first = "对位工具";
	toolPair.second = ToolNameList;
	ToolNamesVec.push_back(toolPair);
	ToolNameList.clear();
	//几何工具
	LoadPlugins("几何工具");  //加载几何工具
	toolPair.first = "几何工具";
	toolPair.second = ToolNameList;
	ToolNamesVec.push_back(toolPair);
	ToolNameList.clear();
	//几何测量
	LoadPlugins("几何测量");  //加载几何测量	
	toolPair.first = "几何测量";
	toolPair.second = ToolNameList;
	ToolNamesVec.push_back(toolPair);
	ToolNameList.clear();
	//三维检测	
	LoadPlugins("三维检测");  //加载三维检测
	toolPair.first = "三维检测";
	toolPair.second = ToolNameList;
	ToolNamesVec.push_back(toolPair);
	ToolNameList.clear();
	//逻辑工具
	LoadPlugins("逻辑工具");  //加载逻辑工具	
	toolPair.first = "逻辑工具";
	toolPair.second = ToolNameList;
	ToolNamesVec.push_back(toolPair);
	ToolNameList.clear();
	//通讯工具
	LoadPlugins("通讯工具");  //加载通讯工具	
	toolPair.first = "通讯工具";
	toolPair.second = ToolNameList;
	ToolNamesVec.push_back(toolPair);
	ToolNameList.clear();
	//系统工具
	LoadPlugins("系统工具");  //加载系统工具	
	toolPair.first = "系统工具";
	toolPair.second = ToolNameList;
	ToolNamesVec.push_back(toolPair);
	ToolNameList.clear();
	//循环处理图标	
	for (int i = 0; i < ToolNamesVec.size(); i++)
	{
		QString toolBoxName = ToolNamesVec[i].first;
		QStringList toolList = QStringList::fromStdList(ToolNamesVec[i].second);
		QTreeWidgetItem* itemParent = new QTreeWidgetItem(ToolTreeWidget);
		itemParent->setText(0, toolBoxName);
		itemParent->setIcon(0, QIcon(":/res/ico/tool.ico"));
		itemParent->setFlags(Qt::NoItemFlags | Qt::ItemIsEnabled);
		for (int j = 0; j < toolList.count(); j++)
		{
			QTreeWidgetItem* treeItem = new QTreeWidgetItem(itemParent, QStringList(toolList[j]));
			QPixmap pixmap = GetIcon(GetIconName(toolList[j]));
			treeItem->setIcon(0, pixmap);
		}
	}
}

/*
 * @function PluginsApp::LoadPlugins 加载插件、插件放在Plugins文件夹下
 * @return 返回插件的个数
 */
int frmToolsTree::LoadPlugins(QString str)
{
	int count = 0;
	QDir pluginsDir = QDir(qApp->applicationDirPath());
	if (!pluginsDir.cd("Plugins")) return -1;
	QStringList filters;
	filters << "*.dll"; //可以多种格式或者单个格式
	pluginsDir.setFilter(QDir::Files | QDir::NoSymLinks); //设置类型过滤器，只为文件格式
	pluginsDir.setNameFilters(filters);  //设置文件名称过滤器
	foreach(QString fileName, pluginsDir.entryList())
	{
		QPluginLoader pluginLoader(pluginsDir.absoluteFilePath(fileName));
		QObject* plugin = pluginLoader.instance();
		if (plugin)
		{
			auto treeInterface = qobject_cast<MainInterface*>(plugin);
			if (treeInterface)
			{
				++count;
				//加载插件后生成Tree
				PopulateTree(str, plugin, treeInterface);
			}
		}
	}
	return count;
}

/*
 * @function PluginsApp::PopulateTree 根据插件生成Tree
 * @param pluginInterface 插件
 * @param i 插件实现的接口
 */
void frmToolsTree::PopulateTree(QString str, QObject* pluginInterface, MainInterface* i)
{
	try
	{
		if (i->information() == str)
		{
			ToolNameList.push_back(i->name());
		}
	}
	catch(...){}	
}

// 获取图标
QPixmap frmToolsTree::GetIcon(QString fileName)
{
	return IconHelper::Instance()->getPixmap(fileName);
}

// 获取图标路径
QString frmToolsTree::GetIconName(QString Name)
{
	QString IconName;
	if (Name.contains("获取图像")) IconName = ":/res/ico/image_source.png";
	if (Name.contains("图像显示")) IconName = ":/res/ico/image_view.ico";
	if (Name.contains("导出图像")) IconName = ":/res/ico/export_image.png";	
	if (Name.contains("预处理")) IconName = ":/res/ico/morphology.png";
	if (Name.contains("图像拼接")) IconName = ":/res/ico/image_splice.png";	
	if (Name.contains("图像修复")) IconName = ":/res/ico/repair.png";	
	if (Name.contains("图像细化")) IconName = ":/res/ico/skeleton.png";
	if (Name.contains("图像翻转")) IconName = ":/res/ico/flip.png";
	if (Name.contains("图像旋转")) IconName = ":/res/ico/rotate.png";
	if (Name.contains("透视变换")) IconName = ":/res/ico/perspective.png";	
	if (Name.contains("裁切图像")) IconName = ":/res/ico/crop.png";
	if (Name.contains("创建ROI")) IconName = ":/res/ico/roi.png";
	if (Name.contains("条形码识别")) IconName = ":/res/ico/barcode.png";
	if (Name.contains("斑点分析")) IconName = ":/res/ico/blob.png";
	if (Name.contains("二维码识别")) IconName = ":/res/ico/qrcode.png";
	if (Name.contains("字符识别")) IconName = ":/res/ico/ocr.png";
	if (Name.contains("分类器")) IconName = ":/res/ico/classifier.png";
	if (Name.contains("颜色识别")) IconName = ":/res/ico/color_r.png";	
	if (Name.contains("亮度检测")) IconName = ":/res/ico/brightness.png";
	if (Name.contains("图像清晰度")) IconName = ":/res/ico/clarity.png";		
	if (Name.contains("轮廓特征选择")) IconName = ":/res/ico/shape.png";
	if (Name.contains("N点标定")) IconName = ":/res/ico/ert_calib.png";
	if (Name.contains("畸变标定")) IconName = ":/res/ico/distortion_calib.png";
	if (Name.contains("测量标定")) IconName = ":/res/ico/measure_calib.png";
	if (Name.contains("灰度匹配")) IconName = ":/res/ico/match.png";	
	if (Name.contains("形状匹配")) IconName = ":/res/ico/shape_match.png";
	if (Name.contains("目标跟踪")) IconName = ":/res/ico/track.png";
	if (Name.contains("线性计算")) IconName = ":/res/ico/affine.png";
	if (Name.contains("寻找圆")) IconName = ":/res/ico/find_circle.png";
	if (Name.contains("寻找直线")) IconName = ":/res/ico/find_line.png";
	if (Name.contains("拟合圆")) IconName = ":/res/ico/fit_circle.png";
	if (Name.contains("拟合椭圆")) IconName = ":/res/ico/fit_ellipse.png";
	if (Name.contains("拟合直线")) IconName = ":/res/ico/fit_line.png";
	if (Name.contains("获取边界点")) IconName = ":/res/ico/border_point.png";
	if (Name.contains("线圆交点")) IconName = ":/res/ico/line_circle.png";
	if (Name.contains("点+点")) IconName = ":/res/ico/point_point.png";
	if (Name.contains("点+线")) IconName = ":/res/ico/point_l.png";
	if (Name.contains("线线交点")) IconName = ":/res/ico/intersection.png";
	if (Name.contains("查找圆缺角")) IconName = ":/res/ico/rounded_c.png";	
	if (Name.contains("边缘宽度测量")) IconName = ":/res/ico/edge.png";
	if (Name.contains("拟合平面")) IconName = ":/res/ico/flatness.png";	
	if (Name.contains("扩展库")) IconName = ":/res/ico/extension_library.png";
	if (Name.contains("跳转语句")) IconName = ":/res/ico/goto.png";
	if (Name.contains("判断语句")) IconName = ":/res/ico/logic_judge.png";
	if (Name.contains("结束语句")) IconName = ":/res/ico/end.png";
	if (Name.contains("脚本编辑")) IconName = ":/res/ico/script_edit.png";
	if (Name.contains("TCP/IP服务器")) IconName = ":/res/ico/server.png";	
	if (Name.contains("TCP/IP客户端")) IconName = ":/res/ico/client.png";
	if (Name.contains("PLC通信")) IconName = ":/res/ico/plc_communicate.png";
	if (Name.contains("串口通信")) IconName = ":/res/ico/serial_port.png";
	if (Name.contains("通用I/O")) IconName = ":/res/ico/general_io.png";
	if (Name.contains("延时")) IconName = ":/res/ico/delay.png";	
	if (Name.contains("导出CSV")) IconName = ":/res/ico/export_csv.png";
	return IconName;
}