140 lines
4.9 KiB
Python
140 lines
4.9 KiB
Python
"""
|
|
|
|
RenderPipeline
|
|
|
|
Copyright (c) 2014-2016 tobspr <tobias.springer1@gmail.com>
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in
|
|
all copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
THE SOFTWARE.
|
|
|
|
"""
|
|
|
|
from rplibs.six.moves import range # pylint: disable=import-error
|
|
|
|
from panda3d.core import Vec3, CurveFitter
|
|
|
|
|
|
class SmoothConnectedCurve(object):
|
|
|
|
""" Interface to a curve which also manages connecting the end of the
|
|
curve with the beginning. """
|
|
|
|
def __init__(self):
|
|
self._curve = None
|
|
self._modified = False
|
|
self._border_points = 1
|
|
self._color = (0, 0, 0)
|
|
self._cv_points = [(0, 0), (0.5, 0), (1.0, 0)]
|
|
self.build_curve()
|
|
|
|
@property
|
|
def modified(self):
|
|
""" Returns whether the curve was modified since the creation """
|
|
return self._modified
|
|
|
|
@property
|
|
def control_points(self):
|
|
""" Returns a list of all controll points """
|
|
return self._cv_points
|
|
|
|
@control_points.setter
|
|
def control_points(self, points):
|
|
""" Sets the cv points to the given list of points """
|
|
self._cv_points = points
|
|
self._modified = True
|
|
self.build_curve()
|
|
|
|
@property
|
|
def color(self):
|
|
""" Returns the display color of the curve """
|
|
return self._color
|
|
|
|
@color.setter
|
|
def color(self, rgb):
|
|
""" Sets the display color of the curve """
|
|
self._color = rgb
|
|
|
|
def set_single_value(self, val):
|
|
""" Sets the curve to be linear, and only use a single value """
|
|
self._cv_points = [[0.5, val]]
|
|
self._modified = False
|
|
self.build_curve()
|
|
|
|
def append_cv(self, x, y):
|
|
""" Appends a new cv and returns the index of the attached cv """
|
|
self._cv_points.append([x, y])
|
|
self.build_curve()
|
|
self._modified = True
|
|
return len(self._cv_points) - 1
|
|
|
|
def remove_cv(self, index):
|
|
""" Attempts to remove the cv at the given index, does nothing if only
|
|
one control point is left """
|
|
if len(self._cv_points) > 1:
|
|
del self._cv_points[index]
|
|
self._modified = True
|
|
self.build_curve()
|
|
|
|
def build_curve(self):
|
|
""" Rebuilds the curve based on the controll point values """
|
|
sorted_points = sorted(self._cv_points, key=lambda v: v[0])
|
|
first_point = sorted_points[0]
|
|
fitter = CurveFitter()
|
|
|
|
# Duplicate curve at the beginning
|
|
for i in range(self._border_points):
|
|
end_point = self._cv_points[(-i + self._border_points - 1) % len(self._cv_points)]
|
|
end_point = first_point
|
|
fitter.add_xyz(0.0, Vec3(0, end_point[1], 0))
|
|
|
|
# Append the actual points
|
|
for point in self._cv_points:
|
|
# Clamp point x position to avoid artifacts at the beginning
|
|
point_t = max(0.01, point[0])
|
|
fitter.add_xyz(point_t, Vec3(point_t, point[1], 0))
|
|
|
|
# Duplicate curve at the end
|
|
for i in range(self._border_points):
|
|
start_point = self._cv_points[i % len(self._cv_points)]
|
|
start_point = first_point
|
|
fitter.add_xyz(1.0, Vec3(1, start_point[1], 0))
|
|
|
|
fitter.sort_points()
|
|
fitter.wrap_hpr()
|
|
fitter.compute_tangents(1.0)
|
|
|
|
self._curve = fitter.make_hermite()
|
|
|
|
def set_cv_value(self, index, x_value, y_value):
|
|
""" Updates the cv point at the given index """
|
|
self._cv_points[index] = [x_value, y_value]
|
|
self._modified = True
|
|
|
|
def get_value(self, offset):
|
|
""" Returns the value on the curve ranging whereas the offset should be
|
|
from 0 to 1 (0 denotes the start of the curve). The returned value will
|
|
be a value from 0 to 1 as well. """
|
|
point = Vec3(0)
|
|
self._curve.evaluate_xyz(offset, point)
|
|
return max(0.0, min(1.0, point.y))
|
|
|
|
def serialize(self):
|
|
""" Returns the value of the curve as yaml list """
|
|
points = ["[{:5.10f},{:5.10f}]".format(float(a), float(b)) for a, b in self._cv_points]
|
|
return "[" + ','.join(points) + "]"
|