#pragma once

#ifndef FALLOUT_SCRIPT_H
#define FALLOUT_SCRIPT_H

#include "StartupCode.h"
#include "ProcTable.h"
#include "Namespace.h"
#include "Node.h"


// Constants

// CFalloutScript
class CFalloutScript : public CObject {
public:
	CFalloutScript();
	virtual ~CFalloutScript();

public:
	virtual void Serialize(CArchive& ar);

	void Dump(CArchive& ar);

	void InitDefinitions();
	void ProcessCode();

	void StoreSource(CArchive& ar);
	void StoreTree(CArchive& ar);

private:
	enum Assoc {
		NON_ASSOC,
		LEFT_ASSOC,
		RIGHT_ASSOC,
	};

private:
	void ExtractCodeElements(COpcodeArray& Source, COpcodeArray& Destination, WORD wDelimeter, int nSizeOfCodeItem, LPCTSTR lpszErrorMessage, BOOL (CFalloutScript::*pCheckFunc)(WORD, INT_PTR));
	BOOL CheckExportVarCode(WORD wOperator, INT_PTR nIndex);
	BOOL CheckSetExportedVarValueCode(WORD wOperator, INT_PTR nIndex);
	BOOL CheckExportProcCode(WORD wOperator, INT_PTR nIndex);

    INT_PTR GetIndexOfProc(const char* lpszName);
	INT_PTR GetIndexOfProc(ULONG ulNameOffset);
	INT_PTR GetIndexOfExportedVariable(ULONG ulNameOffset);

	void SetExternalVariable(ULONG ulNameOffset);
	void TryRenameGlobalVariables();
	void TryRenameImportedVariables();

    INT_PTR NextNodeIndex( CNodeArray& NodeArray, INT_PTR nCurrentIndex, INT_PTR nSteep);
	BOOL CheckSequenceOfNodes(CNodeArray& NodeArray, INT_PTR nStartIndex, const WORD wSequence[], INT_PTR nSequenceLen);
	BOOL RemoveSequenceOfNodes(CNodeArray& NodeArray,INT_PTR nStartIndex, INT_PTR nCount, const WORD wSequence[], INT_PTR nSequenceLen);

	void InitialReduce();
	void BuildTree(CNodeArray& NodeArray);
	void ExtractAndReduceCondition(CNodeArray& Source, CNodeArray& Destination, INT_PTR nStartIndex);
	void SetBordersOfBlocks(CNodeArray& NodeArray);
	ULONG BuildTreeBranch(CNodeArray& NodeArray, ULONG nStartIndex, ULONG ulEndOffset);
	void ReduceConditionalExpressions(CNodeArray& NodeArray);
	BOOL IsOmittetArgsAllowed(WORD wOpcode);

	void StoreDefinitions(CArchive& ar);
	void StoreDeclarations(CArchive& ar);

    CString GetSource( CNode& node, BOOL bLabel, ULONG ulNumArgs);
    CString GetSource( CNode& node, BOOL bLabel, ULONG ulNumArgs, ULONG aulProcArg[], ULONG ulProcArgCount);
	bool ArgNeedParens(const CNode& node, const CNode& argument, CFalloutScript::Assoc assoc = CFalloutScript::NON_ASSOC);
	CString GetIndentString(INT_PTR nLevel);

	int GetPriority(WORD wOperator);
	Assoc GetAssociation(WORD wOperator);

private:
	// CDefObject
	class CDefObject {
	public:
		enum ObjectType {
			OBJECT_VARIABLE,
			OBJECT_PROCEDURE
		};

	public:
		CDefObject(ObjectType type = OBJECT_VARIABLE, ULONG ulAttributes = 0, ULONG ulObjectData = 0);

	public:
		ObjectType m_ObjectType;
		ULONG m_ulAttributes;

		union {
			ULONG m_ulVarValue;
			ULONG m_ulProcIndex;
		};
	};

private:
	// CMapULongToDefObject
	typedef CMap<ULONG, ULONG, CDefObject, CDefObject&> CMapULongToDefObject;

	CStartupCode m_StartupCode;
	CProcTable   m_ProcTable;
	CNamespace   m_Namespace;
	CNamespace   m_Stringspace;

	COpcodeArray m_GlobalVar;
	COpcodeArray m_ExportedVar;
	COpcodeArray m_ExportedVarValue;
	COpcodeArray m_ExportedProc;

	CArrayOfNodeArray m_ProcBodies;
	CArrayOfNodeArray m_Conditions;

	CMapULongToDefObject m_Definitions;
	CStringArray m_GlobalVarsNames;
};

#endif