1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
import numpy as np
from dipy.viz import fos
import dipy.core.geometry as geometry
import matplotlib.pyplot as mplp
def plot_sphere(v,key):
r = fos.ren()
fos.add(r,fos.point(v,fos.green, point_radius= 0.01))
fos.show(r, title=key, size=(1000,1000))
def plot_lambert(v,key,centre=np.array([0,0])):
lamb = geometry.lambert_equal_area_projection_cart(*v.T).T
(y1,y2) = lamb
radius = np.sum(lamb**2,axis=0) < 1
#print inner
#print y1[inner]
#print y1[-inner]
fig = mplp.figure(facecolor='w')
current = fig.add_subplot(111)
current.patch.set_color('k')
current.plot(y1[radius],y2[radius],'.g')
current.plot(y1[-radius],y2[-radius],'.r')
current.plot([0.],[0.],'ob')
#current.patches.Circle(*centre, radius=50, color='w', fill=True, alpha=0.7)
current.axes.set_aspect(aspect = 'equal', adjustable = 'box')
current.title.set_text(key)
fig.show()
fig.waitforbuttonpress()
mplp.close()
|
