class World:
__metaclass__ = ABCMeta
def __init__(self, config):
self.firms = []
self.workers = []
self.stats = Stats()
self.money = config['global']['initial_money']
self.steps = config['global']['steps']
self.firm_algorithms = config['algorithms']
workers_count = config['global']['workers_count']
self.config = config
firm_counter = 0
for class_, count in self.firm_algorithms.items():
for i in range(int(count)):
firm_class = getattr(algorithms, class_)
firm = firm_class(firm_counter)
self.firms.append(firm)
firm_counter += 1
firm_count = firm_counter
# initial worker distribution
for i in range(workers_count):
worker = Worker(i)
self.workers.append(worker)
firm_id = i % firm_count
firm = self.firms[firm_id]
firm.add_worker(worker, firm.current_salary)
self.history = History(self.steps, self.firms)
self.firm_actions = [0] * firm_count
self.firm_results = [0] * firm_count
self.firm_labormarket_actions = [0] * firm_count
self.firm_goodmarket_actions = [0] * firm_count
self.firm_labormarket_results = [0] * firm_count
self.firm_goodmarket_results = [0] * firm_count
def manage_firm_actions(self, firm_actions):
pass
def manage_sales(self):
pass
def manage_job_offers(self):
pass
def compute_stats(self):
self.stats.f = len(self.firms)
self.stats.price = 0
self.stats.stock = 0
self.stats.sales = 0
self.stats.sold = 0
self.stats.salary = 0
employed = 0
for firm in self.firms:
self.stats.stock += firm.stock
self.stats.sold += firm.sold
self.stats.sales += firm.sales
employed += len(firm.workers)
for worker in firm.workers:
self.stats.salary += worker.salary
if self.stats.sold > 0:
self.stats.price = self.stats.sales / self.stats.sold
else:
self.stats.price = 0
if employed > 0:
self.stats.salary /= employed
else:
self.stats.salary = 0
unemployed = 0
for worker in self.workers:
if worker.employer is None:
unemployed += 1
if len(self.workers) > 0:
self.stats.unemployment_rate = unemployed / len(self.workers)
self.stats.money = self.money
self.stats.expected_sales_growth = self.config['global']['money_growth']/self.money
self.stats.expected_sold_growth = 0.05
def go(self):
print("It's alive!!")
birth_rate = self.config['global']['birth_rate']
money_growth = self.config['global']['money_growth']
for step in range(self.steps):
# print("Step:", step)
for i, firm in enumerate(self.firms):
# @todo: enable bankrupt
# if firm.money < self.config['global']['bankrupt_rate']:
# firm.bankrupt()
# del self.firms[i]
# del self.firm_actions[i]
# continue
# print(firm)
firm.work()
# print(firm)
self.firm_goodmarket_actions[firm.id] = firm.decide_price(self.stats)
self.manage_sales()
for firm_id, firm_action in enumerate(self.firm_goodmarket_actions):
firm = self.firms[firm_id]
firm.apply_goodmarket_result(self.firm_goodmarket_results[firm_id])
self.history.add_record(step, firm)
self.compute_stats()
self.history.add_stats(step, self.stats) # needs to be rewritten with proper history object in mind
for j in range(birth_rate):
worker = Worker(len(self.workers))
self.workers.append(worker)
for i, firm in enumerate(self.firms):
self.firm_labormarket_actions[firm.id] = firm.decide_salary(self.stats)
self.manage_job_offers()
for firm_id, firm_action in enumerate(self.firm_labormarket_actions):
firm = self.firms[firm_id]
firm.apply_labormarket_result(self.firm_labormarket_results[firm_id])
self.money += money_growth
return self.history
class World:
__metaclass__ = ABCMeta
def __init__(self, config):
self.firms = []
self.workers = []
self.stats = Stats()
self.money = config['global']['initial_money']
self.steps = config['global']['steps']
self.firm_algorithms = config['algorithms']
workers_count = config['global']['workers_count']
self.config = config
firm_counter = 0
for class_, count in self.firm_algorithms.items():
for i in range(int(count)):
firm_class = getattr(algorithms, class_)
firm = firm_class(firm_counter)
self.firms.append(firm)
firm_counter += 1
firm_count = firm_counter
# initial worker distribution
for i in range(workers_count):
worker = Worker(i)
self.workers.append(worker)
firm_id = i % firm_count
firm = self.firms[firm_id]
firm.add_worker(worker, firm.current_salary)
self.history = History(self.steps, self.firms)
self.firm_actions = [0] * firm_count
self.firm_results = [0] * firm_count
self.firm_labormarket_actions = [0] * firm_count
self.firm_goodmarket_actions = [0] * firm_count
self.firm_labormarket_results = [0] * firm_count
self.firm_goodmarket_results = [0] * firm_count
def manage_firm_actions(self, firm_actions):
pass
def manage_sales(self):
pass
def manage_job_offers(self):
pass
def compute_stats(self):
self.stats.f = len(self.firms)
self.stats.price = 0
self.stats.stock = 0
self.stats.sales = 0
self.stats.sold = 0
self.stats.salary = 0
employed = 0
for firm in self.firms:
self.stats.stock += firm.stock
self.stats.sold += firm.sold
self.stats.sales += firm.sales
employed += len(firm.workers)
for worker in firm.workers:
self.stats.salary += worker.salary
if self.stats.sold > 0:
self.stats.price = self.stats.sales / self.stats.sold
else:
self.stats.price = 0
if employed > 0:
self.stats.salary /= employed
else:
self.stats.salary = 0
unemployed = 0
for worker in self.workers:
if worker.employer is None:
unemployed += 1
if len(self.workers) > 0:
self.stats.unemployment_rate = unemployed / len(self.workers)
self.stats.money = self.money
self.stats.expected_sales_growth = self.config['global'][
'money_growth'] / self.money
self.stats.expected_sold_growth = 0.05
def go(self):
print("It's alive!!")
birth_rate = self.config['global']['birth_rate']
money_growth = self.config['global']['money_growth']
for step in range(self.steps):
# print("Step:", step)
for i, firm in enumerate(self.firms):
# @todo: enable bankrupt
# if firm.money < self.config['global']['bankrupt_rate']:
# firm.bankrupt()
# del self.firms[i]
# del self.firm_actions[i]
# continue
# print(firm)
firm.work()
# print(firm)
self.firm_goodmarket_actions[firm.id] = firm.decide_price(
self.stats)
self.manage_sales()
for firm_id, firm_action in enumerate(
self.firm_goodmarket_actions):
firm = self.firms[firm_id]
firm.apply_goodmarket_result(
self.firm_goodmarket_results[firm_id])
self.history.add_record(step, firm)
self.compute_stats()
self.history.add_stats(
step, self.stats
) # needs to be rewritten with proper history object in mind
for j in range(birth_rate):
worker = Worker(len(self.workers))
self.workers.append(worker)
for i, firm in enumerate(self.firms):
self.firm_labormarket_actions[firm.id] = firm.decide_salary(
self.stats)
self.manage_job_offers()
for firm_id, firm_action in enumerate(
self.firm_labormarket_actions):
firm = self.firms[firm_id]
firm.apply_labormarket_result(
self.firm_labormarket_results[firm_id])
self.money += money_growth
return self.history
