def get_qaxis_and_gos(self, ienergy, qmin, qmax):
qgosi = self.gos_array[ienergy, :]
if qmax > self.qaxis[-1]:
# Linear extrapolation
g1, g2 = qgosi[-2:]
q1, q2 = self.qaxis[-2:]
gosqmax = get_linear_interpolation((q1, g1), (q2, g2), qmax)
qaxis = np.hstack((self.qaxis, qmax))
qgosi = np.hstack((qgosi, gosqmax))
else:
index = self.qaxis.searchsorted(qmax)
g1, g2 = qgosi[index - 1:index + 1]
q1, q2 = self.qaxis[index - 1:index + 1]
gosqmax = get_linear_interpolation((q1, g1), (q2, g2), qmax)
qaxis = np.hstack((self.qaxis[:index], qmax))
qgosi = np.hstack((qgosi[:index, ], gosqmax))
if qmin > 0:
index = self.qaxis.searchsorted(qmin)
g1, g2 = qgosi[index - 1:index + 1]
q1, q2 = qaxis[index - 1:index + 1]
gosqmin = get_linear_interpolation((q1, g1), (q2, g2), qmin)
qaxis = np.hstack((qmin, qaxis[index:]))
qgosi = np.hstack((
gosqmin,
qgosi[index:],
))
return qaxis, qgosi.clip(0)
def get_qaxis_and_gos(self, ienergy, qmin, qmax):
qgosi = self.gos_array[ienergy, :]
if qmax > self.qaxis[-1]:
# Linear extrapolation
g1, g2 = qgosi[-2:]
q1, q2 = self.qaxis[-2:]
gosqmax = get_linear_interpolation((q1, g1), (q2, g2), qmax)
qaxis = np.hstack((self.qaxis, qmax))
qgosi = np.hstack((qgosi, gosqmax))
else:
index = self.qaxis.searchsorted(qmax)
g1, g2 = qgosi[index - 1:index + 1]
q1, q2 = self.qaxis[index - 1: index + 1]
gosqmax = get_linear_interpolation((q1, g1), (q2, g2), qmax)
qaxis = np.hstack((self.qaxis[:index], qmax))
qgosi = np.hstack((qgosi[:index, ], gosqmax))
if qmin > 0:
index = self.qaxis.searchsorted(qmin)
g1, g2 = qgosi[index - 1:index + 1]
q1, q2 = qaxis[index - 1:index + 1]
gosqmin = get_linear_interpolation((q1, g1), (q2, g2), qmin)
qaxis = np.hstack((qmin, qaxis[index:]))
qgosi = np.hstack((gosqmin, qgosi[index:],))
return qaxis, qgosi.clip(0)