229 lines
7.3 KiB
Python
229 lines
7.3 KiB
Python
from direct.directnotify import DirectNotifyGlobal
|
|
from panda3d.core import *
|
|
|
|
from . import NametagGlobals
|
|
|
|
|
|
class ChatBalloon:
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('ChatBalloon')
|
|
|
|
def __init__(self, node=None):
|
|
if isinstance(node, NodePath):
|
|
node = node.node()
|
|
|
|
self.m_copy_node = None
|
|
self.m_top_node = None
|
|
self.m_top_mat = None
|
|
self.m_middle_node = None
|
|
self.m_middle_mat = None
|
|
self.m_bottom_node = None
|
|
self.m_bottom_mat = None
|
|
|
|
self.m_hscale = 0
|
|
self.m_text_height = 0
|
|
self.m_text_frame = Vec4(0)
|
|
|
|
self.scan(node)
|
|
|
|
@staticmethod
|
|
def find_geom_node(node):
|
|
if node.isGeomNode():
|
|
return node
|
|
|
|
for i in range(node.getNumChildren()):
|
|
n = ChatBalloon.find_geom_node(node.getChild(i))
|
|
if n:
|
|
return n
|
|
|
|
return None
|
|
|
|
@staticmethod
|
|
def find_middle_geom(node):
|
|
if not node.getNumChildren():
|
|
return None
|
|
|
|
child = None
|
|
for i in range(node.getNumChildren()):
|
|
child = node.getChild(i)
|
|
if child.getName() == 'middle':
|
|
return n
|
|
|
|
n = ChatBalloon.find_middle_geom(child)
|
|
if n:
|
|
return n
|
|
|
|
return ChatBalloon.find_geom_node(child)
|
|
|
|
def scan(self, node):
|
|
if node.getName() == 'chatBalloon':
|
|
return self.scan_balloon(node)
|
|
|
|
for i in range(node.getNumChildren()):
|
|
if self.scan(node.getChild(i)):
|
|
return True
|
|
|
|
return False
|
|
|
|
def scan_balloon(self, node):
|
|
self.m_copy_node = node.copySubgraph()
|
|
|
|
for i in range(node.getNumChildren()):
|
|
child = node.getChild(i)
|
|
if child.getName() == 'top':
|
|
self.m_top_node = child
|
|
self.m_top_mat = child.getTransform().getMat()
|
|
|
|
elif child.getName() == 'middle':
|
|
self.m_middle_node = child
|
|
self.m_middle_mat = child.getTransform().getMat()
|
|
|
|
elif child.getName() == 'bottom':
|
|
self.m_bottom_node = child
|
|
self.m_bottom_mat = child.getTransform().getMat()
|
|
|
|
if self.m_top_node and self.m_middle_node and self.m_bottom_node:
|
|
return True
|
|
|
|
else:
|
|
self.notify.warning('ChatBalloon geometry does not include top, middle, and bottom nodes.')
|
|
return False
|
|
|
|
def generate(self, text, font, wordwrap, text_color, balloon_color, for_3d,
|
|
has_draw_order, draw_order, page_button, space_for_button,
|
|
reversed, new_button): # new_button is a pointer, let's use a list hack here
|
|
chat_node = PandaNode('chat')
|
|
chat_node.setAttrib(CullFaceAttrib.make(0))
|
|
text_node = NametagGlobals.getTextNode()
|
|
text_node.setFont(font)
|
|
text_node.setWordwrap(wordwrap)
|
|
text_node.setAlign(TextNode.ALeft)
|
|
text_node.setText(text)
|
|
|
|
v116 = NametagGlobals._balloon_text_origin[0]
|
|
if reversed:
|
|
v116 = v116 + 9.0
|
|
|
|
v27 = (text_node.getRight() - text_node.getLeft()) * 0.11111111
|
|
self.m_hscale = v27
|
|
|
|
if v27 < 0.25:
|
|
self.m_hscale = 0.25
|
|
text_node.setAlign(TextNode.ACenter)
|
|
|
|
v29 = v116
|
|
if not reversed:
|
|
v116 = v29 + 4.5
|
|
|
|
else:
|
|
v116 = v29 - 4.5
|
|
|
|
elif reversed:
|
|
self.m_hscale = -self.m_hscale
|
|
|
|
self.m_text_frame = text_node.getCardActual()
|
|
_space = 0.2 if space_for_button else 0.0
|
|
num_rows = max(1, text_node.getNumRows())
|
|
_num_rows = num_rows
|
|
line_h = text_node.getFont().getLineHeight()
|
|
|
|
num_rows_minus_1 = num_rows - 1
|
|
subgraph_copy_mat = Mat4(self.m_hscale, 0, 0, 0,
|
|
0, 1.0, 0, 0,
|
|
0, 0, 1.0, 0,
|
|
0, 0, 0, 1.0)
|
|
text_h = _num_rows * line_h + _space
|
|
self.m_text_height = text_h
|
|
|
|
v132 = num_rows_minus_1 * line_h + _space
|
|
|
|
middle_mat = Mat4(1, 0, 0, 0,
|
|
0, 1, 0, 0,
|
|
0, 0, text_h, 0,
|
|
0, 0, 0, 1) * self.m_middle_mat
|
|
top_mat = Mat4(1, 0, 0, 0,
|
|
0, 1, 0, 0,
|
|
0, 0, 1, 0,
|
|
0, 0, text_h - 1.0, 1) * self.m_top_mat
|
|
|
|
v137 = v116 * self.m_hscale
|
|
v138 = 0.0
|
|
v139 = NametagGlobals._balloon_text_origin[2] + v132 + 0.2
|
|
|
|
self.m_text_frame += Vec4(v137, v137, v139, v139)
|
|
|
|
'''
|
|
Correct code:
|
|
Python won't let us edit this transform, we'll have to copy it all
|
|
|
|
if self.m_top_node:
|
|
self.m_top_node.setTransform(TransformState.makeMat(top_mat))
|
|
|
|
if self.m_middle_node:
|
|
self.m_middle_node.setTransform(TransformState.makeMat(middle_mat))
|
|
|
|
subgraph_copy = self.m_copy_node.copySubgraph()
|
|
chat_node.addChild(subgraph_copy)
|
|
subgraph_copy.setTransform(TransformState.makeMat(subgraph_copy_mat))
|
|
'''
|
|
|
|
# BEGIN PYTHON CODE
|
|
subgraph_copy = self.m_copy_node.copySubgraph()
|
|
NodePath.anyPath(subgraph_copy).find('**/top').node().setTransform(TransformState.makeMat(top_mat))
|
|
NodePath.anyPath(subgraph_copy).find('**/middle').node().setTransform(TransformState.makeMat(middle_mat))
|
|
|
|
chat_node.addChild(subgraph_copy)
|
|
subgraph_copy.setTransform(TransformState.makeMat(subgraph_copy_mat))
|
|
# END PYTHON CODE
|
|
|
|
if has_draw_order:
|
|
bin = config.GetString('nametag-fixed-bin', 'fixed')
|
|
subgraph_copy.setAttrib(CullBinAttrib.make(bin, draw_order))
|
|
|
|
subgraph_copy.setAttrib(ColorAttrib.makeFlat(balloon_color))
|
|
if balloon_color[3] != 1.0:
|
|
subgraph_copy.setAttrib(TransparencyAttrib.make(1))
|
|
|
|
reducer = SceneGraphReducer()
|
|
reducer.applyAttribs(subgraph_copy)
|
|
reducer.flatten(chat_node, -1)
|
|
|
|
generated_text = text_node.generate()
|
|
if for_3d:
|
|
v86 = ChatBalloon.find_middle_geom(chat_node)
|
|
if not v86:
|
|
v86 = ChatBalloon.find_geom_node(chat_node)
|
|
|
|
v86.setEffect(DecalEffect.make())
|
|
|
|
else:
|
|
v86 = chat_node
|
|
if has_draw_order:
|
|
bin = config.GetString('nametag-fixed-bin', 'fixed')
|
|
generated_text.setAttrib(CullBinAttrib.make(bin, draw_order + 1))
|
|
|
|
np = NodePath.anyPath(v86)
|
|
v144 = np.attachNewNode(generated_text)
|
|
v144.setPos((v137, v138, v139))
|
|
v144.setColor(text_color)
|
|
v144.setY(-0.01) # Panda3D 1.10 hack to prevent z-fighting.
|
|
if text_color[3] != 1.0:
|
|
v144.setTransparency(1)
|
|
|
|
if page_button:
|
|
v116 = ModelNode('button')
|
|
new_button[0] = np.attachNewNode(v116)
|
|
button_copy = page_button.copyTo(new_button[0])
|
|
if reversed:
|
|
button_copy.setPos(self.m_hscale * 1.7, 0, 1.8)
|
|
|
|
else:
|
|
button_copy.setPos(self.m_hscale * 9.0, 0, 1.8)
|
|
|
|
button_copy.setScale(8.0, 8.0, 8.0)
|
|
button_copy.setY(-0.01) # Panda3D 1.10 hack to prevent z-fighting.
|
|
|
|
reducer = SceneGraphReducer()
|
|
reducer.applyAttribs(generated_text)
|
|
reducer.flatten(chat_node, 1)
|
|
|
|
return chat_node
|