#ifndef __SHAMAP__
#define __SHAMAP__

#include <list>

#include <boost/shared_ptr.hpp>

#include "uint256.h"
#include "ScopedLock.h"

class SHAMapNodeID
{
public:
	int	mDepth;
	uint256	mNodeID;
	
        bool operator<(const SHAMapNodeID &) const;
        bool operator>(const SHAMapNodeID &) const;
        bool operator==(const SHAMapNodeID &) const;
        bool operator!=(const SHAMapNodeID &) const;
        bool operator<=(const SHAMapNodeID &) const;
        bool operator>=(const SHAMapNodeID &) const;
};

class SHAMapLeafNode
{
public:
	typedef boost::shared_ptr<SHAMapLeafNode> pointer;

private:
	uint256			mNodeID;
	std::list<uint256>	mHashes;
	uint256			mNodeHash;

public:
	SHAMapLeafNode(const uint256& NodeID);

	const uint256& GetNodeID(void) const { return mNodeID; }
	bool HasHash(const uint256& Hash) const;

	uint256& GetNodeHash();

	bool AddHash(const uint256& Hash);
	bool DelHash(const uint256& Hash);
};

class SHAMapInnerNode
{
public:
	typedef boost::shared_ptr<SHAMapInnerNode> pointer;

private:
	int		mDepth;
	uint256		mNodeID;
	uint256		mHash;
	uint256		mHashes[32];

public:
	SHAMapInnerNode(int Depth, const uint256 &NodeID);

	int GetDepth() const { return mDepth; }
	const uint256& GetNodeID() const { return mNodeID; }
	const uint256& GetChildHash(int m) const;
	const uint256& GetNodeHash() const;

	bool IsRootNode() const { return mDepth==0; }
	bool IsLastInnerNode() const { return mDepth==9; }

	uint256 GetChildNodeID(int m) const;
	void SetChildHash(int m, const uint256& Hash);
	void SetChildHash(const SHAMapInnerNode &mChild);
	void SetChildHash(const SHAMapLeafNode &mLeaf);
};

// Not all nodes need to be resident in memory, the class can hold subsets
// This class does not handle any inter-node logic because that requires reads/writes

class SHAMap
{
public:
	typedef boost::shared_ptr<SHAMap> pointer;

private:
	mutable boost::mutex mLock;
	std::map<SHAMapNodeID, SHAMapLeafNode> mLeaves;
	std::multimap<SHAMapNodeID, SHAMapInnerNode> mInnerNodes;

public:
	SHAMap();

	ScopedLock Lock() const { return ScopedLock(mLock); }

	bool HasInnerNode(int m, const uint160 &nodeID);
	bool GiveInnerNode(SHAMapInnerNode::pointer);
	SHAMapInnerNode::pointer GetInnerNode(int m, const uint160 &nodeID);

	bool HasLeafNode(int m, const uint160 &nodeID);
	bool GiveLeafNode(SHAMapLeafNode::pointer);
	SHAMapLeafNode::pointer GetLeafNode(int m, const uint160 &nodeID);
};

#endif