/*
Open Tracker License

Terms and Conditions

Copyright (c) 1991-2000, Be Incorporated. All rights reserved.

Permission is hereby granted, free of charge, to any person obtaining a copy of
this software and associated documentation files (the "Software"), to deal in
the Software without restriction, including without limitation the rights to
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
of the Software, and to permit persons to whom the Software is furnished to do
so, subject to the following conditions:

The above copyright notice and this permission notice applies to all licensees
and shall be included in all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF TITLE, MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
BE INCORPORATED BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN
AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF, OR IN CONNECTION
WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

Except as contained in this notice, the name of Be Incorporated shall not be
used in advertising or otherwise to promote the sale, use or other dealings in
this Software without prior written authorization from Be Incorporated.

Tracker(TM), Be(R), BeOS(R), and BeIA(TM) are trademarks or registered trademarks
of Be Incorporated in the United States and other countries. Other brand product
names are registered trademarks or trademarks of their respective holders.
All rights reserved.
*/
#ifndef _NU_ICON_CACHE_H
#define _NU_ICON_CACHE_H


// Icon cache is used for drawing node icons; it caches icons
// and reuses them for successive draws


#include <Bitmap.h>
#include <ObjectList.h>
#include <Mime.h>
#include <String.h>

#include "AutoLock.h"
#include "HashSet.h"
#include "Utilities.h"


// Icon cache splits icons into two caches - the shared cache, likely to
// get the most hits and the node cache. Every icon that is found in a
// mime-based structure goes into the shared cache, only files that have
// their own private icon use the node cache;
// Entries are only deleted from the shared cache if an icon for a mime type
// changes, this makes async icon drawing easier. Node cache deletes it's
// entries whenever a file gets deleted.

// if a view ever uses the cache to draw in async mode, it needs to call
// it when it is being destroyed

namespace BPrivate {

class Model;
class ModelNodeLazyOpener;
class LazyBitmapAllocator;
class SharedIconCache;
class SharedCacheEntry;
class GenerateThumbnailJob;

enum IconDrawMode {
	// Different states of icon drawing
	kSelected 					= 0x01,
	kNotFocused					= 0x02,		// Tracker window
	kOpen						= 0x04,		// open folder, trash
	kNotEmpty					= 0x08,		// full trash
	kDisabled					= 0x10,		// inactive nav menu entry
	kActive						= 0x20,		// active home dir, boot volume
	kLink						= 0x40,		// symbolic link
	kTrackerSpecialized			= 0x80,

	// some common combinations
	kNormalIcon						= 0,
	kSelectedIcon					= kSelected,
	kSelectedInBackgroundIcon		= kSelected | kNotFocused,
	kOpenIcon						= kOpen,
	kOpenSelectedIcon				= kSelected | kOpen,
	kOpenSelectedInBackgroundIcon	= kSelected | kNotFocused | kOpen,
	kFullIcon						= kNotEmpty,
	kFullSelectedIcon				= kNotEmpty | kOpen,
	kDimmedIcon
};


#define NORMAL_ICON_ONLY kNormalIcon
	// replace use of these defines with mode once the respective getters
	// can get non-plain icons


// Where did an icon come from
enum IconSource {
	kUnknownSource,
	kUnknownNotFromNode,	// icon origin not known but determined not
							// to be from the node itself
	kTrackerDefault,		// file has no type, Tracker provides generic,
							// folder, symlink or app
	kTrackerSupplied,		// home directory, boot volume, trash, etc.
	kMetaMime,				// from BMimeType
	kPreferredAppForType,	// have a preferred application for a type,
							// has an icon
	kPreferredAppForNode,	// have a preferred application for this node,
							// has an icon
	kVolume,
	kNode
};


template<typename Class>
struct SelfHashing {
	typedef typename Class::HashKeyType KeyType;
	typedef Class ValueType;

	size_t HashKey(KeyType key) const
	{
		return Class::Hash(key);
	}

	size_t Hash(ValueType* value) const
	{
		return value->Hash();
	}

	bool Compare(KeyType key, ValueType* value) const
	{
		return *value == key;
	}

	ValueType*& GetLink(ValueType* value) const
	{
		return value->HashNext();
	}
};


class IconCacheEntry {
	// aliased entries don't own their icons, just point
	// to some other entry that does

	// This is used for icons that are defined by a preferred app for
	// a metamime, types that do not have an icon get to point to
	// generic, etc.

public:
	IconCacheEntry();
	~IconCacheEntry();

