|
46 | 46 | from ..dialog.GeometryDialog import GeometryDialog |
47 | 47 | from ..dialog.IntegrationMethodDialog import IntegrationMethodDialog |
48 | 48 | from ...utils import float_, str_, get_ui_file |
49 | | -from ...units import RADIAL_UNITS, to_unit |
| 49 | +from ...units import RADIAL_UNITS, to_unit, Unit, UnitFiber |
50 | 50 | from ..model.GeometryModel import GeometryModel |
51 | 51 | from ..model.DataModel import DataModel |
52 | 52 | from ..utils import units |
@@ -130,13 +130,22 @@ def __init__(self, parent=None): |
130 | 130 | npt_validator.setBottom(1) |
131 | 131 | self.nbpt_rad.setValidator(npt_validator) |
132 | 132 | self.nbpt_azim.setValidator(npt_validator) |
133 | | - self.radial_unit.setUnits(RADIAL_UNITS.values()) |
| 133 | + |
| 134 | + radial_units = [v for v in RADIAL_UNITS.values() if (isinstance(v, Unit) and not isinstance(v, UnitFiber))] |
| 135 | + self.radial_unit.setUnits(radial_units) |
134 | 136 | self.radial_unit.model().setValue(RADIAL_UNITS["2th_deg"]) |
135 | 137 |
|
136 | 138 | self.radial_unit.setShortNameDisplay(True) |
137 | 139 | self.radial_unit.model().changed.connect(self.__radialUnitUpdated) |
138 | 140 | self.__radialUnitUpdated() |
139 | 141 |
|
| 142 | + fiber_units = [v for v in RADIAL_UNITS.values() if isinstance(v, UnitFiber)] |
| 143 | + self.ip_unit.setUnits(fiber_units) |
| 144 | + self.ip_unit.model().setValue(RADIAL_UNITS["qip_nm^-1"]) |
| 145 | + |
| 146 | + self.oop_unit.setUnits(fiber_units) |
| 147 | + self.oop_unit.model().setValue(RADIAL_UNITS["qoop_nm^-1"]) |
| 148 | + |
140 | 149 | doubleOrEmptyValidator = validators.AdvancedDoubleValidator(self) |
141 | 150 | doubleOrEmptyValidator.setAllowEmpty(True) |
142 | 151 | self.normalization_factor.setValidator(doubleOrEmptyValidator) |
|
0 commit comments