1 files changed, 15 insertions, 3 deletions
diff --git a/main.c b/main.c
index 8718978..c279805 100644
--- a/main.c
+++ b/main.c
@@ -20,10 +20,15 @@
 #define LOG_MODULE "wbg"
 #define LOG_ENABLE_DBG 1
 #include "log.h"
-#include "png-wbg.h"
-#include "jpg.h"
 #include "shm.h"
 
+#if defined(WBG_HAVE_PNG)
+ #include "png-wbg.h"
+#endif
+#if defined(WBG_HAVE_JPG)
+ #include "jpg.h"
+#endif
+
 /* Top-level globals */
 static struct wl_display *display;
 static struct wl_registry *registry;
@@ -312,9 +317,16 @@ main(int argc, const char *const *argv)
     }
 
     const char *image_path = argv[1];
-    image = jpg_load(image_path);
+    image = NULL;
+
+#if defined(WBG_HAVE_JPG)
+    if (image == NULL)
+        image = jpg_load(image_path);
+#endif
+#if defined(WBG_HAVE_PNG)
     if (image == NULL)
         image = png_load(image_path);
+#endif
     if (image == NULL) {
         LOG_ERR("%s: failed to load", image_path);
         return EXIT_FAILURE;