Commitc92f106

committed

Add parameter to use the center or a corner of the rectangle for the position

1 parent4ff193c commitc92f106Copy full SHA for c92f106

File tree

+32

-4

lines changed

+32

-4

lines changed

Lines changed: 9 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -1876,14 +1876,22 @@ class RectangleCollection(_CollectionWithWidthHeightAngle):`
`1876`	`1876`	`refer to the offsets data units. 'xy' differs from all others in`
`1877`	`1877`	`that the angle as plotted varies with the aspect ratio, and equals`
`1878`	`1878`	`the specified angle only when the aspect ratio is unity. Hence`
`1879`		- it behaves the same as the `~.patches.Ellipse` with
	`1879`	+ it behaves the same as the `~.patches.Rectangle` with
`1880`	`1880`	``axes.transData`` as its transform.
	`1881`	`+ centered : bool`
	`1882`	`+ Whether to use the center or the corner of the rectangle to`
	`1883`	`+ define the position of the rectangles. Default is False.`
`1881`	`1884`	`**kwargs`
`1882`	`1885`	Forwarded to `Collection`.
`1883`	`1886`
`1884`	`1887`	`"""`
`1885`	`1888`	`_path_generator=mpath.Path.unit_rectangle`
`1886`	`1889`
	`1890`	`+def__init__(self,args,*kwargs):`
	`1891`	`+ifkwargs.pop("centered",False):`
	`1892`	`+self._path_generator=mpath.Path.unit_rectangle_centered`
	`1893`	`+super().__init__(args,*kwargs)`
	`1894`	`+`
`1887`	`1895`
`1888`	`1896`	`classPatchCollection(Collection):`
`1889`	`1897`	`"""`

Lines changed: 14 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -741,6 +741,20 @@ def unit_rectangle(cls):`
`741`	`741`	`closed=True,readonly=True)`
`742`	`742`	`returncls._unit_rectangle`
`743`	`743`
	`744`	`+_unit_rectangle_centered=None`
	`745`	`+`
	`746`	`+@classmethod`
	`747`	`+defunit_rectangle_centered(cls):`
	`748`	`+"""`
	`749`	+ Return a `Path` instance of the unit rectangle from (-0.5, -0.5) to (0.5, 0.5).
	`750`	`+ """`
	`751`	`+ifcls._unit_rectangle_centeredisNone:`
	`752`	`+cls._unit_rectangle_centered=cls(`
	`753`	`+ [[-0.5,-0.5], [0.5,-0.5], [0.5,0.5], [-0.5,0.5], [-0.5,-0.5]],`
	`754`	`+closed=True,readonly=True`
	`755`	`+ )`
	`756`	`+returncls._unit_rectangle_centered`
	`757`	`+`
`744`	`758`	`_unit_regular_polygons=WeakValueDictionary()`
`745`	`759`
`746`	`760`	`@classmethod`

Lines changed: 9 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -431,10 +431,12 @@ def test_RectangleCollection():`
`431`	`431`
`432`	`432`
`433`	`433`	`@pytest.mark.parametrize(`
`434`		`-'Class,scale',`
`435`		`- [(mcollections.EllipseCollection,0.5), (mcollections.RectangleCollection,1)]`
	`434`	`+'Class, scale, centered',`
	`435`	`+ [(mcollections.EllipseCollection,0.5,None),`
	`436`	`+ (mcollections.RectangleCollection,1,False),`
	`437`	`+ (mcollections.RectangleCollection,1,True)]`
`436`	`438`	`)`
`437`		`-deftest_WidthHeightAngleCollection_setter_getter(Class,scale):`
	`439`	`+deftest_WidthHeightAngleCollection_setter_getter(Class,scale,centered):`
`438`	`440`	`# Test widths, heights and angle setter`
`439`	`441`	`rng=np.random.default_rng(0)`
`440`	`442`
`@@ -445,13 +447,17 @@ def test_WidthHeightAngleCollection_setter_getter(Class, scale):`
`445`	`447`
`446`	`448`	`fig,ax=plt.subplots()`
`447`	`449`
	`450`	`+kwargs= {}`
	`451`	`+ifcenteredisnotNone:`
	`452`	`+kwargs["centered"]=centered`
`448`	`453`	`ec=Class(`
`449`	`454`	`widths=widths,`
`450`	`455`	`heights=heights,`
`451`	`456`	`angles=angles,`
`452`	`457`	`offsets=offsets,`
`453`	`458`	`units='x',`
`454`	`459`	`offset_transform=ax.transData,`
	`460`	`+**kwargs,`
`455`	`461`	`)`
`456`	`462`
`457`	`463`	`assert_array_almost_equal(ec._widths,np.array(widths).ravel()*scale)`

Comments

(0)