class AxisRayW(QtGui.QWidget):
r"""A widget for creating axis aligned rays from a preselected YtObject.
Parameters
----------
parent : ActiveAcquisitionW
The widget which ultimately gave rise to this widget. The widget
containing data objects.
parent_widget : QtGui.QWidget
The widget which will display this widget as one of its children."""
axis_dict = {'x': 0, 'y': 1, 'z': 2}
def __init__(self, parent, parent_widget):
super(AxisRayW, self).__init__()
self.parent = parent
self.parent_widget = parent_widget
axis_label = QtGui.QLabel('Axis of Alignment:')
self.axisw = QtGui.QComboBox()
self.axisw.addItems(list(self.axis_dict.keys()))
axis_super_w = QtGui.QWidget()
axis_layout = QtGui.QHBoxLayout()
axis_layout.addWidget(axis_label)
axis_layout.addWidget(self.axisw)
axis_super_w.setLayout(axis_layout)
self.coord_units_w = CoordinateUnitsW()
coord_label = QtGui.QLabel('Coordinate Ray Intersects:')
self.set_coord_widgets(None)
coord_super_w = QtGui.QWidget()
coord_layout = QtGui.QHBoxLayout()
coord_layout.addWidget(coord_label)
coord_layout.addWidget(self.coord1_w)
coord_layout.addWidget(self.coord2_w)
coord_super_w.setLayout(coord_layout)
self.axisw.activated.connect(self.set_coord_widgets)
self.data_source = DataSourceW(self.parent)
self.field_parameters = FieldParametersW()
self.object_name = NameW(self.parent.active_data_object.name +
' axis aligned ray')
self.generate_object_btn = QtGui.QPushButton('Generate Object')
self.generate_object_btn.clicked.connect(self.generate_object)
layout = QtGui.QVBoxLayout()
layout.addWidget(axis_super_w)
layout.addWidget(self.coord_units_w)
layout.addWidget(coord_super_w)
layout.addWidget(self.field_parameters)
layout.addWidget(self.data_source)
layout.addWidget(self.object_name)
layout.addWidget(self.generate_object_btn)
self.setLayout(layout)
self.parent_widget.layout.addWidget(self)
self.parent_widget.show()
def generate_object(self):
r"""This function generates an axis aligned ray from a YtObject with
parameters specified by the user in the widget."""
axis = self.axis_dict[self.axisw.currentText()]
units = self.coord_units_w.get_unit()
coord1 = self.coord1_w.get_coordinate()
coord2 = self.coord1_w.get_coordinate()
source = self.parent.active_data_object.data
name = self.object_name.get_name()
if units is not None:
coord = YTArray([coord1, coord2], units)
coord = source.arr(coord).in_units('code_length')
else:
coord = [coord1, coord2]
if self.field_parameters.get_field_parameters() == 'None':
if self.data_source.get_data_source() is None:
ray = source.ortho_ray(axis, coord)
else:
dsource = self.data_source.get_data_source()
ray = source.ortho_ray(axis, coord, data_source=dsource)
new_object = YtDataObject(ray, name)
self.parent.add_data_object(new_object)
def set_coord_widgets(self, index):
r"""This function sets the labels for the axes coordinate widgets
based on user input or startup conditions.
Parameters
----------
index : int
The index of the selected axis."""
if index is None:
axis = self.axis_dict[self.axisw.currentText()]
else:
axis = index
plane_axes = [key for key in list(self.axis_dict.keys())
if self.axis_dict[key] != axis]
if hasattr(self, 'coord1_w') and hasattr(self, 'coord2_w'):
self.coord1_w.set_label(plane_axes[0])
self.coord2_w.set_label(plane_axes[1])
else:
self.coord1_w = CoordinateW(plane_axes[0])
self.coord2_w = CoordinateW(plane_axes[1])
class SphereW(QtGui.QWidget):
r"""A widget for generating a YTSphere from a preselected YtObject.
Parameters
----------
parent : ActiveAcquisitionW
The widget which ultimately gave rise to this widget. The widget
containing data objects.
parent_widget : QtGui.QWidget
The widget which will display this widget as one of its children."""
def __init__(self, parent, parent_widget):
super(SphereW, self).__init__()
self.parent = parent
self.parent_widget = parent_widget
self.center_units = CoordinateUnitsW()
self.center_units.label.setText("Center Coordinate Units:")
self.center = CartCoordinateComboW()
self.center.coord1.set_label('Center X:')
self.center.coord2.set_label('Center Y:')
self.center.coord3.set_label('Center Z:')
self.radius_units = CoordinateUnitsW()
self.radius_units.label.setText("Radius Length Units:")
self.radius = CoordinateW('Radius:')
self.field_parameters = FieldParametersW()
self.data_source = DataSourceW(self.parent)
self.name = NameW(self.parent.active_data_object.name + "_sphere")
self.generate_btn = QtGui.QPushButton('Generate Object')
self.generate_btn.clicked.connect(self.generate_object)
layout = QtGui.QVBoxLayout()
layout.addWidget(self.center_units)
layout.addWidget(self.center)
layout.addWidget(self.radius_units)
layout.addWidget(self.radius)
layout.addWidget(self.field_parameters)
layout.addWidget(self.data_source)
layout.addWidget(self.name)
layout.addWidget(self.generate_btn)
self.setLayout(layout)
self.parent_widget.layout.addWidget(self)
self.parent_widget.show()
def generate_object(self):
r"""Generates a YTSphere based on the parameters provided by the user
in the widget."""
source = self.parent.active_data_object.data
center = self.center.get_coordinates()
center_units = self.center_units.get_unit()
radius = self.radius.get_coordinate()
radius_units = self.radius_units.get_unit()
field_params = self.field_parameters.get_field_parameters()
dsource = self.data_source.get_data_source()
name = self.name.get_name()
if center_units is not None:
center = YTArray(center, center_units)
if radius_units is not None:
radius = YTArray(radius, radius_units)
if field_params == 'None':
sphere = source.sphere(center, radius, data_source=dsource)
new_object = YtDataObject(sphere, name)
self.parent.add_data_object(new_object)
class OffAxisRayW(QtGui.QWidget):
r"""A widget for creating off axis rays from a YtObject.
Parameters
----------
parent : ActiveAcquisitionW
The widget which ultimately gave rise to this widget. The widget
containing data objects.
parent_widget : QtGui.QWidget
The widget which will display this widget as one of its children."""
def __init__(self, parent, parent_widget):
super(OffAxisRayW, self).__init__()
self.parent = parent
self.parent_widget = parent_widget
self.coordinate_units_w = CoordinateUnitsW()
start_coord_label = QtGui.QLabel("Ray Starting Coordinate:")
self.start_coord_w = CartCoordinateComboW()
start_coordc = QtGui.QWidget()
start_coordc_layout = QtGui.QHBoxLayout()
start_coordc_layout.addWidget(start_coord_label)
start_coordc_layout.addWidget(self.start_coord_w)
start_coordc.setLayout(start_coordc_layout)
end_coord_label = QtGui.QLabel("Ray Ending Coordinate:")
self.end_coord_w = CartCoordinateComboW()
end_coordc = QtGui.QWidget()
end_coordc_layout = QtGui.QHBoxLayout()
end_coordc_layout.addWidget(end_coord_label)
end_coordc_layout.addWidget(self.end_coord_w)
end_coordc.setLayout(end_coordc_layout)
self.field_parameters_w = FieldParametersW()
self.data_source_w = DataSourceW(self.parent)
self.object_name = NameW(self.parent.active_data_object.name +
'_arbitrary_ray')
self.generate_object_btn = QtGui.QPushButton('Generate Object')
self.generate_object_btn.clicked.connect(self.generate_object)
layout = QtGui.QVBoxLayout()
layout.addWidget(self.coordinate_units_w)
layout.addWidget(start_coordc)
layout.addWidget(end_coordc)
layout.addWidget(self.field_parameters_w)
layout.addWidget(self.data_source_w)
layout.addWidget(self.object_name)
layout.addWidget(self.generate_object_btn)
self.setLayout(layout)
self.parent_widget.layout.addWidget(self)
self.parent_widget.show()
def generate_object(self):
r"""Generates an off axis ray through a preselected YtObject based
on the parameters specified by the user in the widget."""
source = self.parent.active_data_object.data
unit = self.coordinate_units_w.get_unit()
sp = self.start_coord_w.get_coordinates()
ep = self.end_coord_w.get_coordinates()
name = self.object_name.get_name()
if unit is not None:
sp = YTArray(sp, unit)
ep = YTArray(ep, unit)
sp = source.arr(sp).in_units('code_length')
ep = source.arr(ep).in_units('code_length')
fp = self.field_parameters_w.get_field_parameters()
ds = self.data_source_w.get_data_source()
if fp == 'None':
if ds is None:
ray = source.ray(sp, ep)
else:
ray = source.ray(sp, ep, data_source=ds)
new_object = YtDataObject(ray, name)
self.parent.add_data_object(new_object)
class AxisSliceW(QtGui.QWidget):
r"""A widget for generating axis aligned slices from a preselected YtObject.
Parameters
----------
parent : ActiveAcquisitionW
The widget which ultimately gave rise to this widget. The widget
containing data objects.
parent_widget : QtGui.QWidget
The widget which will display this widget as one of its children."""
axis_dict = {"x": 0, "y": 1, "z": 2}
def __init__(self, parent, parent_widget):
super(AxisSliceW, self).__init__()
self.parent = parent
self.parent_widget = parent_widget
axis_label = QtGui.QLabel('Axis Along Which To Slice:')
self.axisw = QtGui.QComboBox()
self.axisw.addItems(list(self.axis_dict.keys()))
self.axisw.activated.connect(self.set_new_center_coord)
axiscw = QtGui.QWidget()
axiscw_layout = QtGui.QHBoxLayout()
axiscw_layout.addWidget(axis_label)
axiscw_layout.addWidget(self.axisw)
axiscw.setLayout(axiscw_layout)
self.slice_point_unit_w = CoordinateUnitsW()
self.slice_point_unit_w.label.setText('Units of Slice Point' +
'Coordinate:')
self.coordinate_w = CoordinateW('Point of Slice on Axis:')
center_label = QtGui.QLabel("Center of Slice:")
self.center_toggle_w = QtGui.QComboBox()
self.center_toggle_w.addItems(['None', 'Custom'])
self.center_toggle_w.currentIndexChanged.connect(self.add_coord_widget)
center_t_w = QtGui.QWidget()
center_t_w_layout = QtGui.QHBoxLayout()
center_t_w_layout.addWidget(center_label)
center_t_w_layout.addWidget(self.center_toggle_w)
center_t_w.setLayout(center_t_w_layout)
self.sec_unit_w = CoordinateUnitsW()
self.sec_unit_w.label.setText('Units of Center Coordinates:')
self.center_coord1_w = CoordinateW('y')
self.center_coord2_w = CoordinateW('z')
center_coord_combo_w = QtGui.QWidget()
center_coord_combo_layout = QtGui.QHBoxLayout()
center_coord_combo_layout.addWidget(self.center_coord1_w)
center_coord_combo_layout.addWidget(self.center_coord2_w)
center_coord_combo_w.setLayout(center_coord_combo_layout)
self.center_coord_cw = QtGui.QWidget()
center_coord_layout = QtGui.QVBoxLayout()
center_coord_layout.addWidget(self.sec_unit_w)
center_coord_layout.addWidget(center_coord_combo_w)
self.center_coord_cw.setLayout(center_coord_layout)
self.center_coord_cw.hide()
self.field_parameters_w = FieldParametersW()
self.data_source_w = DataSourceW(self.parent)
self.name_w = NameW(self.parent.active_data_object.name +
self.axisw.currentText() + '_axis_slice')
self.generate_object_btn = QtGui.QPushButton('Generate Object')
self.generate_object_btn.clicked.connect(self.generate_object)
layout = QtGui.QVBoxLayout()
layout.addWidget(axiscw)
layout.addWidget(self.slice_point_unit_w)
layout.addWidget(self.coordinate_w)
layout.addWidget(center_t_w)
layout.addWidget(self.center_coord_cw)
layout.addWidget(self.field_parameters_w)
layout.addWidget(self.data_source_w)
layout.addWidget(self.name_w)
layout.addWidget(self.generate_object_btn)
self.setLayout(layout)
self.parent_widget.layout.addWidget(self)
self.parent_widget.show()
def add_coord_widget(self, index):
r"""Displays a coordinate widget if the user wants to define a custom
center for the slice that is generated.
Parameters
----------
index : int
Argument provided for compatibility with Qt slot:signal design.
Not used."""
if self.center_toggle_w.currentText() == 'Custom':
self.center_coord_cw.show()
else:
self.center_coord_cw.hide()
def set_new_center_coord(self, index):
r"""This function takes care of setting the labels for the center
coordinates if the user changes the axis along which the slice is
being made.
Parameters
----------
index : int
Provided for compatibility with Qt slot:signal design. Not used."""
new_axes = [key for key in list(self.axis_dict.keys())
if key != self.axisw.itemText(index)]
self.center_coord1_w.set_label(new_axes[0])
self.center_coord2_w.set_label(new_axes[1])
def generate_object(self):
r"""Generates an axis aligned object based on the user input from the
widget."""
axis = self.axis_dict[self.axisw.currentText()]
coord_unit = self.slice_point_unit_w.get_unit()
coord = self.coordinate_w.get_coordinate()
source = self.parent.active_data_object.get_data()
name = self.name_w.get_name()
field_params = self.field_parameters_w.get_field_parameters()
dsource = self.data_source_w.get_data_source()
if coord_unit is not None:
coord = YTArray([coord], coord_unit)
coord = source.arr(coord).in_units('code_length').item(0)
if self.center_toggle_w.currentText() == 'Custom':
c_coord = [self.center_coord1_w.get_coordinate(),
self.center_coord2_w.get_coordinate()]
c_coord_unit = self.sec_unit_w.get_unit()
if c_coord_unit is not None:
c_coord = YTArray(c_coord, c_coord_unit)
else:
c_coord = None
if field_params == 'None':
new_slice = source.slice(axis, coord, center=c_coord,
field_parameters=None,
data_source=dsource)
new_object = YtDataObject(new_slice, name)
self.parent.add_data_object(new_object)
class PointW(QtGui.QWidget):
r"""A widget for creating YTPoint objects from a selected object.
Parameters
----------
parent : AcquisitionActiveW
The widget which is ultimately the source of this widget.
parent_widget : QtGui.QWidget
The widget which will display this widget as one of its children."""
def __init__(self, parent, parent_widget):
super(PointW, self).__init__()
self.parent = parent
self.parent_widget = parent_widget
self.coordinate_unit_w = CoordinateUnitsW()
# Assuming Cartesian, terrible I know, but deal with it.
self.coord_combo_w = CartCoordinateComboW()
# Currently, inputting values here will do nothing.
self.field_parametersw = FieldParametersW()
self.data_sourcew = DataSourceW(self.parent)
self.object_name = NameW(self.parent.active_data_object.name +
'_point')
self.generate_object_btn = QtGui.QPushButton('Generate Object')
self.generate_object_btn.clicked.connect(self.generate_object)
layout = QtGui.QVBoxLayout()
layout.addWidget(self.coordinate_unit_w)
layout.addWidget(self.coord_combo_w)
layout.addWidget(self.field_parametersw)
layout.addWidget(self.data_sourcew)
layout.addWidget(self.object_name)
layout.addWidget(self.generate_object_btn)
self.setLayout(layout)
self.parent_widget.layout.addWidget(self)
self.parent_widget.show()
def generate_object(self):
r"""Generates a YTPoint from the selected yt object, with parameters
specified by the user in the widget."""
unit = self.coordinate_unit_w.get_unit()
coord = self.coord_combo_w.get_coordinates()
if unit is not None:
coord = YTArray(coord, unit)
name = self.object_name.get_name()
source = self.parent.active_data_object.data
if self.field_parametersw.get_field_parameters() == 'None':
if self.data_sourcew.get_data_source() is None:
point = source.point(coord)
new_object = YtDataObject(point, name)
else:
dsource = self.data_source.get_data_source()
point = source.point(coord, data_source=dsource)
new_object = YtDataObject(point, name)
self.parent.add_data_object(new_object)