Trajectron-plus-plus/trajectron/model/dynamics/dynamic.py

31 lines
776 B
Python
Raw Normal View History

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