physics/pdme
2
0
Fork 0
Code Issues Pull Requests 9 Projects Releases Wiki Activity

feat: adds methods for converting pdme objects into flatter numpy arrays

Browse Source
This commit is contained in:
Deepak Mallubhotla
2022-04-03 17:53:42 -05:00
parent a390282e45
commit 5c6c4c79d1
2 changed files with 31 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pdme/measurement/oscillating_dipole.py
View File

@@ -55,6 +55,9 @@ class OscillatingDipole():
def s_for_dot_pair(self, r1: numpy.ndarray, r2: numpy.ndarray, f: float) -> float:
return self._alpha(r1) * self._alpha(r2) * self._b(f)
def to_flat_array(self) -> numpy.ndarray:
return numpy.concatenate([self.p, self.s, numpy.array([self.w])])
class OscillatingDipoleArrangement():
'''
@@ -112,3 +115,9 @@ class OscillatingDipoleArrangement():
For a series of pairs of points, each with three coordinates and a frequency, and also a lower error range and upper error range, return a list of the corresponding DotPairRangeMeasurements.
'''
return [self.get_percent_range_dot_pair_measurement(pair_input, low_percent, high_percent) for pair_input in pair_inputs]
def to_numpy_array(self) -> numpy.ndarray:
'''
Returns a numpy array with the canonical representation of each dipole in a nx7 numpy array.
'''
return numpy.array([dipole.to_flat_array() for dipole in self.dipoles])