creator.py

# %%
from lib2to3.pytree import Node
from vnfplacement.problem_network import ProblemNetwork
from vnfplacement.defines import NodeProperty, PropertyType, LinkProperty
import networkx as nx


# %% settings
init_seed = 333
graph_size = 8
edge_prob = 0.4
# %%
net = ProblemNetwork.from_random_gen(graph_size, edge_prob, init_seed)


node_res = {NodeProperty.MEMORY : 5,  NodeProperty.CPU: 1}
node_costs = {NodeProperty.MEMORY : 1,  NodeProperty.CPU: 1}
for n in list(net.nodes):
    net.set_node_properties(n, PropertyType.RESOURCE, node_res)
    net.set_node_properties(n, PropertyType.COST, node_costs)

for e in net.links:
    net.set_link_properties(e, PropertyType.RESOURCE, {LinkProperty.BANDWIDTH : 1})
    net.set_link_properties(e, PropertyType.COST, {LinkProperty.BANDWIDTH : 1})
    net.set_link_properties(e, PropertyType.DRAWBACK, {LinkProperty.DELAY : 0.2})

net.draw()
# %%


nx.write_gpickle(net._pnet, "test.gpickle")
net_copy = nx.read_gpickle("test.gpickle")

print(net._pnet.nodes())
print(net_copy.nodes())
#net_copy.nodes()
# %%

# %%