pyedb 0.15.dev0__py3-none-any.whl → 0.17.0__py3-none-any.whl

pyedb/dotnet/edb_core/utilities/siwave_simulation_setup.py

@@ -0,0 +1,369 @@
+import warnings
+
+from pyedb.dotnet.edb_core.general import (
+    convert_netdict_to_pydict,
+    convert_pydict_to_netdict,
+)
+from pyedb.dotnet.edb_core.sim_setup_data.data.sim_setup_info import SimSetupInfo
+from pyedb.dotnet.edb_core.sim_setup_data.data.siw_dc_ir_settings import (
+    SiwaveDCIRSettings,
+)
+from pyedb.dotnet.edb_core.sim_setup_data.io.siwave import (
+    AdvancedSettings,
+    DCAdvancedSettings,
+    DCSettings,
+)
+from pyedb.dotnet.edb_core.utilities.simulation_setup import SimulationSetup
+from pyedb.generic.general_methods import is_linux
+
+
+def _parse_value(v):
+    """Parse value in C sharp format."""
+    #  duck typing parse of the value 'v'
+    if v is None or v == "":
+        pv = v
+    elif v == "true":
+        pv = True
+    elif v == "false":
+        pv = False
+    else:
+        try:
+            pv = int(v)
+        except ValueError:
+            try:
+                pv = float(v)
+            except ValueError:
+                if isinstance(v, str) and v[0] == v[-1] == "'":
+                    pv = v[1:-1]
+                else:
+                    pv = v
+    return pv
+
+
+def clone_edb_sim_setup_info(source, target):
+    string = source.ToString().replace("\t", "").split("\r\n")
+    if is_linux:
+        string = string[0].split("\n")
+    keys = [i.split("=")[0] for i in string if len(i.split("=")) == 2 and "SourceTermsToGround" not in i]
+    values = [i.split("=")[1] for i in string if len(i.split("=")) == 2 and "SourceTermsToGround" not in i]
+    for val in string:
+        if "SourceTermsToGround()" in val:
+            break
+        elif "SourceTermsToGround" in val:
+            sources = {}
+            val = val.replace("SourceTermsToGround(", "").replace(")", "").split(",")
+            for v in val:
+                source = v.split("=")
+                sources[source[0]] = int(source[1].replace("'", ""))
+            target.SimulationSettings.DCIRSettings.SourceTermsToGround = convert_pydict_to_netdict(sources)
+            break
+    for k in keys:
+        value = _parse_value(values[keys.index(k)])
+        setter = None
+        if k in dir(target.SimulationSettings):
+            setter = target.SimulationSettings
+        elif k in dir(target.SimulationSettings.AdvancedSettings):
+            setter = target.SimulationSettings.AdvancedSettings
+
+        elif k in dir(target.SimulationSettings.DCAdvancedSettings):
+            setter = target.SimulationSettings.DCAdvancedSettings
+        elif "DCIRSettings" in dir(target.SimulationSettings) and k in dir(target.SimulationSettings.DCIRSettings):
+            setter = target.SimulationSettings.DCIRSettings
+        elif k in dir(target.SimulationSettings.DCSettings):
+            setter = target.SimulationSettings.DCSettings
+        elif k in dir(target.SimulationSettings.AdvancedSettings):
+            setter = target.SimulationSettings.AdvancedSettings
+        if setter:
+            try:
+                setter.__setattr__(k, value)
+            except TypeError:
+                try:
+                    setter.__setattr__(k, str(value))
+                except:
+                    pass
+
+
+class SiwaveSimulationSetup(SimulationSetup):
+    """Manages EDB methods for SIwave simulation setup."""
+
+    def __init__(self, pedb, edb_object=None, name: str = None):
+        super().__init__(pedb, edb_object)
+        self._simulation_setup_builder = self._pedb._edb.Utility.SIWaveSimulationSetup
+        if edb_object is None:
+            self._name = name
+            sim_setup_info = SimSetupInfo(self._pedb, sim_setup=self, setup_type="kSIwave", name=name)
+            self._edb_object = self._simulation_setup_builder(sim_setup_info._edb_object)
+            self._update_setup()
+
+    def create(self, name=None):
+        """Create a SIwave SYZ setup.
+
+        Returns
+        -------
+        :class:`SiwaveDCSimulationSetup`
+        """
+        self._name = name
+        self._create(name, simulation_setup_type="kSIwave")
+        self.si_slider_position = 1
+
+        return self
+
+    def get_configurations(self):
+        """Get SIwave SYZ simulation settings.
+
+        Returns
+        -------
+        dict
+            Dictionary of SIwave SYZ simulation settings.
+        """
+        return {
+            "pi_slider_position": self.pi_slider_position,
+            "si_slider_position": self.si_slider_position,
+            "use_custom_settings": self.use_si_settings,
+            "use_si_settings": self.use_si_settings,
+            "advanced_settings": self.advanced_settings.get_configurations(),
+        }
+
+    @property
+    def advanced_settings(self):
+        """SIwave advanced settings."""
+        return AdvancedSettings(self)
+
+    @property
+    def sim_setup_info(self):
+        """Overrides the default sim_setup_info object."""
+        return SimSetupInfo(self._pedb, sim_setup=self, edb_object=self.get_sim_setup_info)
+
+    @sim_setup_info.setter
+    def sim_setup_info(self, sim_setup_info):
+        self._edb_object = self._simulation_setup_builder(sim_setup_info._edb_object)
+
+    @property
+    def get_sim_setup_info(self):
+        """Get simulation information from the setup."""
+
+        sim_setup_info = SimSetupInfo(self._pedb, sim_setup=self, setup_type="kSIwave", name=self._edb_object.GetName())
+        clone_edb_sim_setup_info(source=self._edb_object, target=sim_setup_info._edb_object)
+        return sim_setup_info._edb_object
+
+    def set_pi_slider(self, value):
+        """Set SIwave PI simulation accuracy level.
+        Options are:
+        - ``0``: Optimal speed
+        - ``1``:  Balanced
+        - ``2``: Optimal accuracy
+
+        .. deprecated:: 0.7.5
+           Use :property:`pi_slider_position` property instead.
+
+        """
+        warnings.warn("`set_pi_slider` is deprecated. Use `pi_slider_position` property instead.", DeprecationWarning)
+        self.pi_slider_position = value
+
+    def set_si_slider(self, value):
+        """Set SIwave SI simulation accuracy level.
+
+        Options are:
+        - ``0``: Optimal speed;
+        - ``1``:  Balanced;
+        - ``2``: Optimal accuracy```.
+        """
+        self.use_si_settings = True
+        self.use_custom_settings = False
+        self.si_slider_position = value
+        self.advanced_settings.set_si_slider(value)
+
+    @property
+    def pi_slider_position(self):
+        """PI solider position. Values are from ``1`` to ``3``."""
+        return self.get_sim_setup_info.SimulationSettings.PISliderPos
+
+    @pi_slider_position.setter
+    def pi_slider_position(self, value):
+        edb_setup_info = self.get_sim_setup_info
+        edb_setup_info.SimulationSettings.PISliderPos = value
+        self._edb_object = self._set_edb_setup_info(edb_setup_info)
+        self._update_setup()
+
+        self.use_si_settings = False
+        self.use_custom_settings = False
+        self.advanced_settings.set_pi_slider(value)
+
+    @property
+    def si_slider_position(self):
+        """SI slider position. Values are from ``1`` to ``3``."""
+        return self.get_sim_setup_info.SimulationSettings.SISliderPos
+
+    @si_slider_position.setter
+    def si_slider_position(self, value):
+        edb_setup_info = self.get_sim_setup_info
+        edb_setup_info.SimulationSettings.SISliderPos = value
+        self._edb_object = self._set_edb_setup_info(edb_setup_info)
+        self._update_setup()
+
+        self.use_si_settings = True
+        self.use_custom_settings = False
+        self.advanced_settings.set_si_slider(value)
+
+    @property
+    def use_custom_settings(self):
+        """Custom settings to use.
+
+        Returns
+        -------
+        bool
+        """
+        return self.get_sim_setup_info.SimulationSettings.UseCustomSettings
+
+    @use_custom_settings.setter
+    def use_custom_settings(self, value):
+        edb_setup_info = self.get_sim_setup_info
+        edb_setup_info.SimulationSettings.UseCustomSettings = value
+        self._edb_object = self._set_edb_setup_info(edb_setup_info)
+        self._update_setup()
+
+    @property
+    def use_si_settings(self):
+        """Whether to use SI Settings.
+
+        Returns
+        -------
+        bool
+        """
+        return self.get_sim_setup_info.SimulationSettings.UseSISettings
+
+    @use_si_settings.setter
+    def use_si_settings(self, value):
+        edb_setup_info = self.get_sim_setup_info
+        edb_setup_info.SimulationSettings.UseSISettings = value
+        self._edb_object = self._set_edb_setup_info(edb_setup_info)
+        self._update_setup()
+
+
+class SiwaveDCSimulationSetup(SimulationSetup):
+    """Manages EDB methods for SIwave DC simulation setup."""
+
+    def __init__(self, pedb, edb_object=None, name: str = None):
+        super().__init__(pedb, edb_object)
+        self._simulation_setup_builder = self._pedb._edb.Utility.SIWaveDCIRSimulationSetup
+        self._mesh_operations = {}
+        if edb_object is None:
+            self._name = name
+            sim_setup_info = SimSetupInfo(self._pedb, sim_setup=self, setup_type="kSIwaveDCIR", name=name)
+            self._edb_object = self._simulation_setup_builder(sim_setup_info._edb_object)
+            self._update_setup()
+
+    def create(self, name=None):
+        """Create a SIwave DCIR setup.
+
+        Returns
+        -------
+        :class:`SiwaveDCSimulationSetup`
+        """
+        self._name = name
+        self._create(name)
+        self.set_dc_slider(1)
+        return self
+
+    @property
+    def sim_setup_info(self):
+        """Overrides the default sim_setup_info object."""
+        return SimSetupInfo(self._pedb, sim_setup=self, edb_object=self.get_sim_setup_info)
+
+    @sim_setup_info.setter
+    def sim_setup_info(self, sim_setup_info):
+        self._edb_object = self._simulation_setup_builder(sim_setup_info._edb_object)
+
+    @property
+    def get_sim_setup_info(self):
+        """Get simulation information from the setup."""
+
+        sim_setup_info = SimSetupInfo(
+            self._pedb, sim_setup=self, setup_type="kSIwaveDCIR", name=self._edb_object.GetName()
+        )
+        clone_edb_sim_setup_info(source=self._edb_object, target=sim_setup_info._edb_object)
+        return sim_setup_info._edb_object
+
+    @property
+    def dc_ir_settings(self):
+        """DC IR settings."""
+        return SiwaveDCIRSettings(self)
+
+    def get_configurations(self):
+        """Get SIwave DC simulation settings.
+
+        Returns
+        -------
+        dict
+            Dictionary of SIwave DC simulation settings.
+        """
+        return {
+            "dc_settings": self.dc_settings.get_configurations(),
+            "dc_advanced_settings": self.dc_advanced_settings.get_configurations(),
+        }
+
+    def set_dc_slider(self, value):
+        """Set DC simulation accuracy level.
+
+        Options are:
+
+        - ``0``: Optimal speed
+        - ``1``: Balanced
+        - ``2``: Optimal accuracy
+        """
+        self.use_custom_settings = False
+        self.dc_settings.dc_slider_position = value
+        self.dc_advanced_settings.set_dc_slider(value)
+
+    @property
+    def dc_settings(self):
+        """SIwave DC setting."""
+        return DCSettings(self)
+
+    @property
+    def dc_advanced_settings(self):
+        """Siwave DC advanced settings.
+
+        Returns
+        -------
+        :class:`pyedb.dotnet.edb_core.edb_data.siwave_simulation_setup_data.SiwaveDCAdvancedSettings`
+        """
+        return DCAdvancedSettings(self)
+
+    @property
+    def source_terms_to_ground(self):
+        """Dictionary of grounded terminals.
+
+        Returns
+        -------
+        Dictionary
+            {str, int}, keys is source name, value int 0 unspecified, 1 negative node, 2 positive one.
+
+        """
+        return convert_netdict_to_pydict(self.get_sim_setup_info.SimulationSettings.DCIRSettings.SourceTermsToGround)
+
+    def add_source_terminal_to_ground(self, source_name, terminal=0):
+        """Add a source terminal to ground.
+
+        Parameters
+        ----------
+        source_name : str,
+            Source name.
+        terminal : int, optional
+            Terminal to assign. Options are:
+
+             - 0=Unspecified
+             - 1=Negative node
+             - 2=Positive none
+
+        Returns
+        -------
+        bool
+
+        """
+        terminals = self.source_terms_to_ground
+        terminals[source_name] = terminal
+        self.get_sim_setup_info.SimulationSettings.DCIRSettings.SourceTermsToGround = convert_pydict_to_netdict(
+            terminals
+        )
+        return self._update_setup()

pyedb/misc/siw_feature_config/xtalk_scan/fd_xtalk_scan_config.py

@@ -0,0 +1,91 @@
+# Copyright (C) 2023 - 2024 ANSYS, Inc. and/or its affiliates.
+# SPDX-License-Identifier: MIT
+#
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+from pyedb.generic.general_methods import ET
+from pyedb.misc.siw_feature_config.xtalk_scan.net import SingleEndedNet
+
+
+class CrosstalkFrequency:
+    """Siwave frequency domain crosstalk configuration handler."""
+
+    def __init__(self, pedb):
+        self._pedb = pedb
+        self.min_transmission_line_segment_length = "0.25mm"
+        self.frequency = "2e9Hz"
+        self.nets = {}
+
+    def extend_xml(self, parent):
+        """Write class xml section.
+
+        Parameters
+        ----------
+        parent : :class xml.etree.cElementTree object
+        Parent object.
+
+        """
+        freq_scan = ET.SubElement(parent, "FdXtalkConfig")
+        freq_scan.set("MinTlineSegmentLength", self.min_transmission_line_segment_length)
+        freq_scan.set("XtalkFrequency", self.frequency)
+        nets = ET.SubElement(freq_scan, "SingleEndedNets")
+        for net in list(self.nets.values()):
+            net.extend_xml(nets)
+
+    def add_single_ended_net(
+        self,
+        name,
+        next_warning_threshold=5.0,
+        next_violation_threshold=10.0,
+        fext_warning_threshold_warning=5.0,
+        fext_violation_threshold=5.0,
+    ):
+        """Add single ended net.
+
+        Parameters
+        ----------
+        name : str
+            Net name.
+        next_warning_threshold : flot or str
+            Near end crosstalk warning threshold value. Default value is ``5.0``.
+        next_violation_threshold : float, str
+            Near end crosstalk violation threshold value. Default value is ``10.0
+
+        fext_violation_threshold : float, str
+            Far end crosstalk violation threshold value, Default value is ``5.0``
+        fext_warning_threshold_warning : float, str
+            Far end crosstalk warning threshold value, Default value is ``5.0``
+
+        Returns
+        -------
+        bool
+        """
+        if name and name not in self.nets:
+            net = SingleEndedNet()
+            net.name = name
+            net.next_warning_threshold = next_warning_threshold
+            net.next_violation_threshold = next_violation_threshold
+            net.fext_warning_threshold = fext_warning_threshold_warning
+            net.fext_violation_threshold = fext_violation_threshold
+            self.nets[name] = net
+            return True
+        else:
+            self._pedb.logger.error(f"Net {name} already assigned.")
+            return False

