//////////////////////////////////////////////////////////////////////////
// Name:	    chars_segment Header
// Version:		1.0
// Date:	    2014-09-19
// Author:	    liuruoze
// Copyright:   liuruoze
// Reference:	Mastering OpenCV with Practical Computer Vision Projects
// Reference:	CSDN Bloger taotao1233
// Desciption:
// Defines CCharsSegment
//////////////////////////////////////////////////////////////////////////
#ifndef __CHARS_SEGMENT_H__
#define __CHARS_SEGMENT_H__

#include "core_func.h"

/*! \namespace easypr
Namespace where all the C++ EasyPR functionality resides
*/
namespace easypr {

class CCharsSegment {
 public:
  CCharsSegment();

  //! 字符分割
  int charsSegment(Mat, std::vector<Mat>&);

  //! 字符尺寸验证
  bool verifyCharSizes(Mat r);

  //! 字符预处理
  Mat preprocessChar(Mat in);

  //! 根据特殊车牌来构造猜测中文字符的位置和大小
  Rect GetChineseRect(const Rect rectSpe);

  //! 找出指示城市的字符的Rect，例如苏A7003X，就是A的位置
  int GetSpecificRect(const std::vector<Rect>& vecRect);

  //! 这个函数做两个事情
  //  1.把特殊字符Rect左边的全部Rect去掉，后面再重建中文字符的位置。
  //  2.从特殊字符Rect开始，依次选择6个Rect，多余的舍去。
  int RebuildRect(const std::vector<Rect>& vecRect, std::vector<Rect>& outRect,
                  int specIndex);

  //! 将Rect按位置从左到右进行排序
  int SortRect(const std::vector<Rect>& vecRect, std::vector<Rect>& out);

  //! 设置变量
  inline void setLiuDingSize(int param) { m_LiuDingSize = param; }
  inline void setColorThreshold(int param) { m_ColorThreshold = param; }

  inline void setBluePercent(float param) { m_BluePercent = param; }
  inline float getBluePercent() const { return m_BluePercent; }
  inline void setWhitePercent(float param) { m_WhitePercent = param; }
  inline float getWhitePercent() const { return m_WhitePercent; }

  //! 是否开启调试模式常量，默认0代表关闭
  static const int DEFAULT_DEBUG = 1;

  //! preprocessChar所用常量
  static const int CHAR_SIZE = 20;
  static const int HORIZONTAL = 1;
  static const int VERTICAL = 0;

  //! preprocessChar所用常量
  static const int DEFAULT_LIUDING_SIZE = 7;
  static const int DEFAULT_MAT_WIDTH = 136;
  static const int DEFAULT_COLORTHRESHOLD = 150;

  //! 是否开启调试模式
  inline void setDebug(int param) { m_debug = param; }

  //! 获取调试模式状态
  inline int getDebug() { return m_debug; }

 private:
  //！柳钉判断参数
  int m_LiuDingSize;

  //！车牌大小参数
  int m_theMatWidth;

  //！车牌颜色判断参数
  int m_ColorThreshold;
  float m_BluePercent;
  float m_WhitePercent;

  //! 是否开启调试模式，0关闭，非0开启
  int m_debug;
};

} /* \namespace easypr  */

#endif /* endif __CHARS_SEGMENT_H__ */