mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-10-11 12:41:57 +08:00
52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
# Copyright (c) 2025 UltiMaker
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
import os
|
|
from PyQt6.QtGui import QImage
|
|
|
|
from UM.PluginRegistry import PluginRegistry
|
|
from UM.View.View import View
|
|
from UM.Scene.Selection import Selection
|
|
from UM.View.GL.OpenGL import OpenGL
|
|
from UM.i18n import i18nCatalog
|
|
|
|
catalog = i18nCatalog("cura")
|
|
|
|
|
|
class PaintView(View):
|
|
"""View for model-painting."""
|
|
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
self._paint_shader = None
|
|
self._paint_texture = None
|
|
|
|
# FIXME: When the texture UV-unwrapping is done, these two values will need to be set to a proper value (suggest 4096 for both).
|
|
self._tex_width = 512
|
|
self._tex_height = 512
|
|
|
|
def _checkSetup(self):
|
|
if not self._paint_shader:
|
|
shader_filename = os.path.join(PluginRegistry.getInstance().getPluginPath("PaintTool"), "paint.shader")
|
|
self._paint_shader = OpenGL.getInstance().createShaderProgram(shader_filename)
|
|
if not self._paint_texture:
|
|
self._paint_texture = OpenGL.getInstance().createTexture(self._tex_width, self._tex_height)
|
|
self._paint_shader.setTexture(0, self._paint_texture)
|
|
|
|
def addStroke(self, stroke_image: QImage, start_x: int, start_y: int) -> None:
|
|
self._paint_texture.setSubImage(stroke_image, start_x, start_y)
|
|
|
|
def getUvTexDimensions(self):
|
|
return self._tex_width, self._tex_height
|
|
|
|
def beginRendering(self) -> None:
|
|
renderer = self.getRenderer()
|
|
self._checkSetup()
|
|
paint_batch = renderer.createRenderBatch(shader=self._paint_shader)
|
|
renderer.addRenderBatch(paint_batch)
|
|
|
|
node = Selection.getAllSelectedObjects()[0]
|
|
if node is None:
|
|
return
|
|
paint_batch.addItem(node.getWorldTransformation(copy=False), node.getMeshData(), normal_transformation=node.getCachedNormalMatrix())
|