Added PDF fileoutput for fitting results

Author: zunzun

FittingResultsViewer.py

@@ -9,7 +9,6 @@
 f = open("pickledEquationFile", "rb")
 unPickledEquation = pickle.load(f)
 f.close()
-os.remove("pickledEquationFile")
 resultsFrame = CustomDialogs.ResultsFrame(None, '', "Fitting Results (resizable dialog)", equation=unPickledEquation)
 resultsFrame.Show()
 app.MainLoop()

guifiles/CustomDialogs.py

@@ -498,94 +498,141 @@ def __init__(self, parent, inText):
 
 # see the included wxNestedTabsExample.py file
 class TopLevelResultsNotebook(wx.Notebook):
-    def __init__(self, parent, equation):
+    def __init__(self, parent, inEquation):
         wx.Notebook.__init__(self, parent, id=wx.ID_ANY, style=wx.BK_DEFAULT)
+        self.equation = inEquation
 
+        self.graphReportsListForPDF = []
+        self.textReportsListForPDF = []
+        self.sourceCodeReportsListForPDF = []
+
         # graphs
         graphReportsTab = wx.Notebook(self)
         self.AddPage(graphReportsTab, "Graph Reports")
 
-        if equation.GetDimensionality() == 2:
-            modelConfidenceScatter = Report_ModelScatterConfidenceGraph(graphReportsTab)
-            modelConfidenceScatter.draw(equation)
-            graphReportsTab.AddPage(modelConfidenceScatter, "Model With 95%Confidence")
+        if self.equation.GetDimensionality() == 2:
+            report = Report_ModelScatterConfidenceGraph(graphReportsTab)
+            report.draw(self.equation)
+            reportName = "Model With 95%Confidence"
+            graphReportsTab.AddPage(report, reportName)
+            self.graphReportsListForPDF.append([report.figure, reportName])
         else:
-            surfacePlot = Report_SurfacePlot(graphReportsTab)
-            surfacePlot.draw(equation)
-            graphReportsTab.AddPage(surfacePlot, "Surface Plot")
+            report = Report_SurfacePlot(graphReportsTab)
+            report.draw(self.equation)
+            reportName = "Surface Plot"
+            graphReportsTab.AddPage(report, reportName)
+            self.graphReportsListForPDF.append([report.figure, reportName])
 
-            contourPlot = Report_ContourPlot(graphReportsTab)
-            contourPlot.draw(equation)
-            graphReportsTab.AddPage(contourPlot, "Contour Plot")
-
-        absoluteErrorGraph = Report_AbsoluteErrorGraph(graphReportsTab)
-        absoluteErrorGraph.draw(equation)
-        graphReportsTab.AddPage(absoluteErrorGraph, "Absolute Error")
-
-        absoluteErrorHistogram = Report_AbsoluteErrorHistogram(graphReportsTab)
-        absoluteErrorHistogram.draw(equation)
-        graphReportsTab.AddPage(absoluteErrorHistogram, "Absolute Error Histogram")
-
-        if equation.dataCache.DependentDataContainsZeroFlag != 1:
-            percentErrorGraph = Report_PercentErrorGraph(graphReportsTab)
-            percentErrorGraph.draw(equation)
-            graphReportsTab.AddPage(percentErrorGraph, "Percent Error")
+            report = Report_ContourPlot(graphReportsTab)
+            report.draw(self.equation)
+            reportName = "Contour Plot"
+            graphReportsTab.AddPage(report, reportName)
+            self.graphReportsListForPDF.append([report.figure, reportName])
+
+        report = Report_AbsoluteErrorGraph(graphReportsTab)
+        report.draw(self.equation)
+        reportName = "Absolute Error"
+        graphReportsTab.AddPage(report, reportName)
+        self.graphReportsListForPDF.append([report.figure, reportName])
+
+        report = Report_AbsoluteErrorHistogram(graphReportsTab)
+        report.draw(self.equation)
+        reportName = "Absolute Error Histogram"
+        graphReportsTab.AddPage(report, reportName)
+        self.graphReportsListForPDF.append([report.figure, reportName])
+
+        if self.equation.dataCache.DependentDataContainsZeroFlag != 1:
+            report = Report_PercentErrorGraph(graphReportsTab)
+            report.draw(self.equation)
+            reportName = "Percent Error"
+            graphReportsTab.AddPage(report, reportName)
+            self.graphReportsListForPDF.append([report.figure, reportName])
 
