Retro68/Rez/ResourceCompiler.h

#ifndef RESOURCECOMPILER_H
#define RESOURCECOMPILER_H

#include "Expression.h"
#include "ResourceDefinitions.h"

class Field;
class RezWorld;
class Diagnostic;

class Subscripts
{
	std::vector<int> subscripts;
public:
	Subscripts();
	~Subscripts();

	void addSubscript(int x);
	void popSubscript();
	bool operator<(const Subscripts& other) const;
	bool empty() const { return subscripts.empty(); }
};

class BinaryOutput
{
protected:
	int currentOffset;
	std::vector<unsigned char> data;
	std::vector<unsigned char> prePassData;
	bool verboseFlag;
	bool prePass;
public:
	BinaryOutput();
	void reset(bool prePass);

	std::string resourceData();

	void reserve(int nBits) { write(nBits, 0); }
	void write(int nBits, int value);
	int tell() { return currentOffset; }

	int peek(int bitPos, int size);

	bool isPrePass() { return prePass; }
};

class ResourceCompiler : public BinaryOutput
{
	RezWorld& world;
	TypeDefinitionPtr typeDefinition;
	CompoundExprPtr body;
	std::map<std::pair<std::string, Subscripts>, ExprPtr> labelValues;
	std::map<std::pair<std::string, Subscripts>, int> arrayCounts;
	std::map<std::string, int> curArrayIndices;
	Field* currentField;
	Subscripts currentSubscripts;

	void beginArrayScope(std::string& arrayName, int index);
public:
	ResourceCompiler(RezWorld& world, TypeDefinitionPtr type, CompoundExprPtr body, bool verboseFlag);

	ExprPtr lookupIdentifier(std::string name, const Subscripts& sub = Subscripts());

	void defineLabel(const std::string& name);
	void compile();

	int getArrayCount(const std::string& arrayName);
	int getArrayIndex(const std::string& arrayName);


	class FieldScope
	{
		ResourceCompiler *compiler;
	public:
		FieldScope(ResourceCompiler* compiler, Field *field)
			: compiler(compiler) { compiler->currentField = field; }
		~FieldScope() { compiler->currentField = nullptr; }
	};

	class ArrayScope
	{
		ResourceCompiler *compiler;
	public:
		ArrayScope(ResourceCompiler* compiler, std::string& arrayName, int index)
			: compiler(compiler) { compiler->beginArrayScope(arrayName, index); }
		~ArrayScope() { compiler->currentSubscripts.popSubscript(); }
	};

	void problem(Diagnostic d);
};



#endif // RESOURCECOMPILER_H