escript/py_src/projector.py

# $Id$

"""
@brief A simple projector for rank 0 and rank 1 Data to a continuous
function space.
"""

from esys.escript import *
from esys.linearPDEs import LinearPDE
import numarray

class Projector:

  def __init__(self, domain, rank = 0, reduce = False):
    """
    @brief Create a continuous function space projector for a domain.

    @param domain Domain of the projection.
    @param rank   Rank of data to project.
    @param reduce Flag to reduce projection order.
    """
    dim = rank * domain.getDim() + (1 - rank)
    self.pde = LinearPDE(domain, numEquations = dim, numSolutions = dim)
    self.pde.setLumping(True)
    self.pde.setReducedOrderTo(reduce)
    if rank == 0:
      D = 1
    elif rank == 1:
      D = numarray.identity(domain.getDim())
    else:
      raise Exception("Projection restricted to rank 0 or rank 1 Data.")
    self.pde._setValue(D = D)
    return

  def __del__(self):
    return

  def __call__(self, value):
    """
    @brief Execute a projection to a continous function space on value.

    @param value  The value to be projected.
    """
    self.pde._setValue(Y = value)
    return self.pde.getSolution()
1	# $Id$
2
3	"""
4	@brief A simple projector for rank 0 and rank 1 Data to a continuous
5	function space.
6	"""
7
8	from esys.escript import *
9	from esys.linearPDEs import LinearPDE
10	import numarray
11
12	class Projector:
13
14	def __init__(self, domain, rank = 0, reduce = False):
15	"""
16	@brief Create a continuous function space projector for a domain.
17
18	@param domain Domain of the projection.
19	@param rank Rank of data to project.
20	@param reduce Flag to reduce projection order.
21	"""
22	dim = rank * domain.getDim() + (1 - rank)
23	self.pde = LinearPDE(domain, numEquations = dim, numSolutions = dim)
24	self.pde.setLumping(True)
25	self.pde.setReducedOrderTo(reduce)
26	if rank == 0:
27	D = 1
28	elif rank == 1:
29	D = numarray.identity(domain.getDim())
30	else:
31	raise Exception("Projection restricted to rank 0 or rank 1 Data.")
32	self.pde._setValue(D = D)
33	return
34
35	def __del__(self):
36	return
37
38	def __call__(self, value):
39	"""
40	@brief Execute a projection to a continous function space on value.
41
42	@param value The value to be projected.
43	"""
44	self.pde._setValue(Y = value)
45	return self.pde.getSolution()
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision