matplotlib/boilerplate.py at master · ivanov/matplotlib · GitHub

Repository navigation

History

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

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

226

227

228

229

230

231

232

233

234

235

236

237

238

239

240

241

242

243

244

245

246

247

248

249

250

251

252

253

254

255

256

257

258

259

260

261

262

263

264

265

266

267

268

269

270

271

272

273

274

275

276

277

278

279

280

281

282

283

284

285

286

287

288

289

290

291

292

293

294

295

296

297

298

299

300

301

302

303

304

305

306

307

308

309

310

311

312

313

"""

Script to autogenerate pyplot wrappers.

When this script is run, the current contents of pyplot are

split into generatable and non-generatable content (via the magic header

:attr:`PYPLOT_MAGIC_HEADER`) and the generatable content is overwritten.

Hence, the non-generatable content should be edited in the pyplot.py file

itself, whereas the generatable content must be edited via templates in

this file.

"""

# We did try to do the wrapping the smart way,

# with callable functions and new.function, but could never get the

# docstrings right for python2.2. See

# http://groups.google.com/group/comp.lang.python/browse_frm/thread/dcd63ec13096a0f6/1b14640f3a4ad3dc?#1b14640f3a4ad3dc

# For some later history, see

# http://thread.gmane.org/gmane.comp.python.matplotlib.devel/7068

import os

import inspect

import random

import types

import textwrap

# import the local copy of matplotlib, not the installed one

#sys.path.insert(0, './lib')

from matplotlib.axes import Axes

# this is the magic line that must exist in pyplot, after which the boilerplate content will be

# appended

PYPLOT_MAGIC_HEADER = '################# REMAINING CONTENT GENERATED BY boilerplate.py ##############\n'

PYPLOT_PATH = os.path.join(os.path.dirname(__file__), 'lib',

'matplotlib', 'pyplot.py')

AUTOGEN_MSG = """

# This function was autogenerated by boilerplate.py. Do not edit as

# changes will be lost"""

PLOT_TEMPLATE = AUTOGEN_MSG + """

@_autogen_docstring(Axes.%(func)s)

def %(func)s(%(argspec)s):

%(ax)s = gca()

# allow callers to override the hold state by passing hold=True|False

%(washold)s = %(ax)s.ishold()

%(sethold)s

if hold is not None:

%(ax)s.hold(hold)

try:

%(ret)s = %(ax)s.%(func)s(%(call)s)

draw_if_interactive()

finally:

%(ax)s.hold(%(washold)s)

%(mappable)s

return %(ret)s

"""

# Used for misc functions such as cla/legend etc.

MISC_FN_TEMPLATE = AUTOGEN_MSG + """

@docstring.copy_dedent(Axes.%(func)s)

def %(func)s(%(argspec)s):

%(ret)s = gca().%(func)s(%(call)s)

draw_if_interactive()

return %(ret)s

"""

# Used for colormap functions

CMAP_TEMPLATE = AUTOGEN_MSG + """

def {name}():

'''

set the default colormap to {name} and apply to current image if any.

See help(colormaps) for more information

'''

rc('image', cmap='{name}')

im = gci()

if im is not None:

im.set_cmap(cm.{name})

draw_if_interactive()

"""

def boilerplate_gen():

"""Generator of lines for the automated part of pyplot."""

# these methods are all simple wrappers of Axes methods by the same

# name.

_plotcommands = (

'acorr',

'arrow',

'axhline',

'axhspan',

'axvline',

'axvspan',

'bar',

'barh',

'broken_barh',

'boxplot',

'cohere',

'clabel',

'contour',

'contourf',

'csd',

'errorbar',

'fill',

'fill_between',

'fill_betweenx',

'hexbin',

'hist',

'hist2d',

'hlines',

'imshow',

'loglog',

'pcolor',

'pcolormesh',

'pie',

'plot',

'plot_date',

'psd',

'quiver',

'quiverkey',

'scatter',

'semilogx',

'semilogy',

'specgram',

#'spy',

'stackplot',

'stem',

'step',

'streamplot',

'tricontour',

'tricontourf',

'tripcolor',

'triplot',

'vlines',

'xcorr',

'barbs',

)

_misccommands = (

'cla',

'grid',

'legend',

'table',

'text',

'annotate',

'ticklabel_format',

'locator_params',

'tick_params',

'margins',

'autoscale',

)

cmappable = {

'contour' : 'if %(ret)s._A is not None: sci(%(ret)s)',

'contourf': 'if %(ret)s._A is not None: sci(%(ret)s)',

'hexbin' : 'sci(%(ret)s)',

'scatter' : 'sci(%(ret)s)',

'pcolor' : 'sci(%(ret)s)',

'pcolormesh': 'sci(%(ret)s)',

'hist2d' : 'sci(%(ret)s[-1])',

'imshow' : 'sci(%(ret)s)',

#'spy' : 'sci(%(ret)s)', ### may return image or Line2D

'quiver' : 'sci(%(ret)s)',

'specgram' : 'sci(%(ret)s[-1])',

'streamplot' : 'sci(%(ret)s)',

'tricontour' : 'if %(ret)s._A is not None: sci(%(ret)s)',

'tricontourf': 'if %(ret)s._A is not None: sci(%(ret)s)',

'tripcolor' : 'sci(%(ret)s)',

}