pyedb/misc/siw_feature_config/xtalk_scan/impedance_scan_config.py

@@ -0,0 +1,70 @@
+# Copyright (C) 2023 - 2024 ANSYS, Inc. and/or its affiliates.
+# SPDX-License-Identifier: MIT
+#
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+from pyedb.generic.general_methods import ET
+from pyedb.misc.siw_feature_config.xtalk_scan.net import SingleEndedNet
+
+
+class ImpedanceScan:
+    def __init__(self, pedb):
+        self._pedb = pedb
+        self.min_transmission_line_segment_length = "0.25mm"
+        self.frequency = "2e9Hz"
+        self.nets = {}
+
+    def extend_xml(self, parent):
+        """Write object wml section"""
+        z_scan = ET.SubElement(parent, "Z0ScanConfig")
+        z_scan.set("MinTlineSegmentLength", self.min_transmission_line_segment_length)
+        z_scan.set("Z0Frequency", self.frequency)
+        single_ended_nets = ET.SubElement(z_scan, "SingleEndedNets")
+        for net in list(self.nets.values()):
+            net.extend_xml(single_ended_nets)
+
+    def add_single_ended_net(self, name, nominal_impedance=50.0, warning_threshold=17.0, violation_threshold=32.0):
+        """Add single ended net.
+
+        Parameters
+        ----------
+        name : str
+            Net name.
+        nominal_impedance : flot or str
+            nominal impedance in ohm. Default value is ``50,0``..
+        warning_threshold : float, str
+            Warning threshold value. Default value is ``17.0
+        violation_threshold : float, str
+            Violation threshold value, Default value is ``5.0``
+
+        Returns
+        -------
+        bool
+        """
+        if name and name not in self.nets:
+            net = SingleEndedNet()
+            net.name = name
+            net.nominal_impedance = nominal_impedance
+            net.warning_threshold = warning_threshold
+            net.violation_threshold = violation_threshold
+            self.nets[name] = net
+        else:
+            self._pedb.logger.error(f"Net {name} already assigned")
+            return False

