57 lines
1.3 KiB
Python
57 lines
1.3 KiB
Python
from abc import ABCMeta, abstractmethod
|
|
|
|
|
|
class AbstractProblem(metaclass=ABCMeta):
|
|
|
|
@property
|
|
def input_variables(self):
|
|
variables = []
|
|
|
|
if hasattr(self, 'spatial_variables'):
|
|
variables += self.spatial_variables
|
|
if hasattr(self, 'temporal_variable'):
|
|
variables += self.temporal_variable
|
|
if hasattr(self, 'parameters'):
|
|
variables += self.parameters
|
|
if hasattr(self, 'custom_variables'):
|
|
variables += self.custom_variables
|
|
|
|
return variables
|
|
|
|
@property
|
|
def domain(self):
|
|
|
|
domains = [
|
|
getattr(self, f'{t}_domain')
|
|
for t in ['spatial', 'temporal', 'parameter']
|
|
if hasattr(self, f'{t}_domain')
|
|
]
|
|
|
|
if len(domains) == 1:
|
|
return domains[0]
|
|
elif len(domains) == 0:
|
|
raise RuntimeError
|
|
|
|
if len(set(map(type, domains))) == 1:
|
|
domain = domains[0].__class__({})
|
|
[domain.update(d) for d in domains]
|
|
return domain
|
|
else:
|
|
raise RuntimeError('different domains')
|
|
|
|
|
|
|
|
@input_variables.setter
|
|
def input_variables(self, variables):
|
|
raise RuntimeError
|
|
|
|
@property
|
|
@abstractmethod
|
|
def output_variables(self):
|
|
pass
|
|
|
|
@property
|
|
@abstractmethod
|
|
def conditions(self):
|
|
pass
|