import types

from ScEpTIC.config.base_config import ScEpTICBaseConfig

class VirtualMemoryConfig(ScEpTICBaseConfig):
    """
    Configuration for the Virtual Memory Transformation
    """
    _config_domain = {
        'n_min_function': {'class': types.FunctionType, 'mode': 'set'},
        'optimize_saved_registers': {'class': bool, 'mode': 'set'},
        'optimize_memory': {'class': bool, 'mode': 'set'}
    }

    def __init__(self, main_config):
        super().__init__(main_config)
        self.n_min_function = None
        self.optimize_saved_registers = True
        self.optimize_memory = False


# Returns the configuration
def get_config(main_config):
    return VirtualMemoryConfig(main_config)