curvature2d.py 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. # ##### BEGIN GPL LICENSE BLOCK #####
  2. #
  3. # This program is free software; you can redistribute it and/or
  4. # modify it under the terms of the GNU General Public License
  5. # as published by the Free Software Foundation; either version 2
  6. # of the License, or (at your option) any later version.
  7. #
  8. # This program is distributed in the hope that it will be useful,
  9. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11. # GNU General Public License for more details.
  12. #
  13. # You should have received a copy of the GNU General Public License
  14. # along with this program; if not, write to the Free Software Foundation,
  15. # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  16. #
  17. # ##### END GPL LICENSE BLOCK #####
  18. # Filename : curvature2d.py
  19. # Author : Stephane Grabli
  20. # Date : 04/08/2005
  21. # Purpose : The stroke points are colored in gray levels and depending
  22. # on the 2d curvature value
  23. from freestyle.chainingiterators import ChainSilhouetteIterator
  24. from freestyle.predicates import (
  25. NotUP1D,
  26. QuantitativeInvisibilityUP1D,
  27. TrueUP1D,
  28. )
  29. from freestyle.shaders import (
  30. ConstantThicknessShader,
  31. py2DCurvatureColorShader,
  32. )
  33. from freestyle.types import Operators, Stroke
  34. Operators.select(QuantitativeInvisibilityUP1D(0))
  35. Operators.bidirectional_chain(ChainSilhouetteIterator(), NotUP1D(QuantitativeInvisibilityUP1D(0)))
  36. shaders_list = [
  37. ConstantThicknessShader(5),
  38. py2DCurvatureColorShader()
  39. ]
  40. Operators.create(TrueUP1D(), shaders_list)