mesa/computation.py

41 lines
1.3 KiB
Python
Raw Normal View History

2024-08-24 11:20:13 +08:00
import os
import datetime
2024-08-24 16:13:37 +08:00
from mesa import Model
2024-08-24 11:20:13 +08:00
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from controller_db import ControllerDB
class Computation:
def __init__(self, c_db: 'ControllerDB'):
# 控制不同进程 计算不同的样本 但使用同一个 数据库 c_db
self.c_db = c_db
self.pid = os.getpid()
def run(self, str_code='0', s_id=None):
sample_random = self.c_db.fetch_a_sample(s_id)
if sample_random is None:
return True
# lock this row by update is_done_flag to 0 将运行后的样本设置为 flag 0
self.c_db.lock_the_sample(sample_random)
print(
f"Pid {self.pid} ({str_code}) is running "
f"sample {sample_random.id} at {datetime.datetime.now()}")
# 将sample 对应的 experiment 的一系列值 和 参数值 传入 模型 中 包括列名 和 值
dct_exp = {column: getattr(sample_random.experiment, column)
for column in sample_random.experiment.__table__.c.keys()}
# 删除不需要的 主键
del dct_exp['id']
dct_sample_para = {'sample': sample_random,
'seed': sample_random.seed,
**dct_exp}
model = Model(dct_sample_para)
2024-08-24 16:13:37 +08:00
for i in range(100):
model.step()
2024-08-24 11:20:13 +08:00
return False