1 files changed, 110 insertions, 0 deletions
diff --git a/plugins/ExtraIcons/src/usedIcons.cpp b/plugins/ExtraIcons/src/usedIcons.cpp
new file mode 100644
index 0000000000..de58069293
--- /dev/null
+++ b/plugins/ExtraIcons/src/usedIcons.cpp
@@ -0,0 +1,110 @@
+/*
+ Copyright (C) 2009 Ricardo Pescuma Domenecci
+
+ This is free software; you can redistribute it and/or
+ modify it under the terms of the GNU Library General Public
+ License as published by the Free Software Foundation; either
+ version 2 of the License, or (at your option) any later version.
+
+ This is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ Library General Public License for more details.
+
+ You should have received a copy of the GNU Library General Public
+ License along with this file; see the file license.txt.  If
+ not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ Boston, MA 02111-1307, USA.
+ */
+
+#include "commons.h"
+
+struct Icon
+{
+	string name;
+	int refCount;
+	HANDLE hImage;
+
+	Icon(const char *icolibName) :
+		name(icolibName), refCount(0), hImage(NULL)
+	{
+	}
+};
+
+static vector<Icon> usedIcons;
+
+static Icon * FindIcon(const char *icolibName)
+{
+	Icon *icon = NULL;
+
+	for (unsigned int i = 0; i < usedIcons.size(); ++i)
+	{
+		Icon *tmp = &usedIcons[i];
+		if (tmp->name != icolibName)
+			continue;
+
+		icon = tmp;
+		break;
+	}
+
+	if (icon == NULL)
+	{
+		usedIcons.push_back(Icon(icolibName));
+		icon = &usedIcons[usedIcons.size() - 1];
+	}
+
+	if (icon->hImage == NULL)
+	{
+		HICON hIcon = IcoLib_LoadIcon(icon->name.c_str());
+		if (hIcon != NULL)
+		{
+			icon->hImage = (HANDLE) CallService(MS_CLIST_EXTRA_ADD_ICON, (WPARAM) hIcon, 0);
+			if (icon->hImage == (HANDLE) -1)
+				icon->hImage = NULL;
+
+			IcoLib_ReleaseIcon(hIcon);
+		}
+	}
+
+	return icon;
+}
+
+HANDLE GetIcon(const char *icolibName)
+{
+	return FindIcon(icolibName)->hImage;
+}
+
+HANDLE AddIcon(const char *icolibName)
+{
+	Icon *icon = FindIcon(icolibName);
+	icon->refCount++;
+	return icon->hImage;
+}
+
+void RemoveIcon(const char *icolibName)
+{
+	for (unsigned int i = 0; i < usedIcons.size(); ++i)
+	{
+		Icon *icon = &usedIcons[i];
+
+		if (icon->name != icolibName)
+			continue;
+
+		icon->refCount--;
+		break;
+	}
+}
+
+static bool NotUsedIcon(const Icon &icon) 
+{
+	return icon.refCount <= 0;
+}
+
+void ResetIcons()
+{
+	usedIcons.erase(std::remove_if(usedIcons.begin(), usedIcons.end(), NotUsedIcon), usedIcons.end());
+
+	for (unsigned int i = 0; i < usedIcons.size(); ++i)
+		usedIcons[i].hImage = NULL;
+}
+