This is a demo of the use of ffc to generate kernels. It solves the identity
equation on a quadrilateral domain. It requires the pyop2 branch of ffc,
which can be obtained with:
bzr branch lp:~mapdes/ffc/pyop2
This may also depend on development trunk versions of other FEniCS programs.
"""
from pyop2 import op2, utils
from pyop2.ffc_interface import compile_form
from ufl import *
import ffc
import numpy as np
parser = utils.parser(group=True, description=__doc__)
parser.add_argument('-s', '--save-output',
action='store_true',
help='Save the output of the run (used for testing)')
opt = vars(parser.parse_args())
op2.init(**opt)
# Set up finite element identity problem
E = FiniteElement("Lagrange", "triangle", 1)
v = TestFunction(E)
u = TrialFunction(E)
f = Coefficient(E)
a = v*u*dx
p_bound(op2.READ))
# Update flow field
rms = op2.Global(1, 0.0, np.double, "rms")
op2.par_loop(update, cells,
p_qold(op2.READ),
p_q(op2.WRITE),
p_res(op2.RW),
p_adt(op2.READ),
rms(op2.INC))
# Print iteration history
rms = sqrt(rms.data / cells.size)
if i % 100 == 0:
print " %d %10.5e " % (i, rms)
if __name__ == '__main__':
parser = utils.parser(group=True, description="PyOP2 airfoil demo")
parser.add_argument('-m', '--mesh', default='meshes/new_grid.h5',
help='HDF5 mesh file to use (default: meshes/new_grid.h5)')
parser.add_argument('-p', '--profile', action='store_true',
help='Create a cProfile for the run')
opt = vars(parser.parse_args())
op2.init(**opt)
if opt['profile']:
import cProfile
filename = 'airfoil.%s.cprofile' % os.path.split(opt['mesh'])[-1]
cProfile.run('main(opt)', filename=filename)
else:
main(opt)
toc('diffusion')
# Perform 1 iteration to warm up plan cache then reset initial condition
timestep_iteration()
op2.par_loop(i_cond, nodes, coords(op2.IdentityMap, op2.READ),
tracer(op2.IdentityMap, op2.WRITE))
reset()
# Timed iteration
t1 = clock()
while current_time < endtime:
timestep_iteration()
current_time += dt
runtime = clock() - t1
print "/fluidity :: %f" % runtime
summary('profile_pyop2_%s_%s.csv' %
(opt['mesh'].split('/')[-1], opt['backend']))
if __name__ == '__main__':
from parameters import *
parser = utils.parser(group=True,
description="PyOP2 P1 advection-diffusion demo")
parser.add_argument(
'-m',
'--mesh',
help=
'Base name of triangle mesh (excluding the .ele or .node extension)')
opt = vars(parser.parse_args())
run(diffusivity, current_time, dt, endtime, **opt)
# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
"""PyOP2 Jacobi demo
Port of the Jacobi demo from OP2-Common.
"""
from pyop2 import op2, utils
import numpy as np
from math import sqrt
parser = utils.parser(group=True, description=__doc__)
parser.add_argument('-s',
'--single',
action='store_true',
help='single precision floating point mode')
parser.add_argument('-n',
'--niter',
action='store',
default=2,
type=int,
help='set the number of iteration')
opt = vars(parser.parse_args())
op2.init(**opt)
fp_type = np.float32 if opt['single'] else np.float64
p_bound(op2.IdentityMap, op2.READ))
# Update flow field
rms = op2.Global(1, 0.0, np.double, "rms")
op2.par_loop(update, cells,
p_qold(op2.IdentityMap, op2.READ),
p_q (op2.IdentityMap, op2.WRITE),
p_res (op2.IdentityMap, op2.RW),
p_adt (op2.IdentityMap, op2.READ),
rms(op2.INC))
# Print iteration history
rms = sqrt(rms.data/cells.size)
if i%100 == 0:
print " %d %10.5e " % (i, rms)
if __name__ == '__main__':
parser = utils.parser(group=True, description="PyOP2 airfoil demo (vector map version)")
parser.add_argument('-m', '--mesh', default='meshes/new_grid.h5',
help='HDF5 mesh file to use (default: meshes/new_grid.h5)')
parser.add_argument('-p', '--profile', action='store_true',
help='Create a cProfile for the run')
opt = vars(parser.parse_args())
op2.init(**opt)
if opt['profile']:
import cProfile
filename = 'adv_diff.%s.cprofile' % os.path.split(opt['mesh'])[-1]
cProfile.run('main(opt)', filename=filename)
else:
main(opt)
"""
This demo verifies that the integral of a unit cube is 1.
The cube will be unstructured in the 2D plane and structured vertically.
"""
from pyop2 import op2, utils
from triangle_reader import read_triangle
from ufl import *
from pyop2.computeind import compute_ind_extr
import numpy as np
import time
parser = utils.parser(group=True, description="PyOP2 2D mass equation demo")
parser.add_argument('-m', '--mesh', action='store', type=str, required=True,
help='Base name of triangle mesh \
(excluding the .ele or .node extension)')
parser.add_argument('-ll', '--layers', action='store', type=str, required=True,
help='Number of extruded layers.')
parser.add_argument('-p', '--partsize', action='store', type=str,
required=False, default=1024,
help='Partition size in the base mesh.')
opt = vars(parser.parse_args())
op2.init(**opt)
mesh_name = opt['mesh']
layers = int(opt['layers'])
partition_size = int(opt['partsize'])
# Generate code for kernel
# OF THE POSSIBILITY OF SUCH DAMAGE.
"""
This demo verifies that the integral of a unit cube is 1.
The cube will be unstructured in the 2D plane and structured vertically.
"""
from pyop2 import op2, utils
from triangle_reader import read_triangle
from ufl import *
from pyop2.computeind import compute_ind_extr
import numpy as np
import time
parser = utils.parser(group=True, description="PyOP2 2D mass equation demo")
parser.add_argument('-m',
'--mesh',
action='store',
type=str,
required=True,
help='Base name of triangle mesh \
(excluding the .ele or .node extension)')
parser.add_argument('-ll',
'--layers',
action='store',
type=str,
required=True,
help='Number of extruded layers.')
parser.add_argument('-p',
'--partsize',
# Update flow field
rms = op2.Global(1, 0.0, np.double, "rms")
op2.par_loop(update, cells,
p_qold(op2.READ),
p_q(op2.WRITE),
p_res(op2.RW),
p_adt(op2.READ),
rms(op2.INC))
# Print iteration history
rms = sqrt(rms.data / cells.size)
if i % 100 == 0:
print " %d %10.5e " % (i, rms)
if __name__ == '__main__':
parser = utils.parser(group=True,
description="PyOP2 airfoil demo (vector map version)")
parser.add_argument('-m', '--mesh', default='meshes/new_grid.h5',
help='HDF5 mesh file to use (default: meshes/new_grid.h5)')
parser.add_argument('-p', '--profile', action='store_true',
help='Create a cProfile for the run')
opt = vars(parser.parse_args())
op2.init(**opt)
if opt['profile']:
import cProfile
filename = 'adv_diff.%s.cprofile' % os.path.split(opt['mesh'])[-1]
cProfile.run('main(opt)', filename=filename)
else:
main(opt)
p_bound(op2.READ))
# Update flow field
rms = op2.Global(1, 0.0, np.double, "rms")
op2.par_loop(update, cells,
p_qold(op2.READ),
p_q(op2.WRITE),
p_res(op2.RW),
p_adt(op2.READ),
rms(op2.INC))
# Print iteration history
rms = sqrt(rms.data / cells.size)
if i % 100 == 0:
print " %d %10.5e " % (i, rms)
if __name__ == '__main__':
parser = utils.parser(group=True, description="PyOP2 airfoil demo")
parser.add_argument('-m', '--mesh', default='meshes/new_grid.h5',
help='HDF5 mesh file to use (default: meshes/new_grid.h5)')
parser.add_argument('-p', '--profile', action='store_true',
help='Create a cProfile for the run')
opt = vars(parser.parse_args())
op2.init(**opt)
if opt['profile']:
import cProfile
filename = 'airfoil.%s.cprofile' % os.path.split(opt['mesh'])[-1]
cProfile.run('main(opt)', filename=filename)
else:
main(opt)
)
toc("assembly")
tic("solve")
op2.solve(mat, tracer, b)
toc("solve")
toc("diffusion")
# Perform 1 iteration to warm up plan cache then reset initial condition
timestep_iteration()
op2.par_loop(i_cond, nodes, coords(op2.IdentityMap, op2.READ), tracer(op2.IdentityMap, op2.WRITE))
reset()
# Timed iteration
t1 = clock()
while current_time < endtime:
timestep_iteration()
current_time += dt
runtime = clock() - t1
print "/fluidity :: %f" % runtime
summary("profile_pyop2_%s_%s.csv" % (opt["mesh"].split("/")[-1], opt["backend"]))
if __name__ == "__main__":
from parameters import *
parser = utils.parser(group=True, description="PyOP2 P1 advection-diffusion demo")
parser.add_argument("-m", "--mesh", help="Base name of triangle mesh (excluding the .ele or .node extension)")
opt = vars(parser.parse_args())
run(diffusivity, current_time, dt, endtime, **opt)
