Actions and functions and stuff

This commit is contained in:
Geoffrey Frogeye 2016-02-11 19:56:36 +01:00
parent 9eb24388ff
commit 1c42ca8ccb

62
main.py
View file

@ -13,12 +13,16 @@ P = 0 # Nb products
W = 0 # Nb warehouses W = 0 # Nb warehouses
C = 0 # Nb customers C = 0 # Nb customers
t = 0 # Turn
Dp = []; # Positions of drones Dp = []; # Positions of drones
# (x, y) # (x, y)
Di = []; # Items of drones Di = []; # Items of drones
# {product number: qt} # {product number: qt}
Dd = []; # Turn avaibility of drone Dd = []; # Turn avaibility of drone
# int # int
Da = []; # Avaibles drones
# int
Pw = []; # Weight of products Pw = []; # Weight of products
# int # int
@ -33,7 +37,65 @@ Cp = []; # Positions of customers
Ci = []; # Needs of customers Ci = []; # Needs of customers
# {product number: qt} # {product number: qt}
# Utilities
def distance(A, B): def distance(A, B):
return math.ceil(sqrt(pow((B[0] - A[0], 2) + pow(B[1] - A[1], 2)))) return math.ceil(sqrt(pow((B[0] - A[0], 2) + pow(B[1] - A[1], 2))))
def weight(d):
return sum([Pw[i] for i in Di[d]])
# Actions
def load(d, w, p, q):
# drone number, warehouse, product, qt
assert(d in Da)
if (Dp[d] != Wp[w]):
Dd[d] += distance(Dp[d], Wp[w])
Wi[w][p] += -q
Di[d][p] += +q
assert(Wp[w][p] >= 0)
assert(weight(d) <= M)
assert(Dd[d] <= T);
print("Drone", d, "loads", q, "of", p, "from warehouse", w, "", Dd[d])
def unload(d, w, p, q):
# drone number, warehouse, product, qt
assert(d in Da)
if (Dp[d] != Wp[w]):
Dd[d] += distance(Dp[d], Wp[w])
Wi[w][p] += +q
Di[d][p] += -q
assert(Dd[d] <= T);
print("Drone", d, "unloads", q, "of", p, "from warehouse", w, "", Dd[d])
def deliver(d, c, p, q):
# drone number, customer, product, qt
assert(d in Da)
if (Dp[d] != Cp[c]):
Dd[d] += distance(Dp[d], Cp[c])
Ci[w][p] += +q
Di[d][p] += -q
Dd[d] += 1
assert(Dd[d] <= T);
print("Drone", d, "delivers", q, "of", p, "to client", w, "", Dd[d])
def wait(d, w=1):
assert(d in Da)
Dd[d] += w
print("Drone", d, "waits", w, "turn" + ('s' if w >= 2 else ''), "", Dd[d])
# Control
def newTurn():
t += 1
print("--- Turn", t)
for d in Da:
wait(d)
assert(Da.len == 0)
for d in range(D):
if Dd[d] <= t:
Da.push(d)
print("Drones", ", ".join(Da), "(", len(Da), ")", "are avaible")
def end():
print("--- End!")
# IA