Clean repo
This commit is contained in:
wqy
Binary file not shown.
207
demo.py
207
demo.py
@@ -1,57 +1,184 @@
|
|||||||
from tjnetwork import *
|
from tjnetwork import *
|
||||||
|
|
||||||
JUNCTION = 0
|
|
||||||
RESERVOIR = 1
|
|
||||||
TANK = 2
|
|
||||||
PIPE = 1
|
|
||||||
NODE_COUNT = 0
|
|
||||||
LINK_COUNT = 2
|
|
||||||
|
|
||||||
create_project("net1")
|
def demo_snapshot():
|
||||||
open_project("net1")
|
p = "demo_snapshot"
|
||||||
|
|
||||||
add_node("net1", "node-1", JUNCTION)
|
print(p)
|
||||||
|
|
||||||
undo("net1")
|
if is_project_open(p):
|
||||||
redo("net1")
|
close_project(p)
|
||||||
|
|
||||||
add_node("net1", "node-2", RESERVOIR)
|
if have_project(p):
|
||||||
add_node("net1", "node-3", TANK)
|
delete_project(p)
|
||||||
|
|
||||||
set_node_coord("net1", "node-3", 10.0, 10.0)
|
create_project(p)
|
||||||
|
open_project(p)
|
||||||
|
|
||||||
delete_node("net1", "node-3")
|
add_junction(p, 'j-1', 10.0, 20.0, 30.0)
|
||||||
undo("net1")
|
add_junction(p, 'j-2', 10.0, 20.0, 30.0)
|
||||||
|
add_junction(p, 'j-3', 10.0, 20.0, 30.0)
|
||||||
|
add_junction(p, 'j-4', 10.0, 20.0, 30.0)
|
||||||
|
take_snapshot(p, "1-2-3-4")
|
||||||
|
|
||||||
set_node_coord("net1", "node-1", 10.0, 10.0)
|
undo(p)
|
||||||
undo("net1")
|
undo(p)
|
||||||
redo("net1")
|
undo(p)
|
||||||
|
undo(p)
|
||||||
|
|
||||||
set_node_coord("net1", "node-2", 20.0, 20.0)
|
add_junction(p, 'j-5', 10.0, 20.0, 30.0)
|
||||||
|
add_junction(p, 'j-6', 10.0, 20.0, 30.0)
|
||||||
|
add_junction(p, 'j-7', 10.0, 20.0, 30.0)
|
||||||
|
add_junction(p, 'j-8', 10.0, 20.0, 30.0)
|
||||||
|
take_snapshot(p, "5-6-7-8")
|
||||||
|
|
||||||
add_link("net1", "link-1", PIPE, "node-1", "node-2")
|
print("before checkout, it should be 5, 6, 7, 8")
|
||||||
|
print(get_nodes(p))
|
||||||
|
|
||||||
add_link("net1", "link-2", PIPE, "node-1", "node-3")
|
pick_snapshot(p, "1-2-3-4")
|
||||||
undo("net1")
|
|
||||||
redo("net1")
|
|
||||||
|
|
||||||
add_link("net1", "link-3", PIPE, "node-2", "node-3")
|
print("after checkout, it should be 1, 2, 3, 4")
|
||||||
undo("net1")
|
print(get_nodes(p))
|
||||||
|
|
||||||
# this could fail since link-1 & link-2 ref node-1
|
close_project(p)
|
||||||
delete_node("net1", "node-1")
|
delete_project(p)
|
||||||
|
|
||||||
node_count = get_count("net1", NODE_COUNT)
|
|
||||||
print(node_count.value)
|
|
||||||
for i in range(1, node_count.value + 1):
|
|
||||||
node_id = get_node_id("net1", i)
|
|
||||||
print(node_id.value)
|
|
||||||
|
|
||||||
link_count = get_count("net1", LINK_COUNT)
|
def demo_transaction():
|
||||||
print(link_count.value)
|
p = "demo_transaction"
|
||||||
for i in range(1, link_count.value + 1):
|
|
||||||
link_id = get_link_id("net1", i)
|
|
||||||
print(link_id.value)
|
|
||||||
|
|
||||||
close_project("net1")
|
print(p)
|
||||||
delete_project("net1")
|
|
||||||
|
if is_project_open(p):
|
||||||
|
close_project(p)
|
||||||
|
|
||||||
|
if have_project(p):
|
||||||
|
delete_project(p)
|
||||||
|
|
||||||
|
create_project(p)
|
||||||
|
open_project(p)
|
||||||
|
|
||||||
|
add_junction(p, 'j-1', 10.0, 20.0, 30.0)
|
||||||
|
take_snapshot(p, "1")
|
||||||
|
add_junction(p, 'j-2', 10.0, 20.0, 30.0)
|
||||||
|
take_snapshot(p, "2")
|
||||||
|
|
||||||
|
start_transaction(p)
|
||||||
|
|
||||||
|
add_junction(p, 'j-3', 10.0, 20.0, 30.0)
|
||||||
|
take_snapshot(p, "3")
|
||||||
|
add_junction(p, 'j-4', 10.0, 20.0, 30.0)
|
||||||
|
take_snapshot(p, "4")
|
||||||
|
|
||||||
|
print("before rollback, it should be 1, 2, 3, 4")
|
||||||
|
print(get_nodes(p))
|
||||||
|
|
||||||
|
print("after rollback, it should be 1, 2")
|
||||||
|
abort_transaction(p)
|
||||||
|
|
||||||
|
print(get_nodes(p))
|
||||||
|
|
||||||
|
print(f"have snapshot 1: {have_snapshot(p, '1')}")
|
||||||
|
print(f"have snapshot 2: {have_snapshot(p, '2')}")
|
||||||
|
print(f"have snapshot 3: {have_snapshot(p, '3')}")
|
||||||
|
print(f"have snapshot 4: {have_snapshot(p, '4')}")
|
||||||
|
|
||||||
|
close_project(p)
|
||||||
|
delete_project(p)
|
||||||
|
|
||||||
|
|
||||||
|
def demo_1_title():
|
||||||
|
p = "demo_1_title"
|
||||||
|
|
||||||
|
print(p)
|
||||||
|
|
||||||
|
if is_project_open(p):
|
||||||
|
close_project(p)
|
||||||
|
|
||||||
|
if have_project(p):
|
||||||
|
delete_project(p)
|
||||||
|
|
||||||
|
create_project(p)
|
||||||
|
open_project(p)
|
||||||
|
|
||||||
|
print(get_title(p)) #
|
||||||
|
|
||||||
|
set_title(p, "title")
|
||||||
|
print(get_title(p)) # title
|
||||||
|
|
||||||
|
set_title(p, "test")
|
||||||
|
print(get_title(p)) # test
|
||||||
|
|
||||||
|
undo(p)
|
||||||
|
print(get_title(p)) # title
|
||||||
|
|
||||||
|
undo(p)
|
||||||
|
print(get_title(p)) #
|
||||||
|
|
||||||
|
close_project(p)
|
||||||
|
delete_project(p)
|
||||||
|
|
||||||
|
|
||||||
|
def demo_2_junctions():
|
||||||
|
p = "demo_2_junctions"
|
||||||
|
|
||||||
|
print(p)
|
||||||
|
|
||||||
|
if is_project_open(p):
|
||||||
|
close_project(p)
|
||||||
|
|
||||||
|
if have_project(p):
|
||||||
|
delete_project(p)
|
||||||
|
|
||||||
|
create_project(p)
|
||||||
|
open_project(p)
|
||||||
|
|
||||||
|
j = 'j-1'
|
||||||
|
print(get_junction_coord(p, j)) # None
|
||||||
|
print(get_junction_elevation(p, j)) # None
|
||||||
|
print(get_junction_demand(p, j)) # None
|
||||||
|
print(get_junction_pattern(p, j)) # None
|
||||||
|
|
||||||
|
add_junction(p, j, 10.0, 20.0, 30.0)
|
||||||
|
print(get_junction_coord(p, j)) # {'x': 10.0, 'y': 20.0}
|
||||||
|
print(get_junction_elevation(p, j)) # 30.0
|
||||||
|
print(get_junction_demand(p, j)) # NULL
|
||||||
|
print(get_junction_pattern(p, j)) # NULL
|
||||||
|
|
||||||
|
set_junction_demand(p, j, 100.0)
|
||||||
|
print(get_junction_demand(p, j)) # 100.0
|
||||||
|
|
||||||
|
undo(p)
|
||||||
|
print(get_junction_demand(p, j)) # NULL
|
||||||
|
|
||||||
|
undo(p)
|
||||||
|
print(get_junction_coord(p, j)) # None
|
||||||
|
print(get_junction_elevation(p, j)) # None
|
||||||
|
print(get_junction_demand(p, j)) # None
|
||||||
|
print(get_junction_pattern(p, j)) # None
|
||||||
|
|
||||||
|
add_junction(p, j, 10.0, 20.0, 30.0)
|
||||||
|
print(get_junction_coord(p, j)) # {'x': 10.0, 'y': 20.0}
|
||||||
|
print(get_junction_elevation(p, j)) # 30.0
|
||||||
|
print(get_junction_demand(p, j)) # NULL
|
||||||
|
print(get_junction_pattern(p, j)) # NULL
|
||||||
|
|
||||||
|
set_junction_coord(p, j, 100.0, 200.0)
|
||||||
|
print(get_junction_coord(p, j)) # {'x': 100.0, 'y': 200.0}
|
||||||
|
|
||||||
|
undo(p)
|
||||||
|
print(get_junction_coord(p, j)) # {'x': 10.0, 'y': 20.0}
|
||||||
|
|
||||||
|
redo(p)
|
||||||
|
print(get_junction_coord(p, j)) # {'x': 100.0, 'y': 200.0}
|
||||||
|
|
||||||
|
close_project(p)
|
||||||
|
delete_project(p)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
demo_snapshot()
|
||||||
|
demo_transaction()
|
||||||
|
demo_1_title()
|
||||||
|
demo_2_junctions()
|
||||||
|
pass
|
||||||
|
|||||||
BIN
epanet2.dll
BIN
epanet2.dll
Binary file not shown.
2
main.py
2
main.py
@@ -7,7 +7,7 @@ from fastapi import FastAPI, File, UploadFile
|
|||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
from starlette.responses import FileResponse, JSONResponse
|
from starlette.responses import FileResponse, JSONResponse
|
||||||
from fastapi import FastAPI, Response, status
|
from fastapi import FastAPI, Response, status
|
||||||
from tjnetwork_new import *
|
from tjnetwork import *
|
||||||
|
|
||||||
JUNCTION = 0
|
JUNCTION = 0
|
||||||
RESERVOIR = 1
|
RESERVOIR = 1
|
||||||
|
|||||||
184
new_demo.py
184
new_demo.py
@@ -1,184 +0,0 @@
|
|||||||
from tjnetwork_new import *
|
|
||||||
|
|
||||||
|
|
||||||
def demo_snapshot():
|
|
||||||
p = "demo_snapshot"
|
|
||||||
|
|
||||||
print(p)
|
|
||||||
|
|
||||||
if is_project_open(p):
|
|
||||||
close_project(p)
|
|
||||||
|
|
||||||
if have_project(p):
|
|
||||||
delete_project(p)
|
|
||||||
|
|
||||||
create_project(p)
|
|
||||||
open_project(p)
|
|
||||||
|
|
||||||
add_junction(p, 'j-1', 10.0, 20.0, 30.0)
|
|
||||||
add_junction(p, 'j-2', 10.0, 20.0, 30.0)
|
|
||||||
add_junction(p, 'j-3', 10.0, 20.0, 30.0)
|
|
||||||
add_junction(p, 'j-4', 10.0, 20.0, 30.0)
|
|
||||||
take_snapshot(p, "1-2-3-4")
|
|
||||||
|
|
||||||
undo(p)
|
|
||||||
undo(p)
|
|
||||||
undo(p)
|
|
||||||
undo(p)
|
|
||||||
|
|
||||||
add_junction(p, 'j-5', 10.0, 20.0, 30.0)
|
|
||||||
add_junction(p, 'j-6', 10.0, 20.0, 30.0)
|
|
||||||
add_junction(p, 'j-7', 10.0, 20.0, 30.0)
|
|
||||||
add_junction(p, 'j-8', 10.0, 20.0, 30.0)
|
|
||||||
take_snapshot(p, "5-6-7-8")
|
|
||||||
|
|
||||||
print("before checkout, it should be 5, 6, 7, 8")
|
|
||||||
print(get_nodes(p))
|
|
||||||
|
|
||||||
pick_snapshot(p, "1-2-3-4")
|
|
||||||
|
|
||||||
print("after checkout, it should be 1, 2, 3, 4")
|
|
||||||
print(get_nodes(p))
|
|
||||||
|
|
||||||
close_project(p)
|
|
||||||
delete_project(p)
|
|
||||||
|
|
||||||
|
|
||||||
def demo_transaction():
|
|
||||||
p = "demo_transaction"
|
|
||||||
|
|
||||||
print(p)
|
|
||||||
|
|
||||||
if is_project_open(p):
|
|
||||||
close_project(p)
|
|
||||||
|
|
||||||
if have_project(p):
|
|
||||||
delete_project(p)
|
|
||||||
|
|
||||||
create_project(p)
|
|
||||||
open_project(p)
|
|
||||||
|
|
||||||
add_junction(p, 'j-1', 10.0, 20.0, 30.0)
|
|
||||||
take_snapshot(p, "1")
|
|
||||||
add_junction(p, 'j-2', 10.0, 20.0, 30.0)
|
|
||||||
take_snapshot(p, "2")
|
|
||||||
|
|
||||||
start_transaction(p)
|
|
||||||
|
|
||||||
add_junction(p, 'j-3', 10.0, 20.0, 30.0)
|
|
||||||
take_snapshot(p, "3")
|
|
||||||
add_junction(p, 'j-4', 10.0, 20.0, 30.0)
|
|
||||||
take_snapshot(p, "4")
|
|
||||||
|
|
||||||
print("before rollback, it should be 1, 2, 3, 4")
|
|
||||||
print(get_nodes(p))
|
|
||||||
|
|
||||||
print("after rollback, it should be 1, 2")
|
|
||||||
abort_transaction(p)
|
|
||||||
|
|
||||||
print(get_nodes(p))
|
|
||||||
|
|
||||||
print(f"have snapshot 1: {have_snapshot(p, '1')}")
|
|
||||||
print(f"have snapshot 2: {have_snapshot(p, '2')}")
|
|
||||||
print(f"have snapshot 3: {have_snapshot(p, '3')}")
|
|
||||||
print(f"have snapshot 4: {have_snapshot(p, '4')}")
|
|
||||||
|
|
||||||
close_project(p)
|
|
||||||
delete_project(p)
|
|
||||||
|
|
||||||
|
|
||||||
def demo_1_title():
|
|
||||||
p = "demo_1_title"
|
|
||||||
|
|
||||||
print(p)
|
|
||||||
|
|
||||||
if is_project_open(p):
|
|
||||||
close_project(p)
|
|
||||||
|
|
||||||
if have_project(p):
|
|
||||||
delete_project(p)
|
|
||||||
|
|
||||||
create_project(p)
|
|
||||||
open_project(p)
|
|
||||||
|
|
||||||
print(get_title(p)) #
|
|
||||||
|
|
||||||
set_title(p, "title")
|
|
||||||
print(get_title(p)) # title
|
|
||||||
|
|
||||||
set_title(p, "test")
|
|
||||||
print(get_title(p)) # test
|
|
||||||
|
|
||||||
undo(p)
|
|
||||||
print(get_title(p)) # title
|
|
||||||
|
|
||||||
undo(p)
|
|
||||||
print(get_title(p)) #
|
|
||||||
|
|
||||||
close_project(p)
|
|
||||||
delete_project(p)
|
|
||||||
|
|
||||||
|
|
||||||
def demo_2_junctions():
|
|
||||||
p = "demo_2_junctions"
|
|
||||||
|
|
||||||
print(p)
|
|
||||||
|
|
||||||
if is_project_open(p):
|
|
||||||
close_project(p)
|
|
||||||
|
|
||||||
if have_project(p):
|
|
||||||
delete_project(p)
|
|
||||||
|
|
||||||
create_project(p)
|
|
||||||
open_project(p)
|
|
||||||
|
|
||||||
j = 'j-1'
|
|
||||||
print(get_junction_coord(p, j)) # None
|
|
||||||
print(get_junction_elevation(p, j)) # None
|
|
||||||
print(get_junction_demand(p, j)) # None
|
|
||||||
print(get_junction_pattern(p, j)) # None
|
|
||||||
|
|
||||||
add_junction(p, j, 10.0, 20.0, 30.0)
|
|
||||||
print(get_junction_coord(p, j)) # {'x': 10.0, 'y': 20.0}
|
|
||||||
print(get_junction_elevation(p, j)) # 30.0
|
|
||||||
print(get_junction_demand(p, j)) # NULL
|
|
||||||
print(get_junction_pattern(p, j)) # NULL
|
|
||||||
|
|
||||||
set_junction_demand(p, j, 100.0)
|
|
||||||
print(get_junction_demand(p, j)) # 100.0
|
|
||||||
|
|
||||||
undo(p)
|
|
||||||
print(get_junction_demand(p, j)) # NULL
|
|
||||||
|
|
||||||
undo(p)
|
|
||||||
print(get_junction_coord(p, j)) # None
|
|
||||||
print(get_junction_elevation(p, j)) # None
|
|
||||||
print(get_junction_demand(p, j)) # None
|
|
||||||
print(get_junction_pattern(p, j)) # None
|
|
||||||
|
|
||||||
add_junction(p, j, 10.0, 20.0, 30.0)
|
|
||||||
print(get_junction_coord(p, j)) # {'x': 10.0, 'y': 20.0}
|
|
||||||
print(get_junction_elevation(p, j)) # 30.0
|
|
||||||
print(get_junction_demand(p, j)) # NULL
|
|
||||||
print(get_junction_pattern(p, j)) # NULL
|
|
||||||
|
|
||||||
set_junction_coord(p, j, 100.0, 200.0)
|
|
||||||
print(get_junction_coord(p, j)) # {'x': 100.0, 'y': 200.0}
|
|
||||||
|
|
||||||
undo(p)
|
|
||||||
print(get_junction_coord(p, j)) # {'x': 10.0, 'y': 20.0}
|
|
||||||
|
|
||||||
redo(p)
|
|
||||||
print(get_junction_coord(p, j)) # {'x': 100.0, 'y': 200.0}
|
|
||||||
|
|
||||||
close_project(p)
|
|
||||||
delete_project(p)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
demo_snapshot()
|
|
||||||
demo_transaction()
|
|
||||||
demo_1_title()
|
|
||||||
demo_2_junctions()
|
|
||||||
pass
|
|
||||||
Binary file not shown.
BIN
pytjnetwork.pyd
BIN
pytjnetwork.pyd
Binary file not shown.
220
tjnetwork.py
220
tjnetwork.py
@@ -1,98 +1,170 @@
|
|||||||
import ctypes
|
import api
|
||||||
|
|
||||||
max_name_len = 32
|
|
||||||
|
|
||||||
server = ctypes.windll.LoadLibrary("./TJNetworkServer.dll")
|
############################################################
|
||||||
|
# ChangeSet
|
||||||
|
############################################################
|
||||||
|
|
||||||
class LinkNodes:
|
ChangeSet = api.ChangeSet
|
||||||
def __init__(self, from_node, to_node):
|
|
||||||
self.from_node = from_node
|
|
||||||
self.to_node = to_node
|
|
||||||
|
|
||||||
class Point:
|
|
||||||
def __init__(self, x, y):
|
|
||||||
self.x = x
|
|
||||||
self.y = y
|
|
||||||
|
|
||||||
class Result:
|
############################################################
|
||||||
def __init__(self, ec, v):
|
# enum
|
||||||
self.error_code = ec
|
############################################################
|
||||||
self.value = v
|
|
||||||
|
|
||||||
def c_str(value):
|
JUNCTION = api.JUNCTION
|
||||||
return ctypes.c_char_p(value.encode())
|
RESERVOIR = api.RESERVOIR
|
||||||
|
TANK = api.TANK
|
||||||
|
PIPE = api.PIPE
|
||||||
|
PUMP = api.PUMP
|
||||||
|
VALVE = api.VALVE
|
||||||
|
|
||||||
def c_ref(value):
|
|
||||||
return ctypes.byref(value)
|
|
||||||
|
|
||||||
# have database
|
############################################################
|
||||||
def have_project(name):
|
# project
|
||||||
yes = 406
|
############################################################
|
||||||
ec = server.TJ_haveproject(c_str(name))
|
|
||||||
return True if ec == yes else False
|
|
||||||
|
|
||||||
# create database
|
def have_project(name: str) -> bool:
|
||||||
def create_project(name):
|
return api.have_project(name)
|
||||||
return server.TJ_createproject(c_str(name))
|
|
||||||
|
|
||||||
# delete database
|
def create_project(name: str) -> None:
|
||||||
def delete_project(name):
|
return api.create_project(name)
|
||||||
return server.TJ_deleteproject(c_str(name))
|
|
||||||
|
|
||||||
# create runtime project and connect to database
|
def delete_project(name: str) -> None:
|
||||||
def open_project(name):
|
return api.delete_project(name)
|
||||||
return server.TJ_openproject(c_str(name))
|
|
||||||
|
|
||||||
# delete runtime project and disconnect from database
|
def is_project_open(name: str) -> bool:
|
||||||
def close_project(name):
|
return api.is_project_open(name)
|
||||||
return server.TJ_closeproject(c_str(name))
|
|
||||||
|
|
||||||
def undo(name):
|
def open_project(name: str) -> None:
|
||||||
return server.TJ_undo(c_str(name))
|
return api.open_project(name)
|
||||||
|
|
||||||
def redo(name):
|
def close_project(name: str) -> None:
|
||||||
return server.TJ_redo(c_str(name))
|
return api.close_project(name)
|
||||||
|
|
||||||
def get_count(name, object_type):
|
def copy_project(source: str, new: str) -> None:
|
||||||
count = ctypes.c_int()
|
return api.copy_project(source, new)
|
||||||
ec = server.TJ_getcount(c_str(name), object_type, c_ref(count))
|
|
||||||
return Result(ec, count.value)
|
|
||||||
|
|
||||||
def add_node(name, node_id, node_type):
|
|
||||||
return server.TJ_addnode(c_str(name), c_str(node_id), node_type)
|
|
||||||
|
|
||||||
def delete_node(name, node_id):
|
############################################################
|
||||||
return server.TJ_deletenode(c_str(name), c_str(node_id))
|
# operation
|
||||||
|
############################################################
|
||||||
|
|
||||||
def get_node_id(name, node_index):
|
def undo(name: str) -> None:
|
||||||
id = ctypes.create_string_buffer(max_name_len)
|
return api.undo(name)
|
||||||
ec = server.TJ_getnodeid(c_str(name), node_index, c_ref(id))
|
|
||||||
return Result(ec, id.value)
|
|
||||||
|
|
||||||
def get_node_coord(name, node_id):
|
def redo(name: str) -> None:
|
||||||
x = ctypes.c_double()
|
return api.redo(name)
|
||||||
y = ctypes.c_double()
|
|
||||||
ec = server.TJ_getnodecoord(c_str(name), c_str(node_id), c_ref(x), c_ref(y))
|
|
||||||
return Result(ec, Point(x.value, y.value))
|
|
||||||
|
|
||||||
def set_node_coord(name, node_id, x, y):
|
def have_snapshot(name: str, tag: str) -> bool:
|
||||||
func = server.TJ_setnodecoord
|
return api.have_snapshot(name, tag)
|
||||||
func.argtypes = [ctypes.c_char_p, ctypes.c_char_p, ctypes.c_double, ctypes.c_double]
|
|
||||||
return func(c_str(name), c_str(node_id), x, y)
|
|
||||||
|
|
||||||
def add_link(name, link_id, link_type, from_node, to_node):
|
def take_snapshot(name: str, tag: str) -> None:
|
||||||
return server.TJ_addlink(c_str(name), c_str(link_id), link_type, c_str(from_node), c_str(to_node))
|
return api.take_snapshot(name, tag)
|
||||||
|
|
||||||
def delete_link(name, link_id):
|
def pick_snapshot(name: str, tag: str) -> None:
|
||||||
return server.TJ_deletelink(c_str(name), c_str(link_id))
|
return api.pick_snapshot(name, tag)
|
||||||
|
|
||||||
def get_link_id(name, link_index):
|
def have_transaction(name: str) -> bool:
|
||||||
id = ctypes.create_string_buffer(max_name_len)
|
return api.have_transaction(name)
|
||||||
ec = server.TJ_getlinkid(c_str(name), link_index, c_ref(id))
|
|
||||||
return Result(ec, id.value)
|
|
||||||
|
|
||||||
def get_link_nodes(name, link_id):
|
def start_transaction(name: str, strict: bool = False) -> None:
|
||||||
from_node = ctypes.create_string_buffer(max_name_len)
|
return api.start_transaction(name, strict)
|
||||||
to_node = ctypes.create_string_buffer(max_name_len)
|
|
||||||
ec = server.TJ_getlinknodes(c_str(name), c_str(link_id), c_ref(from_node), c_ref(to_node))
|
def commit_transaction(name: str) -> None:
|
||||||
return Result(ec, LinkNodes(from_node.value, to_node.value))
|
return api.commit_transaction(name)
|
||||||
|
|
||||||
|
def abort_transaction(name: str) -> None:
|
||||||
|
return api.abort_transaction(name)
|
||||||
|
|
||||||
|
############################################################
|
||||||
|
# type
|
||||||
|
############################################################
|
||||||
|
|
||||||
|
def is_node(name: str, node_id: str) -> bool:
|
||||||
|
return api.is_node(name, node_id)
|
||||||
|
|
||||||
|
def is_junction(name: str, node_id: str) -> bool:
|
||||||
|
return api.is_junction(name, node_id)
|
||||||
|
|
||||||
|
def is_reservoir(name: str, node_id: str) -> bool:
|
||||||
|
return api.is_reservoir(name, node_id)
|
||||||
|
|
||||||
|
def is_tank(name: str, node_id: str) -> bool:
|
||||||
|
return api.is_tank(name, node_id)
|
||||||
|
|
||||||
|
def is_link(name: str, link_id: str) -> bool:
|
||||||
|
return api.is_link(name, link_id)
|
||||||
|
|
||||||
|
def is_pipe(name: str, link_id: str) -> bool:
|
||||||
|
return api.is_pipe(name, link_id)
|
||||||
|
|
||||||
|
def is_pump(name: str, link_id: str) -> bool:
|
||||||
|
return api.is_pump(name, link_id)
|
||||||
|
|
||||||
|
def is_valve(name: str, link_id: str) -> bool:
|
||||||
|
return api.is_valve(name, link_id)
|
||||||
|
|
||||||
|
def is_curve(name: str, curve_id: str) -> bool:
|
||||||
|
return api.is_curve(name, curve_id)
|
||||||
|
|
||||||
|
def is_pattern(name: str, pattern_id: str) -> bool:
|
||||||
|
return api.is_pattern(name, pattern_id)
|
||||||
|
|
||||||
|
def get_nodes(name: str) -> list[str]:
|
||||||
|
return api.get_nodes(name)
|
||||||
|
|
||||||
|
def get_links(name: str) -> list[str]:
|
||||||
|
return api.get_links(name)
|
||||||
|
|
||||||
|
def get_curves(name: str) -> list[str]:
|
||||||
|
return api.get_curves(name)
|
||||||
|
|
||||||
|
def get_patterns(name: str) -> list[str]:
|
||||||
|
return api.get_patterns(name)
|
||||||
|
|
||||||
|
|
||||||
|
############################################################
|
||||||
|
# title 1.[TITLE]
|
||||||
|
############################################################
|
||||||
|
|
||||||
|
def set_title(name: str, title: str) -> ChangeSet:
|
||||||
|
return api.set_title(name, title)
|
||||||
|
|
||||||
|
def get_title(name: str) -> str:
|
||||||
|
return api.get_title(name)
|
||||||
|
|
||||||
|
|
||||||
|
############################################################
|
||||||
|
# junction 2.[JUNCTIONS]
|
||||||
|
############################################################
|
||||||
|
|
||||||
|
def add_junction(name: str, junction_id: str, x: float, y: float, elevation: float) -> ChangeSet:
|
||||||
|
return api.add_junction(name, junction_id, x, y, elevation)
|
||||||
|
|
||||||
|
def delete_junction(name: str, junction_id: str) -> ChangeSet:
|
||||||
|
return api.delete_junction(name, junction_id)
|
||||||
|
|
||||||
|
def get_junction_elevation(name: str, junction_id: str) -> float | None:
|
||||||
|
return api.get_junction_elevation(name, junction_id)
|
||||||
|
|
||||||
|
def get_junction_demand(name: str, junction_id: str) -> float | str | None:
|
||||||
|
return api.get_junction_demand(name, junction_id)
|
||||||
|
|
||||||
|
def get_junction_pattern(name: str, junction_id: str) -> str | None:
|
||||||
|
return api.get_junction_pattern(name, junction_id)
|
||||||
|
|
||||||
|
def get_junction_coord(name: str, junction_id: str) -> dict[str, float] | None:
|
||||||
|
return api.get_junction_coord(name, junction_id)
|
||||||
|
|
||||||
|
def set_junction_elevation(name: str, junction_id: str, elevation: float) -> ChangeSet:
|
||||||
|
return api.set_junction_elevation(name, junction_id, elevation)
|
||||||
|
|
||||||
|
def set_junction_demand(name: str, junction_id: str, demand: float) -> ChangeSet:
|
||||||
|
return api.set_junction_demand(name, junction_id, demand)
|
||||||
|
|
||||||
|
def set_junction_pattern(name: str, junction_id: str, pattern: str) -> ChangeSet:
|
||||||
|
return api.set_junction_pattern(name, junction_id, pattern)
|
||||||
|
|
||||||
|
def set_junction_coord(name: str, junction_id: str, x: float, y: float) -> ChangeSet:
|
||||||
|
return api.set_junction_coord(name, junction_id, x, y)
|
||||||
|
|||||||
170
tjnetwork_new.py
170
tjnetwork_new.py
@@ -1,170 +0,0 @@
|
|||||||
import api
|
|
||||||
|
|
||||||
|
|
||||||
############################################################
|
|
||||||
# ChangeSet
|
|
||||||
############################################################
|
|
||||||
|
|
||||||
ChangeSet = api.ChangeSet
|
|
||||||
|
|
||||||
|
|
||||||
############################################################
|
|
||||||
# enum
|
|
||||||
############################################################
|
|
||||||
|
|
||||||
JUNCTION = api.JUNCTION
|
|
||||||
RESERVOIR = api.RESERVOIR
|
|
||||||
TANK = api.TANK
|
|
||||||
PIPE = api.PIPE
|
|
||||||
PUMP = api.PUMP
|
|
||||||
VALVE = api.VALVE
|
|
||||||
|
|
||||||
|
|
||||||
############################################################
|
|
||||||
# project
|
|
||||||
############################################################
|
|
||||||
|
|
||||||
def have_project(name: str) -> bool:
|
|
||||||
return api.have_project(name)
|
|
||||||
|
|
||||||
def create_project(name: str) -> None:
|
|
||||||
return api.create_project(name)
|
|
||||||
|
|
||||||
def delete_project(name: str) -> None:
|
|
||||||
return api.delete_project(name)
|
|
||||||
|
|
||||||
def is_project_open(name: str) -> bool:
|
|
||||||
return api.is_project_open(name)
|
|
||||||
|
|
||||||
def open_project(name: str) -> None:
|
|
||||||
return api.open_project(name)
|
|
||||||
|
|
||||||
def close_project(name: str) -> None:
|
|
||||||
return api.close_project(name)
|
|
||||||
|
|
||||||
def copy_project(source: str, new: str) -> None:
|
|
||||||
return api.copy_project(source, new)
|
|
||||||
|
|
||||||
|
|
||||||
############################################################
|
|
||||||
# operation
|
|
||||||
############################################################
|
|
||||||
|
|
||||||
def undo(name: str) -> None:
|
|
||||||
return api.undo(name)
|
|
||||||
|
|
||||||
def redo(name: str) -> None:
|
|
||||||
return api.redo(name)
|
|
||||||
|
|
||||||
def have_snapshot(name: str, tag: str) -> bool:
|
|
||||||
return api.have_snapshot(name, tag)
|
|
||||||
|
|
||||||
def take_snapshot(name: str, tag: str) -> None:
|
|
||||||
return api.take_snapshot(name, tag)
|
|
||||||
|
|
||||||
def pick_snapshot(name: str, tag: str) -> None:
|
|
||||||
return api.pick_snapshot(name, tag)
|
|
||||||
|
|
||||||
def have_transaction(name: str) -> bool:
|
|
||||||
return api.have_transaction(name)
|
|
||||||
|
|
||||||
def start_transaction(name: str, strict: bool = False) -> None:
|
|
||||||
return api.start_transaction(name, strict)
|
|
||||||
|
|
||||||
def commit_transaction(name: str) -> None:
|
|
||||||
return api.commit_transaction(name)
|
|
||||||
|
|
||||||
def abort_transaction(name: str) -> None:
|
|
||||||
return api.abort_transaction(name)
|
|
||||||
|
|
||||||
############################################################
|
|
||||||
# type
|
|
||||||
############################################################
|
|
||||||
|
|
||||||
def is_node(name: str, node_id: str) -> bool:
|
|
||||||
return api.is_node(name, node_id)
|
|
||||||
|
|
||||||
def is_junction(name: str, node_id: str) -> bool:
|
|
||||||
return api.is_junction(name, node_id)
|
|
||||||
|
|
||||||
def is_reservoir(name: str, node_id: str) -> bool:
|
|
||||||
return api.is_reservoir(name, node_id)
|
|
||||||
|
|
||||||
def is_tank(name: str, node_id: str) -> bool:
|
|
||||||
return api.is_tank(name, node_id)
|
|
||||||
|
|
||||||
def is_link(name: str, link_id: str) -> bool:
|
|
||||||
return api.is_link(name, link_id)
|
|
||||||
|
|
||||||
def is_pipe(name: str, link_id: str) -> bool:
|
|
||||||
return api.is_pipe(name, link_id)
|
|
||||||
|
|
||||||
def is_pump(name: str, link_id: str) -> bool:
|
|
||||||
return api.is_pump(name, link_id)
|
|
||||||
|
|
||||||
def is_valve(name: str, link_id: str) -> bool:
|
|
||||||
return api.is_valve(name, link_id)
|
|
||||||
|
|
||||||
def is_curve(name: str, curve_id: str) -> bool:
|
|
||||||
return api.is_curve(name, curve_id)
|
|
||||||
|
|
||||||
def is_pattern(name: str, pattern_id: str) -> bool:
|
|
||||||
return api.is_pattern(name, pattern_id)
|
|
||||||
|
|
||||||
def get_nodes(name: str) -> list[str]:
|
|
||||||
return api.get_nodes(name)
|
|
||||||
|
|
||||||
def get_links(name: str) -> list[str]:
|
|
||||||
return api.get_links(name)
|
|
||||||
|
|
||||||
def get_curves(name: str) -> list[str]:
|
|
||||||
return api.get_curves(name)
|
|
||||||
|
|
||||||
def get_patterns(name: str) -> list[str]:
|
|
||||||
return api.get_patterns(name)
|
|
||||||
|
|
||||||
|
|
||||||
############################################################
|
|
||||||
# title 1.[TITLE]
|
|
||||||
############################################################
|
|
||||||
|
|
||||||
def set_title(name: str, title: str) -> ChangeSet:
|
|
||||||
return api.set_title(name, title)
|
|
||||||
|
|
||||||
def get_title(name: str) -> str:
|
|
||||||
return api.get_title(name)
|
|
||||||
|
|
||||||
|
|
||||||
############################################################
|
|
||||||
# junction 2.[JUNCTIONS]
|
|
||||||
############################################################
|
|
||||||
|
|
||||||
def add_junction(name: str, junction_id: str, x: float, y: float, elevation: float) -> ChangeSet:
|
|
||||||
return api.add_junction(name, junction_id, x, y, elevation)
|
|
||||||
|
|
||||||
def delete_junction(name: str, junction_id: str) -> ChangeSet:
|
|
||||||
return api.delete_junction(name, junction_id)
|
|
||||||
|
|
||||||
def get_junction_elevation(name: str, junction_id: str) -> float | None:
|
|
||||||
return api.get_junction_elevation(name, junction_id)
|
|
||||||
|
|
||||||
def get_junction_demand(name: str, junction_id: str) -> float | str | None:
|
|
||||||
return api.get_junction_demand(name, junction_id)
|
|
||||||
|
|
||||||
def get_junction_pattern(name: str, junction_id: str) -> str | None:
|
|
||||||
return api.get_junction_pattern(name, junction_id)
|
|
||||||
|
|
||||||
def get_junction_coord(name: str, junction_id: str) -> dict[str, float] | None:
|
|
||||||
return api.get_junction_coord(name, junction_id)
|
|
||||||
|
|
||||||
def set_junction_elevation(name: str, junction_id: str, elevation: float) -> ChangeSet:
|
|
||||||
return api.set_junction_elevation(name, junction_id, elevation)
|
|
||||||
|
|
||||||
def set_junction_demand(name: str, junction_id: str, demand: float) -> ChangeSet:
|
|
||||||
return api.set_junction_demand(name, junction_id, demand)
|
|
||||||
|
|
||||||
def set_junction_pattern(name: str, junction_id: str, pattern: str) -> ChangeSet:
|
|
||||||
return api.set_junction_pattern(name, junction_id, pattern)
|
|
||||||
|
|
||||||
def set_junction_coord(name: str, junction_id: str, x: float, y: float) -> ChangeSet:
|
|
||||||
return api.set_junction_coord(name, junction_id, x, y)
|
|
||||||
Binary file not shown.
Reference in New Issue
Block a user