Trajectron-plus-plus/trajectron/model/dynamics/dynamic.py
2020-04-05 21:43:49 -04:00

30 lines
776 B
Python

class Dynamic(object):
def __init__(self, dt, dyn_limits, device, model_registrar, xz_size, node_type):
self.dt = dt
self.device = device
self.dyn_limits = dyn_limits
self.initial_conditions = None
self.model_registrar = model_registrar
self.node_type = node_type
self.init_constants()
self.create_graph(xz_size)
def set_initial_condition(self, init_con):
self.initial_conditions = init_con
def init_constants(self):
pass
def create_graph(self, xz_size):
pass
def integrate_samples(self, s, x):
raise NotImplementedError
def integrate_distribution(self, dist, x):
raise NotImplementedError
def create_graph(self, xz_size):
pass