#ifndef _H_WHEELMODEL_H_
#define _H_WHEELMODEL_H_

#include "IWheelModel.h"
#include "ICompareModel.h"

class WheelModel : public IWheelModel {
public:
	WheelModel();
	~WheelModel();

	void readJson(class QJsonObject *);
	void saveJson(class QJsonObject *);

	virtual bool initTmpImage(const QString&);
	virtual bool initComModel(const QString&);
	virtual bool reloadTmpImage(const QString&);

	virtual QString getModelID() const;
	virtual void setModelID(QString id) ;
	virtual void clearCount();
	virtual void increCount(int n /*= 1*/);
	virtual int getCount() const;
	virtual double getDiameter() const ;
	virtual void setDiameter(double dValue);
	virtual int getChannel() const;
	virtual void setChannel(int nValue);
	virtual double getThickness() const;
	virtual void setThickness(double dValue) ;
	virtual bool getDetectState() const;
	virtual void setDetectState(bool bValue);	
	virtual void setRepeatNum(int nValue);
	virtual int getRepeatNum() const;
	virtual QString getPicPath() const;
	virtual void setPicPath(QString str);

	virtual ICompareModel *getImageComModel() const;
	virtual void setImageComModel(ICompareModel *);
	virtual TempImage *getTempImage() const;
	virtual int getImageModel()const;
	virtual int getImgCount() const;
	virtual bool getAddTrainFlag()const { return m_bAddTrain; }
	virtual void setTrainFlag(bool bFlag) { m_bAddTrain = bFlag; }
public:
	QString m_strModelID;
	double m_dHeight;      //mm 厚度
	double m_dDiameter;    //mm 直径
	int m_passageway;
	int m_nRotate;
	bool m_bAddTrain{ true };//是否加入训练  默认true 加入训练 / false 不加入训练
	QString m_strPath;
	
	bool bDetect;
	int m_nCount;
	class ChannelInfo *m_pChannelInfo;
	class TempImage *m_pTempImage;
	class ICompareModel *m_pDetectModel;
};

#endif