From feb4feaae781e0ba42fc7bb39ee5c9f310faf449 Mon Sep 17 00:00:00 2001
From: John Schember <john@nachtimwald.com>
Date: Wed, 21 Oct 2009 21:11:17 -0400
Subject: [PATCH] PML Output: Use set cover as cover or use first image as
 cover if no cover is specified.

---
 src/calibre/ebooks/pml/pmlml.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/calibre/ebooks/pml/pmlml.py b/src/calibre/ebooks/pml/pmlml.py
index b6a62e7c1f..7b1813256e 100644
--- a/src/calibre/ebooks/pml/pmlml.py
+++ b/src/calibre/ebooks/pml/pmlml.py
@@ -95,6 +95,9 @@ class PMLMLizer(object):
 
     def get_cover_page(self):
         output = u''
+        if 'cover' in self.oeb_book.guide:
+            output += '\\m="cover.png"\n'
+            self.image_hrefs[self.oeb_book.guide['cover'].href] = 'cover.png'
         if 'titlepage' in self.oeb_book.guide:
             self.log.debug('Generating title page...')
             href = self.oeb_book.guide['titlepage'].href
@@ -191,7 +194,10 @@ class PMLMLizer(object):
         if tag in IMAGE_TAGS:
             if elem.attrib.get('src', None):
                 if page.abshref(elem.attrib['src']) not in self.image_hrefs.keys():
-                    self.image_hrefs[page.abshref(elem.attrib['src'])] = image_name('%s' % len(self.image_hrefs.keys()), self.image_hrefs.keys()).strip('\x00')
+                    if len(self.image_hrefs.keys()) == 0:
+                        self.image_hrefs[page.abshref(elem.attrib['src'])] = 'cover.png'
+                    else:
+                        self.image_hrefs[page.abshref(elem.attrib['src'])] = image_name('%s' % len(self.image_hrefs.keys()), self.image_hrefs.keys()).strip('\x00')
                 text.append('\\m="%s"' % self.image_hrefs[page.abshref(elem.attrib['src'])])
         if tag == 'hr':
             w = '\\w'