Source code for plugins.plot2d

#
##
##  This file is part of pyFormex 2.0  (Mon Sep 14 12:29:05 CEST 2020)
##  pyFormex is a tool for generating, manipulating and transforming 3D
##  geometrical models by sequences of mathematical operations.
##  Home page: http://pyformex.org
##  Project page:  http://savannah.nongnu.org/projects/pyformex/
##  Copyright 2004-2020 (C) Benedict Verhegghe (benedict.verhegghe@ugent.be)
##  Distributed under the GNU General Public License version 3 or later.
##
##  This program is free software: you can redistribute it and/or modify
##  it under the terms of the GNU General Public License as published by
##  the Free Software Foundation, either version 3 of the License, or
##  (at your option) any later version.
##
##  This program is distributed in the hope that it will be useful,
##  but WITHOUT ANY WARRANTY; without even the implied warranty of
##  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
##  GNU General Public License for more details.
##
##  You should have received a copy of the GNU General Public License
##  along with this program.  If not, see http://www.gnu.org/licenses/.
##
"""plot2d.py

Generic 2D plotting functions for pyFormex.
"""

import numpy as np

import pyformex as pf
from pyformex import utils
from pyformex import arraytools as at


plot2d_system = None
plots = []


def init_plot2d_system(plot2d=None):
    global plt, plot2d_system
    if plot2d_system is not None and plot2d == plot2d_system:
        # already initialised
        pass
    else:
        # need initialisation
        plot2d = pf.cfg['plot2d']
        if utils.Module.has(plot2d):
            if plot2d == 'gnuplot':
                import Gnuplot as plt
            elif plot2d == 'qwt':
                # currently not supported
                #from gui.Qwt5.qplt import *
                plot2d = None
            elif plot2d == 'matplotlib':
                import matplotlib.pyplot as plt
        else:
            plot2d = None
            pf.error("I can not draw the plot because I could not load the requested 2d plot library '%s' for your Python version (%s). You need to install one of the supported plot libraries and/or set the appropriate preference in your pyFormex configuration file or via the Settings->Settings Dialog menu item." % (plot2d_system, 3))

    plot2d_system = plot2d
    return plot2d_system


[docs]def showStepPlot(x, y, xlabel='', ylabel='', label='', title=None): """ Show a step plot of x,y data. """ if plot2d_system is None and init_plot2d_system() is None: return if title is None: title = 'TriSurface statistics plot: %s' % label maxlen = min(len(x), len(y)) x = x[:maxlen] y = y[:maxlen] if plot2d_system == 'gnuplot': data = plt.Data(x, y, title=label, with_='steps') g = plt.Gnuplot(persist=0) plots.append(g) if xlabel: g.xlabel(xlabel) if ylabel: g.ylabel(ylabel) g.title(title) g.plot(data) elif plot2d_system == 'matplotlib': g = plt.figure() # create new fig plots.append(g) #plt.close() # close the current figure (if any) plt.step(x, y, where='post', label=label) if xlabel: plt.xlabel(xlabel) if ylabel: plt.ylabel(ylabel) plt.title(title) plt.legend() plt.show()
[docs]def showHistogram(x, y, cumulative=False, xlabel='', ylabel='', label='', title=''): """ Show a histogram of x,y data. """ if cumulative: fill = y[-1] else: fill = y[0] y = at.growAxis(y, len(x)-len(y), fill=fill) if cumulative: ysum = y[-1] else: ysum = y.sum() ylabel = 'occurrences (total=%s)' % ysum showStepPlot(x, y, xlabel=xlabel, ylabel=ylabel, label=label, title=title)
[docs]def createHistogram(data, cumulative=False, **kargs): """ Create a histogram from data """ y, x = np.histogram(data, **kargs) if cumulative: y = y.cumsum() return y, x
def closeAllPlots(): while plots: p = plots.pop(0) if plot2d_system == 'gnuplot': p.close() elif plot2d_system == 'matplotlib': plt.close(p) if __name__ == '__draw__': mi = 0 ma = 10 n = 100 x = np.arange(mi, ma) y = np.random.randint(mi, ma, n) title = "%s random numbers in range(%s,%s)" % (n, mi, ma) showHistogram(x, y, xlabel='values', title=title) # End