96 lines
2.6 KiB
Python
96 lines
2.6 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)
|