pyedb/misc/siw_feature_config/xtalk_scan/net.py

@@ -0,0 +1,69 @@
+# Copyright (C) 2023 - 2024 ANSYS, Inc. and/or its affiliates.
+# SPDX-License-Identifier: MIT
+#
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+from pyedb.generic.general_methods import ET
+
+
+class SingleEndedNet:
+    """Single ended net class handler."""
+
+    def __init__(self):
+        self.name = None
+        self.nominal_impedance = None
+        self.warning_threshold = None
+        self.violation_threshold = None
+        self.fext_warning_threshold = None
+        self.fext_violation_threshold = None
+        self.next_warning_threshold = None
+        self.next_violation_threshold = None
+        self.driver_rise_time = None
+        self.voltage = None
+        self.driver_impedance = None
+        self.termination_impedance = None
+
+    def extend_xml(self, parent):
+        """Write XMl object section."""
+        net = ET.SubElement(parent, "Net")
+        if self.name is not None:
+            net.set("Name", self.name)
+        if self.nominal_impedance is not None:
+            net.set("NominalZ0", str(self.nominal_impedance))
+        if self.warning_threshold is not None:
+            net.set("WarningThreshold", str(self.warning_threshold))
+        if self.violation_threshold is not None:
+            net.set("ViolationThreshold", str(self.violation_threshold))
+        if self.fext_warning_threshold is not None:
+            net.set("FEXTWarningThreshold", str(self.fext_warning_threshold))
+        if self.fext_violation_threshold is not None:
+            net.set("FEXTViolationThreshold", str(self.fext_violation_threshold))
+        if self.next_warning_threshold is not None:
+            net.set("NEXTWarningThreshold", str(self.next_warning_threshold))
+        if self.next_violation_threshold is not None:
+            net.set("NEXTViolationThreshold", str(self.next_violation_threshold))
+        if self.driver_rise_time is not None:
+            net.set("DriverRiseTime", str(self.driver_rise_time))
+        if self.voltage is not None:
+            net.set("Voltage", str(self.voltage))
+        if self.driver_impedance is not None:
+            net.set("DriverImpedance", str(self.driver_impedance))
+        if self.termination_impedance is not None:
+            net.set("TerminationImpedance", str(self.termination_impedance))

