Legend of the Gold Box... A game written for the LOWREZJAM 2018 game jam
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

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"]