예제 #1
0
 def __eq__(self, o):
     if isinstance(o, Function):
         for a, b, order in self._mesh.iter_elems():
             fekete_points = _fekete.get_fekete_points_phys(order, a, b)
             for p in fekete_points:
                 if not feq(self(p), o(p)):
                     return False
         for a, b, order in o._mesh.iter_elems():
             fekete_points = _fekete.get_fekete_points_phys(order, a, b)
             for p in fekete_points:
                 if not feq(self(p), o(p)):
                     return False
         return True
     else:
         return False
예제 #2
0
파일: fekete.py 프로젝트: tmatsuza/hermes
 def __eq__(self, o):
     if isinstance(o, Function):
         for a, b, order in self._mesh.iter_elems():
             fekete_points = _fekete.get_fekete_points_phys(order, a, b)
             for p in fekete_points:
                 if not feq(self(p), o(p)):
                     return False
         for a, b, order in o._mesh.iter_elems():
             fekete_points = _fekete.get_fekete_points_phys(order, a, b)
             for p in fekete_points:
                 if not feq(self(p), o(p)):
                     return False
         return True
     else:
         return False
예제 #3
0
 def __pow__(self, o):
     if isinstance(o, (int, long)):
         pts = self._mesh._points
         orders = empty(len(self._mesh._orders), dtype="int")
         values = []
         for n, (a, b, order) in enumerate(self._mesh.iter_elems()):
             order = o*order
             x = _fekete.get_fekete_points_phys(order, a, b)
             vals = _fekete.eval_poly(x, self._values[n], a, b)**o
             values.append(vals)
             orders[n] = order
         mesh = Mesh1D(pts, orders)
         return Function(values, mesh)
     else:
         return NotImplemented
예제 #4
0
파일: fekete.py 프로젝트: tmatsuza/hermes
 def __pow__(self, o):
     if isinstance(o, (int, long)):
         pts = self._mesh._points
         orders = empty(len(self._mesh._orders), dtype="int")
         values = []
         for n, (a, b, order) in enumerate(self._mesh.iter_elems()):
             order = o * order
             x = _fekete.get_fekete_points_phys(order, a, b)
             vals = _fekete.eval_poly(x, self._values[n], a, b)**o
             values.append(vals)
             orders[n] = order
         mesh = Mesh1D(pts, orders)
         return Function(values, mesh)
     else:
         return NotImplemented
예제 #5
0
 def plot(self, call_show=True):
     try:
         from jsplot import plot, show
     except ImportError:
         from pylab import plot, show
     odd = False
     for a, b, order in self.iter_elems():
         fekete_points = _fekete.get_fekete_points_phys(order, a, b)
         if odd:
             format = "y-"
         else:
             format = "k-"
         odd = not odd
         plot([a, a, b, b], [0, order, order, 0], format, lw=2)
     if call_show:
         show()
예제 #6
0
파일: fekete.py 프로젝트: tmatsuza/hermes
 def plot(self, call_show=True):
     try:
         from jsplot import plot, show
     except ImportError:
         from pylab import plot, show
     odd = False
     for a, b, order in self.iter_elems():
         fekete_points = _fekete.get_fekete_points_phys(order, a, b)
         if odd:
             format = "y-"
         else:
             format = "k-"
         odd = not odd
         plot([a, a, b, b], [0, order, order, 0], format, lw=2)
     if call_show:
         show()
예제 #7
0
 def project_onto_union(self, mesh):
     """
     The same as project_onto, only "mesh" is a subset of self._mesh.
     """
     if mesh == self._mesh:
         return self
     values = []
     n = 0
     for a, b, order in mesh.iter_elems():
         if a >= self._mesh._points[n+1]:
             n += 1
         fekete_points = _fekete.get_fekete_points_phys(order, a, b)
         elem_values = []
         # Note: this is not a projection (it only evaluates obj in
         # fekete points), so the result is not the best
         # approximation possible:
         elem_values = self.get_values_in_element(n, fekete_points)
         values.append(elem_values)
     return Function(values, mesh)
예제 #8
0
파일: fekete.py 프로젝트: tmatsuza/hermes
 def project_onto_union(self, mesh):
     """
     The same as project_onto, only "mesh" is a subset of self._mesh.
     """
     if mesh == self._mesh:
         return self
     values = []
     n = 0
     for a, b, order in mesh.iter_elems():
         if a >= self._mesh._points[n + 1]:
             n += 1
         fekete_points = _fekete.get_fekete_points_phys(order, a, b)
         elem_values = []
         # Note: this is not a projection (it only evaluates obj in
         # fekete points), so the result is not the best
         # approximation possible:
         elem_values = self.get_values_in_element(n, fekete_points)
         values.append(elem_values)
     return Function(values, mesh)
예제 #9
0
    def __init__(self, obj, mesh=None):
        if not isinstance(mesh, Mesh1D):
            raise Exception("You need to specify a mesh.")
        self._mesh = mesh
        if isinstance(obj, (tuple, list, ndarray)):
            self._values = obj
        else:
            self._values = []
            for a, b, order in mesh.iter_elems():
                fekete_points = _fekete.get_fekete_points_phys(order, a, b)
                # Note: this is not a projection (it only evaluates obj in
                # fekete points), so the result is not the best
                # approximation possible:
                elem_values = [obj(p) for p in fekete_points]
                self._values.append(array(elem_values))

        self._poly_coeffs = {}
        self._fe_sol = None

        self.logger = _logger_Function
예제 #10
0
파일: fekete.py 프로젝트: tmatsuza/hermes
    def __init__(self, obj, mesh=None):
        if not isinstance(mesh, Mesh1D):
            raise Exception("You need to specify a mesh.")
        self._mesh = mesh
        if isinstance(obj, (tuple, list, ndarray)):
            self._values = obj
        else:
            self._values = []
            for a, b, order in mesh.iter_elems():
                fekete_points = _fekete.get_fekete_points_phys(order, a, b)
                # Note: this is not a projection (it only evaluates obj in
                # fekete points), so the result is not the best
                # approximation possible:
                elem_values = [obj(p) for p in fekete_points]
                self._values.append(array(elem_values))

        self._poly_coeffs = {}
        self._fe_sol = None

        self.logger = _logger_Function
예제 #11
0
 def plot(self, call_show=True):
     try:
         from jsplot import plot, show
     except ImportError:
         from pylab import plot, show
     odd = False
     for n, (a, b, order) in enumerate(self._mesh.iter_elems()):
         fekete_points = _fekete.get_fekete_points_phys(order, a, b)
         vals = self._values[n]
         assert len(vals) == len(fekete_points)
         x = arange(a, b, 0.1)
         y = [self(_x) for _x in x]
         if odd:
             format = "g-"
         else:
             format = "r-"
         odd = not odd
         plot(x, y, format)
         plot(fekete_points, vals, "ko")
     if call_show:
         show()
예제 #12
0
파일: fekete.py 프로젝트: tmatsuza/hermes
 def plot(self, call_show=True):
     try:
         from jsplot import plot, show
     except ImportError:
         from pylab import plot, show
     odd = False
     for n, (a, b, order) in enumerate(self._mesh.iter_elems()):
         fekete_points = _fekete.get_fekete_points_phys(order, a, b)
         vals = self._values[n]
         assert len(vals) == len(fekete_points)
         x = arange(a, b, 0.1)
         y = [self(_x) for _x in x]
         if odd:
             format = "g-"
         else:
             format = "r-"
         odd = not odd
         plot(x, y, format)
         plot(fekete_points, vals, "ko")
     if call_show:
         show()