#pragma once
#include <vector>
#include <map>
#include <string>
#include <mutex>
#include <sqlite3/sqlite3.h>

#define MOBILE_FACE_FEATURE_FLOAT_LEN 512
#define MOBILE_FACE_FEATURE_BYTE_LEN (MOBILE_FACE_FEATURE_FLOAT_LEN*sizeof(float))

#define MAX_FEATURES_FOR_ONE_NAME 1                                //一个人限制录入1条数据
#define MAX_FEATURE_FOR_GROUP   2000                            //每组限制5000条数据
#define MAX_GROUP_NUM			500								//最多只能添加500组

#define FS_DB_QUERY_SIZE_BY_NAME "select count(*) from table_features where Name=?;"
#define FS_DB_QUERY_SIZE_OF_NAMES "select count(distinct Name) from table_features;"
#define FS_DB_QUERY_SIZE_INALL "select count(*) from table_features;"

enum DB_ERR_CODE
{	
	DB_ERR_BIND_SQL_DATA = -100,//绑定sql数据错误
	DB_ERR_SQL,					//sql错误
	DB_ERR_SQL_EXE,				//执行SQL错误
	DB_ERR_FEATURE_LEN,			//人脸特征长度错误
	DB_ERR_NAME_FEATURE_EXIST,	//该名字已经注册特征
	DB_ERR_GROUP_FEATURE_FULL,	//分组内注册人脸数达到上限
	DB_ERR_GROUP_FULL,			//创建的分组达到上限
	DB_ERR_ADD_FACE_FEATURE,	//录入人脸特征失败
	DB_ERR_NON_GROUP,			//组名不存在
	DB_ERR_GROUP_REPEAT,		//分组名字重复
	DB_ERR_ADD_GROUP,			//添加分组数据失败
	DB_ERR_NO_INIT,				//人脸数据库未初始化

	DB_ERR_NO_DATA = -1,			//未查询到数据
	DB_ERR_OK = 0					//无错误

};

struct DB_GROUP_OBJ
{
	int group_id;
	std::string group_name;
	std::string create_time;
	std::string remarks;			//分组备注信息
};
struct DB_FACE_OBJ
{
	DB_GROUP_OBJ group;

	std::string vip_id;
	std::string name;
	std::string enter_time;
	unsigned char byteFeature[MOBILE_FACE_FEATURE_BYTE_LEN];
};

struct SEARCH_FACE_RESULT {
	DB_FACE_OBJ face_obj;
	float score;
};


class FaceDbMgr
{
	class AutoDeleteStmt
	{
	public:
		explicit AutoDeleteStmt(sqlite3_stmt * pstmt)
		{
			m_pStmt = pstmt;
		}
		~AutoDeleteStmt()
		{
			if (m_pStmt)
			{
				sqlite3_finalize(m_pStmt);
			}
		}

	private:
		sqlite3_stmt * m_pStmt;
	};

private:
	explicit FaceDbMgr(char * pcDbName = "DeepCamFaceService.db");
	~FaceDbMgr();

public:
	static FaceDbMgr* GetInstance();
	/*********************** 业务接口 *****************************/
	bool IsDbInit();//获取数据库是否初始化成功的状态

	std::string GetErrInfo(DB_ERR_CODE);
	DB_ERR_CODE EnterFace(const std::vector<float>& feature, std::string face_name, std::string vid, std::string group_name, std::vector<unsigned char> & imgData);
	DB_ERR_CODE FaceSearch(const std::vector<float> & feature, const std::string group_name, DB_FACE_OBJ& face_obj, std::vector<uint8_t>& img_data, float & fSimilarity, float fThreshold);
	DB_ERR_CODE DeleteFaceByName(std::string faceName, std::string group_name);
	DB_ERR_CODE DeleteAllFace(std::string group_name);
	DB_ERR_CODE AddGroup(std::string group_name, std::string remarks = "");
	DB_ERR_CODE DeleteGroup(std::string group_name);
	//查询所有的组数据
	DB_ERR_CODE QueryGroupDatas(std::vector<DB_GROUP_OBJ>& objs);
public:
	/*********************** 基础API *****************************/
	float FaceDbMgr::FeatureCompare(const float* feature1, const float* feature2);

	int FeatureDBMgrExec(const char * strSql);//执行一些不需要返回表数据的语句、只需要返回是否执行成功的结果。
	
	//返回查询表数据条数，大于0才表示有数据
	int QueryTableSize(const char * strSql);
	//返回查询表数据条数，大于0才表示有数据
	int QueryTableSizeByName(const char * strSql, const char * strPersonName);
	//查询所有的人脸特征数量
	int	QueryAllFeatureSize();
	//查询分组内所有的人脸特征数量
	int QueryGroupAllFeatureSize(int group_id);
	//通过名字查询组内注册人脸特征数量
	int QueryNameFeatureInGroupSize(std::string groupName, std::string faceName);
	//查询所有分组数量
	int	QueryAllGroupSize();
	//通过组名字查询组ID
	int QueryGroupIdByName(std::string groupName);
	int QueryGroupByName(std::string groupName, std::vector<DB_GROUP_OBJ>& objs);
	int QueryGroupByID(int group_id, std::vector<DB_GROUP_OBJ>& objs);

	//查询组内所有的特征数据
	int QueryFeatrueByGroup(DB_GROUP_OBJ& group, std::vector<DB_FACE_OBJ>& face_db);
	//通过姓名查询人脸特征库
	int QueryFeatureByName(std::string face_name, std::string group_name, std::vector<DB_FACE_OBJ>& face_db);
	//通过姓名查询人脸图片
	int QueryImageByName(std::string face_name, int group_id, std::vector<uint8_t>& img_data);

private:
	sqlite3 * m_db;
	bool m_bDbInit;
	std::mutex m_db_lock;
	std::map<DB_ERR_CODE, std::string> m_err_info;				//错误代码映射表
	//static std::map<std::string, int> m_groups;					//组ID <=> 组名 映射
	static std::vector<DB_GROUP_OBJ> m_groups;
	static std::map<int, std::vector<DB_FACE_OBJ>> m_GroupDb;	//人脸数据集   组ID <=> 单组数据集

	static FaceDbMgr* m_instance;
};