bmtool 0.3.7__py3-none-any.whl → 0.3.9__py3-none-any.whl

bmtool/bmplot.py

@@ -409,12 +409,13 @@ def cell_rotation_3d(**kwargs):
     populations_list = kwargs["populations"]
     config = kwargs["config"]
     group_keys = kwargs["group_by"]
-    title = kwargs["title"]
+    title = kwargs.get("title")
     save_file = kwargs["save_file"]
     quiver_length = kwargs["quiver_length"]
     arrow_length_ratio = kwargs["arrow_length_ratio"]
     group = kwargs["group"]
-    max_cells = kwargs["max_cells"]
+    max_cells = kwargs.get("max_cells",999999999)
+    init_vector = kwargs.get("init_vector","1,0,0")
 
     nodes = util.load_nodes_from_config(config)
 
@@ -480,8 +481,19 @@ def cell_rotation_3d(**kwargs):
             #Convert to arrow direction
             from scipy.spatial.transform import Rotation as R
             uvw = pd.DataFrame([U,V,W]).T
-            rots = R.from_euler('zyx', uvw).as_rotvec().T
+            init_vector = init_vector.split(',')
+            init_vector = np.repeat([init_vector],len(X),axis=0)
+            
+            # To get the final cell orientation after rotation, 
+            # you need to use function Rotaion.apply(init_vec), 
+            # where init_vec is a vector of the initial orientation of a cell
+            #rots = R.from_euler('xyz', uvw).apply(init_vector.astype(float))
+            #rots = R.from_euler('xyz', pd.DataFrame([rots[:,0],rots[:,1],rots[:,2]]).T).as_rotvec().T
+
+            rots = R.from_euler('zyx', uvw).apply(init_vector.astype(float)).T
             h = ax.quiver(X, Y, Z, rots[0],rots[1],rots[2],color=color,label=group_name, arrow_length_ratio = arrow_length_ratio, length=quiver_length)
+
+            #h = ax.quiver(X, Y, Z, rots[0],rots[1],rots[2],color=color,label=group_name, arrow_length_ratio = arrow_length_ratio, length=quiver_length)
             ax.scatter(X,Y,Z,color=color,label=group_name)
             handles.append(h)
     if not handles:

bmtool/plot_commands.py

@@ -136,15 +136,17 @@ def cell(ctx,title,save_file):
 @click.option('--max-cells', type=click.INT, default=999999999, help="max number of cells to display [default: 999999999]")
 @click.option('--quiver-length', type=click.FLOAT, default=10, help="how long the arrows should be [default: 10]")
 @click.option('--arrow-length-ratio', type=click.FLOAT, default=0.2, help="ratio for the arrow of the quiver [default: 0.2]")
+@click.option('--init-vector', type=click.STRING, default="1,0,0", help="comma delimited initial rotation vector specified by pt3dadd [default: 0,0,1]")
 @click.pass_context
-def rotation_3d(ctx,populations,group_by,group,max_cells,quiver_length,arrow_length_ratio):
+def rotation_3d(ctx,populations,group_by,group,max_cells,quiver_length,arrow_length_ratio,init_vector):
     cell_rotation_3d(config=ctx.obj['config'],**ctx.obj['cell'],
                      populations=populations,
                      group_by=group_by,
                      group=group,
                      max_cells=max_cells,
                      quiver_length=quiver_length,
-                     arrow_length_ratio=arrow_length_ratio)
+                     arrow_length_ratio=arrow_length_ratio,
+                     init_vector=init_vector)
     if ctx.obj['display']:
         plt.show()
 

