1 files changed, 202 insertions, 0 deletions
diff --git a/Src/Plugins/Input/in_wmvdrm/AlbumArt.cpp b/Src/Plugins/Input/in_wmvdrm/AlbumArt.cpp
new file mode 100644
index 00000000..a1dc08d2
--- /dev/null
+++ b/Src/Plugins/Input/in_wmvdrm/AlbumArt.cpp
@@ -0,0 +1,202 @@
+#include "main.h"
+#include "../nu/AutoWide.h"
+#include "AlbumArt.h"
+#include "util.h"
+#include <shlwapi.h>
+#include <strsafe.h>
+
+bool ASF_AlbumArtProvider::IsMine(const wchar_t *filename)
+{
+	const wchar_t *ext = PathFindExtension(filename);
+	if (ext && *ext)
+	{
+		ext++;
+		return fileTypes.GetAVType(ext) != -1;
+	}
+	return false;
+}
+
+int ASF_AlbumArtProvider::ProviderType()
+{
+	return ALBUMARTPROVIDER_TYPE_EMBEDDED;
+}
+
+bool NameToAPICType(const wchar_t *name, int &num)
+{
+	if (!name || !*name) // default to cover
+		num=0x3;
+	else if (!_wcsicmp(name, L"fileicon")) // 	32x32 pixels 'file icon' (PNG only)
+		num=0x1;
+	else if (!_wcsicmp(name, L"icon")) // 	Other file icon
+		num=0x2;
+	else if (!_wcsicmp(name, L"cover")) // Cover (front)
+		num=0x3;
+	else if (!_wcsicmp(name, L"back")) // Cover (back)
+		num=0x4;
+	else if (!_wcsicmp(name, L"leaflet")) // Leaflet page
+		num=0x5;
+	else if (!_wcsicmp(name, L"media")) // Media (e.g. lable side of CD)
+		num=0x6;
+	else if (!_wcsicmp(name, L"leadartist")) //Lead artist/lead performer/soloist
+		num=0x7;
+	else if (!_wcsicmp(name, L"artist")) // Artist/performer
+		num=0x8;
+	else if (!_wcsicmp(name, L"conductor")) // Conductor
+		num=0x9;
+	else if (!_wcsicmp(name, L"band")) // Band/Orchestra
+		num=0xA;
+	else if (!_wcsicmp(name, L"composer"))  // Composer
+		num=0xB;
+	else if (!_wcsicmp(name, L"lyricist")) // Lyricist/text writer
+		num=0xC;
+	else if (!_wcsicmp(name, L"location")) // Recording Location
+		num=0xD;
+	else if (!_wcsicmp(name, L"recording")) // During recording
+		num=0xE;
+	else if (!_wcsicmp(name, L"performance")) // During performance
+		num=0xF;
+	else if (!_wcsicmp(name, L"preview")) // Movie/video screen capture
+		num=0x10;
+	else if (!_wcsicmp(name, L"fish")) // A bright coloured fish
+		num=0x11;
+	else if (!_wcsicmp(name, L"illustration")) // Illustration
+		num=0x12;
+	else if (!_wcsicmp(name, L"artistlogo")) // Band/artist logotype
+		num=0x13;
+	else if (!_wcsicmp(name, L"publisherlogo")) // Publisher/Studio logotype
+		num=0x14;
+	else
+		return false;
+	return true;
+}
+
+
+int ASF_AlbumArtProvider::GetAlbumArtData(const wchar_t *filename, const wchar_t *type, void **bits, size_t *len, wchar_t **mimeType)
+{
+	int pictype;
+	if (NameToAPICType(type, pictype))
+	{
+		WMInformation wm(filename);
+		if (wm.GetPicture(bits, len, mimeType, pictype))
+			return ALBUMARTPROVIDER_SUCCESS;
+	}
+
+	return ALBUMARTPROVIDER_FAILURE;
+}
+
+int ASF_AlbumArtProvider::SetAlbumArtData(const wchar_t *filename, const wchar_t *type, void *bits, size_t len, const wchar_t *mimeType)
+{
+	int pictype;
+	if (NameToAPICType(type, pictype))
+	{
+		WMInformation wm(filename);
+		if (!wm.MakeWritable(filename))
+			return ALBUMARTPROVIDER_READONLY; // can't write
+
+		if (wm.SetPicture(bits, len, mimeType, pictype))
+		{
+			wm.Flush();
+			return ALBUMARTPROVIDER_SUCCESS;
+		}
+	}
+
+	return ALBUMARTPROVIDER_FAILURE;
+}
+
+int ASF_AlbumArtProvider::DeleteAlbumArt(const wchar_t *filename, const wchar_t *type)
+{
+	int pictype;
+	if (NameToAPICType(type, pictype))
+	{
+		WMInformation wm(filename);
+		if (!wm.MakeWritable(filename))
+		{
+			if (wm.HasPicture(pictype))
+				return ALBUMARTPROVIDER_READONLY; // can't write
+			else
+				return ALBUMARTPROVIDER_FAILURE;
+		}
+
+		if (wm.DeletePicture(pictype))
+		{
+			wm.Flush();
+			return ALBUMARTPROVIDER_SUCCESS;
+		}
+	}
+
+	return ALBUMARTPROVIDER_FAILURE;
+}
+
+#define CBCLASS ASF_AlbumArtProvider
+START_DISPATCH;
+CB(SVC_ALBUMARTPROVIDER_PROVIDERTYPE, ProviderType);
+CB(SVC_ALBUMARTPROVIDER_GETALBUMARTDATA, GetAlbumArtData);
+CB(SVC_ALBUMARTPROVIDER_SETALBUMARTDATA, SetAlbumArtData);
+CB(SVC_ALBUMARTPROVIDER_DELETEALBUMART, DeleteAlbumArt);
+CB(SVC_ALBUMARTPROVIDER_ISMINE, IsMine);
+END_DISPATCH;
+#undef CBCLASS
+
+static ASF_AlbumArtProvider albumArtProvider;
+
+// {B4184902-EE79-4015-B9A4-76209C6153FA}
+static const GUID asf_albumartproviderGUID = 
+{ 0xb4184902, 0xee79, 0x4015, { 0xb9, 0xa4, 0x76, 0x20, 0x9c, 0x61, 0x53, 0xfa } };
+
+
+FOURCC AlbumArtFactory::GetServiceType()
+{
+	return svc_albumArtProvider::SERVICETYPE;
+}
+
+const char *AlbumArtFactory::GetServiceName()
+{
+	return "ASF Album Art Provider";
+}
+
+GUID AlbumArtFactory::GetGUID()
+{
+	return asf_albumartproviderGUID;
+}
+
+void *AlbumArtFactory::GetInterface(int global_lock)
+{
+	return &albumArtProvider;
+}
+
+int AlbumArtFactory::SupportNonLockingInterface()
+{
+	return 1;
+}
+
+int AlbumArtFactory::ReleaseInterface(void *ifc)
+{
+	//plugin.service->service_unlock(ifc);
+	return 1;
+}
+
+const char *AlbumArtFactory::GetTestString()
+{
+	return 0;
+}
+
+int AlbumArtFactory::ServiceNotify(int msg, int param1, int param2)
+{
+	return 1;
+}
+
+#ifdef CBCLASS
+#undef CBCLASS
+#endif
+
+#define CBCLASS AlbumArtFactory
+START_DISPATCH;
+CB(WASERVICEFACTORY_GETSERVICETYPE, GetServiceType)
+CB(WASERVICEFACTORY_GETSERVICENAME, GetServiceName)
+CB(WASERVICEFACTORY_GETGUID, GetGUID)
+CB(WASERVICEFACTORY_GETINTERFACE, GetInterface)
+CB(WASERVICEFACTORY_SUPPORTNONLOCKINGGETINTERFACE, SupportNonLockingInterface)
+CB(WASERVICEFACTORY_RELEASEINTERFACE, ReleaseInterface)
+CB(WASERVICEFACTORY_GETTESTSTRING, GetTestString)
+CB(WASERVICEFACTORY_SERVICENOTIFY, ServiceNotify)
+END_DISPATCH;