-            percentErrorHistogram = Report_PercentErrorHistogram(graphReportsTab)
-            percentErrorHistogram.draw(equation)
-            graphReportsTab.AddPage(percentErrorHistogram, "Percent Error Histogram")
+            report = Report_PercentErrorHistogram(graphReportsTab)
+            report.draw(self.equation)
+            reportName = "Percent Error Histogram"
+            graphReportsTab.AddPage(report, reportName)
+            self.graphReportsListForPDF.append([report.figure, reportName])
 
         textReportsTab = wx.Notebook(self)
         self.AddPage(textReportsTab, "Text Reports")
 
-        textReport1 = CoefficientAndFitStatisticsReport(textReportsTab, equation)
-        textReportsTab.AddPage(textReport1, "Coefficient And Fit Statistics")
+        report = CoefficientAndFitStatisticsReport(textReportsTab, self.equation)
+        reportName = "Coefficient And Fit Statistics"
+        textReportsTab.AddPage(report, reportName)
+        self.textReportsListForPDF.append([str(report.text.GetValue()), reportName])
 
-        textReport2 = CoefficientsReport(textReportsTab, equation)
-        textReportsTab.AddPage(textReport2, "Coefficient Listing")
+        report = CoefficientsReport(textReportsTab, self.equation)
+        reportName = "Coefficient Listing"
+        textReportsTab.AddPage(report, reportName)
+        self.textReportsListForPDF.append([str(report.text.GetValue()), reportName])
 
-        textReport3 = DataArrayStatisticsReport(textReportsTab, 'Absolute Error Statistics', equation.modelAbsoluteError)
-        textReportsTab.AddPage(textReport3, "Absolute Error Statistics")
+        report = DataArrayStatisticsReport(textReportsTab, 'Absolute Error Statistics', self.equation.modelAbsoluteError)
+        reportName = "Absolute Error Statistics"
+        textReportsTab.AddPage(report, reportName)
+        self.textReportsListForPDF.append([str(report.text.GetValue()), reportName])
 
-        if equation.dataCache.DependentDataContainsZeroFlag != 1:
-            textReport4 = DataArrayStatisticsReport(textReportsTab, 'Percent Error Statistics', equation.modelPercentError)
-            textReportsTab.AddPage(textReport4, "Percent Error Statistics")
+        if self.equation.dataCache.DependentDataContainsZeroFlag != 1:
+            report = DataArrayStatisticsReport(textReportsTab, 'Percent Error Statistics', self.equation.modelPercentError)
+            reportName = "Percent Error Statistics"
+            textReportsTab.AddPage(report, reportName)
+            self.textReportsListForPDF.append([str(report.text.GetValue()), reportName])
 
         sourceCodeTab = wx.Notebook(self)
         self.AddPage(sourceCodeTab, "Source Code")
 
-        sourcecode1 = SourceCodeReport(sourceCodeTab, equation, 'CPP')
-        sourceCodeTab.AddPage(sourcecode1, "C++")
+        report = SourceCodeReport(sourceCodeTab, self.equation, 'CPP')
+        reportName = "C++"
+        sourceCodeTab.AddPage(report, reportName)
+        self.sourceCodeReportsListForPDF.append([str(report.text.GetValue()), reportName])
 
-        sourcecode2 = SourceCodeReport(sourceCodeTab, equation, 'CSHARP')
-        sourceCodeTab.AddPage(sourcecode2, "CSHARP")
+        report = SourceCodeReport(sourceCodeTab, self.equation, 'CSHARP')
+        reportName = "CSHARP"
+        sourceCodeTab.AddPage(report, reportName)
+        self.sourceCodeReportsListForPDF.append([str(report.text.GetValue()), reportName])
 
