make_orientation_historgram for calibrated vectors

py4dstem · Apr 15, 2024 · 03312a3 · 03312a3
1 parent a673684
commit 03312a3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/py4DSTEM/process/diffraction/flowlines.py b/py4DSTEM/process/diffraction/flowlines.py
@@ -16,7 +16,7 @@
 
 
 def make_orientation_histogram(
- bragg_peaks: PointListArray = None,
+ bragg_peaks=None,
  radial_ranges: np.ndarray = None,
  orientation_map=None,
  orientation_ind: int = 0,
@@ -38,7 +38,7 @@ def make_orientation_histogram(
  texture OrientationMap generated by the ACOM module of py4DSTEM.
 
  Args:
- bragg_peaks (PointListArray): 2D of pointlists containing centered peak locations.
+ bragg_peaks (BraggVectors):  bragg_vectos containing centered peak locations.
  radial_ranges (np array): Size (N x 2) array for N radial bins, or (2,) for a single bin.
  orientation_map (OrientationMap): Class containing the Euler angles to generate a flowline map.
  orientation_ind (int): Index of the orientation map (default 0)
@@ -110,7 +110,7 @@ def make_orientation_histogram(
  add_data = False
 
  if orientation_map is None:
- p = bragg_peaks.get_pointlist(rx, ry)
+ p = bragg_peaks.cal[rx, ry]
  r2 = p.data["qx"] ** 2 + p.data["qy"] ** 2
  sub = np.logical_and(
  r2 >= radial_ranges_2[a0, 0], r2 < radial_ranges_2[a0, 1]