105 lines
2.7 KiB
Python
105 lines
2.7 KiB
Python
import os
|
|
import numpy as np
|
|
import numpy.random as rnd
|
|
import sys
|
|
import importlib
|
|
# import imp
|
|
import ctypes
|
|
import ctypes.util
|
|
from numpy.ctypeslib import ndpointer
|
|
import event_gen as eg
|
|
|
|
|
|
PPP = 60
|
|
claim_types = {'bronze': {'bw': 1, 'value': 2, 'damage': 4, 'duration': 100, 'pr': 0.4},
|
|
'silver': {'bw': 3, 'value': 5, 'damage': 6, 'duration': 50, 'pr': 0.3},
|
|
'gold': {'bw': 10, 'value': 30, 'damage': 80, 'duration': 70, 'pr': 0.1}}
|
|
|
|
|
|
class Claim:
|
|
def __init__(self, ty, time):
|
|
self.ty = ty
|
|
self.bw = claim_types[ty]['bw']
|
|
self.val = claim_types[ty]['value']
|
|
self.dam = claim_types[ty]['damage']
|
|
self.dur = claim_types[ty]['duration']
|
|
self.init_time = time
|
|
self.end_time = time + self.dur
|
|
|
|
|
|
class Sim:
|
|
def __init__(self, bw):
|
|
self.queue = []
|
|
self.bw = bw
|
|
self.free = bw
|
|
self.budget = (-1) * PPP * self.bw
|
|
|
|
def step(self, lista):
|
|
self.budget -= sum([self.queue[i].dam for i in range(len(self.queue)) if lista[i] == 0])
|
|
self.queue = [self.queue[i] for i in range(len(self.queue)) if lista[i] == 1]
|
|
|
|
def new(self, item):
|
|
if item.bw <= self.free:
|
|
self.queue.append(item)
|
|
|
|
def process(self, time):
|
|
for i in range(len(self.queue) - 1, -1, -1):
|
|
if self.queue[i].end_time <= time:
|
|
self.budget += self.queue[i].val
|
|
del self.queue[i]
|
|
self.free = self.bw
|
|
for item in self.queue:
|
|
self.free -= item.bw
|
|
|
|
def get_queue_list(self, time):
|
|
lista = []
|
|
for item in self.queue:
|
|
lista.append([item.ty, item.end_time - time])
|
|
return lista
|
|
|
|
|
|
results = {}
|
|
folder = 'Strats'
|
|
|
|
|
|
sys.path.append(folder)
|
|
strats = os.listdir(folder)
|
|
strats_py = [x.split('.')[0] for x in strats if x.split('.')[-1] == "py"]
|
|
# strats_py=['tut_mymy']
|
|
|
|
|
|
for _ in range(10):
|
|
events = eg.event_gen(2000)
|
|
"""
|
|
os.system('python event_gen.py >event1.txt')
|
|
with open('event1.txt') as inf:
|
|
for line in inf:
|
|
a=line.split()
|
|
# print a
|
|
if a:
|
|
events.append([int(a[0]),a[1]])
|
|
|
|
#print events
|
|
"""
|
|
for strat in strats_py:
|
|
mod = importlib.import_module(strat)
|
|
sim = Sim(max(0, mod.init_bw()))
|
|
for e in events:
|
|
[time, claim] = e
|
|
sim.process(time)
|
|
[new, lista] = mod.decision(claim, sim.get_queue_list(time))
|
|
sim.step(lista)
|
|
if new == 1:
|
|
sim.new(Claim(claim, time))
|
|
sim.process(time + 1000)
|
|
if not strat in results:
|
|
results[strat] = 0
|
|
results[strat] += sim.budget
|
|
|
|
|
|
for key in results:
|
|
results[key] /= 10.0
|
|
|
|
|
|
print(results)
|