/scenes/scenekit-demo.py

https://github.com/tdamdouni/Pythonista · Python · 57 lines · 48 code · 7 blank · 2 comment · 0 complexity · 6ca6b56fdd4734008b71b1dbde74e761 MD5 · raw file 

    

  1. # coding: utf-8
    2. 

  2. 

  3. # https://forum.omz-software.com/topic/1686/3d-in-pythonista/7
    4. 

  4. 

  5. from objc_util import *
    6. 

  6. import ui
    7. 

  7. import math
    8. 

  8. 

  9. SCNView, SCNScene, SCNBox, SCNText, SCNNode, SCNLight, SCNAction, UIFont = map(ObjCClass, ['SCNView', 'SCNScene', 'SCNBox', 'SCNText', 'SCNNode', 'SCNLight', 'SCNAction', 'UIFont'])
    10. 

  10. 

  11. class SCNVector3 (Structure):
    12. 

  12. 	_fields_ = [('x', c_float), ('y', c_float), ('z', c_float)]
    13. 

  13. 	
    14. 

  14. @on_main_thread
    15. 

  15. def demo():
    16. 

  16. 	main_view = ui.View()
    17. 

  17. 	main_view_objc = ObjCInstance(main_view)
    18. 

  18. 	scene_view = SCNView.alloc().initWithFrame_options_(((0, 0),(400, 400)), None).autorelease()
    19. 

  19. 	scene_view.setAutoresizingMask_(18)
    20. 

  20. 	scene_view.setAllowsCameraControl_(True)
    21. 

  21. 	scene = SCNScene.scene()
    22. 

  22. 	root_node = scene.rootNode()
    23. 

  23. 	text_mesh = SCNText.textWithString_extrusionDepth_('@tdamdouni', 6.0)
    24. 

  24. 	text_mesh.setFlatness_(0.2)
    25. 

  25. 	text_mesh.setChamferRadius_(0.4)
    26. 

  26. 	text_mesh.setFont_(UIFont.fontWithName_size_('HelveticaNeue-Bold', 18))
    27. 

  27. 	bbox_min, bbox_max = SCNVector3(), SCNVector3()
    28. 

  28. 	text_mesh.getBoundingBoxMin_max_(byref(bbox_min), byref(bbox_max), restype=None, argtypes=[POINTER(SCNVector3), POINTER(SCNVector3)])
    29. 

  29. 	text_width = bbox_max.x - bbox_min.x
    30. 

  30. 	text_node = SCNNode.nodeWithGeometry_(text_mesh)
    31. 

  31. 	text_node.setCastsShadow_(True)
    32. 

  32. 	text_container = SCNNode.node()
    33. 

  33. 	text_container.addChildNode_(text_node)
    34. 

  34. 	text_container.setPosition_((0, 40, 0))
    35. 

  35. 	text_node.setPosition_((-text_width/2, 0, 0))
    36. 

  36. 	box = SCNBox.boxWithWidth_height_length_chamferRadius_(100, 4, 100, 1)
    37. 

  37. 	box_node = SCNNode.nodeWithGeometry_(box)
    38. 

  38. 	root_node.addChildNode_(box_node)
    39. 

  39. 	rotate_action = SCNAction.repeatActionForever_(SCNAction.rotateByX_y_z_duration_(0, math.pi*2, math.pi*2, 10))
    40. 

  40. 	text_container.runAction_(rotate_action)
    41. 

  41. 	root_node.addChildNode_(text_container)
    42. 

  42. 	light_node = SCNNode.node()
    43. 

  43. 	light_node.setPosition_((0, 100, 10))
    44. 

  44. 	light_node.setRotation_((1, 0, 0, -math.pi/2))
    45. 

  45. 	light = SCNLight.light()
    46. 

  46. 	light.setType_('spot')
    47. 

  47. 	light.setCastsShadow_(True)
    48. 

  48. 	light.setColor_(UIColor.cyanColor().CGColor())
    49. 

  49. 	light_node.setLight_(light)
    50. 

  50. 	root_node.addChildNode_(light_node)
    51. 

  51. 	scene_view.setScene_(scene)
    52. 

  52. 	main_view_objc.addSubview_(scene_view)
    53. 

  53. 	main_view.name = 'SceneKit Demo'
    54. 

  54. 	main_view.present()
    55. 

  55. 	
    56. 

  56. demo()
    57.