Skip to content

Commit

Permalink
Revert "Adjust scalebar logic to combine any specified parameters wit…
Browse files Browse the repository at this point in the history
…h defaults"
  • Loading branch information
smribet authored Sep 23, 2024
1 parent e796f3c commit 34a9658
Showing 1 changed file with 4 additions and 10 deletions.
14 changes: 4 additions & 10 deletions py4DSTEM/visualize/show.py
Original file line number Diff line number Diff line change
Expand Up @@ -186,7 +186,7 @@ def show(
and
>>> show(dp, calibration=calibration, scalebar={'length':0.5,'width':2,
'position':'ul','label':True})
'position':'ul','label':True'})
will display a more customized scalebar.
Expand Down Expand Up @@ -316,7 +316,7 @@ def show(
if returnfig==False (default), the figure is plotted and nothing is returned.
if returnfig==True, return the figure and the axis.
"""
if scalebar is True or scalebar is None:
if scalebar is True:
scalebar = {}

# Alias dep
Expand Down Expand Up @@ -428,31 +428,25 @@ def show(
er = ".calibration attribute must be a Calibration instance"
assert isinstance(cal, Calibration), er
if isinstance(ar, DiffractionSlice):
defaultscalebar = {
scalebar = {
"Nx": ar.data.shape[0],
"Ny": ar.data.shape[1],
"pixelsize": cal.get_Q_pixel_size(),
"pixelunits": cal.get_Q_pixel_units(),
"space": "Q",
"position": "br",
}
for key, value in defaultscalebar.items():
if key not in scalebar.keys():
scalebar[key] = value
pixelsize = cal.get_Q_pixel_size()
pixelunits = cal.get_Q_pixel_units()
elif isinstance(ar, RealSlice):
defaultscalebar = {
scalebar = {
"Nx": ar.data.shape[0],
"Ny": ar.data.shape[1],
"pixelsize": cal.get_R_pixel_size(),
"pixelunits": cal.get_R_pixel_units(),
"space": "Q",
"position": "br",
}
for key, value in defaultscalebar.items():
if key not in scalebar.keys():
scalebar[key] = value
pixelsize = cal.get_R_pixel_size()
pixelunits = cal.get_R_pixel_units()
# get the data
Expand Down

0 comments on commit 34a9658

Please sign in to comment.