def set_new_sweep(self, file_list_widget=None):
file_list_item = self.file_list.currentItem()
sweep_name = file_list_item.text()
sweep_path = self.sweep_dict[sweep_name]['path']
self.sweep = Sweep(sweep_path)
self.sweep_path = sweep_path
self.sweep.set_pdata(self.pcols.name_func_dict)
for mpl_layout in self.mpl_layouts:
title_wrapped = self.wrap_title(sweep_name, mpl_layout)
mpl_layout.set_title(title_wrapped)
mpl_layout.reset_and_plot(self.sweep)
self.sweep_name = sweep_name
class CDT(object):
def __init__(self, polyline):
self.sweep_context_ = SweepContext(polyline)
self.sweep_ = Sweep()
def __del__(self):
self.sweep_context_ = None
self.sweep_ = None
#TODO:添加洞
def AddHole(self, polyline):
self.sweep_context_.AddHole(polyline)
#TODO:添加一个点
def AddPoint(self, point):
self.sweep_context_.AddPoint(point)
#TODO:三角形点
def Triangulate(self):
self.sweep_.Triangulate(self.sweep_context_)
#TODO:得到Delaunay 三角形
def GetTriangles(self):
return self.sweep_context_.GetTriangles()
#TODO:得到三角形map
def GetMap(self):
return self.sweep_context_.GetMap()
def setUp(self):
path = '../data/2016-09-19#015'
self.sweep = Sweep(path)
data = self.sweep.data
x = data['mL']
y = data['mR']
self.data_h = data_handler_factory(x, y)
def load_sweeps_in_dir(self):
self.sweep_dict = {}
dir_walker = os.walk(self.dir_path, followlinks=False)
for sub_dir_path, _, fnames in dir_walker:
try:
meta = Sweep.load_dir(sub_dir_path, meta_only=True)
except FileNotFoundError:
continue
time_stamp = os.path.split(sub_dir_path)[-1]
sweep_name = time_stamp + ' ' + meta['name']
self.sweep_dict[sweep_name] = {}
self.sweep_dict[sweep_name]['path'] = sub_dir_path
self.sweep_dict[sweep_name]['time_stamp'] = time_stamp
def __init__(self, tello: Tello):
QtWidgets.QVBoxLayout.__init__(self)
self.tello = tello
self.video = Video(tello)
self.sweep = Sweep(tello)
# buttons
self.addWidget(Button("Take Off", tello.takeoff))
self.addWidget(Button("Land", tello.land))
self.addWidget(Button("Stream Video", self.video.start))
self.addWidget(Button("Perimeter sweep", self.sweep.start))
self.addWidget(Button("Manual override", self.sweep.pause))
self.addWidget(
Button("Forward", lambda: tello.move_forward(self.distance)))
self.addWidget(Button("Back", lambda: tello.move_back(self.distance)))
self.addWidget(Button("Left", lambda: tello.move_left(self.distance)))
self.addWidget(Button("Right",
lambda: tello.move_right(self.distance)))
# style
self.setSpacing(20)
import sys
sys.path.append({path_to_folderbrowser_dir})
import numpy as np
from plothandler import plot_handler_factory
from datahandler import data_handler_factory
from sweep import Sweep
import matplotlib.pyplot as plt
import importlib.util
pcols_path = {pcols_path}
spec = importlib.util.spec_from_file_location('', pcols_path)
pcols = importlib.util.module_from_spec(spec)
spec.loader.exec_module(pcols)
sweep = Sweep({sweep_path})
sweep.set_pdata(pcols.name_func_dict)
x = sweep.get_data({x_name})
y = sweep.get_data({y_name})
z = {z_data_code}
data_h = data_handler_factory(x, y, z)
fig, ax = plt.subplots()
plot_h = plot_handler_factory(ax, data_h, plot_dim=1)
ax.ticklabel_format(style='sci', axis='both',
scilimits={scilimits}, useOffset=False)
ax.set_xlabel({xlabel})
ax.set_ylabel({ylabel})
ax.set_xlim({xlim})
ax.set_ylim({ylim})
def lockin_measure():
time.sleep(self.tc_delay)
return np.mean([self.lock.ch1 for i in range(self.averages)])
self.frequency.set_method(self.lock.set_frequency)
self.noise.set_method(lockin_measure)
def instruments_shutdown(self):
self.lock.channel_1_type = 'X'
if __name__ == '__main__':
proc = FieldTest()
# Define a sweep over prarameters
sw = Sweep(proc)
values = np.append(
np.append(np.arange(0.1, 105, 5), np.arange(200, 1100, 50)),
np.arange(2000, 7000, 500)).tolist()
sw.add_parameter_hack(proc.frequency, values)
# Define a writer
sw.add_writer('SweepFrequency.h5', 'NoiseVsFreq-30ms', proc.noise)
proc.instruments_init()
for i in sw:
logging.info("Freq, Noise: %f, %g" %
(proc.frequency.value, proc.noise.value))
proc.instruments_shutdown()
import numpy as np
import pandas as pd
import random as rand
import time
if __name__ == '__main__':
"""
capacity : capacity
graph : Graph
delivery_demand : Delivery demands
"""
capacity, graph, delivery_demand, cityref, datafile = dataset(
'data_3_240.txt')
#-------------------------
tempdata = pd.read_csv("../helpers/raw_data/" + datafile)
citydata = tempdata.set_index("x")
for vehicle_capacity in [1000, 1500, 2500, 2000, 2745, 4000, 11000, 15000]:
sweepAlg = Sweep(vehicle_capacity, delivery_demand, cityref, citydata)
start_time = time.time()
sweepAlg.set_graph(graph)
bestSol = sweepAlg.process()
elasped_time = (time.time() - start_time)
if bestSol is not None:
plot_paths(graph, cityref, bestSol, elasped_time, False)
class FolderBrowser(QMainWindow):
"""
This class is the main window.
Parameters
----------
n_layouts : integer
Number of desired MplLayouts. Cannot be changed after the window has
loaded.
dir_path : string
Fully qualified path to the directory containing data folders.
pcols_path : string
Fully qualified path to the .py file containing functions to calculate
pseudocolumns.
window_title : string
Title of the window.
"""
def __init__(self,
n_layouts,
dir_path,
pcols_path,
window_title='FolderBrowser'):
super().__init__()
self.n_layouts = n_layouts
self.dir_path = dir_path
self.pcols_path = pcols_path
self.assert_exists(dir_path)
self.assert_exists(pcols_path)
self.set_pcols()
self.sweep_name = None
self.sweep = None
self.setWindowTitle(window_title)
self.dock_widgets = []
self.init_statusbar()
self.init_mpl_layouts()
self.load_sweeps_in_dir()
self.init_file_list()
self.setDockNestingEnabled(True)
self.setAttribute(QtCore.Qt.WA_DeleteOnClose)
self.set_hotkeys()
self.set_icon()
self.show()
@show_loading
def set_new_sweep(self, file_list_widget=None):
file_list_item = self.file_list.currentItem()
sweep_name = file_list_item.text()
sweep_path = self.sweep_dict[sweep_name]['path']
self.sweep = Sweep(sweep_path)
self.sweep_path = sweep_path
self.sweep.set_pdata(self.pcols.name_func_dict)
for mpl_layout in self.mpl_layouts:
title_wrapped = self.wrap_title(sweep_name, mpl_layout)
mpl_layout.set_title(title_wrapped)
mpl_layout.reset_and_plot(self.sweep)
self.sweep_name = sweep_name
def init_statusbar(self):
self.statusBar = QtWidgets.QStatusBar()
self.setStatusBar(self.statusBar)
def init_mpl_layouts(self):
self.mpl_layouts = [None] * self.n_layouts
for i in range(self.n_layouts):
self.mpl_layouts[i] = MplLayout(statusBar=self.statusBar,
parent=self)
for i, mpl_layout in enumerate(self.mpl_layouts):
title = 'Plot {}'.format(i)
dock_widget = CustomDockWidget(title, self)
dock_widget.setWidget(mpl_layout)
dock_widget.setAllowedAreas(QtCore.Qt.AllDockWidgetAreas)
self.addDockWidget(QtCore.Qt.TopDockWidgetArea, dock_widget)
self.dock_widgets.append(dock_widget)
self.set_active_layout(self.mpl_layouts[0])
def load_sweeps_in_dir(self):
self.sweep_dict = {}
dir_walker = os.walk(self.dir_path, followlinks=False)
for sub_dir_path, _, fnames in dir_walker:
try:
meta = Sweep.load_dir(sub_dir_path, meta_only=True)
except FileNotFoundError:
continue
time_stamp = os.path.split(sub_dir_path)[-1]
sweep_name = time_stamp + ' ' + meta['name']
self.sweep_dict[sweep_name] = {}
self.sweep_dict[sweep_name]['path'] = sub_dir_path
self.sweep_dict[sweep_name]['time_stamp'] = time_stamp
def init_file_list(self):
names = self.sweep_dict.keys()
self.file_list = FileList(names)
self.file_list.itemClicked.connect(self.set_new_sweep)
self.file_list.itemActivated.connect(self.set_new_sweep)
dock_widget = QDockWidget('Browser', self)
dock_widget.setWidget(self.file_list)
dock_widget.setAllowedAreas(QtCore.Qt.AllDockWidgetAreas)
self.addDockWidget(QtCore.Qt.BottomDockWidgetArea, dock_widget)
self.dock_widgets.append(dock_widget)
def reload_file_list(self):
self.file_list.reload_items()
msg = 'File list reloaded.'
self.statusBar.showMessage(msg, 1000)
def set_active_layout(self, layout):
try:
inactive_str = 'background-color: 10; border: none'
self.active_layout.navi_toolbar.setStyleSheet(inactive_str)
except AttributeError:
pass
active_str = 'background-color: lightblue; border: none'
layout.navi_toolbar.setStyleSheet(active_str)
self.active_layout = layout
def set_hotkeys(self):
self.open_folder_hotkey = QShortcut(QKeySequence('F2'), self)
self.open_folder_hotkey.activated.connect(self.code_to_clipboard)
self.open_folder_hotkey = QShortcut(QKeySequence('F5'), self)
self.open_folder_hotkey.activated.connect(self.reload_file_list)
self.open_folder_hotkey = QShortcut(QKeySequence('F6'), self)
self.open_folder_hotkey.activated.connect(self.reload_pcols)
self.copy_fig_hotkey = QShortcut(QKeySequence('Ctrl+c'), self)
self.copy_fig_hotkey.activated.connect(self.copy_active_fig)
self.open_folder_hotkey = QShortcut(QKeySequence('Ctrl+t'), self)
self.open_folder_hotkey.activated.connect(self.show_text_for_copying)
self.open_folder_hotkey = QShortcut(QKeySequence('Ctrl+w'), self)
self.open_folder_hotkey.activated.connect(self.close)
self.open_folder_hotkey = QShortcut(QKeySequence('Ctrl+Shift+o'), self)
self.open_folder_hotkey.activated.connect(self.open_folder)
def copy_active_fig(self):
self.active_layout.copy_fig_to_clipboard()
active_dock_widget = self.active_layout.parentWidget()
title = active_dock_widget.windowTitle()
msg = 'Figure in ' + title + ' copied to clipboard'
self.statusBar.showMessage(msg, 1000)
def open_folder(self):
if platform.system() != 'Windows':
err_msg = '"Open folder" hotkey only implemented on Windows'
self.statusBar.showMessage(err_msg)
return
norm_path = os.path.normpath(self.sweep_path)
cmd = ['explorer', norm_path]
subprocess.Popen(cmd)
def show_text_for_copying(self):
if self.sweep is None:
msg = 'No sweep selected. Select a sweep to show its information.'
self.statusBar.showMessage(msg, 3000)
return
lay = self.active_layout
title = self.sweep_name
time_stamp = self.sweep_dict[self.sweep_name]['time_stamp']
name = lay.title
diag = TextForCopying(title, time_stamp, name, *lay.labels)
diag.setWindowModality(QtCore.Qt.ApplicationModal)
diag.setModal(True)
diag.exec_()
def set_icon(self):
app_icon = QtGui.QIcon()
icons_dir = os.path.join(os.path.dirname(__file__), 'icons')
for size in (16, 24, 32, 48, 256):
fname = '{}x{}.png'.format(size, size)
fpath = os.path.join(icons_dir, fname)
app_icon.addFile(fpath, QtCore.QSize(size, size))
self.setWindowIcon(app_icon)
def set_pcols(self):
spec = importlib.util.spec_from_file_location('', self.pcols_path)
pcols = importlib.util.module_from_spec(spec)
spec.loader.exec_module(pcols)
self.pcols = pcols
def reload_pcols(self):
self.set_pcols()
self.sweep.set_pdata(self.pcols.name_func_dict)
msg = 'pcols reloaded.'
self.statusBar.showMessage(msg, 1000)
def code_to_clipboard(self):
if self.sweep is None:
msg = 'No sweep selected. Select a sweep to copy its code.'
self.statusBar.showMessage(msg, 3000)
return
own_path = os.path.dirname(os.path.realpath(__file__))
lay = self.active_layout
sel_col_names = lay.sel_col_names
try:
z_data_code = "sweep.get_data('{}')".format(sel_col_names[2])
except IndexError:
z_data_code = None
foo = {
'path_to_folderbrowser_dir': self.prep_path_for_template(own_path),
'pcols_path': self.prep_path_for_template(self.pcols_path),
'sweep_path': self.prep_path_for_template(self.sweep_path),
'x_name': self.pad_str(sel_col_names[0]),
'y_name': self.pad_str(sel_col_names[1]),
'z_data_code': z_data_code,
'plot_dim': lay.plot_dim,
'scilimits': lay.scilimits,
'xlabel': self.pad_str(lay.labels[0]),
'ylabel': self.pad_str(lay.labels[1]),
'xlim': lay.lims[0],
'ylim': lay.lims[1],
'aspect': self.pad_str(lay.aspect),
}
if lay.plot_is_2D:
template_name = 'plot_2D.py'
foo['plot_2D_type'] = self.pad_str(lay.plot_2D_type)
foo['cmap_name'] = self.pad_str(lay.cmap_name)
foo['zlim'] = lay.lims[2]
foo['zlabel'] = self.pad_str(lay.labels[2])
else:
template_name = 'plot_1D.py'
template_path = os.path.join(own_path, 'templates', template_name)
with open(template_path, 'r') as file:
tmpl_str = file.read()
out_str = tmpl_str.format(**foo)
QtWidgets.QApplication.clipboard().setText(out_str)
msg = 'Code for figure copied to clipboard.'
self.statusBar.showMessage(msg, 1000)
@staticmethod
def wrap_title(title, mpl_layout):
"""
The wrapping functionality is not tested against changes in anything,
e.g., font, font size, screen resolution.
"""
fig = mpl_layout.canvas.figure
ax = fig.axes[0]
ext_pixels = ax.get_window_extent()
ext_inches = ext_pixels.transformed(fig.dpi_scale_trans.inverted())
magic_number = 10
letters_per_line = int(ext_inches.width * magic_number)
title_wrapped = '\n'.join(textwrap.wrap(title, letters_per_line))
return title_wrapped
@classmethod
def prep_path_for_template(cls, path):
path_norm = os.path.normpath(path)
path_norm_fslash = path_norm.replace('\\', '/')
padded_path_norm_fslash = cls.pad_str(path_norm_fslash)
return padded_path_norm_fslash
@staticmethod
def pad_str(string):
if isinstance(string, str):
return "'{}'".format(string)
return string
@staticmethod
def assert_exists(path):
if not os.path.exists(path):
raise ValueError('Path {} does not exist'.format(path))
self.noise.assign_method(lockin_measure)
def run(self):
for param in self._parameters:
self._parameters[param].push()
for quant in self._quantities:
self._quantities[quant].measure()
def shutdown_instruments(self):
self.lock.channel_1_type = 'R'
if __name__ == '__main__':
proc = NoiseTest()
# Define a sweep over prarameters
sw = Sweep(proc)
sw.add_parameter(proc.frequency, np.logspace(2, 5, 100))
proc.time_constant.value = 100e-3
# Define a writer
sw.add_plotter('X Noise vs. Frequency', proc.frequency, proc.noise, color="navy", line_width=2)
#sw.add_writer('SweepFrequencyTC.h5', 'NoiseVsFreqAndTC', proc.noise)
sw.run()
# proc.instruments_init()
# for i in sw:
# logging.info("Freq, TC, Noise: %f, %g, %g" % (proc.frequency.value, proc.time_constant.value, proc.noise.value) )
# proc.instruments_shutdown()
def __init__(self, polyline):
self.sweep_context_ = SweepContext(polyline)
self.sweep_ = Sweep()
for param in self._parameters:
self._parameters[param].push()
for quant in self._quantities:
self._quantities[quant].measure()
def instruments_shutdown(self):
self.bop.current = 0.0
if __name__ == '__main__':
proc = CurrentLoop()
proc.field.value = 390.0
# Define a sweep over prarameters
sw = Sweep(proc)
values = np.append(np.arange(-3.0, 3.01, 0.1), np.arange(3.0, -3.01,
-0.1)).tolist()
sw.add_parameter(proc.dc_bias, values)
# Define a writer
sw.add_writer('data/CurrentLoops.h5', 'SWS2129(2,0)G-(011,09)', 'Loop',
proc.voltage)
# sw.add_writer('data/CurrentLoops.h5', 'Test', proc.voltage)
proc.instruments_init()
for i in sw:
logging.info("Bias (V), Lockin Magnitude (V): {:f}, {:g}".format(
proc.dc_bias.value, proc.voltage.value))
proc.instruments_shutdown()