-        sourcecode3 = SourceCodeReport(sourceCodeTab, equation, 'VBA')
-        sourceCodeTab.AddPage(sourcecode3, "VBA")
+        report = SourceCodeReport(sourceCodeTab, self.equation, 'VBA')
+        reportName = "VBA"
+        sourceCodeTab.AddPage(report, reportName)
+        self.sourceCodeReportsListForPDF.append([str(report.text.GetValue()), reportName])
 
-        sourcecode4 = SourceCodeReport(sourceCodeTab, equation, 'PYTHON')
-        sourceCodeTab.AddPage(sourcecode4, "PYTHON")
+        report = SourceCodeReport(sourceCodeTab, self.equation, 'PYTHON')
+        reportName = "PYTHON"
+        sourceCodeTab.AddPage(report, reportName)
+        self.sourceCodeReportsListForPDF.append([str(report.text.GetValue()), reportName])
 
-        sourcecode5 = SourceCodeReport(sourceCodeTab, equation, 'JAVA')
-        sourceCodeTab.AddPage(sourcecode5, "JAVA")
+        report = SourceCodeReport(sourceCodeTab, self.equation, 'JAVA')
+        reportName = "JAVA"
+        sourceCodeTab.AddPage(report, reportName)
+        self.sourceCodeReportsListForPDF.append([str(report.text.GetValue()), reportName])
 
-        sourcecode6 = SourceCodeReport(sourceCodeTab, equation, 'JAVASCRIPT')
-        sourceCodeTab.AddPage(sourcecode6, "JAVASCRIPT")
+        report = SourceCodeReport(sourceCodeTab, self.equation, 'JAVASCRIPT')
+        reportName = "JAVASCRIPT"
+        sourceCodeTab.AddPage(report, reportName)
+        self.sourceCodeReportsListForPDF.append([str(report.text.GetValue()), reportName])
 
-        sourcecode7 = SourceCodeReport(sourceCodeTab, equation, 'JULIA')
-        sourceCodeTab.AddPage(sourcecode7, "JULIA")
+        report = SourceCodeReport(sourceCodeTab, self.equation, 'JULIA')
+        reportName = "JULIA"
+        sourceCodeTab.AddPage(report, reportName)
+        self.sourceCodeReportsListForPDF.append([str(report.text.GetValue()), reportName])
 
-        sourcecode8 = SourceCodeReport(sourceCodeTab, equation, 'SCILAB')
-        sourceCodeTab.AddPage(sourcecode8, "SCILAB")
+        report = SourceCodeReport(sourceCodeTab, self.equation, 'SCILAB')
+        reportName = "SCILAB"
+        sourceCodeTab.AddPage(report, reportName)
+        self.sourceCodeReportsListForPDF.append([str(report.text.GetValue()), reportName])
 
-        sourcecode9 = SourceCodeReport(sourceCodeTab, equation, 'MATLAB')
-        sourceCodeTab.AddPage(sourcecode9, "MATLAB")
+        report = SourceCodeReport(sourceCodeTab, self.equation, 'MATLAB')
+        reportName = "MATLAB"
+        sourceCodeTab.AddPage(report, reportName)
+        self.sourceCodeReportsListForPDF.append([str(report.text.GetValue()), reportName])
 
-        sourcecode10 = SourceCodeReport(sourceCodeTab, equation, 'FORTRAN90')
-        sourceCodeTab.AddPage(sourcecode10, "FORTRAN90")
+        report = SourceCodeReport(sourceCodeTab, self.equation, 'FORTRAN90')
+        reportName = "FORTRAN90"
+        sourceCodeTab.AddPage(report, reportName)
+        self.sourceCodeReportsListForPDF.append([str(report.text.GetValue()), reportName])
 
         # equation list
-        dim = equation.GetDimensionality()
+        dim = self.equation.GetDimensionality()
         equationList = EquationListReport(self, dim)
         self.AddPage(equationList, "List Of Standard " + str(dim) + "D Equations")
 
@@ -602,6 +649,37 @@ def __init__(self, parent, equation):
         info = AdditionalInfoReport(additionalInfoTab, AdditionalInfo.links)
         additionalInfoTab.AddPage(info, "Web Links")
 
