// $Revision: 1.1 $
// SimnpleDragAndDropItem.C
// Copyright (C) 1994, 1995 Taligent, Inc. All rights reserved.

#ifndef Taligent_SIMPLEDRAGANDDROPITEM
#include "SimpleDragAndDropItem.h"
#endif

#ifndef Taligent_ASSERTIONS
#include <Taligent/Assertions.h>
#endif

MCollectibleDefinitionsMacro(TSimpleDragAndDropItem, kOriginalVersion);

TSimpleDragAndDropItem::TSimpleDragAndDropItem()
	: TDragAndDropItem(),
	fDictionary(new TOperatorKeyValuePairComparator<TTypeDescription, TScrapItem>,
		new TPolymorphicPairStreamer<TKeyValuePair<TTypeDescription, TScrapItem> >),
	fScraps(new TOperatorComparator<TScrapItem>, new TPolymorphicStreamer<TScrapItem>)
{
}

TSimpleDragAndDropItem::TSimpleDragAndDropItem(const TViewHandle &viewForActions)
	: TDragAndDropItem(viewForActions),
	fDictionary(new TOperatorKeyValuePairComparator<TTypeDescription, TScrapItem>,
		new TPolymorphicPairStreamer<TKeyValuePair<TTypeDescription, TScrapItem> >),
	fScraps(new TOperatorComparator<TScrapItem>, new TPolymorphicStreamer<TScrapItem>)
{
}

TSimpleDragAndDropItem::TSimpleDragAndDropItem(const TSimpleDragAndDropItem& source)
	: TDragAndDropItem(source),
	fDictionary(new TOperatorKeyValuePairComparator<TTypeDescription, TScrapItem>,
		new TPolymorphicPairStreamer<TKeyValuePair<TTypeDescription, TScrapItem> >),
	fScraps(new TOperatorComparator<TScrapItem>, new TPolymorphicStreamer<TScrapItem>)
{
	CopyDictionaryAndScraps(source.fScraps);
}

TSimpleDragAndDropItem::~TSimpleDragAndDropItem()
{
	fDictionary.DeleteAllKeys();
	fScraps.DeleteAll();
}

TSimpleDragAndDropItem&
TSimpleDragAndDropItem::operator=(const TSimpleDragAndDropItem& source)
{
	if (&source != this) {
		TDragAndDropItem::operator=(source);
		fDictionary.DeleteAllKeys();
		fScraps.DeleteAll();
		CopyDictionaryAndScraps(source.fScraps);
	}
	return *this;
}

TStream&
TSimpleDragAndDropItem::operator>>=(TStream& toStream) const
{
	::WriteVersion(toStream);
	TDragAndDropItem::operator>>=(toStream);
	fDictionary >>= toStream;
	fScraps >>= toStream;
	return toStream;
}

TStream&
TSimpleDragAndDropItem::operator<<=(TStream& fromStream)
{
	::ReadVersion(fromStream, kOriginalVersion, kOriginalVersion);

	TDragAndDropItem::operator<<=(fromStream);
	fDictionary <<= fromStream;
	fScraps <<= fromStream;

	return fromStream;
}

void
TSimpleDragAndDropItem::CopyAvailableTypes(TSequenceOf<TTypeDescription>& result,
	TMemoryHeap* whichHeap) const
{
	TDeleterFor<TIteratorOver<TKeyValuePair<TTypeDescription, TScrapItem> > > iterator =
		fDictionary.CreateIterator();

	for (TKeyValuePair<TTypeDescription, TScrapItem>* scrap = iterator->First();
		scrap != NIL;
		scrap = iterator->Next()) {

		result.Add(new (TAllocationHeap(whichHeap)) TTypeDescription(*scrap->GetKey()));
	}
}

TScrapItem*
TSimpleDragAndDropItem::FindItemMatchingDescription(const TTypeDescription &theDescription)
{
	TDeleterFor<TIteratorOver<TKeyValuePair<TTypeDescription, TScrapItem> > > iterator =
		fDictionary.CreateIterator();

	for (TKeyValuePair<TTypeDescription, TScrapItem>* scrap = iterator->First();
		scrap != NIL;
		scrap = iterator->Next()) {
	
		if (theDescription == *scrap->GetKey()) {
			return scrap->GetValue();
		}
	}
	::Assertion(false, "Whoa! Bogus type passed to TSimpleDragAndDropItem::FindItemMatchingDescription");
	return NIL;
}

void
TSimpleDragAndDropItem::AdoptScrapItem(TScrapItem* scrapItem)
{
	::Assertion(scrapItem != NIL, "NIL was passed to TSimpleDragAndDropItem::AdoptScrapItem");
	
	TDequeOf<TTypeDescription> types;
	scrapItem->CopyAvailableTypes(types);

	TDeleterFor<TIteratorOver<TTypeDescription> > iterator = types.CreateIterator();

	fScraps.Add(scrapItem);

	for (TTypeDescription* type = iterator->First();
		type != NIL;
		type = iterator->Next()) {

		fDictionary.Add(new TKeyValuePair<TTypeDescription, TScrapItem>(
			new TTypeDescription(*type), scrapItem));
	}
}

void
TSimpleDragAndDropItem::CopyDictionaryAndScraps(const TDequeOf<TScrapItem>& otherScraps)
{
	TDeleterFor<TIteratorOver<TScrapItem> > iterator = otherScraps.CreateIterator();

	for (TScrapItem* scrap = iterator->First();
		scrap != NIL;
		scrap = iterator->Next()) {

		AdoptScrapItem(::Copy(*scrap));
	}
}