import matplotlib.pyplot as plt import numpy as np from plotting.splineplotter import SplinePlotter from color import Color from dataset import Dataset fig = plt.figure(figsize=[6, 6]) gs = GridSpec(1, 1) ax = fig.add_subplot(gs[0, 0]) dataset = Dataset(1, 1) splineInterval = np.array([0.0, 1.0]) s = SplinePlotter(ax, splineInterval) s.plotGrid(dataset.phi.evaluate, 10, 10, color=Color.DIRECT) xlim = ax.get_xlim() ylim = ax.get_ylim() xextent = xlim[1] - xlim[0] yextent = ylim[1] - ylim[0] xmid = xlim[0] + xextent/2.0 ymid = ylim[0] + yextent/2.0 newextent = max(xextent, yextent) * 1.1 ticks = np.linspace(-1.0, 2.0, 13) ax.set_xticks(ticks) ax.set_yticks(ticks) ax.set_xlim(xmid - newextent/2, xmid + newextent/2)
phi = dataset.phi rho = dataset.rho phiPlane = SplinePlane(phi, splineInterval, newtonTolerance) boundingBox = phiPlane.createBoundingBox() splineModel = SplineModel(None, phiPlane, rho) s = SplinePlotter(ax, splineInterval) s.pointMarker = 'o' s.pointColor = Color.POINT s.rayColor = 'k' p = ParamPlotter(ax2, splineInterval) p.gridColor = Color.DIRECT s.plotGrid(phi.evaluate, 10, 10, color=Color.DIRECT) p.plotGrid(10, 10) s.plotBoundingBox(boundingBox, edgecolor=Color.BOUNDINGBOX) p.pointMarker = None p.pointColor = 'k' p.connectPoints = True splineModel.generateScalarMatrix(boundingBox, 1000, texDimSize, newtonTolerance, paramPlotter=p) p.pointMarker = 'o' p.pointColor = Color.POINT p.connectPoints = False samplingScalars = splineModel.generateScalarMatrix(boundingBox, texDimSize, texDimSize, newtonTolerance, paramPlotter=p, geomPlotter=s) ax.set_xlim([-0.3, 1.1])