+        # additional information
+        pdfPanel = wx.Panel(self, id=wx.ID_ANY)
+        self.AddPage(pdfPanel, "Save To PDF File")
+
+        btnCreatePDF = wx.Button(pdfPanel, -1, "Save To PDF")
+        self.Bind(wx.EVT_BUTTON, self.createPDF, btnCreatePDF)
+
+
+    def createPDF(self, evt):
+        try:
+            import reportlab
+        except:
+            wx.MessageBox("\nCould not import reportlab.\n\nPlease install using the command\n\n'pip3 install reportlab'", "Error")
+            return
+
+        fd =wx.FileDialog(self, "PDF file name", "", "",
+                                "PDF files (*.pdf)|*.pdf", wx.FD_SAVE | wx.FD_OVERWRITE_PROMPT)
+                                
+        fName = ''
+        if fd.ShowModal() != wx.CANCEL:
+            fName = fd.GetPath()
+        if fName:
+            from . import pdfCode
+            pdfCode.CreatePDF(fName,
+                              self.equation,
+                              self.graphReportsListForPDF,
+                              self.textReportsListForPDF,
+                              self.sourceCodeReportsListForPDF
+                              )
+            wx.MessageBox("\nSuccessfully created PDF file.", "Success")
+
 
 
 # see the included wxNestedTabsExample.py file

guifiles/pdfCode.py

