modellib/py_src/input.py

# $Id$


from esys.modelframe import Model
from esys.escript import *
from math import log


class GausseanProfile(Model):
    """@brief generates a gaussean profile at center x_c, width width and height A over a domain

           @param domain (in) - domain
           @param x_c (in)  - center of the Gaussean profile (default [0.,0.,0.])
           @param A (in)  - height of the profile. A maybe a vector. (default 1.)
           @param width (in) - width of the profile (default 0.1)
           @param r (in) -  radius of the circle (default = 0)
           @param out (out) - profile


       In the case that the spatial dimension is two, The third component of x_c is dropped
    """
    def __init__(self,debug=False):
        Model.__init__(self,debug=debug)
        self.declareParameter(domain=None, x_c=numarray.zeros([3]),A=1.,width=0.1,r=0)

    def out(self):
        x=self.domain.getX()
        dim=self.domain.getDim()
        l=length(x-self.x_c[:dim])
        m=(l-self.r).whereNegative()
        return (m+(1.-m)*exp(-log(2.)*(l/self.width)**2))*self.A

class InterpolatedTimeProfile(Model):
       """ """

       def __init__(self,debug=False):
           Model.__init__(self,debug=debug)
           self.declareParameter(t=[0.,1.],\
                                 values=[1.,1.],\
                                 out=0.)
       def doInitialization(self,t):
           self.__tn=t

       def doStep(self,dt):
            t=self.__tn+dt
            if t<=self.t[0]:
                self.out=self.values[0]
            else:
               for i in range(1,len(self.t)):
                  if t<self.t[i]:
                    m=(self.values[i-1]-self.values[i])/(self.t[i-1]-self.t[i])
                    self.out=m*(t-self.t[i-1])+self.values[i-1]
               self.out=self.t[len(self.t)-1]
            self.__tn+=dt
1	jgs	127	# $Id$
2
3
4			from esys.modelframe import Model
5			from esys.escript import *
6			from math import log
7
8
9			class GausseanProfile(Model):
10			"""@brief generates a gaussean profile at center x_c, width width and height A over a domain
11
12			@param domain (in) - domain
13			@param x_c (in) - center of the Gaussean profile (default [0.,0.,0.])
14			@param A (in) - height of the profile. A maybe a vector. (default 1.)
15			@param width (in) - width of the profile (default 0.1)
16			@param r (in) - radius of the circle (default = 0)
17			@param out (out) - profile
18
19
20			In the case that the spatial dimension is two, The third component of x_c is dropped
21			"""
22			def __init__(self,debug=False):
23			Model.__init__(self,debug=debug)
24			self.declareParameter(domain=None, x_c=numarray.zeros([3]),A=1.,width=0.1,r=0)
25
26			def out(self):
27			x=self.domain.getX()
28			dim=self.domain.getDim()
29			l=length(x-self.x_c[:dim])
30			m=(l-self.r).whereNegative()
31			return (m+(1.-m)exp(-log(2.)(l/self.width)*2))self.A
32
33			class InterpolatedTimeProfile(Model):
34			""" """
35
36			def __init__(self,debug=False):
37			Model.__init__(self,debug=debug)
38			self.declareParameter(t=[0.,1.],\
39			values=[1.,1.],\
40			out=0.)
41			def doInitialization(self,t):
42			self.__tn=t
43
44			def doStep(self,dt):
45			t=self.__tn+dt
46			if t<=self.t[0]:
47			self.out=self.values[0]
48			else:
49			for i in range(1,len(self.t)):
50			if t<self.t[i]:
51			m=(self.values[i-1]-self.values[i])/(self.t[i-1]-self.t[i])
52			self.out=m*(t-self.t[i-1])+self.values[i-1]
53			self.out=self.t[len(self.t)-1]
54			self.__tn+=dt
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision