## Copyright (C) 2013 Marco Vassallo <gedeone-octave@users.sourceforge.net>
##
## This program is free software; you can redistribute it and/or modify it under
## the terms of the GNU General Public License as published by the Free Software
## Foundation; either version 3 of the License, or (at your option) any later
## version.
##
## This program is distributed in the hope that it will be useful, but WITHOUT
## ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
## FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
## details.
##
## You should have received a copy of the GNU General Public License along with
## this program; if not, see <http://www.gnu.org/licenses/>.


V = VectorElement("CG", triangle, 2)
Q = FiniteElement("CG", triangle, 1)
P0 = FiniteElement("DG", triangle, 0)
# Define trial and test functions
u = TrialFunction(V)
v = TestFunction(V)

# Define coefficients
k  = Constant(triangle)
u0 = Coefficient(V)
p0 = Coefficient(Q)
f  = Coefficient(V)
sig = Coefficient(P0)
nu = Coefficient(P0)

# Define bilinear and linear forms
eq = (1/k)*inner(u - u0, v)*dx + inner(grad(u)*u0, v)*dx + \
    nu*inner(grad(u), grad(v))*dx + sig * (inner (u, v))*dx \
    - inner(f, v)*dx - div(v)*p0*dx
a  = lhs(eq)
L  = rhs(eq)