@@ -0,0 +1,146 @@
+import time, os, glob
+import reportlab
+import reportlab.platypus
+from reportlab.pdfgen import canvas
+from reportlab.lib.units import mm
+import reportlab.lib.pagesizes
+
+
+# from http://code.activestate.com/recipes/576832-improved-reportlab-recipe-for-page-x-of-y/
+class NumberedCanvas(canvas.Canvas):
+    def __init__(self, *args, **kwargs):
+        canvas.Canvas.__init__(self, *args, **kwargs)
+        self._saved_page_states = []
+
+    def showPage(self):
+        self._saved_page_states.append(dict(self.__dict__))
+        self._startPage()
+
+    def save(self):
+        """add page info to each page (page x of y)"""
+        num_pages = len(self._saved_page_states)
+        for state in self._saved_page_states:
+            self.__dict__.update(state)
+            self.draw_page_number(num_pages)
+            canvas.Canvas.showPage(self)
+        canvas.Canvas.save(self)
+
+    def draw_page_number(self, page_count):
+        self.setFont("Helvetica", 7)
+        self.drawRightString(200*mm, 20*mm, "Page %d of %d" % (self._pageNumber, page_count))
+        self.drawCentredString(25*mm, 20*mm, 'https://github.com/zunzun/wxPythonFit')
+
+
+
+def CreatePDF(inFileAndPathName, inEquation, inGraphList, inTextList, inSourceCodeList):
+    pageElements = []
+
+    styles = reportlab.lib.styles.getSampleStyleSheet()
+
+    styles.add(reportlab.lib.styles.ParagraphStyle(name='CenteredBodyText', parent=styles['BodyText'], alignment=reportlab.lib.enums.TA_CENTER))
+    styles.add(reportlab.lib.styles.ParagraphStyle(name='SmallCode', parent=styles['Code'], fontSize=6, alignment=reportlab.lib.enums.TA_LEFT)) # 'Code' and wordwrap=CJK causes problems
+
+    myTableStyle = [('ALIGN', (1,1), (-1,-1), 'CENTER'),
+                    ('VALIGN', (1,1), (-1,-1), 'MIDDLE')]
+
+    tableRow = ['wxPythonFit'] # originally included images that are now unused
+
+    table = reportlab.platypus.Table([tableRow], style=myTableStyle)
+
+    pageElements.append(table)
+
+    pageElements.append(reportlab.platypus.XPreformatted('<br/><br/><br/><br/>', styles['CenteredBodyText']))
+
+    pageElements.append(reportlab.platypus.Paragraph(inEquation.GetDisplayName(), styles['CenteredBodyText']))
+    pageElements.append(reportlab.platypus.XPreformatted('<br/><br/>', styles['CenteredBodyText']))
+
+    pageElements.append(reportlab.platypus.XPreformatted('<br/><br/>', styles['CenteredBodyText']))
+    pageElements.append(reportlab.platypus.Paragraph("Created " + time.asctime(time.localtime()), styles['CenteredBodyText']))
+
+    pageElements.append(reportlab.platypus.PageBreak())
+
+    # make a page for each report output, with report name as page header
+    filenamePrefix = 'temp_'
+    index = 1
+    for report in inGraphList:
+        pageElements.append(reportlab.platypus.XPreformatted('<br/><br/>', styles['CenteredBodyText']))
+        
+        # I could not get io.BytesIO and ImageReader to work, using files instead
+        # each image file name must be unique for reportlab's use - delete when done
+        fname = filenamePrefix + str(index) + '.png'
+        report[0].savefig(fname, format='png')
+        im = reportlab.platypus.Image(fname)
+        im._restrictSize(600, 600) # if image is too large for one page
+        im.hAlign = 'CENTER'
+        pageElements.append(im)
+        
+        index += 1
+
+        pageElements.append(reportlab.platypus.PageBreak())
+    
+    for report in inTextList + inSourceCodeList:
+        pageElements.append(reportlab.platypus.Preformatted(report[1], styles['SmallCode']))
+        pageElements.append(reportlab.platypus.XPreformatted('<br/><br/><br/>', styles['CenteredBodyText']))
+
+        replacedText = report[0]
+        
+        if -1 != report[1].find('Coefficients'):
+            reportText = reportText.replace('<sup>', '^')
+            reportText = reportText.replace('<SUP>', '^')
+
+        replacedText = replacedText.replace('\t', '    ') # convert tabs to four spaces
+        replacedText = replacedText.replace('\r\n', '\n')
+
+        rebuiltText = ''
+        for line in replacedText.split('\n'):
+            if line == '':
+                rebuiltText += '\n'
+            else:
+                if line[0] == '<':
+                    splitLine = line.split('>')
+                    if len(splitLine) > 1:
+                        newLine = splitLine[len(splitLine)-1]
+                    else:
+                        newLine = ''
+                else:
+                    newLine = line
+
+                # crude line wrapping
+                if len(newLine) > 500:
+                    rebuiltText += newLine[:100] + '\n'
+                    rebuiltText += newLine[100:200] + '\n'
+                    rebuiltText += newLine[200:300] + '\n'
+                    rebuiltText += newLine[300:400] + '\n'
+                    rebuiltText += newLine[400:500] + '\n'
+                    rebuiltText += newLine[500:] + '\n'
+                elif len(newLine) > 400:
+                    rebuiltText += newLine[:100] + '\n'
+                    rebuiltText += newLine[100:200] + '\n'
+                    rebuiltText += newLine[200:300] + '\n'
+                    rebuiltText += newLine[300:400] + '\n'
+                    rebuiltText += newLine[400:] + '\n'
+                elif len(newLine) > 300:
+                    rebuiltText += newLine[:100] + '\n'
+                    rebuiltText += newLine[100:200] + '\n'
+                    rebuiltText += newLine[200:300] + '\n'
+                    rebuiltText += newLine[300:] + '\n'
+                elif len(newLine) > 200:
+                    rebuiltText += newLine[:100] + '\n'
+                    rebuiltText += newLine[100:200] + '\n'
+                    rebuiltText += newLine[200:] + '\n'
+                elif len(newLine) > 100:
+                    rebuiltText += newLine[:100] + '\n'
+                    rebuiltText += newLine[100:] + '\n'
+                else:
+                    rebuiltText += newLine + '\n'
+                    
+        pageElements.append(reportlab.platypus.Preformatted(rebuiltText, styles['SmallCode']))
+
+        pageElements.append(reportlab.platypus.PageBreak())
+        
+    doc = reportlab.platypus.SimpleDocTemplate(inFileAndPathName, pagesize=reportlab.lib.pagesizes.letter)
+    doc.build(pageElements, canvasmaker=NumberedCanvas)
+
+    # Done, now delete the temporary image files
+    for fname in glob.glob(filenamePrefix + "*.png"):
+        os.remove(fname)