cSAXS/cdtools
0
0
Fork 0
mirror of https://github.com/cdtools-developers/cdtools.git synced 2026-09-30 13:52:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add the frantic work over the past few weeks, next challenge is to organize and test it

Browse Source
This commit is contained in:
Abe Levitan
2019-04-23 14:50:16 -04:00
parent ace81cc66d
commit bf8dbe086a
14 changed files with 815 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CDTools/tools/plotting.py
+3 -3
View File
@@ -64,7 +64,7 @@ def plot_amplitude(im, fig = None, basis = np.array([[0,-1], [-1,0], [0,0]]), **
if fig is None:
fig = plt.figure()
ax = fig.add_subplot(111, **kwargs)
basis_norm = np.linalg.norm(basis, axis = -1)
basis_norm = np.linalg.norm(basis, axis = 0)
if isinstance(im, t.Tensor):
absolute = cmath.cabs(im).detach().cpu().numpy()
else:
@@ -93,7 +93,7 @@ def plot_phase(im, fig = None, basis = np.array([[0,-1], [-1,0], [0,0]]), **kwa
phase = cmath.cphase(im).detach().cpu().numpy()
else:
phase = np.angle(im)
basis_norm = np.linalg.norm(basis, axis = -1)
basis_norm = np.linalg.norm(basis, axis = 0)
try: plt.imshow(phase, cmap = 'twilight', extent = [0, phase.shape[-1]*basis_norm[1], 0, phase.shape[-2]*basis_norm[0]])
except: plt.imshow(phase, cmap = 'hsv', extent = [0, phase.shape[-1]*basis_norm[1], 0, phase.shape[-2]*basis_norm[0]])
plt.colorbar()
@@ -119,7 +119,7 @@ def plot_colorized(im, fig = None, basis = np.array([[0,-1], [-1,0], [0,0]]), *
ax = fig.add_subplot(111, **kwargs)
if isinstance(im, t.Tensor):
im = cmath.torch_to_complex(im.detach().cpu())
basis_norm = np.linalg.norm(basis, axis = -1)
basis_norm = np.linalg.norm(basis, axis = 0)
colorized = colorize(im)
plt.imshow(colorized, extent = [0, im.shape[-1]*basis_norm[1], 0, im.shape[-2]*basis_norm[0]])
return fig