import numpy as np
import json

from InventorySystem import InventorySystem
from OrderSystem import OrderSystem
from ProduceSystem import ProduceSystem


# Create Company Instance
class Firm:
    the_os: OrderSystem
    the_is: InventorySystem
    the_ps: ProduceSystem

    def __init__(self, env, dct_all_para):
        self.env = env

        # get the parameters here
        self.xv_ary_initial_product_num = np.asarray(dct_all_para['xv_ary_initial_product_num'])  # 初始产成品库存
        self.xv_ary_initial_material_num = np.asarray(dct_all_para['xv_ary_initial_material_num'])  # 初始原材料库存
        self.xv_product_num = int(dct_all_para['xv_product_num'])  # 产成品个数
        self.xv_material_num = int(dct_all_para['xv_material_num'])  # 原材料个数
        self.xv_ary_plan = np.asarray(dct_all_para['xy_ary_plan'])
        self.xv_ary_bom = np.asarray(dct_all_para['xv_ary_bom'])
        self.xv_ary_product_id = np.asarray(dct_all_para['xv_ary_product_id'])
        self.xv_ary_material_id = np.asarray(dct_all_para['xv_ary_material_id'])
        self.xv_ary_s = np.asarray(dct_all_para['xv_ary_s'])
        self.xv_ary_S = np.asarray(dct_all_para['xv_ary_S'])
        # self.dct_status_info = json.loads(dct_all_para['dct_status_info'])  # 生产状态表
        # self.xv_ary_lead_time = np.array([dct_all_para['xv_int_lead_time_a'], dct_all_para['xv_int_lead_time_b']])

        # create agents here
        self.the_os = OrderSystem(env, xv_produt_num=self.xv_product_num, xv_ary_plan=self.xv_ary_plan,
                                  xv_ary_product_id=self.xv_ary_product_id)  # 实例Oss
        self.the_is = InventorySystem(env, xv_ary_initial_material_num=self.xv_ary_initial_material_num,
                                      xv_ary_initial_product_num=self.xv_ary_initial_product_num,
                                      xv_product_num=self.xv_product_num,
                                      xv_material_num=self.xv_material_num,
                                      xv_ary_bom=self.xv_ary_bom, xv_ary_pan=self.xv_ary_plan,
                                      xv_ary_material_id=self.xv_ary_material_id)  # 实例Iss
        self.the_ps = ProduceSystem(env, xv_ary_bom=self.xv_ary_bom, xv_ary_plan=self.xv_ary_plan,
                                    xv_product_num=self.xv_product_num,
                                    xv_ary_product_id=self.xv_ary_product_id,
                                    xv_ary_material_id=self.xv_ary_material_id)  # 实例Pss
        # self.the_fs = FinancialSystem(env, xv_flt_initial_cash=self.xv_flt_initial_cash)

    def operating(self):  # 按步骤进行
        self.the_os.rank_order()
        self.the_os.produce_status()
        self.the_is.material_state_to_use()
        self.the_is.material_check()
        self.the_ps.change_status()
        self.the_ps.run_produce(ev_lst_trans_quan_material=self.the_is.ev_lst_trans_quan_material,
                                ev_ary_product_to_produce=self.the_os.ev_ary_product_to_produce)
        self.the_is.consume_and_store(ev_ary_produced_num=self.the_ps.ev_ary_produce_number,
                                      ev_changed_product=self.the_os.ev_changed_product,
                                      ev_lst_backtrans_material=self.the_ps.ev_lst_backtrans_material)
        self.the_is.material_replenishment()
        self.the_os.do_shipment(ev_ary_current_product=self.the_is.ev_ary_current_product)
        # self.the_is.inventory_cost()
        # self.the_fs.financial_calculating()