Added ability to do segmented peak detection on polar transform of DP #726
Conversation
NJMarchese
commented
Jul 8, 2025
NJMarchese
commented
- Added new function to allow for different peak detection parameters for different q-spacings on the diffraction pattern.
- Also enabled the ability to pass a realspace mask onto find_peaks.
- Added an error catch onto the curve_fit of flowlines to give a more informative error message.
- Added kwargs.get method to find_peaks_single_pattern so that the figure and axes are passed correctly. Before wouldn't handle correctly, and if figax was passed it would give an error saying there were multiple values for figax (the default in the code and the one passed in kwargs)
- Added vmin and vmax kwargs pass to show_amorphous_ring_fit
- Added ability to return data for plot_radial_peaks
py4DSTEM/visualize/vis_special.py
Outdated
| **kwargs, | ||
| ) | ||
| if kwargs.get("vmax") is not None: | ||
| vmax = kwargs.pop("vmax") |
There was a problem hiding this comment.
I think if you want to do this you need add vmin and max as flags to the function
There was a problem hiding this comment.
Changed to add vmin and vmax as flags for the function input, but also had to change other lines of code since variable with names vmin and vmax assigned in line 154.
| kwargs_pass=None, | ||
| ): | ||
| """ | ||
| Peak detection function for polar transformations. Loop through all probe positions, |
There was a problem hiding this comment.
missing doc strings...especially for your new contributions with kwargs_pass, it'd be helpful to add detailed docs here!
There was a problem hiding this comment.
I put in some doc strings, but as this is built off the original find_peaks function and the docstrings for that function is what is in find_peaks_segmented, I'm not sure if it's best for me to fill in those belonging to the original function. I added docstrings for the parameters I added for find_peaks_segmented.
| def find_peaks_segmented( | ||
| self, | ||
| mask=None, | ||
| mask_real=None, |
There was a problem hiding this comment.
Can this be added to the other find_peaks function too?
There was a problem hiding this comment.
Maybe. When I looked at it the first time it seemed like it would break a bunch of things, so made it separate initially to keep everything in tact. I can take another look.
|
Hi @NJMarchese, thanks for contributing these features! I have a few small suggestions to your code. |
- Updated `find_peaks_single_pattern` to restore intensity scaling for scatter plot. - Enhanced `find_peaks_segmented` with new mask parameters for improved peak detection. - Modified `plot_radial_peaks` to allow customizable vertical lines. - Adjusted orientation histogram calculation to ensure angle conversion is in radians. - Added vmin and vmax parameters to `show_amorphous_ring_fit` for better visualization control.
2 participants