	void SetAliasFor(const SharedIconCache* sharedCache,
		const SharedCacheEntry* entry);
	static IconCacheEntry* ResolveIfAlias(const SharedIconCache* sharedCache,
		IconCacheEntry* entry);
	IconCacheEntry* ResolveIfAlias(const SharedIconCache* sharedCache);

	void SetIcon(BBitmap* bitmap, IconDrawMode mode, BSize size);

	bool HaveIconBitmap(IconDrawMode mode, BSize size) const;
	bool CanConstructBitmap(IconDrawMode mode, BSize size) const;
	static bool AlternateModeForIconConstructing(IconDrawMode requestedMode,
		IconDrawMode &alternate, BSize size);
	BBitmap* ConstructBitmap(BBitmap* constructFrom,
		IconDrawMode requestedMode, IconDrawMode constructFromMode,
		BSize size, LazyBitmapAllocator*);
	BBitmap* ConstructBitmap(IconDrawMode requestedMode, BSize size,
		LazyBitmapAllocator*);
		// same as above, always uses normal icon as source

	bool IconHitTest(BPoint, IconDrawMode, BSize) const;
		// given a point, returns true if a non-transparent pixel was hit

	void RetireIcons(BObjectList<BBitmap>* retiredBitmapList);
		// can't just delete icons, they may be still drawing
		// async; instead, put them on the retired list and
		// only delete the list if it grows too much, way after
		// the icon finishes drawing
		//
		// This could fail if we retire a lot of icons (10 * 1024)
		// while we are drawing them, shouldn't be a practical problem

protected:
	BBitmap* IconForMode(IconDrawMode mode, BSize size) const;
	void SetIconForMode(BBitmap* bitmap, IconDrawMode mode, BSize size);

	// list of most common icons
	BBitmap* fLargeIcon;
	BBitmap* fHighlightedLargeIcon;
	BBitmap* fMiniIcon;
	BBitmap* fHighlightedMiniIcon;

	const IconCacheEntry* fAliasTo;

	// list of other icon kinds would be added here

	friend class SharedIconCache;
	friend class NodeIconCache;
};


class SimpleIconCache {
public:
	SimpleIconCache(const char*);
	virtual ~SimpleIconCache() {}

	virtual void Draw(IconCacheEntry*, BView*, BPoint, IconDrawMode mode,
		BSize size, bool async = false) = 0;
	virtual void Draw(IconCacheEntry*, BView*, BPoint, IconDrawMode,
		BSize, void (*)(BView*, BPoint, BBitmap*, void*),
		void* = NULL) = 0;

	bool Lock();
	void Unlock();
	bool IsLocked() const;

private:
	Benaphore fLock;
};


class SharedCacheEntry : public IconCacheEntry {
public:
	SharedCacheEntry();
	SharedCacheEntry(const char* fileType, const char* appSignature = 0);

	void Draw(BView*, BPoint, IconDrawMode mode, BSize size,
		bool async = false);

	void Draw(BView*, BPoint, IconDrawMode, BSize,
		void (*)(BView*, BPoint, BBitmap*, void*), void* = NULL);

	const char* FileType() const;
	const char* AppSignature() const;

public:
	// hash table support
	struct TypeAndSignature {
		const char* type, *signature;
		TypeAndSignature(const char* t, const char* s)
			: type(t), signature(s) {}
	};
	typedef TypeAndSignature HashKeyType;
	static size_t Hash(const TypeAndSignature& typeAndSignature);

	size_t Hash() const;
	SharedCacheEntry*& HashNext() { return fNext; }
	bool operator==(const TypeAndSignature& typeAndSignature) const;

private:
	SharedCacheEntry* fNext;

	BString fFileType;
	BString fAppSignature;

	friend class SharedIconCache;
};


class SharedIconCache : public SimpleIconCache {
	// SharedIconCache is used for icons that come from the mime database
public:
	SharedIconCache();

	virtual void Draw(IconCacheEntry*, BView*, BPoint, IconDrawMode mode,
		BSize size, bool async = false);
	virtual void Draw(IconCacheEntry*, BView*, BPoint, IconDrawMode,
		BSize, void (*)(BView*, BPoint, BBitmap*, void*), void* = NULL);

	SharedCacheEntry* FindItem(const char* fileType,
		const char* appSignature = 0) const;
	SharedCacheEntry* AddItem(const char* fileType,
		const char* appSignature = 0);
	void IconChanged(SharedCacheEntry*);

	void SetAliasFor(IconCacheEntry* entry,
		const SharedCacheEntry* original) const;
	IconCacheEntry* ResolveIfAlias(IconCacheEntry* entry) const;

