ExcelPlotter.py

Go to the documentation of this file.

import win32com.client
import gaudiextra

class Excel:
    def __init__(self):
        self.ex = win32com.client.Dispatch('Excel.Application')
        self.wb = self.ex.Workbooks.Add()
        self.ws = self.wb.WorkSheets.Add()
        self.ws.Name = 'HistoData'
        self.ch = self.wb.Charts.Add()
        self.ch.Name = 'HistoPlot'
    def plot(self, h):
        self.ex.Visible = 0
        heights = h.heights()
        nbin = len(heights)
        xmin, xmax = h.edges()
        for i in range(nbin) :
            self.ws.Cells(i+1,1).value = heights[i]
            self.ws.Cells(i+1,2).value = xmin + i*(xmax-xmin)/nbin 
        self.ch.SetSourceData(self.ws.Range(self.ws.Cells(1,1), self.ws.Cells(nbin+1,1)))
        self.ch.HasTitle = 1
        self.ch.ChartTitle.Text = h.title()
        self.ser = self.ch.SeriesCollection(1)
        self.ser.XValues = self.ws.Range(self.ws.Cells(1,2), self.ws.Cells(nbin+1,2))
        self.ex.Visible = 1

global excel
excel = Excel()