oldschool-toontown/libotp/nametag/ChatBalloon.py

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