def format_value(value):

"""

Format function default values as needed for inspect.formatargspec.

The interesting part is a hard-coded list of functions used

as defaults in pyplot methods.

"""

if isinstance(value, types.FunctionType):

if value.__name__ in ('detrend_none', 'window_hanning'):

return '=mlab.' + value.__name__

if value.__name__ == 'mean':

return '=np.' + value.__name__

raise ValueError(('default value %s unknown to boilerplate.' + \

'formatvalue') % value)

return '='+repr(value)

text_wrapper = textwrap.TextWrapper(break_long_words=False)

for fmt, cmdlist in [(PLOT_TEMPLATE, _plotcommands),

(MISC_FN_TEMPLATE, _misccommands)]:

for func in cmdlist:

# For some commands, an additional line is needed to set the

# color map

if func in cmappable:

mappable = cmappable[func] % locals()

else:

mappable = ''

# Get argspec of wrapped function

args, varargs, varkw, defaults = inspect.getargspec(getattr(Axes, func))

args.pop(0) # remove 'self' argument

if defaults is None:

defaults = ()

# How to call the wrapped function

call = []

for i, arg in enumerate(args):

if len(defaults) < len(args) - i:

call.append('%s' % arg)

else:

call.append('%s=%s' % (arg, arg))

if varargs is not None:

call.append('*'+varargs)

if varkw is not None:

call.append('**'+varkw)

call = ', '.join(call)

text_wrapper.width = 80 - 19 - len(func)

join_with = '\n' + ' ' * (18 + len(func))

call = join_with.join(text_wrapper.wrap(call))

# Add a hold keyword argument if needed (fmt is PLOT_TEMPLATE) and

# possible (if *args is used, we can't just add a hold

# argument in front of it since it would gobble one of the

# arguments the user means to pass via *args)

if varargs:

sethold = "hold = %(varkw)s.pop('hold', None)" % locals()

elif fmt is PLOT_TEMPLATE:

args.append('hold')

defaults = defaults + (None,)

sethold = ''

# Now we can build the argspec for defining the wrapper

argspec = inspect.formatargspec(args, varargs, varkw, defaults,

formatvalue=format_value)

argspec = argspec[1:-1] # remove parens

text_wrapper.width = 80 - 5 - len(func)

join_with = '\n' + ' ' * (5 + len(func))

argspec = join_with.join(text_wrapper.wrap(argspec))

# A gensym-like facility in case some function takes an

# argument named washold, ax, or ret

washold, ret, ax = 'washold', 'ret', 'ax'

bad = set(args) | set((varargs, varkw))

while washold in bad or ret in bad or ax in bad:

washold = 'washold' + str(random.randrange(10**12))

ret = 'ret' + str(random.randrange(10**12))

ax = 'ax' + str(random.randrange(10**12))

# Since we can't avoid using some function names,

# bail out if they are used as argument names

for reserved in ('gca', 'gci', 'draw_if_interactive'):

if reserved in bad:

msg = 'Axes method %s has kwarg named %s' % (func, reserved)

raise ValueError(msg)

yield fmt % locals()

cmaps = (

'autumn',

'bone',

'cool',

'copper',

'flag',

'gray' ,

'hot',

'hsv',

'jet' ,

'pink',

'prism',

'spring',

'summer',

'winter',

'spectral'

)

# add all the colormaps (autumn, hsv, ....)

for name in cmaps:

yield CMAP_TEMPLATE.format(name=name)

yield ''

yield '_setup_pyplot_info_docstrings()'

def build_pyplot():

pyplot_path = os.path.join(os.path.dirname(__file__), 'lib',

'matplotlib', 'pyplot.py')

pyplot_orig = open(pyplot_path, 'r').readlines()

try:

pyplot_orig = pyplot_orig[:pyplot_orig.index(PYPLOT_MAGIC_HEADER)+1]

except IndexError:

raise ValueError('The pyplot.py file *must* have the exact line: %s' % PYPLOT_MAGIC_HEADER)

pyplot = open(pyplot_path, 'w')

pyplot.writelines(pyplot_orig)

pyplot.write('\n')

pyplot.writelines(boilerplate_gen())

if __name__ == '__main__':

# Write the matplotlib.pyplot file

build_pyplot()

matplotlib/boilerplate.py at master · ivanov/matplotlib · GitHub

Repository navigation

Expand file tree

Breadcrumbs

boilerplate.py

Latest commit

History

Breadcrumbs

boilerplate.py

File metadata and controls