home *** CD-ROM | disk | FTP | other *** search
/ Hackers Magazine 57 / CdHackersMagazineNr57.iso / Software / Multimedia / k3d-setup-0.7.11.0.exe / lib / site-packages / cgkit / euleradapter.py < prev    next >
Encoding:
Python Source  |  2007-01-11  |  5.6 KB  |  154 lines
  1. # ***** BEGIN LICENSE BLOCK *****
  2. # Version: MPL 1.1/GPL 2.0/LGPL 2.1
  3. #
  4. # The contents of this file are subject to the Mozilla Public License Version
  5. # 1.1 (the "License"); you may not use this file except in compliance with
  6. # the License. You may obtain a copy of the License at
  7. # http://www.mozilla.org/MPL/
  8. #
  9. # Software distributed under the License is distributed on an "AS IS" basis,
  10. # WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  11. # for the specific language governing rights and limitations under the
  12. # License.
  13. #
  14. # The Original Code is the Python Computer Graphics Kit.
  15. #
  16. # The Initial Developer of the Original Code is Matthias Baas.
  17. # Portions created by the Initial Developer are Copyright (C) 2004
  18. # the Initial Developer. All Rights Reserved.
  19. #
  20. # Contributor(s):
  21. #
  22. # Alternatively, the contents of this file may be used under the terms of
  23. # either the GNU General Public License Version 2 or later (the "GPL"), or
  24. # the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  25. # in which case the provisions of the GPL or the LGPL are applicable instead
  26. # of those above. If you wish to allow use of your version of this file only
  27. # under the terms of either the GPL or the LGPL, and not to allow others to
  28. # use your version of this file under the terms of the MPL, indicate your
  29. # decision by deleting the provisions above and replace them with the notice
  30. # and other provisions required by the GPL or the LGPL. If you do not delete
  31. # the provisions above, a recipient may use your version of this file under
  32. # the terms of any one of the MPL, the GPL or the LGPL.
  33. #
  34. # ***** END LICENSE BLOCK *****
  35. # $Id: euleradapter.py,v 1.1.1.1 2004/12/12 14:30:59 mbaas Exp $
  36.  
  37. ## \file euleradapters.py
  38. ## Contains the EulerAdapter class.
  39.  
  40. import protocols
  41. from Interfaces import *
  42. from component import *
  43. import slots
  44. from cgtypes import *
  45. from math import pi
  46. import _core
  47.  
  48. # EulerAdapter
  49. class EulerAdapter(Component):
  50.     """Euler angle to mat3, mat4 or quat adapter.
  51.  
  52.     This class can be used to convert euler angles either to a mat3, a mat4
  53.     or a quat. The input slots are \c anglex_slot, \c angley_slot and
  54.     \c anglez_slot. The output slot is \c output_slot. The type of the output
  55.     can be determined in the constructor.
  56.     """
  57.  
  58.     protocols.advise(instancesProvide=[ISceneItem])
  59.  
  60.     def __init__(self,
  61.                  anglex = 0,
  62.                  angley = 0,
  63.                  anglez = 0,
  64.                  radians = False,
  65.                  order = "xyz",
  66.                  outtype = "mat3",
  67.                  name="EulerAdapter",
  68.                  auto_insert=True):
  69.         """Constructor.
  70.  
  71.         \param anglex (\c float) Initial angle around x axis
  72.         \param angley (\c float) Initial angle around y axis
  73.         \param anglez (\c float) Initial angle around z axis
  74.         \param radians (\c bool) True = Angles are specified in radians instead of degrees
  75.         \param order (\c str) The rotation order ("xyz", "xzy", ...)
  76.         \param outtpye (\c str) Output type ("mat3", "mat4", "quat")
  77.         \param name (\c str) Component name
  78.         \param auto_insert (\c bool) Auto insert flag
  79.         """
  80.         Component.__init__(self, name=name, auto_insert=auto_insert)
  81.  
  82.         if radians:
  83.             self.factor = 1.0
  84.         else:
  85.             self.factor = pi/180.0
  86.  
  87.         self.anglex_slot = slots.DoubleSlot(anglex)
  88.         self.angley_slot = slots.DoubleSlot(angley)
  89.         self.anglez_slot = slots.DoubleSlot(anglez)
  90.         self.addSlot("anglex", self.anglex_slot)
  91.         self.addSlot("angley", self.angley_slot)
  92.         self.addSlot("anglez", self.anglez_slot)
  93.  
  94.         if outtype=="mat3":
  95.             self.output_slot = slots.ProceduralMat3Slot(self.computeMat3)
  96.         elif outtype=="mat4":
  97.             self.output_slot = slots.ProceduralMat4Slot(self.computeMat4)
  98.         elif outtype=="quat":
  99.             self.output_slot = slots.ProceduralQuatSlot(self.computeQuat)
  100.         else:
  101.             raise ValueError, "Unknown output type: %s"%outtype
  102.         
  103.         self.addSlot("output", self.output_slot)
  104.             
  105.         self.anglex_slot.addDependent(self.output_slot)
  106.         self.angley_slot.addDependent(self.output_slot)
  107.         self.anglez_slot.addDependent(self.output_slot)
  108.  
  109.         # self.fromEuler is the mat3() method that computes the matrix
  110.         # from the euler angles. Which one exactly it is depends on the
  111.         # order
  112.         exec "self.fromEuler = mat3.fromEuler%s"%order.upper()
  113.  
  114.     def protocols(self):
  115.         return [ISceneItem, IComponent]
  116.  
  117.     def computeMat3(self):
  118.         """Slot procedure."""
  119.         f = self.factor
  120.         return self.fromEuler(f*self.anglex_slot.getValue(),
  121.                               f*self.angley_slot.getValue(),
  122.                               f*self.anglez_slot.getValue())
  123.  
  124.     def computeMat4(self):
  125.         """Slot procedure."""
  126.         f = self.factor
  127.         m3 = self.fromEuler(f*self.anglex_slot.getValue(),
  128.                             f*self.angley_slot.getValue(),
  129.                             f*self.anglez_slot.getValue())
  130.         res = mat4(1)
  131.         res.setMat3(m3)
  132.         return res
  133.  
  134.     def computeQuat(self):
  135.         """Slot procedure."""
  136.         f = self.factor
  137.         m3 = self.fromEuler(f*self.anglex_slot.getValue(),
  138.                             f*self.angley_slot.getValue(),
  139.                             f*self.anglez_slot.getValue())
  140.         res = quat().fromMat(m3)
  141.         return res
  142.  
  143.     ## protected:
  144.         
  145.     # angle properties...
  146.     exec slotPropertyCode("anglex")
  147.     exec slotPropertyCode("angley")
  148.     exec slotPropertyCode("anglez")
  149.  
  150.     # "output" property...
  151.     exec slotPropertyCode("output")
  152.     
  153.  
  154.