QBinaryTranslate/TranslateUI/extract_basic.h

#pragma once

#include "common.h"
#include <QHash>

class SizeProvider;
class RuleMatch;
class ScopeFieldsSetter;
class DataAccessContext;


/// <summary>
/// <20><><EFBFBD>뵥Ԫ<EBB5A5><D4AA><EFBFBD><EFBFBD><EFBFBD>ӿ<EFBFBD>
/// </summary>
class ExtractUnit : public Serializable {
public:

	/// <summary>
	/// <20><>Ԫ<EFBFBD><D4AA><EFBFBD><EFBFBD>
	/// </summary>
	/// <returns></returns>
	virtual QString name() const = 0;

	/// <summary>
	/// ע<><D7A2><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
	/// </summary>
	/// <param name="inst"></param>
	virtual void registSubField(std::shared_ptr<ScopeFieldsSetter> inst) = 0;

	/// <summary>
	/// 
	/// </summary>
	/// <returns></returns>
	virtual DataType outType() const = 0;
	/// <summary>
	/// <20><><EFBFBD><EFBFBD>ƫ<EFBFBD><C6AB><EFBFBD>ֽ<EFBFBD><D6BD><EFBFBD><EFBFBD><EFBFBD>
	/// </summary>
	/// <param name="bytes"></param>
	virtual bool setOffsetSpan(int bytes) = 0;
	/// <summary>
	/// <20><><EFBFBD><EFBFBD>һ<EFBFBD><D2BB><EFBFBD><EFBFBD>Ԫ<EFBFBD>ֽ<EFBFBD>ƫ<EFBFBD><C6AB><EFBFBD>ֽ<EFBFBD><D6BD><EFBFBD><EFBFBD><EFBFBD>
	/// </summary>
	/// <returns></returns>
	virtual int offsetSpan() const = 0;
	/// <summary>
	/// <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>õ<EFBFBD><C3B5>ֽ<EFBFBD><D6BD><EFBFBD><EFBFBD><EFBFBD>
	/// </summary>
	/// <returns></returns>
	virtual int countWithin() const = 0;

	/// <summary>
	/// ִ<>н<EFBFBD><D0BD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
	/// </summary>
	/// <param name="bytes"></param>
	/// <returns></returns>
	virtual void parse(const QByteArray& bytes, std::shared_ptr<DataAccessContext> out) const = 0;
};

namespace extract {
	class AbstractExtractor : public ExtractUnit {
	public:
		struct __Private {
			QString name_store = "";
			DataType type_store = DataType::Integer;
			int byte_offset = 0, byte_count = 1;
		} _abs_data;

		AbstractExtractor(const QString& name, DataType data);
		virtual bool setCountWithin(int bytes);

		// ExtractUnit ===========================
		QString name() const override; 
		void registSubField(std::shared_ptr<ScopeFieldsSetter> inst) override;
		virtual DataType outType() const;
		/// <summary>
		/// <20><><EFBFBD><EFBFBD>ƫ<EFBFBD><C6AB><EFBFBD>ֽ<EFBFBD><D6BD><EFBFBD><EFBFBD><EFBFBD>
		/// </summary>
		/// <param name="bytes"></param>
		virtual bool setOffsetSpan(int bytes);
		int offsetSpan() const override;
		int countWithin() const override;

		// Serializable ==============================
		void loadFrom(std::shared_ptr<TranslateBasic> core, const QJsonObject& obj) override;
		void saveTo(QJsonObject& obj) const override;
	};

	/// <summary>
	/// ת<><D7AA>Դ<EFBFBD><D4B4>ֵδ16<31><36><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ֵ<EFBFBD><D6B5>Ĭ<EFBFBD>Ϸָʽ
	/// </summary>
	class BytesAsHex : public AbstractExtractor {
	public:
		BytesAsHex();

		void parse(const QByteArray& bytes, std::shared_ptr<DataAccessContext> out) const override;
		std::shared_ptr<Serializable> newDefault() const override;
	};

	/// <summary>
	/// ת<><D7AA>Դ<EFBFBD><D4B4>ֵΪλ<CEAA><CEBB><EFBFBD><EFBFBD>
	/// </summary>
	class BytesAsBitCombine : public AbstractExtractor {
	public:
		struct __Private {
			QHash<int, QString> _switch_options;
		} _combine;

		BytesAsBitCombine();

		bool setSwitchOption(int bit_index, const QString& keyword);
		QHash<int, QString> switchOptions() const;
		virtual void clearOptions();

		// ExtractUnit ============================
		void parse(const QByteArray& bytes, std::shared_ptr<DataAccessContext> out) const override;

		// Serializable ==============================
		void loadFrom(std::shared_ptr<TranslateBasic> core, const QJsonObject& obj) override;
		void saveTo(QJsonObject& obj) const override;
		std::shared_ptr<Serializable> newDefault() const override;
	};

	/// <summary>
	/// ת<><D7AA>Դ<EFBFBD><D4B4><EFBFBD><EFBFBD>Ϊ<EFBFBD><CEAA><EFBFBD><EFBFBD>
	/// </summary>
	class BytesAsInteger : public AbstractExtractor {
	public:
		BytesAsInteger();

		// ExtractUnit ============================
		bool setCountWithin(int bytes) override;
		void parse(const QByteArray& bytes, std::shared_ptr<DataAccessContext> out) const override;

		// Serializable ==============================
		std::shared_ptr<Serializable> newDefault() const override;
	};

