iqm-exa-common 27.0.0__py3-none-any.whl → 27.2.0__py3-none-any.whl

exa/common/qcm_data/chip_layout.py

@@ -0,0 +1,123 @@
+# Copyright 2025 IQM
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Chip layout class."""
+
+from __future__ import annotations
+
+from functools import cached_property
+
+import numpy as np
+
+from exa.common.errors.exa_error import ExaError
+
+
+class ChipLayout:
+    """The chip layout contains the components and their 2D cartesian coordinates."""
+
+    def __init__(
+        self,
+        qubits: dict[str, tuple[float, float]],
+        couplers: dict[str, tuple[float, float]],
+        computational_resonators: dict[str, tuple[float, float]],
+    ) -> None:
+        self._qubits = list(qubits)
+        self._couplers = list(couplers)
+        self._computational_resonators = list(computational_resonators)
+        self._coordinates = {
+            comp: (x, y) for comp, (x, y) in [*qubits.items(), *couplers.items(), *computational_resonators.items()]
+        }
+
+    @classmethod
+    def from_chip_design_record(cls, record: dict) -> ChipLayout:
+        """Construct the chip layout from a raw chip design record.
+
+        Args:
+            record: The chip design record as returned by station control.
+
+        Returns:
+            The corresponding chip layout.
+
+        """
+        qubits = record["content"]["components"].get("qubit", [])
+        couplers = record["content"]["components"].get("tunable_coupler", [])
+        comprs = record["content"]["components"].get("computational_resonator", [])
+        if all("locations" in component for component in [*qubits, *couplers, *comprs]):
+            return cls(
+                qubits={
+                    qubit["name"]: (qubit["locations"]["metro"]["x"], qubit["locations"]["metro"]["y"])
+                    for qubit in qubits
+                },
+                couplers={
+                    coupler["name"]: (coupler["locations"]["metro"]["x"], coupler["locations"]["metro"]["y"])
+                    for coupler in couplers
+                },
+                computational_resonators={
+                    compr["name"]: (compr["locations"]["metro"]["x"], compr["locations"]["metro"]["y"])
+                    for compr in comprs
+                },
+            )
+        raise ExaError("Chip design record is missing locations.")
+
+    def normalize_coordinates(self, scale: float) -> None:
+        self._coordinates = {comp: (xx * scale, yy * scale) for comp, (xx, yy) in self._coordinates.items()}
+
+    def mirror_yaxis(self) -> None:
+        self._coordinates = {comp: (xx, -yy) for comp, (xx, yy) in self._coordinates.items()}
+
+    def rotate_layout(self) -> None:
+        self._coordinates = {
+            comp: ((xx + yy) / np.sqrt(2), (-xx + yy) / np.sqrt(2)) for comp, (xx, yy) in self._coordinates.items()
+        }
+
+    def move_origin(self) -> None:
+        x_min, y_min = (
+            min([xx for comp, (xx, yy) in self._coordinates.items()]),
+            min([yy for comp, (xx, yy) in self._coordinates.items()]),
+        )
+        self._coordinates = {comp: (xx - x_min, yy - y_min) for comp, (xx, yy) in self._coordinates.items()}
+
+    @property
+    def qubits(self) -> list[str]:
+        return self._qubits
+
+    @property
+    def couplers(self) -> list[str]:
+        return self._couplers
+
+    @property
+    def computational_resonators(self) -> list[str]:
+        return self._computational_resonators
+
+    @cached_property
+    def components(self) -> list[str]:
+        return [*self._qubits, *self._couplers, *self._computational_resonators]
+
+    def get_coordinates(self, component: str) -> tuple[float, float]:
+        """Get the coordinates for the given component.
+
+        Args:
+            component: The name of the component.
+
+        Returns:
+            The 2D cartesian coordinates.
+
+        """
+        if component not in self.components:
+            raise ValueError(f"Component {component} not in chip layout.")
+        return self._coordinates[component]
+
+    def get_all_qubit_coordinates(self) -> dict[str, tuple[float, float]]:
+        """Get the coordinates for all qubits."""
+        return {qubit: self._coordinates[qubit] for qubit in self._qubits}

{iqm_exa_common-27.0.0.dist-info → iqm_exa_common-27.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: iqm-exa-common
-Version: 27.0.0
+Version: 27.2.0
 Summary: Framework for control and measurement of superconducting qubits: common library
 Author-email: IQM Finland Oy <info@meetiqm.com>
 License: Apache License

{iqm_exa_common-27.0.0.dist-info → iqm_exa_common-27.2.0.dist-info}/RECORD

@@ -41,6 +41,7 @@ exa/common/logger/__init__.py,sha256=1bIsGxHzfujXlkgtcAnWToKMkw3dpU5PEd_7LE_NpgQ
 exa/common/logger/logger.py,sha256=TA9HxFZDyFB7ai8C6mCZOm1cx3JMRAhB8DcMCwDAKt0,5713
 exa/common/qcm_data/__init__.py,sha256=VtsYkGoaniSjCkY0oQlqkcYJCtmC2sTDxfrIe_kpqZg,567
 exa/common/qcm_data/chad_model.py,sha256=1vWpB8gsrnIEDqEOqqQAXrofN2jM6c8AypZvOSH5I18,11279
+exa/common/qcm_data/chip_layout.py,sha256=jDEMsujltH_KqLjj2Ja-XEEJubfsMg_HPlqnC4-L9YE,4555
 exa/common/qcm_data/chip_topology.py,sha256=-IsWh0hev_vnAWs0z3iWBvProe6LypgKs1Erp8HCb0M,20027
 exa/common/qcm_data/file_adapter.py,sha256=VAvyV4FrCE93bvy7YP2DDlS_cgHQe-uwPdMQwMlZveQ,2319
 exa/common/qcm_data/immutable_base_model.py,sha256=QXmKIWQbsbWQvovXwKT1d9jtyf2LNJtjQquIwO52zOU,901
@@ -48,8 +49,8 @@ exa/common/qcm_data/qcm_data_client.py,sha256=Rze6yQd0xUeH6eUMv3wduYbiDCTP3C4WEC
 exa/common/sweep/__init__.py,sha256=uEKk5AtzSgSnf8Y0geRPwUpqXIBIXpeCxsN64sX7F1o,591
 exa/common/sweep/database_serialization.py,sha256=9n96OqmEnjy1x2F1HwaNOQgwX_39HjLtE9EUoetSw0E,7493
 exa/common/sweep/util.py,sha256=-QE2AaH-WDkYAVH5-Z-30leLgY0x4efmby4kc1JTCgY,3732
-iqm_exa_common-27.0.0.dist-info/LICENSE.txt,sha256=R6Q7eUrLyoCQgWYorQ8WJmVmWKYU3dxA3jYUp0wwQAw,11332
-iqm_exa_common-27.0.0.dist-info/METADATA,sha256=HSADQ--8rxKTImWGTmuX7t6lDJ1Tajdke8LLVC98trw,14741
-iqm_exa_common-27.0.0.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-iqm_exa_common-27.0.0.dist-info/top_level.txt,sha256=Clphg2toaZ3_jSFRPhjMNEmLurkMNMc4lkK2EFYsSlM,4
-iqm_exa_common-27.0.0.dist-info/RECORD,,
+iqm_exa_common-27.2.0.dist-info/LICENSE.txt,sha256=R6Q7eUrLyoCQgWYorQ8WJmVmWKYU3dxA3jYUp0wwQAw,11332
+iqm_exa_common-27.2.0.dist-info/METADATA,sha256=pUaveDJ3RTxPsMtAOwoM8u7ujPd0vEiwlpw1i0pGLes,14741
+iqm_exa_common-27.2.0.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+iqm_exa_common-27.2.0.dist-info/top_level.txt,sha256=Clphg2toaZ3_jSFRPhjMNEmLurkMNMc4lkK2EFYsSlM,4
+iqm_exa_common-27.2.0.dist-info/RECORD,,