#include "le_entry_table_entry.hpp"

#include "parse.hpp"
#include "output.hpp"

#include <boost/fusion/adapted/struct.hpp>

typedef boost::variant<binparse::Offset16, binparse::Offset32> EntryT;

BOOST_FUSION_ADAPT_STRUCT(
	le::EntryPoint,
	(binparse::Value8, flags)
	(EntryT, entry)
)

BOOST_FUSION_ADAPT_STRUCT(
	le::EntryTableEntry,
	(binparse::Value8, nr_entries)
	(binparse::Value8, flags)
	(binparse::Value16, object_index)
	(std::vector<le::EntryPoint>, entries)
)

namespace le {

EntryTableEntry parse_entry_table_entry(std::istream& is) {
	EntryTableEntry entry;

	entry.nr_entries = binparse::parse<binparse::Value8>(is, "nr_entries");
	entry.flags = binparse::parse<binparse::Value8>(is, "flags");
	entry.object_index = binparse::parse<binparse::Value16>(is, "object_index");
	
	for(Value8 i = Value8(0); i < entry.nr_entries; i++) {
		if(entry.flags & 2) {
			entry.entries.push_back({
				binparse::parse<Value8>(is, "flags"),
				binparse::parse<Offset32>(is, "entry")
			});
		} else {
			entry.entries.push_back({
				binparse::parse<Value8>(is, "flags"),
				binparse::parse<Offset16>(is, "entry")
			});
		}
	}
	
	return entry;
}

std::ostream&operator<<(std::ostream& os, const EntryTableEntry& entry)
{
	return binparse::operator<<(os, entry);
}

}