Skip to content

Commit 146157a

Browse files
removed unnecessary if statement checks for both dyn and stat DDSs
1 parent c2ec2b9 commit 146157a

File tree

1 file changed

+32
-35
lines changed

1 file changed

+32
-35
lines changed

labscript_devices/AD9959DDSSweeper/labscript_devices.py

Lines changed: 32 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -207,48 +207,45 @@ def generate_code(self, hdf5_file):
207207
elif isinstance(output, StaticDDS):
208208
stat_DDSs[channel] = output
209209

210-
if dyn_DDSs:
211-
for connection in dyn_DDSs:
212-
dds = dyn_DDSs[connection]
213-
dds.frequency.raw_output, dds.frequency.scale_factor = self.quantise_freq(dds.frequency.raw_output, dds)
214-
dds.phase.raw_output, dds.phase.scale_factor = self.quantise_phase(dds.phase.raw_output, dds)
215-
dds.amplitude.raw_output, dds.amplitude.scale_factor = self.quantise_amp(dds.amplitude.raw_output, dds)
210+
for connection in dyn_DDSs:
211+
dds = dyn_DDSs[connection]
212+
dds.frequency.raw_output, dds.frequency.scale_factor = self.quantise_freq(dds.frequency.raw_output, dds)
213+
dds.phase.raw_output, dds.phase.scale_factor = self.quantise_phase(dds.phase.raw_output, dds)
214+
dds.amplitude.raw_output, dds.amplitude.scale_factor = self.quantise_amp(dds.amplitude.raw_output, dds)
216215

217-
dyn_dtypes = {'names':['%s%d' % (k, i) for i in dyn_DDSs for k in ['freq', 'amp', 'phase'] ],
218-
'formats':[f for i in dyn_DDSs for f in ('<u4', '<u2', '<u2')]}
216+
dyn_dtypes = {'names':['%s%d' % (k, i) for i in dyn_DDSs for k in ['freq', 'amp', 'phase'] ],
217+
'formats':[f for i in dyn_DDSs for f in ('<u4', '<u2', '<u2')]}
219218

220-
clockline = self.parent_clock_line
221-
pseudoclock = clockline.parent_device
222-
times = pseudoclock.times[clockline]
219+
clockline = self.parent_clock_line
220+
pseudoclock = clockline.parent_device
221+
times = pseudoclock.times[clockline]
223222

224-
dyn_table = np.zeros(len(times), dtype=dyn_dtypes)
223+
dyn_table = np.zeros(len(times), dtype=dyn_dtypes)
225224

226-
for i, dds in dyn_DDSs.items():
227-
dyn_table['freq%d' % i][:] = dds.frequency.raw_output
228-
dyn_table['amp%d' % i][:] = dds.amplitude.raw_output
229-
dyn_table['phase%d' % i][:] = dds.phase.raw_output
225+
for i, dds in dyn_DDSs.items():
226+
dyn_table['freq%d' % i][:] = dds.frequency.raw_output
227+
dyn_table['amp%d' % i][:] = dds.amplitude.raw_output
228+
dyn_table['phase%d' % i][:] = dds.phase.raw_output
230229

231-
if stat_DDSs:
232-
# conversion to AD9959 units
233-
for connection in stat_DDSs:
234-
dds = stat_DDSs[connection]
235-
dds.frequency.raw_output, dds.frequency.scale_factor = self.quantise_freq(dds.frequency.raw_output, dds)
236-
dds.phase.raw_output, dds.phase.scale_factor = self.quantise_phase(dds.phase.raw_output, dds)
237-
dds.amplitude.raw_output, dds.amplitude.scale_factor = self.quantise_amp(dds.amplitude.raw_output, dds)
238-
239-
static_dtypes = {
240-
'names':['%s%d' % (k, i) for i in stat_DDSs for k in ['freq', 'amp', 'phase'] ],
241-
'formats':[f for i in stat_DDSs for f in ('<u4', '<u2', '<u2')]
242-
}
230+
# conversion to AD9959 units
231+
for connection in stat_DDSs:
232+
dds = stat_DDSs[connection]
233+
dds.frequency.raw_output, dds.frequency.scale_factor = self.quantise_freq(dds.frequency.raw_output, dds)
234+
dds.phase.raw_output, dds.phase.scale_factor = self.quantise_phase(dds.phase.raw_output, dds)
235+
dds.amplitude.raw_output, dds.amplitude.scale_factor = self.quantise_amp(dds.amplitude.raw_output, dds)
243236

244-
static_table = np.zeros(1, dtype=static_dtypes)
245-
246-
for connection in list(stat_DDSs.keys()):
247-
sdds = stat_DDSs[connection]
248-
static_table['freq%d' % connection] = sdds.frequency.raw_output[0]
249-
static_table['amp%d' % connection] = sdds.amplitude.raw_output[0]
250-
static_table['phase%d' % connection] = sdds.phase.raw_output[0]
237+
static_dtypes = {
238+
'names':['%s%d' % (k, i) for i in stat_DDSs for k in ['freq', 'amp', 'phase'] ],
239+
'formats':[f for i in stat_DDSs for f in ('<u4', '<u2', '<u2')]
240+
}
241+
242+
static_table = np.zeros(1, dtype=static_dtypes)
251243

244+
for connection in list(stat_DDSs.keys()):
245+
sdds = stat_DDSs[connection]
246+
static_table['freq%d' % connection] = sdds.frequency.raw_output[0]
247+
static_table['amp%d' % connection] = sdds.amplitude.raw_output[0]
248+
static_table['phase%d' % connection] = sdds.phase.raw_output[0]
252249

253250
# if no channels are being used, no need to continue
254251
if not dyn_DDSs and not stat_DDSs:

0 commit comments

Comments
 (0)