{bmtool-0.3.7.dist-info → bmtool-0.3.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: bmtool
-Version: 0.3.7
+Version: 0.3.9
 Summary: BMTool
 Home-page: https://github.com/tjbanks/bmtool
 Download-URL: 
@@ -352,6 +352,34 @@ This will plot cells in the `hippocampus` network, using the `pop_name` as the c
 
 All  `point_process` cell types will be ignored since they do not have physical locations.
 
+### Plot 3d cell location and rotation
+Plot the location and rotation of your cells. Plot all of your cells with a single command
+```
+bmtool plot cell rotation
+```
+![bmtool](./figures/rotation3d_1.png "3d Rotation Figure")
+
+Customize your plot by limiting the cells you want or selecting a max number of cells to plot.
+```
+bmtool plot --config simulation_configECP.json cell rotation --group-by pop_name --group CR --max-cells 100 --quiver-length 100 --arrow-length-ratio 0.25
+```
+![bmtool](./figures/rotation3d_2.png "3d Rotation Figure")
+
+Code
+```
+from bmtool import
+from bmtool import bmplot
+
+bmplot.cell_rotation_3d(config=config,
+                     populations=populations,
+                     group_by=group_by,
+                     group=group,
+                     title=title,
+                     max_cells=max_cells,
+                     quiver_length=quiver_length,
+                     arrow_length_ratio=arrow_length_ratio)
+```
+
 ### Plotting Current Clamp and Spike Train Info
 To plot all current clamp info involved in a simulation, use the following command (uses 'simulation_config.json' as default)
 ```

{bmtool-0.3.7.dist-info → bmtool-0.3.9.dist-info}/RECORD

@@ -1,9 +1,9 @@
 bmtool/__init__.py,sha256=FrcSFNlD5lhNIWDYrX0kZiBYS_lDSK5rBkSXMuDoEWg,141
 bmtool/__main__.py,sha256=61s3Dd4-brkJt_amGpGapdIoV-3QNPbjRvIVQHlfZZI,676
-bmtool/bmplot.py,sha256=0CtiNXOMugzxQhKf8uz9hyEn9ZFN6BnOBiobxlCrXmI,28949
+bmtool/bmplot.py,sha256=pf7TKzjHAEIqRCQiEdzJho3IXCBhr2-VDopRIP54EzQ,29723
 bmtool/manage.py,sha256=ALYltiBaljm1pBtz-0ifvrkrzlPNDiIu7Lr5q3-_oFU,688
 bmtool/plot.py,sha256=C1novNnisc_7IuQbY4p4ZbYcgj11UttUIqnHtpqHoJs,25374
-bmtool/plot_commands.py,sha256=ivA3W7faMMhATgsJlbSxX8zA_yMR1oVvvYz5d-xCqt8,12315
+bmtool/plot_commands.py,sha256=U3up0PLZN8drcjyLsbrhsA4mVh9iXX-5dt0NsXgfjxI,12528
 bmtool/singlecell.py,sha256=3qWaELnCuYuYtxDOQwa-SzmQ8F8L32Q839FGy35LII0,14900
 bmtool/debug/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bmtool/debug/commands.py,sha256=5xtXtAJhYMrlxM9vF6jvfLhE-kOTwDnibmDBIrPhYrA,602
@@ -30,9 +30,9 @@ bmtools/cli/plugins/util/commands.py,sha256=7BRj6PEm5XCQaYk4EhIGvqXNcsCbykw3I9l9
 bmtools/cli/plugins/util/util.py,sha256=ir7omOJgAMO3RxIKI39ZpIzI6to9LqToI6Dck4wkLyw,50126
 bmtools/cli/plugins/util/neuron/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bmtools/cli/plugins/util/neuron/celltuner.py,sha256=d3WTbEq4WW6koHc5TdiUGGm2JWJU3LuPeiAFtUiaMgM,89264
-bmtool-0.3.7.dist-info/LICENSE,sha256=eOgqgvoeIS9FFSJjHbxCyfHyXdHUhk2YHLqQlWWzdXk,1089
-bmtool-0.3.7.dist-info/METADATA,sha256=ymdveEduCuBjnO9l12e6M-H8Vhqw5fD4FGjbuirEh08,17013
-bmtool-0.3.7.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-bmtool-0.3.7.dist-info/entry_points.txt,sha256=0-BHZ6nUnh0twWw9SXNTiRmKjDnb1VO2DfG_-oprhAc,45
-bmtool-0.3.7.dist-info/top_level.txt,sha256=gpd2Sj-L9tWbuJEd5E8C8S8XkNm5yUE76klUYcM-eWM,7
-bmtool-0.3.7.dist-info/RECORD,,
+bmtool-0.3.9.dist-info/LICENSE,sha256=eOgqgvoeIS9FFSJjHbxCyfHyXdHUhk2YHLqQlWWzdXk,1089
+bmtool-0.3.9.dist-info/METADATA,sha256=c5iCJrjQ193_4LcJ1maVtBkBz3hF8y2GwmMlzN4CGIc,17987
+bmtool-0.3.9.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
+bmtool-0.3.9.dist-info/entry_points.txt,sha256=0-BHZ6nUnh0twWw9SXNTiRmKjDnb1VO2DfG_-oprhAc,45
+bmtool-0.3.9.dist-info/top_level.txt,sha256=gpd2Sj-L9tWbuJEd5E8C8S8XkNm5yUE76klUYcM-eWM,7
+bmtool-0.3.9.dist-info/RECORD,,