298 lines
8.0 KiB
Python
Executable File
298 lines
8.0 KiB
Python
Executable File
# %%
|
|
PROJECT_PATH = '/home/md/Work/ligalytics/leagues_stable/'
|
|
import os, sys
|
|
sys.path.insert(0, PROJECT_PATH)
|
|
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "leagues.settings")
|
|
os.environ["DJANGO_ALLOW_ASYNC_UNSAFE"] = "true"
|
|
import random
|
|
|
|
# XPRESS ENVIRONMENT
|
|
os.environ['XPRESSDIR'] = "/opt/xpressmp_8.4"
|
|
os.environ['XPRESS'] = "/opt/xpressmp_8.4/bin"
|
|
os.environ['LD_LIBRARY_PATH'] = os.environ['XPRESSDIR']+"/lib:"+os.environ['LD_LIBRARY_PATH']
|
|
os.environ['DYLD_LIBRARY_PATH'] = os.environ['XPRESSDIR']+"/lib:"
|
|
os.environ['SHLIB_PATH'] = os.environ['XPRESSDIR']+"/lib:"
|
|
os.environ['LIBPATH'] = os.environ['XPRESSDIR']+"/lib:"
|
|
# os.environ['PYTHONPATH'] =os.environ['XPRESSDIR']+"/lib:"+os.environ['PYTHONPATH']
|
|
os.environ['PYTHONPATH'] =os.environ['XPRESSDIR']+"/lib:"
|
|
os.environ['CLASSPATH'] =os.environ['XPRESSDIR']+"/lib/xprs.jar:"
|
|
os.environ['CLASSPATH'] =os.environ['XPRESSDIR']+"/lib/xprb.jar:"+os.environ['CLASSPATH']
|
|
os.environ['CLASSPATH'] =os.environ['XPRESSDIR']+"/lib/xprm.jar:"+os.environ['CLASSPATH']
|
|
os.environ['PATH'] =os.environ['XPRESSDIR']+"/bin:"+os.environ['PATH']
|
|
|
|
|
|
from leagues import settings
|
|
settings.DATABASES['default']['NAME'] = PROJECT_PATH+'/db.sqlite3'
|
|
|
|
import django
|
|
django.setup()
|
|
|
|
from scheduler.models import *
|
|
from pulp import *
|
|
|
|
|
|
import xpress as xp
|
|
xp.controls.outputlog = 1
|
|
|
|
|
|
|
|
scenario = Scenario.objects.get(id=32)
|
|
|
|
|
|
# %%
|
|
|
|
teams = Team.objects.filter(season=scenario.season,active=True).order_by('position')
|
|
team_ids = list(teams.values_list('id',flat=True))
|
|
|
|
|
|
getTeamByID = {
|
|
t.id:t for t in teams
|
|
}
|
|
|
|
# %%
|
|
|
|
pot = {}
|
|
teams_in_pot = {}
|
|
for i in teams.values_list('pot',flat=True).distinct():
|
|
pot[i] = teams.filter(pot=i)
|
|
teams_in_pot[i] = list(teams.filter(pot=i).values_list('id',flat=True))
|
|
|
|
|
|
|
|
# %%
|
|
|
|
|
|
requirements = GameRequirement.objects.filter(scenario=scenario)
|
|
|
|
gamereqs = []
|
|
for req in requirements:
|
|
gamereqs.append((req.team1.id,req.team2.id))
|
|
|
|
# directed version
|
|
# requirement = {
|
|
# 1:[1,-1,2,-2,3,-3,4,-4],
|
|
# 2:[1,-1,2,-2,3,-3,4,-4],
|
|
# 3:[1,-1,2,-2,3,-3,4,-4],
|
|
# 4:[1,-1,2,-2,3,-3,4,-4],
|
|
# }
|
|
# undirected version
|
|
requirement = {
|
|
1:{
|
|
1:2,
|
|
2:2,
|
|
3:2,
|
|
4:2,
|
|
},
|
|
2:{
|
|
1:2,
|
|
2:2,
|
|
3:2,
|
|
4:2,
|
|
},
|
|
3:{
|
|
1:2,
|
|
2:2,
|
|
3:2,
|
|
4:2,
|
|
},
|
|
4:{
|
|
1:2,
|
|
2:2,
|
|
3:2,
|
|
4:2,
|
|
},
|
|
}
|
|
|
|
|
|
# %%
|
|
# model = LpProblem("Draws", LpMinimize)
|
|
|
|
# x = {(t1, t2): LpVariable('x_'+str(t1)+'_'+str(t2), lowBound=0, upBound=1, cat=LpInteger) for t1 in teams for t2 in teams if t1 != t2 and t1.country != t2.country}
|
|
|
|
# # REQUIREMENTS
|
|
# for t in teams:
|
|
# for r,val in requirement[t.pot].items():
|
|
# # if r > 0:
|
|
# # model += lpSum(x[t,t2] for t2 in pot[r] if (t,t2) in x.keys()) == 1, f'home_{t.id}_pot_{r}'
|
|
# # else:
|
|
# # model += lpSum(x[t2,t] for t2 in pot[-r] if (t,t2) in x.keys()) == 1, f'away_{t.id}_pot_{r}'
|
|
# model += lpSum(x[t,t2] + x[t2,t] for t2 in pot[r] if (t,t2) in x.keys()) == val, f'undirected_{t.id}_pot_{r}'
|
|
|
|
# for t1,t2 in x.keys():
|
|
# model += x[t1,t2]+x[t2,t1] <= 1, f'play_once_{t1.id}_{t2.id}'
|
|
|
|
# model += lpSum(random.uniform(0,1)*x[key] for key in x.keys())
|
|
# model.solve()
|
|
|
|
# # %%
|
|
# gamereqs = []
|
|
# for key in x.keys():
|
|
# if x[key].value() > 0 :
|
|
# gamereqs.append((key[0].id,key[1].id))
|
|
|
|
|
|
# %%
|
|
|
|
|
|
# START COMPUTING SCHEDULE
|
|
|
|
nRounds = 8
|
|
rounds = range(1,nRounds+1)
|
|
|
|
|
|
|
|
|
|
# %%
|
|
|
|
# MODEL WITH PATTERNS
|
|
|
|
# pos_patterns = [
|
|
# [1,0,1,1,0,0,1,0 ],
|
|
# [1,0,1,0,1,0,1,0 ],
|
|
# [1,0,1,0,1,0,0,1 ],
|
|
# [1,0,1,0,0,1,1,0 ],
|
|
# [1,0,1,0,0,1,0,1 ],
|
|
# [1,0,0,1,1,0,1,0 ],
|
|
# [1,0,0,1,1,0,0,1 ],
|
|
# [1,0,0,1,0,1,1,0 ],
|
|
# [1,0,0,1,0,1,0,1 ],
|
|
# ]
|
|
|
|
# # add reverse patterns (all those starting with 0,1 )
|
|
# pos_patterns+=[ [ 1-p[i] for i in range(8)] for p in pos_patterns ]
|
|
|
|
|
|
# patterns = {
|
|
# i:p for i,p in enumerate(pos_patterns)
|
|
# }
|
|
# model = xp.problem(name='Patterns', sense=xp.minimize)
|
|
|
|
# x = {}
|
|
# assignPattern = {}
|
|
# for r in rounds:
|
|
# for game in gamereqs:
|
|
# x[r, game[0], game[1]] = xp.var(ub=1, vartype=xp.integer)
|
|
# x[r, game[1], game[0]] = xp.var(ub=1, vartype=xp.integer)
|
|
# for t in team_ids:
|
|
# for p in patterns:
|
|
# assignPattern[t,p] = xp.var(ub=1, vartype=xp.integer)
|
|
|
|
|
|
# model.addVariable(x)
|
|
# model.addVariable(assignPattern)
|
|
|
|
# for (t1,t2) in gamereqs:
|
|
# model.addConstraint(xp.Sum([x[r,t1,t2]+x[r,t2,t1] for r in rounds]) == 1)
|
|
|
|
# for t in team_ids:
|
|
# for r in rounds:
|
|
# model.addConstraint(xp.Sum([x[r,t,t2]+x[r,t2,t] for t2 in team_ids if (r,t,t2) in x.keys()]) == 1)
|
|
# model.addConstraint(xp.Sum([x[r,t,t2] for t2 in team_ids if (r,t,t2) in x.keys()]) <= xp.Sum([assignPattern[t,p] for p in patterns if patterns[p][r-1] == 1]))
|
|
# model.addConstraint(xp.Sum([assignPattern[t,p] for p in patterns]) == 1)
|
|
|
|
# for p in pot:
|
|
# model.addConstraint(xp.Sum([x[r,t,t2] for r in rounds for t2 in team_ids if (r,t,t2) in x.keys() and t2 in teams_in_pot[p]]) == 1)
|
|
|
|
|
|
# =====>>> NOT WORKING WELL
|
|
|
|
# %%
|
|
|
|
# MODEL WITH HOME
|
|
model = xp.problem(name='HomeAssignments', sense=xp.minimize)
|
|
|
|
x = {}
|
|
home = {}
|
|
for r in rounds:
|
|
for game in gamereqs:
|
|
x[r, game[0], game[1]] = xp.var(ub=1, vartype=xp.integer)
|
|
x[r, game[1], game[0]] = xp.var(ub=1, vartype=xp.integer)
|
|
for t in team_ids:
|
|
home[r,t] = xp.var(ub=1, vartype=xp.integer)
|
|
|
|
model.addVariable(x)
|
|
model.addVariable(home)
|
|
|
|
# each game has to be played
|
|
for (t1,t2) in gamereqs:
|
|
model.addConstraint(xp.Sum([x[r,t1,t2]+x[r,t2,t1] for r in rounds]) == 1)
|
|
|
|
for t in team_ids:
|
|
for r in rounds:
|
|
# each team plays once in each round
|
|
model.addConstraint(xp.Sum([x[r,t,t2]+x[r,t2,t] for t2 in team_ids if (r,t,t2) in x.keys()]) == 1)
|
|
# couple homes
|
|
model.addConstraint(home[r,t] == xp.Sum([x[r,t,t2] for t2 in team_ids if (r,t,t2) in x.keys()]))
|
|
|
|
# structural (home) requirements
|
|
for r in range(1,nRounds-1):
|
|
model.addConstraint(xp.Sum([home[r2,t] for r2 in range(r,r+3)]) <= 2)
|
|
model.addConstraint(xp.Sum([1-home[r2,t] for r2 in range(r,r+3)]) <= 2)
|
|
model.addConstraint(xp.Sum([home[r2,t] for r2 in range(1,3)]) == 1)
|
|
model.addConstraint(xp.Sum([home[r2,t] for r2 in range(nRounds-1,nRounds+1)]) == 1)
|
|
|
|
# play home against each pot
|
|
for p in pot:
|
|
model.addConstraint(xp.Sum([x[r,t,t2] for r in rounds for t2 in team_ids if (r,t,t2) in x.keys() and t2 in teams_in_pot[p]]) == 1)
|
|
|
|
start_time = time()
|
|
model.solve()
|
|
comp_time = time()-start_time
|
|
|
|
|
|
if model.getProbStatus() != 6:
|
|
print("INFEASIBLE FOUND")
|
|
|
|
# %%
|
|
|
|
|
|
if model.getProbStatus() == 6:
|
|
home_dict = {}
|
|
away_dict = {}
|
|
for key in x.keys():
|
|
if model.getSolution(x[key]) > 0:
|
|
print(key[0],getTeamByID[key[1]],getTeamByID[key[2]])
|
|
home_dict[key[0],key[1]] = key[2]
|
|
away_dict[key[0],key[2]] = key[1]
|
|
|
|
sol = " \
|
|
<style> \
|
|
table, th, td { \
|
|
border: 1px solid black; \
|
|
border-collapse: collapse; \
|
|
} \
|
|
</style> \
|
|
"
|
|
sol += "<table style='border:1px solid black'>\n"
|
|
sol += "<thead>\n"
|
|
sol += "<tr><th></th>"
|
|
for r in rounds:
|
|
sol += f"<th>{r}</th>"
|
|
sol += "</tr>"
|
|
sol += "</thead>\n"
|
|
sol += "<tbody>\n"
|
|
for t in team_ids:
|
|
tname = getTeamByID[t]
|
|
sol += f"<tr><td>{tname}</td>"
|
|
for r in rounds:
|
|
if (r,t) in home_dict.keys():
|
|
opponent = getTeamByID[home_dict[(r,t)]]
|
|
sol += f"<td style='background-color:lightyellow'>{opponent.pot} - {opponent}</td>"
|
|
elif (r,t) in away_dict.keys():
|
|
opponent = getTeamByID[away_dict[(r,t)]]
|
|
sol += f"<td style='background-color:lightsteelblue'>{opponent.pot} - {opponent}</td>"
|
|
else:
|
|
sol += "<td></td>"
|
|
sol += "</tr>"
|
|
sol += "</tbody>\n"
|
|
sol += "</table>\n"
|
|
sol += "<br><br><br>\n"
|
|
|
|
with open('draw_schedule.html', 'w') as f:
|
|
f.write(sol)
|
|
|
|
# %%
|
|
|
|
games = [(key[1],key[2],key[0]) for key in x.keys() if model.getSolution(x[key]) > 0]
|
|
# %%
|