ModelInfo.h

#pragma once
#include <memory>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <cstdint>
#include "ninja.h"

enum ModelFormat
{
	ModelFormat_Invalid,
	ModelFormat_Basic,
	ModelFormat_Chunk,
	ModelFormat_SA2B
};

class ModelInfo
{
public:
	struct Metadata { uint32_t size; const uint8_t *data; };

	ModelInfo(const char *filename);
	ModelInfo(const wchar_t *filename);
	ModelInfo(const std::string &filename);
	ModelInfo(const std::wstring &filename);
	ModelInfo(std::istream &stream);

	ModelFormat getformat();
	NJS_OBJECT *getmodel();
	const std::string &getauthor();
	const std::string &gettool();
	const std::string &getdescription();
	const uint8_t *getmetadata(uint32_t identifier, uint32_t &size);
	const std::string &getlabel(void *data);
	void *getdata(const std::string &label);
	const std::unordered_map<std::string, void*>* getlabels() const;
	const std::list<std::string> &getanimations();
	const std::list<std::string> &getmorphs();

private:
	static const uint64_t SA1MDL = 0x4C444D314153u;
	static const uint64_t SA2MDL = 0x4C444D324153u;
	static const uint64_t SA2BMDL = 0x4C444D42324153u;
	static const uint64_t FormatMask = 0xFFFFFFFFFFFFFFu;
	static const uint8_t CurrentVersion = 3;
	static const int headersize = 0x10;

	ModelFormat format;
	NJS_OBJECT *model;
	std::string author, tool, description;
	std::unordered_map<uint32_t, Metadata> metadata;
	std::unordered_map<void *, std::string> labels1;
	std::unordered_map<std::string, void *> labels2;
	std::vector<std::shared_ptr<void>>allocatedmem;
	std::unordered_set<void *>fixedpointers;
	std::list<std::string> animations, morphs;

	enum ChunkTypes : uint32_t
	{
		ChunkTypes_Label = 0x4C42414C,
		ChunkTypes_Animation = 0x4D494E41,
		ChunkTypes_Morph = 0x46524F4D,
		ChunkTypes_Author = 0x48545541,
		ChunkTypes_Tool = 0x4C4F4F54,
		ChunkTypes_Description = 0x43534544,
		ChunkTypes_Texture = 0x584554,
		ChunkTypes_End = 0x444E45
	};

	void fixbasicmodelpointers(NJS_MODEL *model, intptr_t base);
	void fixchunkmodelpointers(NJS_CNK_MODEL *model, intptr_t base);
	void fixsa2bmodelpointers(SA2B_Model *model, intptr_t base);
	void fixobjectpointers(NJS_OBJECT *object, intptr_t base);
	void init(std::istream &stream);
};