Skip to content

Commit 0f81320

Browse files
author
y8z
committed
fix the FT param reading issue
1 parent 6da0bd5 commit 0f81320

File tree

2 files changed

+31
-2
lines changed

2 files changed

+31
-2
lines changed

addie/addiedriver.py

Lines changed: 30 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -143,15 +143,43 @@ def calculate_gr(
143143
pdf_filter = True
144144
if pdf_filter != 'lorch':
145145
print(
146-
'[WARNING] PDF filter {0} is not supported.'.format(pdf_filter))
146+
'[WARNING] PDF filter {0} is not supported.'.format(pdf_filter)
147+
)
147148

148149
q_pystog = simpleapi.mtd[sq_ws_name].readX(0)
149150
sq_pystog = simpleapi.mtd[sq_ws_name].readY(0)
150151
if len(sq_pystog) != len(q_pystog):
151152
sq_pystog = np.insert(sq_pystog, 0, sq_pystog[0])
153+
q_pystog_f = list()
154+
sq_pystog_f = list()
155+
for qi, qv in enumerate(q_pystog):
156+
if min_q <= qv <= max_q:
157+
q_pystog_f.append(qv)
158+
sq_pystog_f.append(sq_pystog[qi])
159+
q_pystog_f = np.array(q_pystog_f)
160+
sq_pystog_f = np.array(sq_pystog_f)
152161
transformer = Transformer()
153162
r_pystog = np.arange(0, max_r + delta_r, delta_r)
154-
r_pystog, gr_pystog, _ = transformer.S_to_G(q_pystog, sq_pystog, r_pystog)
163+
r_pystog, gr_pystog, _ = transformer.S_to_G(
164+
q_pystog_f,
165+
sq_pystog_f,
166+
r_pystog
167+
)
168+
169+
if pdf_type == "G(r)":
170+
pass
171+
else:
172+
if rho0 is None:
173+
warn_msg = "[Warning] No number density provided. "
174+
warn_msg += "A dummy value of 1.0 will be used "
175+
warn_msg += "for the conversion."
176+
print(warn_msg)
177+
rho0 = 1.
178+
179+
gr_pystog = gr_pystog / (4. * np.pi * rho0 * r_pystog) + 1.
180+
181+
if pdf_type == "RDF(r)":
182+
gr_pystog = 4. * np.pi * rho0 * r_pystog**2. * gr_pystog
155183

156184
gr_ws_name = '%s(R)_%s_%d' % (prefix, self._currSqWsName, ws_seq_index)
157185
simpleapi.CreateWorkspace(

addie/calculate_gr/event_handler.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,7 @@ def load_sq(main_window):
104104
# set the UI widgets
105105
main_window.calculategr_ui.doubleSpinBoxQmin.setValue(q_min)
106106
main_window.calculategr_ui.doubleSpinBoxQmax.setValue(q_max)
107+
main_window.calculategr_ui.lineEdit_rho.setText("1.0")
107108

108109
# plot S(Q) - TODO why is it getting the name again?
109110
ws_name = main_window._myController.get_current_sq_name()

0 commit comments

Comments
 (0)