1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
|
from .. import PlotItem
from .. import functions as fn
from ..Qt import QtCore, QtWidgets
from .Exporter import Exporter
__all__ = ['MatplotlibExporter']
"""
It is helpful when using the matplotlib Exporter if your
.matplotlib/matplotlibrc file is configured appropriately.
The following are suggested for getting usable PDF output that
can be edited in Illustrator, etc.
backend : Qt4Agg
text.usetex : True # Assumes you have a findable LaTeX installation
interactive : False
font.family : sans-serif
font.sans-serif : 'Arial' # (make first in list)
mathtext.default : sf
figure.facecolor : white # personal preference
# next setting allows pdf font to be readable in Adobe Illustrator
pdf.fonttype : 42 # set fonts to TrueType (otherwise it will be 3
# and the text will be vectorized.
text.dvipnghack : True # primarily to clean up font appearance on Mac
The advantage is that there is less to do to get an exported file cleaned and ready for
publication. Fonts are not vectorized (outlined), and window colors are white.
"""
_symbol_pg_to_mpl = {
'o' : 'o', # circle
's' : 's', # square
't' : 'v', # triangle_down
't1' : '^', # triangle_up
't2' : '>', # triangle_right
't3' : '<', # triangle_left
'd' : 'd', # thin_diamond
'+' : 'P', # plus (filled)
'x' : 'X', # x (filled)
'p' : 'p', # pentagon
'h' : 'h', # hexagon1
'star' : '*', # star
'arrow_up' : 6, # caretup
'arrow_right' : 5, # caretright
'arrow_down' : 7, # caretdown
'arrow_left' : 4, # caretleft
'crosshair' : 'o', # circle
}
class MatplotlibExporter(Exporter):
Name = "Matplotlib Window"
windows = []
def __init__(self, item):
Exporter.__init__(self, item)
def parameters(self):
return None
def cleanAxes(self, axl):
if type(axl) is not list:
axl = [axl]
for ax in axl:
if ax is None:
continue
for loc, spine in ax.spines.items():
if loc in ['left', 'bottom']:
pass
elif loc in ['right', 'top']:
spine.set_color('none')
# do not draw the spine
else:
raise ValueError('Unknown spine location: %s' % loc)
# turn off ticks when there is no spine
ax.xaxis.set_ticks_position('bottom')
def export(self, fileName=None):
if not isinstance(self.item, PlotItem):
raise Exception("MatplotlibExporter currently only works with PlotItem")
mpw = MatplotlibWindow()
MatplotlibExporter.windows.append(mpw)
fig = mpw.getFigure()
xax = self.item.getAxis('bottom')
yax = self.item.getAxis('left')
# get labels from the graphic item
xlabel = xax.label.toPlainText()
ylabel = yax.label.toPlainText()
title = self.item.titleLabel.text
# if axes use autoSIPrefix, scale the data so mpl doesn't add its own
# scale factor label
xscale = yscale = 1.0
if xax.autoSIPrefix:
xscale = xax.autoSIPrefixScale
if yax.autoSIPrefix:
yscale = yax.autoSIPrefixScale
ax = fig.add_subplot(111, title=title)
ax.clear()
self.cleanAxes(ax)
for item in self.item.curves:
x, y = item.getData()
x = x * xscale
y = y * yscale
opts = item.opts
pen = fn.mkPen(opts['pen'])
if pen.style() == QtCore.Qt.PenStyle.NoPen:
linestyle = ''
else:
linestyle = '-'
color = pen.color().getRgbF()
symbol = opts['symbol']
symbol = _symbol_pg_to_mpl.get(symbol, "")
symbolPen = fn.mkPen(opts['symbolPen'])
symbolBrush = fn.mkBrush(opts['symbolBrush'])
markeredgecolor = symbolPen.color().getRgbF()
markerfacecolor = symbolBrush.color().getRgbF()
markersize = opts['symbolSize']
if opts['fillLevel'] is not None and opts['fillBrush'] is not None:
fillBrush = fn.mkBrush(opts['fillBrush'])
fillcolor = fillBrush.color().getRgbF()
ax.fill_between(x=x, y1=y, y2=opts['fillLevel'], facecolor=fillcolor)
ax.plot(x, y, marker=symbol, color=color, linewidth=pen.width(),
linestyle=linestyle, markeredgecolor=markeredgecolor, markerfacecolor=markerfacecolor,
markersize=markersize)
xr, yr = self.item.viewRange()
ax.set_xbound(xr[0]*xscale, xr[1]*xscale)
ax.set_ybound(yr[0]*yscale, yr[1]*yscale)
ax.set_xlabel(xlabel) # place the labels.
ax.set_ylabel(ylabel)
mpw.draw()
MatplotlibExporter.register()
class MatplotlibWindow(QtWidgets.QMainWindow):
def __init__(self):
from ..widgets import MatplotlibWidget
QtWidgets.QMainWindow.__init__(self)
self.mpl = MatplotlibWidget.MatplotlibWidget()
self.setCentralWidget(self.mpl)
self.show()
def __getattr__(self, attr):
return getattr(self.mpl, attr)
def closeEvent(self, ev):
MatplotlibExporter.windows.remove(self)
self.deleteLater()
|