	/// <summary>
	/// ת<><D7AA>Դ<EFBFBD><D4B4><EFBFBD><EFBFBD>Ϊ<EFBFBD><CEAA><EFBFBD><EFBFBD>
	/// </summary>
	class BytesAsUnsigned : public AbstractExtractor {
	public:
		BytesAsUnsigned();

		// ExtractUnit ============================
		bool setCountWithin(int bytes) override;
		void parse(const QByteArray& bytes, std::shared_ptr<DataAccessContext> out) const override;

		// Serializable ==============================
		std::shared_ptr<Serializable> newDefault() const override;
	};

	/// <summary>
	/// ת<><D7AA>Դ<EFBFBD><D4B4><EFBFBD><EFBFBD>Ϊ<EFBFBD>ַ<EFBFBD><D6B7><EFBFBD>
	/// </summary>
	class BytesAsString : public AbstractExtractor {
	public:
		struct __Private {
			QTextCodec* _conv_with = nullptr;
		} _strings;

	public:
		BytesAsString();

		void setStrCodec(QTextCodec* ins);
		QString codecName() const;

		// ExtractUnit ============================
		void parse(const QByteArray& bytes, std::shared_ptr<DataAccessContext> out) const override;

		// Serializable ==============================
		void loadFrom(std::shared_ptr<TranslateBasic> core, const QJsonObject& obj) override;
		void saveTo(QJsonObject& obj) const override;
		std::shared_ptr<Serializable> newDefault() const override;
	};

	class BytesAsFloat : public AbstractExtractor {
	public:
		BytesAsFloat();

		void parse(const QByteArray& bytes, std::shared_ptr<DataAccessContext> out) const override;
		std::shared_ptr<Serializable> newDefault() const override;
	};

	class BytesAsDouble : public AbstractExtractor {
	public:
		BytesAsDouble();

		void parse(const QByteArray& bytes, std::shared_ptr<DataAccessContext> out) const override;
		std::shared_ptr<Serializable> newDefault() const override;
	};


	class BytesAsList : public ExtractUnit {
	public:
		struct __Private {
			std::shared_ptr<ExtractUnit> _bind_unit = nullptr;
			int _bytes_offset = 0;
			std::shared_ptr<SizeProvider> _bind_size_v = nullptr;
		} _list;

	public:
		bool setOffsetSpan(int value);
		std::shared_ptr<ExtractUnit> elementRule() const;
		bool appendElementRule(std::shared_ptr<ExtractUnit> u);
		std::shared_ptr<SizeProvider> sizeProvider() const;
		bool setSizeProvider(std::shared_ptr<SizeProvider> ins);

	public:
		QString name() const override;
		void registSubField(std::shared_ptr<ScopeFieldsSetter> inst) override;
		DataType outType() const override;
		int offsetSpan() const override;
		int countWithin() const override;

		void loadFrom(std::shared_ptr<TranslateBasic> core, const QJsonObject& obj) override;
		void saveTo(QJsonObject& obj) const override;

		void parse(const QByteArray& bytes, std::shared_ptr<DataAccessContext> out) const override;
		std::shared_ptr<Serializable> newDefault() const override;

	};

	class BytesAsUnion : public ExtractUnit {
	public:
		struct __Private {
			int byte_offset = 0, byte_count = 1;
			QList<std::shared_ptr<RuleMatch>> _rule_list;
		} _union;

	public:
		bool setCountWithin(int bytes);

		void clearRules();
		QList<std::shared_ptr<RuleMatch>> elementRules() const;
		bool appendElementRule(std::shared_ptr<RuleMatch> u);

	public:
		QString name() const override;
		void registSubField(std::shared_ptr<ScopeFieldsSetter> inst) override;
		DataType outType() const override;
		bool setOffsetSpan(int value);
		int offsetSpan() const override;
		int countWithin() const override;

		void parse(const QByteArray& bytes, std::shared_ptr<DataAccessContext> out) const override;
		void loadFrom(std::shared_ptr<TranslateBasic> core, const QJsonObject& obj) override;
		void saveTo(QJsonObject& obj) const override;

		std::shared_ptr<Serializable> newDefault() const override;
	};


	class BytesAsRuleSet : public ExtractUnit {
	public:
		struct __Private {
			int _byte_offset = 0, _byte_count = 1;
			QString _name_store;
			QList<std::pair<QString, std::shared_ptr<ExtractUnit>>> sub_units;

			QList<QString> fieldNames() const;
			std::shared_ptr<ExtractUnit> operator[](const QString &field) const;


			std::pair<QString, std::shared_ptr<ExtractUnit>> operator[](int index) const;
			void replace(int index, std::pair<QString, std::shared_ptr<ExtractUnit>> inst);
			void removeAt(int index);
			__Private& append(const QString& nm, std::shared_ptr<ExtractUnit> u);
		}_bind;

		void setCustomName(const QString& name);
		QString customName() const;

	public:
		QString name() const override;
		void registSubField(std::shared_ptr<ScopeFieldsSetter> inst) override;
		DataType outType() const override;
		bool setOffsetSpan(int bytes) override;
		int offsetSpan() const override;
		int countWithin() const override;
		std::shared_ptr<Serializable> newDefault() const override;

		void parse(const QByteArray& bytes, std::shared_ptr<DataAccessContext> out) const override;

		void loadFrom(std::shared_ptr<TranslateBasic> core, const QJsonObject& obj) override;
		void saveTo(QJsonObject& obj) const override;

	};
}