	void RemoveAliasesTo(SharedCacheEntry* alias);

private:
	typedef BOpenHashTable<SelfHashing<SharedCacheEntry> > EntryHashTable;
	EntryHashTable fHashTable;

	BObjectList<BBitmap> fRetiredBitmaps;
		// icons are drawn asynchronously, can't just delete them right away,
		// instead have to place them onto the retired bitmap list and wait
		// for the next sync to delete them
};


class NodeCacheEntry : public IconCacheEntry {
public:
	NodeCacheEntry(bool permanent = false);
	NodeCacheEntry(const node_ref*, bool permanent = false);
	void Draw(BView*, BPoint, IconDrawMode mode, BSize size,
		bool async = false);

	void Draw(BView*, BPoint, IconDrawMode, BSize,
		void (*)(BView*, BPoint, BBitmap*, void*), void* = NULL);

	const node_ref* Node() const;

	bool Permanent() const;

public:
	// hash table support
	typedef const node_ref* HashKeyType;
	static size_t Hash(const node_ref* node);

	size_t Hash() const;
	NodeCacheEntry*& HashNext() { return fNext; }
	bool operator==(const node_ref* ref) const;

private:
	NodeCacheEntry* fNext;

	node_ref fRef;
	bool fPermanent;
		// special cache entry that has to be deleted explicitly

	friend class NodeIconCache;
};


class NodeIconCache : public SimpleIconCache {
	// NodeIconCache is used for nodes that define their own icons
public:
	NodeIconCache();

	virtual void Draw(IconCacheEntry*, BView*, BPoint, IconDrawMode,
		BSize, bool async = false);

	virtual void Draw(IconCacheEntry*, BView*, BPoint, IconDrawMode,
		BSize, void (*)(BView*, BPoint, BBitmap*, void*), void* = 0);

	NodeCacheEntry* FindItem(const node_ref*) const;
	NodeCacheEntry* AddItem(const node_ref*, bool permanent = false);
	void Deleting(const node_ref*);
		// model for this node is getting deleted
		// (not necessarily the node itself)
	void Removing(const node_ref*);
		// used by permanent NodeIconCache entries, when an entry gets deleted
	void Deleting(const BView*);
	void IconChanged(const Model*);

	void RemoveAliasesTo(SharedCacheEntry* alias);

private:
	typedef BOpenHashTable<SelfHashing<NodeCacheEntry> > EntryHashTable;
	EntryHashTable fHashTable;
};


const int32 kColorTransformTableSize = 256;


class IconCache {
public:
	IconCache();

	void Draw(Model*, BView*, BPoint where, IconDrawMode mode,
		BSize size, bool async = false);
		// draw an icon for a model, load the icon from the appropriate
		// location if not cached already

	void SyncDraw(Model*, BView*, BPoint, IconDrawMode,
		BSize, void (*)(BView*, BPoint, BBitmap*, void*),
		void* passThruState = 0);
		// draw an icon for a model, load the icon from the appropriate
		// location if not cached already; only works for sync draws,
		// once the call returns, the bitmap may be deleted

	// preload calls used to ensure successive cache hit for the respective
	// icon, used for common tracker types, etc; Not calling these should only
	// cause a slowdown
	void Preload(Model*, IconDrawMode mode, BSize size,
		bool permanent = false);
	status_t Preload(const char* mimeType, IconDrawMode mode, BSize size);

	void Deleting(const Model*);
		// hook to manage unloading icons for nodes that are going away
	void Removing(const Model* model);
		// used by permanent NodeIconCache entries, when an entry gets
		// deleted
	void Deleting(const BView*);
		// hook to manage deleting draw view caches for views that are
		// going away

	// icon changed calls, used when a node or a file type has an icon changed
	// the icons for the node/file type will be flushed and re-cached during
	// the next draw
	void IconChanged(Model*);
	void IconChanged(const char* mimeType, const char* appSignature);

	bool IsIconFrom(const Model*, const char* mimeType,
		const char* appSignature) const;
		// called when metamime database changed to figure out which models
		// to redraw

	bool IconHitTest(BPoint, const Model*, IconDrawMode, BSize);

	// utility calls for building specialized icons
	BBitmap* MakeSelectedIcon(const BBitmap* normal, BSize,
		LazyBitmapAllocator*);

	static bool NeedsDeletionNotification(IconSource);

