import assimulo.explicit_ode as ode
import assimulo.problem as apr
import assimulo.solvers as aso

class FEM_ODE(ode.Explicit_ODE):

    def __init__(self, p,t,v=None):
        self.p=p
        self.v=v
        self.t=t
        def rhs(x,t):
            if(v==None):
                return p*x
            else:
                return p*x[0:len[x]/2]+v*x[len[x]/2,len[x]]

        ode.Explicit_ODE.__init__(apr.Explicit_Problem(rhs))