topolib 0.4.2__py3-none-any.whl → 0.5.0__py3-none-any.whl

topolib/topology/topology.py

@@ -8,9 +8,14 @@ This file uses NetworkX (BSD 3-Clause License):
 https://github.com/networkx/networkx/blob/main/LICENSE.txt
 """
 
-from typing import List, Dict, Any, Optional
+from typing import List, Optional, Any
+from numpy.typing import NDArray
 import numpy as np
 import networkx as nx
+import json
+import jsonschema
+import csv
+import networkx as nx
 from topolib.elements.node import Node
 from topolib.elements.link import Link
 
@@ -62,7 +67,7 @@ class Topology:
         self.links = links if links is not None else []
         self.name = name
         # Internal NetworkX graph for algorithms and visualization
-        self._graph = nx.Graph()
+        self._graph: nx.DiGraph[Any] = nx.DiGraph()
         for node in self.nodes:
             self._graph.add_node(node.id, node=node)
         for link in self.links:
@@ -78,12 +83,51 @@ class Topology:
         :return: Topology instance loaded from the file.
         :rtype: Topology
         """
-        import json
-        from topolib.elements.node import Node
-        from topolib.elements.link import Link
-
         with open(json_path, "r", encoding="utf-8") as f:
             data = json.load(f)
+
+        # Validation schema for the assets JSON format
+        topology_schema: dict[str, Any] = {
+            "type": "object",
+            "properties": {
+                "name": {"type": "string"},
+                "nodes": {
+                    "type": "array",
+                    "items": {
+                        "type": "object",
+                        "properties": {
+                            "id": {"type": "integer"},
+                            "name": {"type": "string"},
+                            "weight": {"type": "number"},
+                            "latitude": {"type": "number"},
+                            "longitude": {"type": "number"},
+                            "pop": {"type": "integer"},
+                            "DC": {"type": "integer"},
+                            "IXP": {"type": "integer"},
+                        },
+                        "required": ["id", "name", "latitude", "longitude"],
+                    },
+                },
+                "links": {
+                    "type": "array",
+                    "items": {
+                        "type": "object",
+                        "properties": {
+                            "id": {"type": "integer"},
+                            "src": {"type": "integer"},
+                            "dst": {"type": "integer"},
+                            "length": {"type": "number"},
+                        },
+                        "required": ["id", "src", "dst", "length"],
+                    },
+                },
+            },
+            "required": ["nodes", "links"],
+        }
+        try:
+            jsonschema.validate(instance=data, schema=topology_schema)
+        except jsonschema.ValidationError as e:
+            raise ValueError(f"Invalid topology JSON format: {e.message}")
         nodes = [
             Node(
                 n["id"],
@@ -152,7 +196,7 @@ class Topology:
             self._graph.remove_edge(link.source.id, link.target.id)
         self.links = [l for l in self.links if l.id != link_id]
 
-    def adjacency_matrix(self) -> np.ndarray:
+    def adjacency_matrix(self) -> NDArray[np.int_]:
         """
         Return the adjacency matrix of the topology as a numpy array.
 
@@ -170,3 +214,204 @@ class Topology:
         node_ids = [n.id for n in self.nodes]
         mat = nx.to_numpy_array(self._graph, nodelist=node_ids, dtype=int)
         return mat
+
+    def export_to_json(self, file_path: str) -> None:
+        """
+        Exporta la topología actual al formato JSON usado en la carpeta assets.
+
+        :param file_path: Ruta donde se guardará el archivo JSON.
+        :type file_path: str
+
+        **Ejemplo de uso**
+            >>> topo.export_to_json("/ruta/salida.json")
+
+        **Ejemplo de formato de salida**
+        {
+            "name": "Abilene",
+            "nodes": [
+                {
+                    "id": 0,
+                    "name": "Seattle",
+                    "weight": 0,
+                    "longitude": -122.3328481,
+                    "latitude": 47.6061389,
+                    "pop": 780995,
+                    "DC": 58,
+                    "IXP": 5
+                },
+                # ...
+            ],
+            "links": [
+                {
+                    "id": 0,
+                    "src": 0,
+                    "dst": 1,
+                    "length": 1482.26
+                },
+                # ...
+            ]
+        }
+        """
+        nodes_list: list[dict[str, Any]] = []
+        for n in self.nodes:
+            node_dict: dict[str, Any] = {
+                "id": n.id,
+                "name": getattr(n, "name", None),
+                "weight": getattr(n, "weight", 0),
+                "latitude": getattr(n, "latitude", None),
+                "longitude": getattr(n, "longitude", None),
+                "pop": getattr(n, "pop", 0),
+                "DC": getattr(n, "dc", getattr(n, "DC", 0)),
+                "IXP": getattr(n, "ixp", getattr(n, "IXP", 0)),
+            }
+            nodes_list.append(node_dict)
+        links_list: list[dict[str, Any]] = []
+        for l in self.links:
+            link_dict: dict[str, Any] = {
+                "id": l.id,
+                "src": l.source.id,
+                "dst": l.target.id,
+                "length": getattr(l, "length", None),
+            }
+            links_list.append(link_dict)
+        data: dict[str, Any] = {
+            "name": self.name if self.name else "Topology",
+            "nodes": nodes_list,
+            "links": links_list,
+        }
+        with open(file_path, "w", encoding="utf-8") as f:
+            json.dump(data, f, indent=4, ensure_ascii=False)
+
+    def export_to_csv(self, filename_prefix: str) -> None:
+        """
+        Export the topology to two CSV files: one for nodes and one for links.
+        The files will be named as <filename_prefix>_nodes.csv and <filename_prefix>_links.csv.
+
+        :param filename_prefix: Prefix for the output files (e.g., 'topology1').
+        :type filename_prefix: str
+
+        Example:
+            >>> topo.export_to_csv("mytopo")
+            # Generates 'mytopo_nodes.csv' and 'mytopo_links.csv'
+        """
+        # Export nodes
+        nodes_path = f"{filename_prefix}_nodes.csv"
+        with open(nodes_path, "w", newline="", encoding="utf-8") as f:
+            writer = csv.writer(f)
+            # Header
+            writer.writerow(
+                ["id", "name", "weight", "latitude", "longitude", "pop", "DC", "IXP"]
+            )
+            for n in self.nodes:
+                writer.writerow(
+                    [
+                        n.id,
+                        getattr(n, "name", None),
+                        getattr(n, "weight", 0),
+                        getattr(n, "latitude", None),
+                        getattr(n, "longitude", None),
+                        getattr(n, "pop", 0),
+                        getattr(n, "dc", getattr(n, "DC", 0)),
+                        getattr(n, "ixp", getattr(n, "IXP", 0)),
+                    ]
+                )
+        # Export links
+        links_path = f"{filename_prefix}_links.csv"
+        with open(links_path, "w", newline="", encoding="utf-8") as f:
+            writer = csv.writer(f)
+            writer.writerow(["id", "src", "dst", "length"])
+            for l in self.links:
+                writer.writerow(
+                    [
+                        l.id,
+                        l.source.id,
+                        l.target.id,
+                        getattr(l, "length", None),
+                    ]
+                )
+
+    def export_to_flexnetsim_json(self, file_path: str, slots: int) -> None:
+        """
+        Export the current topology to a JSON file compatible with Flex Net Sim.
+
+        :param file_path: Path where the JSON file will be saved.
+        :type file_path: str
+        :param slots: Number of slots for each link.
+        :type slots: int
+
+        The generated format includes the following fields:
+        - alias: short name of the topology (uses self.name if available)
+        - name: full name of the topology (uses self.name if available)
+        - nodes: list of nodes with 'id' field
+        - links: list of links with id, src, dst, length, slots
+        """
+        alias = self.name if self.name else "Topology"
+        name = self.name if self.name else "Topology"
+        nodes_list = [{"id": n.id} for n in self.nodes]
+        links_list = []
+        for l in self.links:
+            link_dict = {
+                "id": l.id,
+                "src": l.source.id,
+                "dst": l.target.id,
+                "length": getattr(l, "length", None),
+                "slots": slots,
+            }
+            links_list.append(link_dict)
+        data = {
+            "alias": alias,
+            "name": name,
+            "nodes": nodes_list,
+            "links": links_list,
+        }
+        with open(file_path, "w", encoding="utf-8") as f:
+            json.dump(data, f, indent=4, ensure_ascii=False)
+
+    def export_to_flexnetsim_ksp_json(self, file_path: str, k: int = 3) -> None:
+        """
+        Export the k-shortest paths between all node pairs to a JSON file compatible with Flex Net Sim.
+
+        :param file_path: Path where the JSON file will be saved.
+        :type file_path: str
+        :param k: Number of shortest paths to compute for each node pair (default: 3).
+        :type k: int
+
+        The output format matches Flex Net Sim's routes.json:
+        {
+            "name": self.name,
+            "alias": self.name,
+            "routes": [
+                {"src": <id>, "dst": <id>, "paths": [[id, ...], ...]},
+                ...
+            ]
+        }
+        """
+
+        # Build a weighted graph using link length as edge weight
+        G = nx.DiGraph()
+        for l in self.links:
+            G.add_edge(l.source.id, l.target.id, weight=getattr(l, "length", 1))
+        routes = []
+        node_ids = [n.id for n in self.nodes]
+        for src in node_ids:
+            for dst in node_ids:
+                if src == dst:
+                    continue
+                try:
+                    # Compute k shortest paths using link length as weight
+                    paths_gen = nx.shortest_simple_paths(G, src, dst, weight="weight")
+                    paths = []
+                    for i, path in enumerate(paths_gen):
+                        if i >= k:
+                            break
+                        paths.append(path)
+                except (nx.NetworkXNoPath, nx.NodeNotFound):
+                    paths = []
+                routes.append({"src": src, "dst": dst, "paths": paths})
+        data = {
+            "name": self.name if self.name else "Topology",
+            "alias": self.name if self.name else "Topology",
+            "routes": routes,
+        }
+        with open(file_path, "w", encoding="utf-8") as f:
+            json.dump(data, f, indent=4, ensure_ascii=False)

{topolib-0.4.2.dist-info → topolib-0.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: topolib
-Version: 0.4.2
+Version: 0.5.0
 Summary: A compact Python library for modeling, analyzing, and visualizing optical network topologies.
 License: MIT
 License-File: LICENSE
@@ -21,6 +21,7 @@ Classifier: Topic :: Scientific/Engineering :: Information Analysis
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Dist: contextily
 Requires-Dist: geopandas
+Requires-Dist: jsonschema (>=4.0)
 Requires-Dist: matplotlib
 Requires-Dist: networkx (>=2.6)
 Requires-Dist: numpy (>=1.21)
@@ -51,7 +52,8 @@ Description-Content-Type: text/markdown
 
 Explore ready-to-run usage examples in the [`examples/`](examples/) folder!  
 - [Show topology on a map](examples/show_topology_in_map.py) 🗺️  
-- More coming soon!
+- [Export topology to CSV and JSON](examples/export_csv_json.py) 📄
+- [Export topology and k-shortest paths for FlexNetSim](examples/export_flexnetsim.py) 🔀
 
 ---
 

topolib-0.5.0.dist-info/RECORD

@@ -0,0 +1,47 @@
+topolib/__init__.py,sha256=iLmy2rOkHS_4KZWMD8BgT7R3tLMKeaTCDVf3B4FyYxM,91
+topolib/analysis/__init__.py,sha256=qvUC9wV_jQNQIlwJXk7LJ-dkTXhH1Ttn0lKWIdwBbrc,52
+topolib/analysis/metrics.py,sha256=2o5PlMVzepDWwSQjzamtKemggqkxh9JuzzCUaeahfok,2555
+topolib/assets/AMRES.json,sha256=PWQLEOOIpQzDelkWg27EUTg331Ewlqm-jIQGMNyIIdo,27082
+topolib/assets/Abilene.json,sha256=5cnPMbjsNDjmpySdQDYQ2yd83zOFvCRjlAnbX1wyzgk,6010
+topolib/assets/Bell_canada.json,sha256=pjsZBZcp6j-IP57P8MOlcBqnfnLkzbintKzgJxcC7ZE,19921
+topolib/assets/Brazil.json,sha256=_4rSY46CKBd6ylpLLigaUuWjLrP83OYiYorg4wChmFU,14876
+topolib/assets/CESNET.json,sha256=i6ebe-cDZJtTCDbbeuBo-dQwb7dhD4SRdc0I0N0Eczo,13739
+topolib/assets/CORONET.json,sha256=dSpBSR-vBdxeG1IPcnjw5YYwpzujKMceBuLghGt3xQQ,42017
+topolib/assets/China.json,sha256=L3co98t0MYzPApUi0Hv9i-QthDPHZQSsGcCvunspsGA,23820
+topolib/assets/DT-14.json,sha256=IyNZIWelc2lW1eOa6nHOH4HRfCIOGZtBdwFUQhiS5is,9924
+topolib/assets/DT-17.json,sha256=LBqGD3e3ad9nSaFlnJnrIWMk1eEVC6uF6zDElFzPUu8,10972
+topolib/assets/DT-50.json,sha256=4Pe6R3eWaiHB_KOWO2Pprm3veHl6ttZA2dPyh_GxEGQ,31869
+topolib/assets/ES-30.json,sha256=gXrrOGNveIlWNvO214FHcGo08pCCs1mElCRHXdvIJGY,20186
+topolib/assets/EURO-16.json,sha256=LEd_oq9nY9XP-5wzrCcmhe3VI0D_iDlaNg1lAjeRHik,9513
+topolib/assets/FR-43.json,sha256=QK4-95ETMw4C98jMzpgkpL8y2vlSkbiS7mN9-9QCEPw,26797
+topolib/assets/FUNET.json,sha256=RtVbWVmJHGO-a8gSqCiGWkuxjfou-Ep1JWV4b9M2qIc,6650
+topolib/assets/GCN-BG.json,sha256=e24fO_IlVUu6x9S2zpSJWBD7S9Obkw39TkHcKGzayDQ,18299
+topolib/assets/GRNET.json,sha256=xJMwR4DGANRBcLdjv_8gWQp3ymuSSCR2Cws9_sAx9eg,36781
+topolib/assets/HyperOne.json,sha256=jP-BT5jmozezD0AgOkwdb6nXyqiiL6TMy6BjjVOKEGE,5264
+topolib/assets/IT-21.json,sha256=GmYsmC8jycG1WYGwRvVJtnrurZQ7uy8LN0Kmb8Zmg6w,13503
+topolib/assets/India.json,sha256=7wiXKgWBIO0THTpV8pNE8TCriV-zfytI2lPKK6Y3Zoc,10839
+topolib/assets/JPN-12.json,sha256=qz4uZAjE4GvLQnlBpKAiC9Zrp197r5hI8029hsRlDEY,6902
+topolib/assets/KOREN.json,sha256=lKW2if-IdGxQsiX6vpysXpYw__3_6OTWeZ9-oD-h5Eg,5951
+topolib/assets/NORDUNet.json,sha256=X934bx4bTyn98ASz8Eb0-UuEBxnR-3vTy14-gFn8Oms,16412
+topolib/assets/NSFNet.json,sha256=LeU7h6H76HXEyIpUhlzuR6cbs6Q61eEgxmyUMaNUeEk,8375
+topolib/assets/PANEURO.json,sha256=Xh_pmTBOwlusDhAUGpPzJz4tSRXbK-FTsN3Dnlbwf48,15969
+topolib/assets/PAVLOV.json,sha256=YRDqlodGj-zqErX1cwk4bX-XuUkgP1CpiOmtlLIBmIk,9816
+topolib/assets/PLN-12.json,sha256=oxKs0MTe-pEqD-kTa_LATWfDRswwAnlEFubeQlXMFRU,7135
+topolib/assets/SANReN.json,sha256=VUe7JFbzmte6ED3lNyfm1cWhefsapeZ4P-zIG6-rZXc,3394
+topolib/assets/SERBIA-MONTENEGRO.json,sha256=83Mk5yNju7B50JolE1uCwy8wQJgsaBzZ71xxhMGYOSc,2897
+topolib/assets/Telefonica-21.json,sha256=Wom5SQn3lLL-z-ESZ3Du_Fz_ZQ2bzP2_bNhzmnmmW_I,13302
+topolib/assets/Turk_Telekom.json,sha256=5yO0cgtJOmh6ulCCIb8g_nDFSmv8Wn52Yt_Oyoc1Y0w,11533
+topolib/assets/UKNet.json,sha256=9BEkRM9YFeMMpih5kyDY34TsOGFzt5KBGsmgWd6KoB0,14210
+topolib/assets/Vega_Telecom.json,sha256=E07ZCvG4exRj1a5DV8lqS3Sdo5tRm_Lc07IzIjP2EW0,17324
+topolib/elements/__init__.py,sha256=ZWSd5uwVtppJtMnZ30zrAdzXhUAYIK62RrUaH9rsWu0,180
+topolib/elements/link.py,sha256=BwjdCR8peh7dDpKunZodlIsjKcfvwjKxyQ3tmp1obb8,3778
+topolib/elements/node.py,sha256=uOnljwA3jcVR4utUfI-om4SIt_QBsEQt8QLW9uvjr3Y,4974
+topolib/topology/__init__.py,sha256=2VRhVm4ZvKBiTDB2T8BDmLZBpwGCVFRF3fvtxxC_d28,86
+topolib/topology/path.py,sha256=oUNwmpBcS6LMMAJIxokROm3MVqr7vRR44M3Fh5ADq_w,2057
+topolib/topology/topology.py,sha256=pRmbn8vaFh35Fi0dtSy-fl_3VNVz6MqW-iqfeh3BvPw,14896
+topolib/visualization/__init__.py,sha256=wv065-KB5uDbTaQIASPVfMMW5sE76Bs-q0oai48vAzk,29
+topolib/visualization/mapview.py,sha256=4iKJMYsH6oFNj9tOcgSKLVRtoYfUuH3XLjShsGwUXcM,2799
+topolib-0.5.0.dist-info/METADATA,sha256=8jZNgjgAyWsX3d63Lxo5bI6T6JDCAmKUbu_PQ2svtXM,4169
+topolib-0.5.0.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+topolib-0.5.0.dist-info/licenses/LICENSE,sha256=kbnIP0XU6f2ualiTjEawdlU81IGPBbwc-_GF3N-1e9E,1081
+topolib-0.5.0.dist-info/RECORD,,

topolib-0.4.2.dist-info/RECORD

@@ -1,18 +0,0 @@
-topolib/__init__.py,sha256=iLmy2rOkHS_4KZWMD8BgT7R3tLMKeaTCDVf3B4FyYxM,91
-topolib/analysis/__init__.py,sha256=qvUC9wV_jQNQIlwJXk7LJ-dkTXhH1Ttn0lKWIdwBbrc,52
-topolib/analysis/metrics.py,sha256=2o5PlMVzepDWwSQjzamtKemggqkxh9JuzzCUaeahfok,2555
-topolib/assets/Abilene_IXP.json,sha256=5cnPMbjsNDjmpySdQDYQ2yd83zOFvCRjlAnbX1wyzgk,6010
-topolib/assets/China_pop.json,sha256=v-bs5qib8-CB81rvBNNM9ssoelrNGuFrPEyRCht9YpY,22262
-topolib/assets/DT-50_pop.json,sha256=hCv1MCvipgdrQ34VT-qXJ68BdATriQ8ZObh0fLPqar4,29719
-topolib/elements/__init__.py,sha256=ZWSd5uwVtppJtMnZ30zrAdzXhUAYIK62RrUaH9rsWu0,180
-topolib/elements/link.py,sha256=BwjdCR8peh7dDpKunZodlIsjKcfvwjKxyQ3tmp1obb8,3778
-topolib/elements/node.py,sha256=uOnljwA3jcVR4utUfI-om4SIt_QBsEQt8QLW9uvjr3Y,4974
-topolib/topology/__init__.py,sha256=2VRhVm4ZvKBiTDB2T8BDmLZBpwGCVFRF3fvtxxC_d28,86
-topolib/topology/path.py,sha256=oUNwmpBcS6LMMAJIxokROm3MVqr7vRR44M3Fh5ADq_w,2057
-topolib/topology/topology.py,sha256=PzPx80TxWMcOwpOCh4fohcWT2YUra-4m04tBjqyMU40,5605
-topolib/visualization/__init__.py,sha256=wv065-KB5uDbTaQIASPVfMMW5sE76Bs-q0oai48vAzk,29
-topolib/visualization/mapview.py,sha256=4iKJMYsH6oFNj9tOcgSKLVRtoYfUuH3XLjShsGwUXcM,2799
-topolib-0.4.2.dist-info/METADATA,sha256=O2V7bL8xtAJOhNVmbLX0mHLj1ic5ZZu_3bbnFFbEzAw,3993
-topolib-0.4.2.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-topolib-0.4.2.dist-info/licenses/LICENSE,sha256=kbnIP0XU6f2ualiTjEawdlU81IGPBbwc-_GF3N-1e9E,1081
-topolib-0.4.2.dist-info/RECORD,,