import pandas as pd

from app.entity.strategy.MarketInfo import *
from app.entity.product.ChartDataBean import *

class SimulationCalculatorDataBean:
    def __init__(self, code='SimulationCalculatorDataBean', entityType='SimulationCalculatorDataBean',
                 scenario=None, strategyNames=[],   
                 axisMin=-1., axisMax=-1., 
                 marketInfoDict={}, scenarioResultsDict={}, chartDataBeanDict = {}):
        self.code = code
        self.entityType = 'StrategyCalculatorDataBean'
        
        self.strategyNames = strategyNames
        
        self.scenario = scenario
        self.marketInfoDict = marketInfoDict
        self.scenarioResultsDict = scenarioResultsDict
        self.axisMin = axisMin
        self.axisMax = axisMax
        self.chartDataBeanDict = chartDataBeanDict
        
    def serialize(self):
        return {
                "code": self.code,
                "entityType": self.entityType,
                "scenario": self.scenario,
                "strategyNames": self.strategyNames,
                "axisMin": self.axisMin,
                "axisMax": self.axisMax,
                "scenarioResultsDict": self.scenarioResultsDict,
                "chartDataBeanDict": self.chartDataBeanDict,
                }