	static IconCache* sIconCache;
	static BSize sMiniIconSize;

private:
	// shared calls
	IconCacheEntry* Preload(AutoLock<SimpleIconCache>* nodeCache,
		AutoLock<SimpleIconCache>* sharedCache,
		AutoLock<SimpleIconCache>** resultingLockedCache,
		Model*, IconDrawMode mode, BSize size, bool permanent);
		// preload uses lazy locking, returning the cache we decided
		// to use to get the icon
		// <resultingLockedCache> may be null if we don't care

	// shared mime-based icon retrieval calls
	IconCacheEntry* GetIconForPreferredApp(const char* mimeTypeSignature,
		const char* preferredApp, IconDrawMode mode, BSize size,
		 LazyBitmapAllocator*, IconCacheEntry*);
	IconCacheEntry* GetIconFromFileTypes(ModelNodeLazyOpener*,
		IconSource &source, IconDrawMode mode, BSize size,
		LazyBitmapAllocator*, IconCacheEntry*);
	IconCacheEntry* GetIconFromMetaMime(const char* fileType,
		IconDrawMode mode, BSize size, LazyBitmapAllocator*,
		IconCacheEntry*);
	IconCacheEntry* GetVolumeIcon(AutoLock<SimpleIconCache>* nodeCache,
		AutoLock<SimpleIconCache>* sharedCache,
		AutoLock<SimpleIconCache>** resultingLockedCache,
		Model*, IconSource&, IconDrawMode mode,
		BSize size, LazyBitmapAllocator*);
	IconCacheEntry* GetRootIcon(AutoLock<SimpleIconCache>* nodeCache,
		AutoLock<SimpleIconCache>* sharedCache,
		AutoLock<SimpleIconCache>** resultingLockedCache,
		Model*, IconSource&, IconDrawMode mode,
		BSize size, LazyBitmapAllocator*);
	IconCacheEntry* GetWellKnownIcon(AutoLock<SimpleIconCache> *nodeCache,
		AutoLock<SimpleIconCache>* sharedCache,
		AutoLock<SimpleIconCache>** resultingLockedCache,
		Model*, IconSource&, IconDrawMode mode,
		BSize size, LazyBitmapAllocator*);
	IconCacheEntry* GetNodeIcon(ModelNodeLazyOpener *,
		AutoLock<SimpleIconCache>* nodeCache,
		AutoLock<SimpleIconCache>** resultingLockedCache,
		Model*, IconSource&, IconDrawMode mode,
		BSize size, LazyBitmapAllocator*, IconCacheEntry*,
		bool permanent);
	IconCacheEntry* GetGenericIcon(AutoLock<SimpleIconCache>* sharedCache,
		AutoLock<SimpleIconCache>** resultingLockedCache,
		Model*, IconSource&, IconDrawMode mode,
		BSize size, LazyBitmapAllocator*, IconCacheEntry*);
	IconCacheEntry* GetFallbackIcon(
		AutoLock<SimpleIconCache>* sharedCacheLocker,
		AutoLock<SimpleIconCache>** resultingOpenCache,
		Model* model, IconDrawMode mode, BSize size,
		LazyBitmapAllocator* lazyBitmap, IconCacheEntry* entry);

	BBitmap* MakeTransformedIcon(const BBitmap*, BSize,
		int32 colorTransformTable [], LazyBitmapAllocator*);

private:
	NodeIconCache fNodeCache;
	SharedIconCache fSharedCache;

	void InitHighlightTable();

	int32 fHighlightTable[kColorTransformTableSize];
	bool fInitHighlightTable;
		// whether or not we need to initialize the highlight table

	friend class BPrivate::GenerateThumbnailJob;
};


class LazyBitmapAllocator {
	// Utility class used when we aren't sure that we will keep a bitmap,
	// need a bitmap or be able to construct it properly
public:
	LazyBitmapAllocator(BSize size,
		color_space colorSpace = kDefaultIconDepth,
		bool preallocate = false);
	~LazyBitmapAllocator();

	BBitmap* Get();
	BBitmap* Adopt();

private:
	BBitmap* fBitmap;
	BSize fSize;
	color_space fColorSpace;
};


// inlines follow

inline const char*
SharedCacheEntry::FileType() const
{
	return fFileType.String();
}


inline const char*
SharedCacheEntry::AppSignature() const
{
	return fAppSignature.String();
}


inline bool
IconCache::NeedsDeletionNotification(IconSource from)
{
	return from == kNode;
}


inline IconCacheEntry*
SharedIconCache::ResolveIfAlias(IconCacheEntry* entry) const
{
	if (entry->fAliasTo == NULL)
		return entry;

	return const_cast<IconCacheEntry*>(entry->fAliasTo);
}


} // namespace BPrivate

using namespace BPrivate;


#endif	// _NU_ICON_CACHE_H