Legend of the Gold Box... A game written for the LOWREZJAM 2018 game jam
Vous ne pouvez pas sélectionner plus de 25 sujets Les noms de sujets doivent commencer par une lettre ou un nombre, peuvent contenir des tirets ('-') et peuvent comporter jusqu'à 35 caractères.

891 lines
29KB

  1. '''
  2. Filename nodes.py
  3. Author: Bryan "ObsidianBlk" Miller
  4. Date Created: 8/1/2018
  5. Python Version: 3.7
  6. '''
  7. from .display import Display
  8. from .events import Events
  9. from .resource import ResourceManager
  10. import pygame
  11. class NodeError(Exception):
  12. pass
  13. class Node:
  14. def __init__(self, name="Node", parent=None):
  15. self._NODE_DATA={
  16. "parent":None,
  17. "name":name,
  18. "children":[],
  19. "resource":None,
  20. "position":(0,0)
  21. }
  22. if parent is not None:
  23. try:
  24. self.parent = parent
  25. except NodeError as e:
  26. raise e
  27. @property
  28. def parent(self):
  29. return self._NODE_DATA["parent"]
  30. @parent.setter
  31. def parent(self, new_parent):
  32. try:
  33. self.parent_to_node(new_parent)
  34. except NodeError as e:
  35. raise e
  36. @property
  37. def root(self):
  38. if self.parent is None:
  39. return self
  40. return self.parent.root
  41. @property
  42. def name(self):
  43. return self._NODE_DATA["name"]
  44. @name.setter
  45. def name(self, value):
  46. if self.parent is not None:
  47. if self.parent.get_node(value) is not None:
  48. raise NodeError("Parent already contains node named '{}'.".format(name))
  49. self._NODE_DATA["name"] = value
  50. @property
  51. def full_name(self):
  52. if self.parent is None:
  53. return self.name
  54. return self.parent.full_name + "." + self.name
  55. @property
  56. def resource(self):
  57. if self._NODE_DATA["resource"] is None:
  58. # Only bother creating the instance if it's being asked for.
  59. # All ResourceManager instances access same data.
  60. self._NODE_DATA["resource"] = ResourceManager()
  61. return self._NODE_DATA["resource"]
  62. @property
  63. def child_count(self):
  64. return len(self._NODE_DATA["children"])
  65. @property
  66. def position(self):
  67. p = self._NODE_DATA["position"]
  68. return (p[0], p[1])
  69. @position.setter
  70. def position(self, pos):
  71. if not isinstance(pos, (list, tuple)):
  72. raise TypeError("Expected a list or tuple.")
  73. if len(pos) != 2:
  74. raise ValueError("Wrong number of values given.")
  75. try:
  76. self.position_x = pos[0]
  77. self.position_y = pos[1]
  78. except Exception as e:
  79. raise e
  80. @property
  81. def position_x(self):
  82. return self._NODE_DATA["position"][0]
  83. @position_x.setter
  84. def position_x(self, v):
  85. if not isinstance(v, (int, float)):
  86. raise TypeError("Excepted an number value.")
  87. self._NODE_DATA["position"] = (float(v), self._NODE_DATA["position"][1])
  88. @property
  89. def position_y(self):
  90. return self._NODE_DATA["position"][1]
  91. @position_y.setter
  92. def position_y(self, v):
  93. if not isinstance(v, (int, float)):
  94. raise TypeError("Excepted an number value.")
  95. self._NODE_DATA["position"] = (self._NODE_DATA["position"][0], float(v))
  96. def get_world_position(self):
  97. if self.parent is None:
  98. return (0,0)
  99. pos = self.position
  100. ppos = self.parent.get_world_position()
  101. return (pos[0] + ppos[0], pos[1] + ppos[1])
  102. def parent_to_node(self, parent, allow_reparenting=False):
  103. if not isinstance(parent, Node):
  104. raise NodeError("Node may only parent to another Node instance.")
  105. if self.parent is None or self.parent != parent:
  106. if self.parent is not None:
  107. if allow_Reparenting == False:
  108. raise NodeError("Node already assigned a parent Node.")
  109. if self.parent.remove_node(self) != self:
  110. raise NodeError("Failed to remove self from current parent.")
  111. try:
  112. parent.attach_node(self)
  113. except NodeError as e:
  114. raise e
  115. def attach_node(self, node, reparent=False, index=-1):
  116. if node.parent is not None:
  117. if node.parent == self:
  118. return # Nothing to do. Given node already parented to this node.
  119. if reparent == False:
  120. raise NodeError("Node already parented.")
  121. if node.parent.remove_node(node) != node:
  122. raise NodeError("Failed to remove given node from it's current parent.")
  123. if self.get_node(node.name) is not None:
  124. raise NodeError("Node with name '{}' already attached.".format(node.name))
  125. node._NODE_DATA["parent"] = self
  126. children = self._NODE_DATA["children"]
  127. if index < 0 or index >= len(children):
  128. children.append(node)
  129. else:
  130. children.insert(index, node)
  131. def remove_node(self, node):
  132. if isinstance(node, (str, unicode)):
  133. n = self.get_node(node)
  134. if n is not None:
  135. try:
  136. return self.remove_node(n)
  137. except NodeError as e:
  138. raise e
  139. elif isinstance(node, Node):
  140. if node.parent != self:
  141. if node.parent == None:
  142. raise NodeError("Cannot remove an unparented node.")
  143. try:
  144. return node.parent.remove_node(node)
  145. except NodeError as e:
  146. raise e
  147. if node in self._NODE_DATA["children"]:
  148. self._NODE_DATA["children"].remove(node)
  149. node._NODE_DATA["parent"] = None
  150. return node
  151. else:
  152. raise NodeError("Expected a Node instance or a string.")
  153. return None
  154. def get_node(self, name):
  155. if self.child_count <= 0:
  156. return None
  157. subnames = name.split(".")
  158. for c in self._NODE_DATA["children"]:
  159. if c.name == subnames[0]:
  160. if len(subnames) > 1:
  161. return c.get_node(".".join(subnames[1:-1]))
  162. return c
  163. return None
  164. def listen(self, signal, callback_fn):
  165. try:
  166. Events.listen(signal, callback_fn)
  167. except Exception as e:
  168. raise e
  169. def unlisten(self, signal, callback_fn):
  170. Events.unlisten(signal, callback_fn)
  171. def emit(self, signal, data={}):
  172. # NOTE: I'm currently forcing the existance of this "NODE" key in the dictionary given. Not sure if I'll keep this.
  173. data["NODE"] = {
  174. "name":self.name,
  175. "n":self
  176. }
  177. Events.emit(signal, data)
  178. def _init(self):
  179. if hasattr(self, "on_init"):
  180. self.on_init()
  181. for c in self._NODE_DATA["children"]:
  182. c._init()
  183. def _close(self):
  184. if hasattr(self, "on_close"):
  185. self.on_close()
  186. for c in self._NODE_DATA["children"]:
  187. c._close()
  188. def _pause(self):
  189. if hasattr(self, "on_pause"):
  190. self.on_pause()
  191. for c in self._NODE_DATA["children"]:
  192. c._pause()
  193. def _start(self):
  194. if hasattr(self, "on_start"):
  195. self.on_start()
  196. for c in self._NODE_DATA["children"]:
  197. c._start()
  198. def _update(self, dt):
  199. if hasattr(self, "on_update"):
  200. self.on_update(dt)
  201. for c in self._NODE_DATA["children"]:
  202. c._update(dt)
  203. def _render(self, surface):
  204. for c in self._NODE_DATA["children"]:
  205. c._render(surface)
  206. class Node2D(Node):
  207. def __init__(self, name="Node2D", parent=None):
  208. try:
  209. Node.__init__(self, name, parent)
  210. except NodeError as e:
  211. raise e
  212. @property
  213. def resolution(self):
  214. p = self.parent
  215. # We don't directly have the answer, but maybe our parent does?
  216. if p is not None:
  217. if hasattr(p, "resolution"):
  218. return p.resolution
  219. # Otherwise the Display object should.
  220. return Display.resolution
  221. def _callOnRender(self, surface):
  222. if hasattr(self, "on_render"):
  223. self._ACTIVE_SURF = surface
  224. self.on_render()
  225. del self._ACTIVE_SURF
  226. def _render(self, surface):
  227. self._callOnRender(surface)
  228. Node._render(self, surface)
  229. def draw_image(self, img, pos=(0,0), rect=None):
  230. if not hasattr(self, "_ACTIVE_SURF"):
  231. return
  232. self._ACTIVE_SURF.blit(img, pos, rect)
  233. def fill(self, color):
  234. if not hasattr(self, "_ACTIVE_SURF"):
  235. return
  236. self._ACTIVE_SURF.fill(color)
  237. def draw_lines(self, points, color, thickness=1, closed=False):
  238. if not hasattr(self, "_ACTIVE_SURF"):
  239. return
  240. pygame.draw.lines(self._ACTIVE_SURF, color, closed, points, thickness)
  241. def draw_rect(self, rect, color, thickness=1):
  242. if not hasattr(self, "_ACTIVE_SURF"):
  243. return
  244. pygame.draw.rect(self._ACTIVE_SURF, color, rect, thickness)
  245. def draw_ellipse(self, rect, color, thickness=1, fill_color=None):
  246. if not hasattr(self, "_ACTIVE_SURF"):
  247. return
  248. if fill_color is not None:
  249. pygame.draw.ellipse(self._ACTIVE_SURF, fill_color, rect)
  250. if thickness > 0:
  251. pygame.draw.ellipse(self._ACTIVE_SURF, color, rect, thickness)
  252. def draw_circle(self, pos, radius, color, thickness=1, fill_color=None):
  253. if not hasattr(self, "_ACTIVE_SURF"):
  254. return
  255. if fill_color is not None:
  256. pygame.draw.circle(self._ACTIVE_SURF, fill_color, pos, radius)
  257. if thickness > 0:
  258. pygame.draw.circle(self._ACTIVE_SURF, color, pos, radius, thickness)
  259. def draw_polygon(self, points, color, thickness=1, fill_color=None):
  260. if not hasattr(self, "_ACTIVE_SURF"):
  261. return
  262. if fill_color is not None:
  263. pygame.draw.polygon(self._ACTIVE_SURF, fill_color, points)
  264. if thickness >= 1:
  265. pygame.draw.polygon(self._ACTIVE_SURF, color, points, thickness)
  266. class NodeSurface(Node2D):
  267. def __init__(self, name="NodeSurface", parent=None):
  268. try:
  269. Node2D.__init__(self, name, parent)
  270. except NodeError as e:
  271. raise e
  272. # TODO: Update this class to use the _NODE*_DATA={} structure.
  273. self._scale = (1.0, 1.0)
  274. self._scaleToDisplay = False
  275. self._scaleDirty = False
  276. self._keepAspectRatio = False
  277. self._alignCenter = False
  278. self._surface = None
  279. self._tsurface = None
  280. self.set_surface()
  281. def _updateTransformSurface(self):
  282. if self._surface is None:
  283. return
  284. self._scaleDirty = False
  285. if self._scaleToDisplay:
  286. dsize = Display.resolution
  287. ssize = self._surface.get_size()
  288. self._scale = (dsize[0] / ssize[0], dsize[1] / ssize[1])
  289. if self._keepAspectRatio:
  290. if self._scale[0] < self._scale[1]:
  291. self._scale = (self._scale[0], self._scale[0])
  292. else:
  293. self._scale = (self._scale[1], self._scale[1])
  294. if self._scale[0] == 1.0 and self._scale[1] == 1.0:
  295. self._tsurface = None
  296. return
  297. size = self._surface.get_size()
  298. nw = size[0] * self._scale[0]
  299. nh = 0
  300. if self._keepAspectRatio:
  301. nh = size[1] * self._scale[0]
  302. else:
  303. nh = size[1] * self._scale[1]
  304. self._tsurface = pygame.Surface((nw, nh), pygame.SRCALPHA, self._surface)
  305. self._tsurface.fill(pygame.Color(0,0,0,0))
  306. @property
  307. def resolution(self):
  308. if self._surface is None:
  309. return super().resolution
  310. return self._surface.get_size()
  311. @resolution.setter
  312. def resolution(self, res):
  313. try:
  314. self.set_surface(res)
  315. except (TypeError, ValueError) as e:
  316. raise e
  317. @property
  318. def width(self):
  319. return self.resolution[0]
  320. @property
  321. def height(self):
  322. return self.resolution[1]
  323. @property
  324. def scale(self):
  325. return self._scale
  326. @scale.setter
  327. def scale(self, scale):
  328. if self._keepAspectRatio:
  329. if not isinstance(scale, (int, float)):
  330. raise TypeError("Expected number value.")
  331. self._scale = (scale, self._scale[1])
  332. else:
  333. if not isinstance(scale, tuple):
  334. raise TypeError("Expected a tuple")
  335. if len(scale) != 2:
  336. raise ValueError("Expected tuple of length two.")
  337. if not isinstance(scale[0], (int, float)) or not isinstance(scale[1], (int, float)):
  338. raise TypeError("Expected number values.")
  339. self._scale = scale
  340. self._updateTransformSurface()
  341. @property
  342. def keep_aspect_ratio(self):
  343. return self._keepAspectRatio
  344. @keep_aspect_ratio.setter
  345. def keep_aspect_ratio(self, keep):
  346. self._keepAspectRatio = (keep == True)
  347. self._updateTransformSurface()
  348. @property
  349. def align_center(self):
  350. return self._alignCenter
  351. @align_center.setter
  352. def align_center(self, center):
  353. self._alignCenter = (center == True)
  354. @property
  355. def scale_to_display(self):
  356. return self._scaleToDisplay
  357. @scale_to_display.setter
  358. def scale_to_display(self, todisplay):
  359. if todisplay == True:
  360. self._scaleToDisplay = True
  361. Events.listen("VIDEORESIZE", self._OnVideoResize)
  362. else:
  363. self._scaleToDisplay = False
  364. Events.unlisten("VIDEORESIZE", self._OnVideoResize)
  365. self._updateTransformSurface()
  366. def scale_to(self, target_resolution):
  367. if self._surface is not None:
  368. size = self._surface.get_size()
  369. nscale = (float(size[0]) / float(target_resolution[0]), float(size[1]) / float(target_resolution[1]))
  370. self.scale = nscale
  371. def set_surface(self, resolution=None):
  372. dsurf = Display.surface
  373. if resolution is None:
  374. if dsurf is not None:
  375. self._surface = dsurf.convert_alpha()
  376. self._surface.fill(pygame.Color(0,0,0,0))
  377. self._updateTransformSurface()
  378. else:
  379. if not isinstance(resolution, tuple):
  380. raise TypeError("Expected a tuple.")
  381. if len(resolution) != 2:
  382. raise ValueError("Expected a tuple of length two.")
  383. if not isinstance(resolution[0], int) or not isinstance(resolution[1], int):
  384. raise TypeError("Tuple expected to contain integers.")
  385. if dsurf is not None:
  386. self._surface = pygame.Surface(resolution, pygame.SRCALPHA, dsurf)
  387. else:
  388. self._surface = pygame.Surface(resolution, pygame.SRCALPHA)
  389. self._surface.fill(pygame.Color(0,0,0,0))
  390. self._updateTransformSurface()
  391. def _render(self, surface):
  392. if self._surface is None:
  393. self.set_surface()
  394. if self._surface is not None:
  395. if self._scaleDirty:
  396. self._updateTransformSurface()
  397. Node2D._render(self, self._surface)
  398. else:
  399. Node2D._render(self, surface)
  400. self._scale_and_blit(surface)
  401. def _scale_and_blit(self, dest):
  402. dsize = dest.get_size()
  403. src = self._surface
  404. if self._tsurface is not None:
  405. pygame.transform.scale(self._surface, self._tsurface.get_size(), self._tsurface)
  406. src = self._tsurface
  407. ssize = src.get_size()
  408. posx = self.position_x
  409. posy = self.position_y
  410. if self._alignCenter:
  411. if dsize[0] > ssize[0]:
  412. posx += (dsize[0] - ssize[0]) * 0.5
  413. if dsize[1] > ssize[1]:
  414. posy += (dsize[1] - ssize[1]) * 0.5
  415. pos = (int(posx), int(posy))
  416. dest.blit(src, pos)
  417. def _OnVideoResize(self, event, data):
  418. if self._scaleToDisplay:
  419. self._scaleDirty = True
  420. class NodeText(Node2D):
  421. def __init__(self, name="NodeText", parent=None):
  422. try:
  423. Node2D.__init__(self, name, parent)
  424. except NodeError as e:
  425. raise e
  426. self._NODETEXT_DATA={
  427. "font_src":"",
  428. "size":26,
  429. "antialias":True,
  430. "color":pygame.Color(255,255,255),
  431. "background":None,
  432. "text":"Some Text",
  433. "surface":None
  434. }
  435. @property
  436. def font_src(self):
  437. return self._NODETEXT_DATA["font_src"]
  438. @font_src.setter
  439. def font_src(self, src):
  440. res = self.resource
  441. if src != "" and src != self._NODETEXT_DATA["font_src"] and res.is_valid("font", src):
  442. self._NODETEXT_DATA["font_src"] = src
  443. if not res.has("font", src):
  444. res.store("font", src)
  445. self._NODETEXT_DATA["surface"] = None
  446. @property
  447. def size(self):
  448. return self._NODETEXT_DATA["size"]
  449. @size.setter
  450. def size(self, size):
  451. if not isinstance(size, int):
  452. raise TypeError("Expected integer value.")
  453. if size <= 0:
  454. raise ValueError("Size must be greater than zero.")
  455. if size != self._NODETEXT_DATA["size"]:
  456. self._NODETEXT_DATA["size"] = size
  457. self._NODETEXT_DATA["surface"] = None
  458. @property
  459. def antialias(self):
  460. return self._NODETEXT_DATA["antialias"]
  461. @antialias.setter
  462. def antialias(self, enable):
  463. enable = (enable == True)
  464. if enable != self._NODETEXT_DATA["antialias"]:
  465. self._NODETEXT_DATA["antialias"] = enable
  466. self._NODETEXT_DATA["surface"] = None
  467. @property
  468. def text(self):
  469. return self._NODETEXT_DATA["text"]
  470. @text.setter
  471. def text(self, text):
  472. if text != self._NODETEXT_DATA["text"]:
  473. self._NODETEXT_DATA["text"] = text
  474. self._NODETEXT_DATA["surface"] = None
  475. def _setColor(self, cname, r, g, b, a):
  476. if r < 0 or r > 255:
  477. raise ValueError("Red value out of bounds.")
  478. if g < 0 or g > 255:
  479. raise ValueError("Green value out of bounds.")
  480. if b < 0 or b > 255:
  481. raise ValueError("Blue value out of bounds.")
  482. if a < 0 or a > 255:
  483. raise ValueError("Alpha value out of bounds.")
  484. color = self._NODETEXT_DATA[cname]
  485. if color.r != r or color.g != g or color.b != b or color.a != a:
  486. self._NODETEXT_DATA[cname] = pygame.Color(r,g,b,a)
  487. self._NODETEXT_DATA["surface"] = None
  488. def _getColor(self, cname):
  489. if self._NODETEXT_DATA[cname] is None:
  490. return (0,0,0,0)
  491. else:
  492. c = self._NODETEXT_DATA[cname]
  493. return (c.r, c.g, c.b, c.a)
  494. def set_color(self, r, g, b, a=255):
  495. try:
  496. self._setColor("color", r, g, b, a)
  497. except ValueError as e:
  498. raise e
  499. return self
  500. def get_color(self):
  501. return self._getColor("color")
  502. def set_background(self, r, g, b, a=255):
  503. try:
  504. self._setColor("background", r, g, b, a)
  505. except ValueError as e:
  506. raise e
  507. return self
  508. def clear_background(self):
  509. if self._NODETEXT_DATA["background"] is not None:
  510. self._NODETEXT_DATA["background"] = None
  511. self._NODETEXT_DATA["surface"] = None
  512. return self
  513. def get_background(self):
  514. c = self._getColor("background")
  515. if c[3] <= 0:
  516. return None
  517. return c
  518. def _render(self, surface):
  519. if self._NODETEXT_DATA["surface"] is None and self._NODETEXT_DATA["text"] != "":
  520. res = self.resource
  521. fnt = res.get("font", self._NODETEXT_DATA["font_src"], {"size":self._NODETEXT_DATA["size"]})
  522. if fnt is not None and fnt() is not None:
  523. surf = None
  524. try:
  525. text = self.text
  526. antialias = self.antialias
  527. color = self._NODETEXT_DATA["color"]
  528. background = self._NODETEXT_DATA["background"]
  529. surf = fnt().render(text, antialias, color, background)
  530. except pygame.error as e:
  531. pass # TODO: Update to send out warning!
  532. self._NODETEXT_DATA["surface"] = surf
  533. # We really don't want to keep the font instance around... just in case we want to use the same font at different sizes.
  534. res.clear("font", self._NODETEXT_DATA["font_src"])
  535. Node2D._render(self, surface)
  536. if self._NODETEXT_DATA["surface"] is not None:
  537. pos = self.position
  538. pos = (int(pos[0]), int(pos[1]))
  539. surface.blit(self._NODETEXT_DATA["surface"], pos)
  540. class NodeSprite(Node2D):
  541. def __init__(self, name="NodeSprite", parent=None):
  542. try:
  543. Node2D.__init__(self, name, parent)
  544. except NodeError as e:
  545. raise e
  546. self._NODESPRITE_DATA={
  547. "rect":[0,0,0,0],
  548. "image":"",
  549. "scale":[1.0, 1.0],
  550. "surface":None
  551. }
  552. @property
  553. def image_width(self):
  554. if self._NODESPRITE_DATA["surface"] is None:
  555. return 0
  556. surf = self._NODESPRITE_DATA["surface"]()
  557. return surf.get_width()
  558. @property
  559. def image_height(self):
  560. if self._NODESPRITE_DATA["surface"] is None:
  561. return 0
  562. surf = self._NODESPRITE_DATA["surface"]()
  563. return surf.get_height()
  564. @property
  565. def sprite_width(self):
  566. return int(self.rect_width * self.scale_x)
  567. @property
  568. def sprite_height(self):
  569. return int(self.rect_height * self.scale_y)
  570. @property
  571. def rect(self):
  572. return (self._NODESPRITE_DATA["rect"][0],
  573. self._NODESPRITE_DATA["rect"][1],
  574. self._NODESPRITE_DATA["rect"][2],
  575. self._NODESPRITE_DATA["rect"][3])
  576. @rect.setter
  577. def rect(self, rect):
  578. if not isinstance(rect, (list, tuple)):
  579. raise TypeError("Expected a list or tuple.")
  580. if len(rect) != 4:
  581. raise ValueError("rect value contains wrong number of values.")
  582. try:
  583. self.rect_x = rect[0]
  584. self.rect_y = rect[1]
  585. self.rect_width = rect[2]
  586. self.rect_height = rect[3]
  587. except Exception as e:
  588. raise e
  589. @property
  590. def rect_x(self):
  591. return self._NODESPRITE_DATA["rect"][0]
  592. @rect_x.setter
  593. def rect_x(self, v):
  594. if not isinstance(v, int):
  595. raise TypeError("Expected integer value.")
  596. self._NODESPRITE_DATA["rect"][0] = v
  597. self._NODESPRITE_ValidateRect()
  598. @property
  599. def rect_y(self):
  600. return self._NODESPRITE_DATA["rect"][1]
  601. @rect_y.setter
  602. def rect_y(self, v):
  603. if not isinstance(v, int):
  604. raise TypeError("Expected integer value.")
  605. self._NODESPRITE_DATA["rect"][1] = v
  606. self._NODESPRITE_ValidateRect()
  607. @property
  608. def rect_width(self):
  609. return self._NODESPRITE_DATA["rect"][2]
  610. @rect_width.setter
  611. def rect_width(self, v):
  612. if not isinstance(v, int):
  613. raise TypeError("Expected integer value.")
  614. self._NODESPRITE_DATA["rect"][2] = v
  615. self._NODESPRITE_ValidateRect()
  616. @property
  617. def rect_height(self):
  618. return self._NODESPRITE_DATA["rect"][3]
  619. @rect_height.setter
  620. def rect_height(self, v):
  621. if not isinstance(v, int):
  622. raise TypeError("Expected integer value.")
  623. self._NODESPRITE_DATA["rect"][3] = v
  624. self._NODESPRITE_ValidateRect()
  625. @property
  626. def center(self):
  627. r = self._NODESPRITE_DATA["rect"]
  628. return (int(r[0] + (r[2] * 0.5)), int(r[1] + (r[3] * 0.5)))
  629. @property
  630. def scale(self):
  631. return (self._NODESPRITE_DATA["scale"][0], self._NODESPRITE_DATA["scale"][1])
  632. @scale.setter
  633. def scale(self, scale):
  634. if not isinstance(scale, (list, tuple)):
  635. raise TypeError("Expected a list or tuple.")
  636. if len(scale) != 2:
  637. raise ValueError("Scale contains wrong number of values.")
  638. try:
  639. self.scale_x = scale[0]
  640. self.scale_y = scale[1]
  641. except Exception as e:
  642. raise e
  643. @property
  644. def scale_x(self):
  645. return self._NODESPRITE_DATA["scale"][0]
  646. @scale_x.setter
  647. def scale_x(self, v):
  648. if not isinstance(v, (int, float)):
  649. raise TypeError("Expected number value.")
  650. self._NODESPRITE_DATA["scale"][0] = float(v)
  651. self._NODESPRITE_DATA["scale_dirty"] = True
  652. @property
  653. def scale_y(self):
  654. return self._NODESPRITE_DATA["scale"][1]
  655. @scale_y.setter
  656. def scale_y(self, v):
  657. if not isinstance(v, (int, float)):
  658. raise TypeError("Expected number value.")
  659. self._NODESPRITE_DATA["scale"][1] = float(v)
  660. self._NODESPRITE_DATA["scale_dirty"] = True
  661. @property
  662. def image(self):
  663. return self._NODESPRITE_DATA["image"]
  664. @image.setter
  665. def image(self, src):
  666. src = src.strip()
  667. if self._NODESPRITE_DATA["image"] == src:
  668. return # Nothing to change... lol
  669. if self._NODESPRITE_DATA["image"] != "":
  670. self._NODESPRITE_DATA["surface"] = None # Clear reference to original surface.
  671. if src != "":
  672. rm = self.resource
  673. try:
  674. if not rm.has("graphic", src):
  675. rm.store("graphic", src)
  676. self._NODESPRITE_DATA["image"] = src
  677. self._NODESPRITE_DATA["surface"] = rm.get("graphic", src)
  678. if self._NODESPRITE_DATA["surface"] is None:
  679. self._NODESPRITE_DATA["image"] = ""
  680. self._NODESPRITE_DATA["rect"]=[0,0,0,0]
  681. else:
  682. # Resetting the rect to identity for the new image.
  683. surf = self._NODESPRITE_DATA["surface"]()
  684. size = surf.get_size()
  685. self._NODESPRITE_DATA["rect"]=[0,0,size[0], size[1]]
  686. except Exception as e:
  687. raise e
  688. else:
  689. self._NODESPRITE_DATA["image"] = ""
  690. self._NODESPRITE_DATA["rect"]=[0,0,0,0]
  691. def _render(self, surface):
  692. # Call the on_render() method, if any
  693. Node2D._callOnRender(self, surface)
  694. # Paint the sprite onto the surface
  695. if self._NODESPRITE_DATA["surface"] is not None:
  696. rect = self._NODESPRITE_DATA["rect"]
  697. scale = self._NODESPRITE_DATA["scale"]
  698. surf = self._NODESPRITE_DATA["surface"]()
  699. # Of course, only bother if we have a surface to work with.
  700. if surf is not None:
  701. # Do some prescaling work, if needed.
  702. if self._NODESPRITE_DATA["scale_dirty"]:
  703. self._NODESPRITE_UpdateScaleSurface(scale, surf)
  704. fsurf = surf # Initial assumption that the surface is also the "frame"
  705. # If we have a "frame" surface, however, let's get it and blit the rect into the frame surface.
  706. if "frame_surf" in self._NODESPRITE_DATA:
  707. fsurf = self._NODESPRITE_DATA["frame_surf"]
  708. fsurf.blit(surf, (0, 0), rect)
  709. # If scaling, then transform (scale) the frame surface into the scale surface and set the frame surface to the scale surface.
  710. if "scale_surf" in self._NODESPRITE_DATA:
  711. ssurf = self._NODESPRITE_DATA["scale_surf"]
  712. pygame.transform.scale(fsurf, ssurf.get_size(), ssurf)
  713. fsurf = ssurf
  714. # Place the sprite! WHEEEEE!
  715. pos = self.position
  716. surface.blit(fsurf, pos)
  717. # Call on all children
  718. Node._render(self, surface)
  719. def _NODESPRITE_UpdateScaleSurface(self, scale, surf):
  720. self._NODESPRITE_DATA["scale_dirty"] = False
  721. ssurf = None
  722. if "scale_surf" in self._NODESPRITE_DATA:
  723. ssurf = self._NODESPRITE_DATA["scale_surf"]
  724. if scale[0] == 1.0 and scale[1] == 1.0:
  725. if ssurf is not None:
  726. del self._NODESPRITE_DATA["scale_surf"]
  727. return
  728. nw = int(self.rect_width * scale[0])
  729. nh = int(self.rect_height * scale[1])
  730. if nw != ssurf.get_width() or nh != ssurf.get_height():
  731. ssurf = pygame.Surface((nw, nh), pygame.SRCALPHA, surf)
  732. ssurf.fill(pygame.Color(0,0,0,0))
  733. self._NODESPRITE_DATA["scale_surf"] = ssurf
  734. def _NODESPRITE_ValidateRect(self):
  735. if self._NODESPRITE_DATA["surface"] is None:
  736. self._NODESPRITE_DATA["rect"] = [0,0,0,0]
  737. else:
  738. rect = self._NODESPRITE_DATA["rect"]
  739. isize = (self.image_width, self.image_height)
  740. if rect[0] < 0:
  741. rect[2] += rect[0]
  742. rect[0] = 0
  743. elif rect[0] >= isize[0]:
  744. rect[0] = isize[0]-1
  745. rect[2] = 0
  746. if rect[1] < 0:
  747. rect[3] += rect[1]
  748. rect[1] = 0
  749. elif rect[1] >= isize[1]:
  750. rect[1] = isize[1]-1
  751. rect[3] = 0
  752. if rect[2] < 0:
  753. rect[2] = 0
  754. elif rect[0] + rect[2] > isize[0]:
  755. rect[2] = isize[0] - rect[0]
  756. if rect[3] < 0:
  757. rect[3] = 0
  758. elif rect[1] + rect[3] > isize[1]:
  759. rect[3] = isize[1] - rect[1]
  760. fssize = [0,0]
  761. if rect[2] > 0 and rect[3] > 0:
  762. if rect[2] < isize[0] or rect[1] < isize[1]:
  763. surf = self._NODESPRITE_DATA["surface"]()
  764. self._NODESPRITE_DATA["frame_surf"] = pygame.Surface((rect[2], rect[3]), pygame.SRCALPHA, surf)
  765. if fssize[0] > 0 and fssize[1] > 0:
  766. pass
  767. elif "frame_surf" in self._NODESPRITE_DATA:
  768. del self._NODESPRITE_DATA["frame_surf"]