From 90eb02133944adf3b8ba2be393b427ca2be2451e Mon Sep 17 00:00:00 2001
From: krustur <krustur@gmail.com>
Date: Thu, 29 Nov 2018 23:07:16 +0100
Subject: [PATCH] Get DiscObject by GetIconTags if V44 enabled

---
 src/main.c | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/main.c b/src/main.c
index fd5a80b..bffd69a 100644
--- a/src/main.c
+++ b/src/main.c
@@ -6,6 +6,7 @@
 #include <proto/exec.h>
 #include <proto/dos.h>
 #include <proto/icon.h>
+#include <workbench/icon.h>
 
 #include <stdlib.h>
 #include <stdio.h>
@@ -371,7 +372,18 @@ unsigned int AlignIcon(unsigned char *diskObjectName)
         fixedDiskObjectName[diskObjectNameLen - 5] = 0;
     }
 
-    struct DiskObject *diskObject = GetDiskObject(fixedDiskObjectName);
+    struct DiskObject *diskObject;
+
+    if (iconLibraryV44Enabled)
+    {
+        Verbose("GetDiscObjectBy: GetIconTags()\n");
+        diskObject = GetIconTags(fixedDiskObjectName, TAG_DONE);//ICONGETA_GetPaletteMappedIcon, FALSE, TAG_DONE);
+    }
+    else
+    {
+        Verbose("GetDiscObjectBy: GetDiskObject()\n");
+        diskObject = GetDiskObject(fixedDiskObjectName);
+    }
     if (diskObject)
     {
         Verbose("Icon found: %s\n", fixedDiskObjectName);