pyedb/misc/siw_feature_config/xtalk_scan/pins.py

@@ -0,0 +1,60 @@
+# Copyright (C) 2023 - 2024 ANSYS, Inc. and/or its affiliates.
+# SPDX-License-Identifier: MIT
+#
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+
+from pyedb.generic.general_methods import ET
+
+
+class DriverPin:
+    """Driver pin class handler."""
+
+    def __init__(self):
+        self.name = None
+        self.ref_des = None
+        self.driver_rise_time = None
+        self.voltage = None
+        self.driver_impedance = None
+
+    def extend_xml(self, parent):
+        """Write  object to xml section."""
+        pin = ET.SubElement(parent, "Pin")
+        pin.set("Name", self.name)
+        pin.set("RefDes", self.ref_des)
+        pin.set("DriverRiseTime", str(self.driver_rise_time))
+        pin.set("Voltage", str(self.voltage))
+        pin.set("DriverImpedance", str(self.driver_impedance))
+
+
+class ReceiverPin:
+    """Receiver pin class handler."""
+
+    def __init__(self):
+        self.name = None
+        self.ref_des = None
+        self.receiver_impedance = None
+
+    def extend_xml(self, parent):
+        """Write object to xml section."""
+        pin = ET.SubElement(parent, "Pin")
+        pin.set("Name", self.name)
+        pin.set("Name", self.name)
+        pin.set("ReceiverImpedance", str(self.receiver_impedance))