Changeset 9fb814a in sasview for invariantview/perspectives/invariant

Timestamp:

Jul 27, 2010 2:40:18 PM (14 years ago)

Author:

Jae Cho <jhjcho@…>

Branches:

master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, release_4.0.1, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

83ee3851

Parents:

aa5617d

Message:

added save report as pdf: pisa-3.0.33, html5lib-0.9.0, and reportlab-2.4 require

File:

• invariantview/perspectives/invariant/report_dialog.py (modified) (5 diffs)

invariantview/perspectives/invariant/report_dialog.py

@@ -13 +13 @@
 Dialog report panel to show and summarize the results of the invariant calculation.
 """
-    
-
 import wx
 import sys,os
 import wx.html as html
-import Image
 
 if sys.platform.count("win32")>0:
@@ -116 +113 @@
         dlg = wx.FileDialog(self, "Choose a file",\
                             wildcard ='HTML files (*.html)|*.html|'+
-                            'Text files (*.txt)|*.txt',
-                            #'PDF files (*.pdf)|*.pdf',
+                            'Text files (*.txt)|*.txt|'+
+                            'PDF files (*.pdf)|*.pdf',
                             style = wx.SAVE|wx.OVERWRITE_PROMPT|wx.CHANGE_DIR)
         dlg.SetFilterIndex(0) #Set .html files to be default
@@ -140 +137 @@
             #save png file using pic_fname
             self.report_list[2].save(pic_fname)
+            
         elif dlg.GetFilterIndex()== 1:
             fName = os.path.splitext(fName)[0] + '.txt'
@@ -149 +147 @@
             f.close()
             
-        else:
+        elif dlg.GetFilterIndex()== 2:
+            fName = os.path.splitext(fName)[0] + '.pdf'
             dlg.Destroy()
+
+            #pic (png) file path/name
+            pic_fname = os.path.splitext(fName)[0] + '_img.png'
+            # save the image for use with pdf writer
+            self.report_list[2].save(pic_fname)
+            
+            # put the image file path in the html data
+            html = self.report_list[0] % str(pic_fname)
+            
+            open_pdf = True
+            # make/open file in case of absence
+            f = open(fName, 'w')
+            f.close()
+            # write pdf as a pdf file
+            pdf = self.HTML2PDF(data=html, filename=fName)
+            
+            #open pdf
+            if open_pdf and pdf:
+                os.startfile(str(fName))
+
+            #delete image file
+            os.remove(pic_fname)
             
     def onPreview(self,event=None):
@@ -179 +200 @@
         
         self.Close()
-          
-        
+    
+    def HTML2PDF(self, data, filename):
+        """
+        Create a PDF file from html source string. 
+        
+        : data: html string
+        : filename: name of file to be saved
+        """
+        import ho.pisa as pisa
+        f = file(filename, "wb")
+        # pisa requires some extra packages, see their web-site
+        pdf = pisa.CreatePDF(data,f)
+        # close the file here otherwise it will be open until quitting the application.
+        f.close()
+
+        return not pdf.err
+
+        
+