ASCEE/lasp
ASCEE/lasp
5
1
Fork 0
Code Issues 5 Pull Requests Actions Releases 14 Activity

Bugfix in bar plot when ylim[0] is not zero

Browse Source
This commit is contained in:
Anne de Jong 2018-06-15 09:09:27 +02:00 committed by J.A. de Jong - ASCEE
parent 1b21cc1873
commit 3311e43ef5
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
lasp/plot/bar.py
View File

@ -279,6 +279,18 @@ class BarScene(QGraphicsScene):
return (n*(self.G*L+dxbars*L) + xL + self.G*L/2) return (n*(self.G*L+dxbars*L) + xL + self.G*L/2)
def getBarRect(self, n, g, yval): def getBarRect(self, n, g, yval):
"""
Returns a bar QRectF.
Args:
n: Bar index (i.e. corresponding to a certain frequency band)
g: Bar group (i.e. corresponding to a certain quantity)
yval: Height of bar, 1 for full lenght, 0 for no length
Returns:
QRectF corresponding to the bar at the right place in the scene
"""
assert yval >= 0 and yval <= 1, "Invalid yval"
Lx = self.xsize-rightoffset-leftoffset Lx = self.xsize-rightoffset-leftoffset
Ly = self.ysize-topoffset - bottomoffset Ly = self.ysize-topoffset - bottomoffset
@ -326,9 +338,12 @@ class BarScene(QGraphicsScene):
assert newydata.shape[0] == N assert newydata.shape[0] == N
assert newydata.shape[1] == G assert newydata.shape[1] == G
# Crop values to be between 0 and 1 # Y-values of the bars should be between 0 and 1.
scalefac = self.ylim[1]-self.ylim[0] scalefac = self.ylim[1]-self.ylim[0]
yvals = np.clip(newydata, self.ylim[0], self.ylim[1])/scalefac yvals = (newydata - self.ylim[0])/scalefac
# Clip values to be between 0 and 1
yvals = np.clip(yvals, 0, 1)
for g in range(G): for g in range(G):
color = self.colors[g] color = self.colors[g]