Fix problem with AFM files that don't specify the font's full name or family name

jkseppan · jkseppan · commit 72c10b46913b · 2008-10-05T10:02:16.000Z
svn path=/branches/v0_91_maint/; revision=6149
diff --git a/API_CHANGES b/API_CHANGES
@@ -1,3 +1,8 @@
+    AFM.get_fullname() and get_familyname() no longer raise an 
+    exception if the AFM file does not specify these optional
+    attributes, but returns a guess based on the required FontName
+    attribute.
+
     Deprecated all mlab2 functions in mlab
 
     matplotlib.image.imread now no longer always returns RGBA -- if
diff --git a/CHANGELOG b/CHANGELOG
@@ -1,3 +1,6 @@
+2008-10-05 Fix problem with AFM files that don't specify the font's
+           full name or family name. - JKS
+
 2008-09-11 Fix use of backticks in PS - MGD
 
 2008-09-07 Changed full arrows slightly to avoid an xpdf rendering
diff --git a/lib/matplotlib/afm.py b/lib/matplotlib/afm.py
@@ -34,7 +34,7 @@
   John D. Hunter <jdhunter@ace.bsd.uchicago.edu>
 """
 
-import sys, os
+import sys, os, re
 from _mathtext_data import uni2type1
 
 #Convert string the a python type
@@ -433,11 +433,21 @@ def get_fontname(self):
 
     def get_fullname(self):
         "Return the font full name, eg, Times-Roman"
-        return self._header['FullName']
+        name = self._header.get('FullName')
+        if name is None: # use FontName as a substitute
+            name = self._header['FontName']
+        return name
 
     def get_familyname(self):
         "Return the font family name, eg, Times"
-        return self._header['FamilyName']
+        name = self._header.get('FamilyName')
+        if name is not None:
+            return name
+
+        # FamilyName not specified so we'll make a guess
+        name = self.get_fullname()
+        extras = r'(?i)([ -](regular|plain|italic|oblique|bold|semibold|light|ultralight|extra|condensed))+$'
+        return re.sub(extras, '', name)
 
     def get_weight(self):
         "Return the font weight, eg, 'Bold' or 'Roman'"
