diff options
Diffstat (limited to 'lib/automata.py')
| -rwxr-xr-x | lib/automata.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/lib/automata.py b/lib/automata.py index 96a724e..2d5c27c 100755 --- a/lib/automata.py +++ b/lib/automata.py @@ -258,7 +258,8 @@ class PTA: def __init__(self, state_names: list = [], accepting_states: list = None, - parameters: list = [], initial_param_values: list = None): + parameters: list = [], initial_param_values: list = None, + instance: str = None): """ Return a new PTA object. @@ -268,10 +269,12 @@ class PTA: accepting_states -- names of accepting states. By default, all states are accepting parameters -- names of PTA parameters initial_param_values -- initial value for each parameter + instance -- class used for generated C++ code """ self.state = dict([[state_name, State(state_name)] for state_name in state_names]) self.accepting_states = accepting_states.copy() if accepting_states else None self.parameters = parameters.copy() + self.instance = instance if initial_param_values: self.initial_param_values = initial_param_values.copy() else: @@ -380,6 +383,9 @@ class PTA: if 'states' in yaml_input: kwargs['state_names'] = yaml_input['states'] + if 'instance' in yaml_input: + kwargs['instance'] = yaml_input['instance'] + pta = cls(**kwargs) for trans_name in sorted(yaml_input['transition'].keys()): |