コード例 #1
0
    def combineChannels(self, switchToMain, *args):
        currentChannel = None
        for mesh in getMeshIter():
            currentChannel = mesh.currentColorSetName()

        validateChannels()

        iter = getMeshIter()
        # combine SUB_COLOR_SETS' into MAIN_COLOR_SET
        for mesh in iter:
            rColors = mesh.getVertexColors(R_CHANNEL)
            gColors = mesh.getVertexColors(G_CHANNEL)
            bColors = mesh.getVertexColors(B_CHANNEL)
            aColors = mesh.getVertexColors(A_CHANNEL)

            colors = om.MColorArray(mesh.numVertices, om.MColor((1, 1, 1, 1)))
            for i in xrange(mesh.numVertices):
                colors[i] = om.MColor(
                    (rColors[i].r, gColors[i].g, bColors[i].b, aColors[i].r))

            mesh.setCurrentColorSetName(MAIN_COLOR_SET)
            mesh.setVertexColors(colors, getIncrementIter(mesh.numVertices))
            mesh.syncObject()

        if switchToMain:
            currentChannel = MAIN_COLOR_SET

        if currentChannel != None:
            for mesh in getMeshIter():
                mesh.setCurrentColorSetName(currentChannel)
            self.updateUi()
コード例 #2
0
def fillColor(color, colorName):
    validateChannels()
    channel = getCurrentColorSetNames()[0]
    colorName = COLOR_SET_NAMES[channel] if colorName == None else colorName
    warnMessage = "Fill %s channel's color to %s?" % (channel, colorName)
    resetAll = channel == MAIN_COLOR_SET
    if resetAll:
        warnMessage = "Fill all channels color to %s?" % (colorName)

    result = cmds.confirmDialog(title='Warning',
                                message=warnMessage,
                                button=['OK', 'Cancel'],
                                defaultButton='OK',
                                cancelButton='Cancel',
                                dismissString='Cancel')

    if result != "OK":
        return

    for mesh in getMeshIter():
        colors = mesh.getVertexColors(MAIN_COLOR_SET)
        numColors = len(colors)
        if resetAll:
            for colorSet in RESERVED_SETS:
                defaultColor = COLOR_SET_COLORS[colorSet]
                newColor = [a * b for a, b in zip(color, defaultColor)]
                newColors = om.MColorArray(numColors, om.MColor(newColor))
                mesh.setCurrentColorSetName(colorSet)
                mesh.setVertexColors(newColors, getIncrementIter(numColors))
        else:
            defaultColor = COLOR_SET_COLORS[channel]
            newColor = [a * b for a, b in zip(color, defaultColor)]
            newColors = om.MColorArray(numColors, om.MColor(newColor))
            mesh.setVertexColors(newColors, getIncrementIter(numColors))
コード例 #3
0
    def updateDG(self):
        '''Retrieve and prepare data for drawing'''

        self.resolution = self.get_resolution()
        self.width = self.resolution
        self.height = self.resolution

        plug = om.MPlug(self.obj, Swatch.inColor)
        in_plugs = plug.connectedTo(True, False)
        if in_plugs:
            in_in_plugs = in_plugs[0].connectedTo(True, False)
            color_plug = in_in_plugs[0] if in_in_plugs else in_plugs[0]

            if color_plug.attribute().apiTypeStr != 'kAttribute3Float':
                self.color = om.MColor([1, 1, 1])
                self.set_texture(None)

            color_node = color_plug.node()
            if color_node.apiTypeStr == 'kFileTexture':
                depfn = om.MFnDependencyNode(color_node)
                file = depfn.findPlug('fileTextureName', False).asString()
                self.set_texture(
                    TEXTURE_MANAGER.acquireTexture('', color_plug, self.width,
                                                   self.height, False))
            else:
                self.set_texture(
                    TEXTURE_MANAGER.acquireTexture('', color_plug, self.width,
                                                   self.height, False))
        else:
            self.set_texture(None)
            self.color = om.MColor(self.get_color())
コード例 #4
0
ファイル: DDHUD.py プロジェクト: DangoWang/DDHUD
 def prepareForDraw(self, obj_path, camera_path, frame_context, old_data):
     """
     """
     data = old_data
     if not isinstance(data, DDHUDData):
         data = DDHUDData()
     fnDagNode = om.MFnDagNode(obj_path)
     # 需要先获取数据
     data.camera_name = fnDagNode.findPlug("camera", False).asString()
     data.text = fnDagNode.findPlug("text", False).asString()
     data.font_name = fnDagNode.findPlug("fontName", False).asString()
     data.x = fnDagNode.findPlug("X", False).asFloat()
     data.y = fnDagNode.findPlug("Y", False).asFloat()
     r = fnDagNode.findPlug("fontColorR", False).asFloat()
     g = fnDagNode.findPlug("fontColorG", False).asFloat()
     b = fnDagNode.findPlug("fontColorB", False).asFloat()
     a = fnDagNode.findPlug("fontAlpha", False).asFloat()
     data.font_color = om.MColor((r, g, b, a))
     data.font_scale = fnDagNode.findPlug("fontScale", False).asFloat()
     data.border = fnDagNode.findPlug("border", False).asBool()
     r = fnDagNode.findPlug("borderColorR", False).asFloat()
     g = fnDagNode.findPlug("borderColorG", False).asFloat()
     b = fnDagNode.findPlug("borderColorB", False).asFloat()
     a = fnDagNode.findPlug("borderAlpha", False).asFloat()
     data.border_color = om.MColor((r, g, b, a))
     data.border_w = fnDagNode.findPlug("borderWidth", False).asFloat()
     data.border_h = fnDagNode.findPlug("borderHeight", False).asFloat()
     return data
コード例 #5
0
    def prepareForDraw(self, objPath, cameraPath, frameContext, oldData):
        depNode = om.MFnDependencyNode(objPath.node())
        obj = depNode.userNode()

        isSelected = isPathSelected(objPath)
        self.xray = obj.xray

        plug = om.MPlug(objPath.node(), zRigHandle.colorAttr)
        self.color = om.MColor(om.MFnNumericData(plug.asMObject()).getData())

        alpha = om.MPlug(objPath.node(), zRigHandle.alphaAttr).asFloat()
        self.color.a = alpha

        if isSelected:
            self.borderColor = omr.MGeometryUtilities.wireframeColor(objPath)
        else:
            plug = om.MPlug(objPath.node(), zRigHandle.borderColorAttr)
            self.borderColor = om.MColor(
                om.MFnNumericData(plug.asMObject()).getData())

            # If no color has been set and we're on the default of (-1,-1,-1), use the main color,
            # so in the common case where you want to use the same color you don't have to set both.
            if self.borderColor.r == -1 and self.borderColor.g == -1 and self.borderColor.b == -1:
                self.borderColor = om.MColor(self.color)

            self.borderColor.a = om.MPlug(
                objPath.node(), zRigHandle.borderAlphaAttr).asFloat()

        self.shape = obj.getShape()
コード例 #6
0
    def addUIDrawables(self, objPath, drawManager, frameContext, data): 
        locatordata = data
        status = OpenMayaUI.M3dView.displayStatus(objPath)

        if not isinstance(locatordata, MntLocatorNodeData):
            return

        elif OpenMayaAnim.MAnimControl.isPlaying() == True:
            return
            
        elif OpenMayaRender.MFrameContext.inUserInteraction() == True and data.interactiveRefresh == False:
            return

        drawManager.beginDrawable(OpenMayaRender.MUIDrawManager.kSelectable)

        if data.iconType < 6:
            color = OpenMaya.MColor((0.1, 0.2, 0.4, data.areaOpacity))
            drawManager.setColor(color)            
            drawManager.box(data.localPosition, OpenMaya.MVector(0, 1, 0), OpenMaya.MVector(1, 0, 0), data.worldSizeX, data.worldSizeY, data.worldSizeZ, True)

            drawManager.beginDrawInXray()
            drawManager.setTexture(data.texture)
            drawManager.rect2d(OpenMaya.MPoint(data.screenPos[0], data.screenPos[1]), OpenMaya.MVector(0, 1), data.screenSize, data.screenSize, True)
            drawManager.endDrawInXray()

            if data.label != '':
                drawManager.setColor(data.color)      
                drawManager.setFontSize(11)      
                drawManager.text2d(OpenMaya.MPoint(data.screenPos[0] + 16, data.screenPos[1]), data.label, alignment = OpenMayaRender.MUIDrawManager.kLeft)

        if data.iconType == 6:
            drawManager.setColor(data.color)        
            drawManager.sphere(data.localPosition, data.size, 16, 8, True)

        if data.iconType == 7:
            drawManager.setColor(data.color)        
            drawManager.circle(data.localPosition, data.upVector, data.size, 32, True)

        if data.iconType == 8:
            drawManager.setColor(data.color)  
            drawManager.mesh(4, self.pointsArray, None, None, self.indicesArray, None)

        if data.hierarchicalLink == True:
            if data.pointArray.__len__() > 1:
                drawManager.beginDrawInXray()
                drawManager.setColor(OpenMaya.MColor((data.lineColor[0], data.lineColor[1], data.lineColor[2],1)))
                drawManager.setLineWidth(data.lineWidth)

                if data.dottedLine == False:
                    drawManager.line(data.pointArray[0], data.pointArray[1])
                else:
                    for i in range(0, data.pointArray.__len__(), 2):
                        drawManager.line(data.pointArray[i], data.pointArray[i + 1]) 
                        
                drawManager.endDrawInXray()
                           
        drawManager.endDrawable()
コード例 #7
0
    def __init__(self):
        super(HelperData, self).__init__(False)  ## don't delete after draw

        self.surface_color = om.MColor([1.0, 0.0, 0.0, 0.25])
        self.wire_color = om.MColor([1.0, 0.0, 0.0])

        self.points = om.MPointArray()
        self.triangles_indices = []
        self.lines_indices = []
コード例 #8
0
    def prepareForDraw(self, obj_path, camera_path, frame_context, old_data):
        """
        """
        data = old_data
        if not isinstance(data, ZShotMaskData):
            data = ZShotMaskData()

        fnDagNode = om.MFnDagNode(obj_path)

        data.camera_name = fnDagNode.findPlug("camera", False).asString()

        data.text_fields = []
        for i in range(0, len(ZShotMaskLocator.TEXT_ATTRS), 2):
            data.text_fields.append(
                fnDagNode.findPlug(ZShotMaskLocator.TEXT_ATTRS[i],
                                   False).asString())

        counter_padding = fnDagNode.findPlug("counterPadding", False).asInt()
        if counter_padding < 1:
            counter_padding = 1
        elif counter_padding > 6:
            counter_padding = 6

        current_time = int(cmds.currentTime(q=True))
        counter_position = fnDagNode.findPlug("counterPosition", False).asInt()
        if counter_position > 0 and counter_position <= len(
                ZShotMaskLocator.TEXT_ATTRS) / 2:
            data.text_fields[counter_position - 1] = "{0}".format(
                str(current_time).zfill(counter_padding))

        data.text_padding = fnDagNode.findPlug("textPadding", False).asInt()

        data.font_name = fnDagNode.findPlug("fontName", False).asString()

        r = fnDagNode.findPlug("fontColorR", False).asFloat()
        g = fnDagNode.findPlug("fontColorG", False).asFloat()
        b = fnDagNode.findPlug("fontColorB", False).asFloat()
        a = fnDagNode.findPlug("fontAlpha", False).asFloat()
        data.font_color = om.MColor((r, g, b, a))

        data.font_scale = fnDagNode.findPlug("fontScale", False).asFloat()

        r = fnDagNode.findPlug("borderColorR", False).asFloat()
        g = fnDagNode.findPlug("borderColorG", False).asFloat()
        b = fnDagNode.findPlug("borderColorB", False).asFloat()
        a = fnDagNode.findPlug("borderAlpha", False).asFloat()
        data.border_color = om.MColor((r, g, b, a))

        data.border_scale = fnDagNode.findPlug("borderScale", False).asFloat()

        data.top_border = fnDagNode.findPlug("topBorder", False).asBool()
        data.bottom_border = fnDagNode.findPlug("bottomBorder", False).asBool()

        return data
コード例 #9
0
    def addUIDrawables(self, objPath, drawManager, frameContext, data):
        if not isinstance(data, GuideSpineData):
            return

        normalVector = OpenMaya.MVector([1.0, 0.0, 0.0])
        bounding = OpenMaya.MBoundingBox(data.boundingBoxCorner1,
                                         data.boundingBoxCorner2)
        up = OpenMaya.MVector.kYaxisVector
        width = bounding.depth / 2.0
        height = bounding.height / 2.0

        dormantBorderColor = OpenMaya.MColor([0.0, 0.0, 0.8, 1.0])
        leadBorderColor = OpenMaya.MColor([1.0, 1.0, 1.0, 1.0])
        activeBorderColor = OpenMaya.MColor([0.4, 0.8, 1.0, 1.0])
        lineColor = OpenMaya.MColor([1.0, 1.0, 1.0, 1.0])
        fillColor = OpenMaya.MColor([0.0, 0.0, 1.0, 1.0])
        fillColor.a = 0.4
        borderWidth = 1.0
        lineWidth = 2.0

        displayStatus = OpenMayaRender.MGeometryUtilities.displayStatus(
            objPath)
        selected = True
        if displayStatus == util.DisplayStatus.Lead:
            borderColor = leadBorderColor
            borderWidth = 2.0
        elif displayStatus == util.DisplayStatus.Active:
            borderColor = activeBorderColor
            borderWidth = 2.0
        else:
            # kActiveAffected, kDormant, kHilite.
            borderColor = dormantBorderColor
            borderWidth = 1.0
            selected = False

        drawManager.beginDrawable()

        drawManager.setColor(borderColor)
        drawManager.setLineWidth(borderWidth)
        drawManager.rect(bounding.center,
                         up,
                         normalVector,
                         width,
                         height,
                         filled=False)
        # drawManager.setColor(fillColor)
        # drawManager.rect(bounding.center, up, normalVector, width, height, filled=True)

        # drawManager.setColor(OpenMaya.MColor([1.0, 1.0, 1.0, 1.0]))
        # for position in data.outputPoints:
        #     drawManager.sphere(position, 0.5, filled=False)

        drawManager.endDrawable()
コード例 #10
0
    def __init__(self):

        """
        Initialize camera render ui manager draw request.
        """

        self.uiDraw = False
        self.uiType = constants.kText
        self.uiSize = 1.0
        self.uiColor = OpenMaya.MColor([0.0, 0.0, 0.0])

        self.uiResolutionGate = constants.kViewportGate
        self.uiDrawResolutionGate = True

        self.region = Canvas(0, 0, 0, 0)
        self.uiRegion = OpenMaya.MVector(0.0, 0.0, 0.0)
        self.uiRegionPosition = OpenMaya.MPoint(0.0, 0.0, 0.0)
        self.uiRegionColor = OpenMaya.MColor([0.0, 0.0, 0.0])
        self.uiTextBackgroundColor = None
        self.uiRegionIsFilled = False
        self.uiDrawRegion = False

        self.uiHorisontalAttach = constants.kAttachHorizontalLeft
        self.uiVerticalAttach = constants.kAttachVerticalTop

        self.uiHorisontalAlignment = constants.kHorizontalAlignmentLeft
        self.uiVerticalAlignment = constants.kVerticalAlignmentTop

        self.uiRadius = 1.0
        self.uiFilled = 1.0
        self.uiLineStyle = constants.kLineSolid
        self.uiLineWidth = 2.0

        self.uiPositionList = []
        self.uiRegionPositionList = []

        self.uiText = "Text"
        self.uiTextDynamic = False
        self.uiFitToResolutionGate = True
        self.uiFontStyle = None
        self.uiFontStyleSize = OpenMayaRender.MUIDrawManager.kDefaultFontSize
        self.uiFontStyleStretch = OpenMayaRender.MUIDrawManager.kStretchUnstretched
        self.uiFontStyleWeight = constants.kFontStyleWeightLight
        self.uiFontStyleIncline = constants.kFontStyleInclineNormal
        self.uiFontStyleLine = constants.kFontStyleLineNone

        self.file = ""
        self.year = ""
        self.month = ""
        self.day = ""
        self.hour = ""
        self.minute = ""
コード例 #11
0
class ValueBarLocatorData(OpenMaya.MUserData):
    """Class for handling node attribute values.

    Instanced to hold cached attribute values.
    Class globals used as attribute default values.
    """

    # default values
    inputValue = 0.0

    barAim = (0.0, 1.0, 0.0)
    barUp = (0.0, 0.0, 1.0)

    barWidth = 0.05
    barColor = (0.0, 1.0, 0.0)
    barAlpha = 0.5

    barHasMin = True
    barMinValue = 0.0
    barMinVisibility = True
    barMinColor = (0.0, 0.0, 1.0)

    barHasMax = True
    barMaxValue = 1.0
    barMaxVisibility = True
    barMaxColor = (1.0, 0.0, 0.0)

    # calculated values
    start_point = OpenMaya.MPoint(0.0, 0.0, 0.0)
    aim_vector = OpenMaya.MVector()
    up_vector = OpenMaya.MVector()
    normal_vector = OpenMaya.MVector()

    start_value = 0.0
    end_value = 0.0
    bar_height = 0.0
    bar_center_point = OpenMaya.MPoint()

    min_bar_height = 0.0
    min_center_point = OpenMaya.MPoint()

    max_bar_height = 0.0
    max_center_point = OpenMaya.MPoint()

    bar_m_color = OpenMaya.MColor()
    min_m_color = OpenMaya.MColor()
    max_m_color = OpenMaya.MColor()

    def __init__(self):
        OpenMaya.MUserData.__init__(self, False)  # don't delete after draw
コード例 #12
0
 def draw(self, view, path, style, status):
     """
     Legacy viewport draw function
     if in legacy - once on init, twice on frame change, selection, navigation
     """
     print "drawing in legacy..."
     # Getting the OpenGL Renderer 
     glRenderer  = OpenMayaRenderv1.MHardwareRenderer.theRenderer()
     # Getting all classes from the renderer
     glFT = glRenderer.glFunctionTable()
     
     view.beginGL()
     glFT.glPushAttrib(OpenMayaRenderv1.MGL_ALL_ATTRIB_BITS)
     glFT.glPushMatrix()
     glFT.glDepthRange(0, 0)
     glFT.glEnable(OpenMayaRenderv1.MGL_LINE_SMOOTH)
     glFT.glEnable(OpenMayaRenderv1.MGL_POINT_SMOOTH)
     glFT.glEnable(OpenMayaRenderv1.MGL_BLEND)
     glFT.glDisable(OpenMayaRenderv1.MGL_LIGHTING)
     glFT.glBlendFunc(OpenMayaRenderv1.MGL_SRC_ALPHA, OpenMayaRenderv1.MGL_ONE_MINUS_SRC_ALPHA)
     
     # Setting a color for Viewport draw
     color1 = OpenMaya.MColor(0.0, 1.0, 0.0, 1.0)
     color2 = OpenMaya.MColor(1.0, 1.0, 1.0, 1.0)
     
     # Draw points
     glFT.glPointSize(12.0)
     glFT.glBegin(OpenMayaRenderv1.MGL_POINTS)
     glFT.glColor4f(color1[0], color1[1], color1[2], 1)
     for point in self.points:
         glFT.glVertex3f(point[0], point[1], point[2])
     glFT.glEnd()
     
     # Draw lines
     glFT.glLineWidth(3.0)
     glFT.glBegin(OpenMayaRenderv1.MGL_LINE_STRIP)
     glFT.glColor4f(color2[0], color2[1], color2[2], 1)
     for point in self.points:
         glFT.glVertex3f(point[0], point[1], point[2])
     glFT.glEnd()
     
     # Wrap
     glFT.glDisable(OpenMayaRenderv1.MGL_BLEND)
     glFT.glDisable(OpenMayaRenderv1.MGL_LINE_SMOOTH)
     glFT.glDisable(OpenMayaRenderv1.MGL_POINT_SMOOTH)
     glFT.glEnable(OpenMayaRenderv1.MGL_LIGHTING)
     glFT.glPopMatrix()
     glFT.glPopAttrib()
     view.endGL()
コード例 #13
0
	def draw(self, view, path, style, status):
		## Get the size
		##
		thisNode = self.thisMObject()
		plug = om.MPlug( thisNode, footPrint.size )
		sizeVal = plug.asMDistance()
		multiplier = sizeVal.asCentimeters()

		global sole, soleCount
		global heel, heelCount

		view.beginGL()

		## drawing in VP1 views will be done using V1 Python APIs:
		import maya.OpenMayaRender as v1omr
		glRenderer = v1omr.MHardwareRenderer.theRenderer()
		glFT = glRenderer.glFunctionTable()

		if ( style == omui.M3dView.kFlatShaded ) or ( style == omui.M3dView.kGouraudShaded ):
			## Push the color settings
			##
			glFT.glPushAttrib( v1omr.MGL_CURRENT_BIT )

			# Show both faces
			glFT.glDisable( v1omr.MGL_CULL_FACE )

			if status == omui.M3dView.kActive:
				view.setDrawColor( 13, omui.M3dView.kActiveColors )
			else:
				view.setDrawColor( 13, omui.M3dView.kDormantColors )

			glFT.glBegin( v1omr.MGL_TRIANGLE_FAN )
			for i in range(soleCount-1):
				glFT.glVertex3f( sole[i][0] * multiplier, sole[i][1] * multiplier, sole[i][2] * multiplier )
			glFT.glEnd()

			glFT.glBegin( v1omr.MGL_TRIANGLE_FAN )
			for i in range(heelCount-1):
				glFT.glVertex3f( heel[i][0] * multiplier, heel[i][1] * multiplier, heel[i][2] * multiplier )
			glFT.glEnd()

			glFT.glPopAttrib()

		## Draw the outline of the foot
		##
		glFT.glBegin( v1omr.MGL_LINES )
		for i in range(soleCount-1):
			glFT.glVertex3f( sole[i][0] * multiplier, sole[i][1] * multiplier, sole[i][2] * multiplier )
			glFT.glVertex3f( sole[i+1][0] * multiplier, sole[i+1][1] * multiplier, sole[i+1][2] * multiplier )

		for i in range(heelCount-1):
			glFT.glVertex3f( heel[i][0] * multiplier, heel[i][1] * multiplier, heel[i][2] * multiplier )
			glFT.glVertex3f( heel[i+1][0] * multiplier, heel[i+1][1] * multiplier, heel[i+1][2] * multiplier )
		glFT.glEnd()

		view.endGL()

		## Draw the name of the footPrint
		view.setDrawColor( om.MColor( (0.1, 0.8, 0.8, 1.0) ) )
		view.drawText( "Footprint", om.MPoint( 0.0, 0.0, 0.0 ), omui.M3dView.kCenter )
コード例 #14
0
    def prepareForDraw(self, objPath, cameraPath, frameContext, oldData):
        data = oldData
        if not isinstance(data, GuideHandleData):
            data = GuideHandleData()

        controlNode = objPath.node()

        data.childPositions = OpenMaya.MPointArray()
        plug = OpenMaya.MPlug(controlNode, self.GuideClass.childPosition)
        fnData = OpenMaya.MFnNumericData()
        for i in xrange(plug.numElements()):
            handle = plug.elementByPhysicalIndex(i)
            fnData.setObject(handle.asMObject())
            pt = OpenMaya.MPoint(fnData.getData())
            data.childPositions.append(pt)

        plug.setAttribute(self.GuideClass.handleMatrix)
        handleMatrix = OpenMaya.MFnMatrixData(plug.asMObject()).matrix()
        data.inverseMatrix = handleMatrix.inverse()

        plug.setAttribute(self.GuideClass.handleColor)
        colorData = OpenMaya.MFnNumericData(plug.asMObject())
        data.handleColor = OpenMaya.MColor(colorData.getData())

        plug.setAttribute(self.GuideClass.jointMatrix)
        jointMatrix = OpenMaya.MFnMatrixData(plug.asMObject()).matrix()
        data.jointMatrix = jointMatrix * data.inverseMatrix

        plug.setAttribute(self.GuideClass.guide)
        data.hasGuide = plug.isConnected

        return data
コード例 #15
0
def get_falloff(vertex_distance_list, vertex_value_list):
    """
    :param vertex_distance_list: float list with distances of nearby vertices. Must be same length as vertex_value list.
    :param vertex_value_list: Can be and array of MVectors, MColors, Floats.
    :return: Sum of the vertex_value_list with weight given to closer vertices.
    """

    if sum(vertex_distance_list) == 0:
        distances_percentages = [(x / 1) for x in vertex_distance_list]
    else:
        distances_percentages = [(x / sum(vertex_distance_list))
                                 for x in vertex_distance_list]
    """Setting this so that the greater the distance value, the less weight it's vertex_value adds."""
    inverse_distance_percentage = [1 - x for x in distances_percentages]
    """Normalizing the inverse distance to multiply with the vertex values"""
    normalize_inverse_percentage = [
        x / sum(inverse_distance_percentage)
        for x in inverse_distance_percentage
    ]
    """Multiply each vertex value with normalized inverse percentage and then gather the sum of that list, this will be 
    the new sampled value. Closer verts with contribute more to the sum."""
    new_vertex_value = \
        [vertex_value_list[x] * normalize_inverse_percentage[x] for x in range(len(normalize_inverse_percentage))]
    value_sum = om2.MColor((0, 0, 0, 1))
    for i in new_vertex_value:

        value_sum += i

    return value_sum
コード例 #16
0
ファイル: footPrintNode.py プロジェクト: jonntd/mayadev-1
    def addUIDrawables(self, objPath, drawManager, frameContext, data):
        locatordata = data
        if not isinstance(locatordata, footPrintData):
            return

        drawManager.beginDrawable()

        ##Draw the foot print solid/wireframe
        drawManager.setColor(locatordata.fColor)
        drawManager.setDepthPriority(5)

        if (frameContext.getDisplayStyle() & omr.MFrameContext.kGouraudShaded):
            drawManager.mesh(omr.MGeometry.kTriangles,
                             locatordata.fSoleTriangleList)
            drawManager.mesh(omr.MGeometry.kTriangles,
                             locatordata.fHeelTriangleList)

        drawManager.mesh(omr.MUIDrawManager.kClosedLine,
                         locatordata.fSoleLineList)
        drawManager.mesh(omr.MUIDrawManager.kClosedLine,
                         locatordata.fHeelLineList)

        ## Draw a text "Foot"
        pos = om.MPoint(0.0, 0.0, 0.0)  ## Position of the text
        textColor = om.MColor((0.1, 0.8, 0.8, 1.0))  ## Text color

        drawManager.setColor(textColor)
        drawManager.setFontSize(omr.MUIDrawManager.kSmallFontSize)
        drawManager.text(pos, "Footprint", omr.MUIDrawManager.kCenter)

        drawManager.endDrawable()
コード例 #17
0
    def addUIDrawables(self, objPath, drawManager, frameContext, data):
        '''This function is the draw call.

        Args:
            objPath: The DAG path to the locator node
            drawManager: The MUIDrawManager object used for drawing
            frameContext: A frame context
            data: The data required to draw
        '''

        drawManager.beginDrawable()

        # Draw the foot print solid/wireframe
        # drawManager.setColor(locatordata.fColor)
        colour = om.MColor()
        colour.g = 1.0
        drawManager.setColor(colour)
        drawManager.setDepthPriority(5)

        if (frameContext.getDisplayStyle() & omr.MFrameContext.kGouraudShaded):
            for centre, radius in zip(self.m_centres, self.m_radii):
                drawManager.sphere(centre, radius, True)

        else:
            for centre, radius in zip(self.m_centres, self.m_radii):
                drawManager.sphere(centre, radius, False)

        drawManager.endDrawable()
コード例 #18
0
    def prepareForDraw(self, obj, cam, frame_context, data, verbose=0):
        """Retrieve data cache (create if does not exist).

        Args:
            obj (MDagPath): path to object being drawn
            cam (MDagPath): path to viewport camera
            frame_context (MFrameContext): frame context
            data (MeshXRayerData): previous data
            verbose (int): print process data

        Returns:
            (MeshXRayerData): node data
        """
        lprint('PREPARE FOR DRAW', verbose=verbose)

        _data = data
        if not isinstance(_data, MeshXRayerData):
            _data = MeshXRayerData()
            lprint(' - USING EXISTING DATA', _data, verbose=verbose)
        lprint(' - DATA', _data, verbose=verbose)

        # Read in_mesh plug
        lprint(' - OBJ', obj, verbose=verbose)
        _node = obj.node()
        _in_mesh_plug = om.MPlug(_node, MeshXRayer.in_mesh)
        lprint(' - IN MESH PLUG', _in_mesh_plug, verbose=verbose)

        _data.mesh_tris.clear()
        if _in_mesh_plug.isNull:
            return None
        if _in_mesh_plug.asMDataHandle().type() != om.MFnData.kMesh:
            return None
        _in_mesh_handle = _in_mesh_plug.asMDataHandle().asMesh()
        _in_mesh = om.MFnMesh(_in_mesh_handle)

        # Read mesh triangles
        _mesh_pts = _in_mesh.getPoints()
        for _poly_id in range(_in_mesh.numPolygons):
            _vtx_ids = _in_mesh.getPolygonVertices(_poly_id)
            for _vtx_id in _vtx_ids[:3]:
                _data.mesh_tris.append(_mesh_pts[_vtx_id])
        lprint(' - IN MESH',
               _in_mesh,
               len(_in_mesh.getPoints()),
               verbose=verbose)

        # Read col/hide_angle + draw toggles
        _col_plug = om.MPlug(_node, MeshXRayer.color)
        _data.color = om.MColor(
            [_col_plug.child(_idx).asFloat() for _idx in range(3)])
        _data.hide_angle = om.MPlug(_node, MeshXRayer.hide_angle).asFloat()
        _data.draw_control = om.MPlug(_node, MeshXRayer.draw_control).asBool()
        _data.draw_mesh = om.MPlug(_node, MeshXRayer.draw_mesh).asBool()

        _obj_pos = hom.HMatrix(obj.inclusiveMatrix()).pos()
        _cam_pos = hom.HMatrix(cam.inclusiveMatrix()).pos()
        _data.mesh_to_cam = _cam_pos - _obj_pos

        return _data
コード例 #19
0
ファイル: footPrintNode.py プロジェクト: jonntd/mayadev-1
    def __init__(self):
        om.MUserData.__init__(self, False)  ## don't delete after draw

        self.fColor = om.MColor()
        self.fSoleLineList = om.MPointArray()
        self.fSoleTriangleList = om.MPointArray()
        self.fHeelLineList = om.MPointArray()
        self.fHeelTriangleList = om.MPointArray()
コード例 #20
0
    def __init__(self):
        om.MUserData.__init__(self, False)  ## don't delete after draw

        ######VECTOR
        self.fColor = om.MColor()
        self.fsourcePt = om.MPoint()
        self.faimPt = om.MPoint()
        self.fWidth = 1.0
コード例 #21
0
    def __init__(self):
        # The false tells it not to delete after its used
        super(LocatorData, self).__init__(False)

        self.shape = None  # The shape index that is being drawn
        self.color = om.MColor()  # Holds the color to draw
        self.lineList = om.MPointArray()  # The list of lines to draw
        self.triangleList = om.MPointArray()  # A list of triangles to draw
コード例 #22
0
def apply_api_colors(colors, indices, obj):
    import maya.api.OpenMaya as api
    colors = [api.MColor(i) for i in colors]
    selectionList = api.MSelectionList()
    selectionList.add(obj)
    nodeDagPath = selectionList.getDagPath(0)
    mfnMesh = api.MFnMesh(nodeDagPath)
    mfnMesh.setVertexColors(colors, indices)
コード例 #23
0
    def addUIDrawables(self, objPath, drawManager, frameContext, data):
        if not isinstance(data, GuideHandleData):
            return

        displayStatus = OpenMayaRender.MGeometryUtilities.displayStatus(objPath)
        selected = displayStatus == util.DisplayStatus.Lead or displayStatus == util.DisplayStatus.Active

        mainColor = data.handleColor
        if selected:
            mainColor = OpenMaya.MColor([1.0, 1.0, 1.0, 1.0])
        connectorColor = OpenMaya.MColor(mainColor)
        connectorColor.a = 0.6

        center = OpenMaya.MPoint(0.0, 0.0, 0.0)
        radius = 1.0
        colorR = OpenMaya.MColor([1.0, 0.0, 0.0, 1.0])
        colorG = OpenMaya.MColor([0.0, 1.0, 0.0, 1.0])
        colorB = OpenMaya.MColor([0.0, 0.0, 1.0, 1.0])

        viewDirection = OpenMaya.MVector(frameContext.getTuple(frameContext.kViewDirection))
        viewDirection *= data.inverseMatrix

        drawManager.beginDrawable()
        drawManager.beginDrawInXray()

        drawManager.setColor(mainColor)
        drawManager.sphere(center, radius, filled=True)

        if data.childPositions:
            self.drawConnectors(drawManager, data.childPositions, viewDirection, connectorColor)

        if data.hasGuide:
            x = OpenMaya.MVector(data.jointMatrix[0], data.jointMatrix[1], data.jointMatrix[2]).normal()
            y = OpenMaya.MVector(data.jointMatrix[4], data.jointMatrix[5], data.jointMatrix[6]).normal()
            z = OpenMaya.MVector(data.jointMatrix[8], data.jointMatrix[9], data.jointMatrix[10]).normal()
            pos = OpenMaya.MPoint(data.jointMatrix[12], data.jointMatrix[13], data.jointMatrix[14])

            drawManager.setColor(colorR)
            drawManager.line(pos, pos + x)
            drawManager.setColor(colorG)
            drawManager.line(pos, pos + y)
            drawManager.setColor(colorB)
            drawManager.line(pos, pos + z)

        drawManager.endDrawInXray()
        drawManager.endDrawable()
コード例 #24
0
def static_compare(mesh, target, clamp=10, world=True):
    """
    Compares two meshes on a given static frame.
    Sets a vertex color value to show the distance values.
    
    :param mesh: The name or path to the mesh to visualize on.
    :param target: The name or path of the mesh we are comparing against.
    :param clamp: The value to clamp against. All values be scaled accordingly.
    :param world: Whether to compare in world space or object space.
    """
    # Get the Dag Paths to the mesh shapes
    mesh = get_shape(mesh)
    target = get_shape(target)

    # Turn on visualization of the vertex colors on the shape
    mesh_path = mesh.fullPathName()
    mc.polyOptions(mesh_path, colorShadedDisplay=True)

    # Get Mesh objects from the shapes.
    mesh_mesh = om.MFnMesh(mesh)
    target_mesh = om.MFnMesh(target)

    # For each mesh, get their points
    space = om.MSpace.kWorld if world else om.MSpace.kObject
    mesh_points = mesh_mesh.getPoints(space)
    target_points = target_mesh.getPoints(space)
    print("mesh points = ", len(mesh_points))
    print("target points = ", len(target_points))

    colors = om.MColorArray()
    ids = []

    # Loop through to calculate the colors
    for i, mpoint in enumerate(mesh_points):
        tpoint = target_points[i]

        # Get the distance between the vertices.
        distance = math.sqrt(((mpoint.x - tpoint.x)**2) +
                             ((mpoint.y - tpoint.y)**2) +
                             ((mpoint.z - tpoint.z)**2))

        # If our distance is not zero, then process it
        if distance:
            # First scale according to the clamp value
            scaled = distance / clamp
            # Then clamp it off at 1
            clamped = min(scaled, 1)
            # Finally we only want half its value since our base color is 50% grey
            distance = clamped / 2.0

        # Start from 50% grey and bias to the red channel as distance grows
        color = om.MColor((0.5 + distance, 0.5 - distance, 0.5 - distance))

        colors.append(color)
        ids.append(i)

    # Apply the colors
    mesh_mesh.setVertexColors(colors, ids)
コード例 #25
0
ファイル: annotate.py プロジェクト: matthieuCantat/python
def glDrawTriangles(drawManager, camMatrix, coords, ssOffsets, color, colors):
    colorsArray = ompy.MColorArray()
    for i in range(0, coords.length()):
        if (colors.length() == 0):
            colorsArray.append(
                ompy.MColor([color[0], color[1], color[2], color[3]]))
        else:
            colorsArray.append(
                ompy.MColor(
                    [colors[i][0], colors[i][1], colors[i][2], colors[i][3]]))

    for i in range(0, coords.length()):
        coords[i] += utils_screenTo3dOffset(
            camMatrix, coords[i], ssOffsets[min(i,
                                                ssOffsets.length() - 1)])

    drawManager.mesh(ompyr.MUIDrawManager.kTriangles, coords, None,
                     colorsArray)
コード例 #26
0
 def draw_text(draw_manager, position, text, alignment, background_size):
     if not len(text):
         return
     draw_manager.text2d(
         position, text, alignment=alignment,
         backgroundSize=background_size,
         backgroundColor=OpenMaya.MColor((0.0, 0.0, 0.0, 0.0)),
         dynamic=False
     )
コード例 #27
0
 def drawText(self, drawManager, position, text, alignment, bgSize):
     """
     """
     if (len(text) > 0):
         drawManager.text2d(position,
                            text,
                            alignment=alignment,
                            backgroundSize=bgSize,
                            backgroundColor=om.MColor((0.0, 0.0, 0.0, 0.0)))
コード例 #28
0
ファイル: templateLocator.py プロジェクト: arunpillaii/Python
    def addUIDrawables(self, objPath, drawManager, frameContext, data):
        # Viewport 2.0 addUIDrawables(Draw) Function
        locatordata = data
        if not isinstance(locatordata, TemplateLocatorData):
            return
        drawManager.beginDrawable()

        textColor = OpenMaya.MColor((1.0, 1.0, 1.0, 1.0))
        drawManager.setColor(textColor)

        drawManager.text(OpenMaya.MPoint(0, 1, 0), "3D SPACE TEXT", OpenMayaRender.MUIDrawManager.kLeft)

        textColor = OpenMaya.MColor((0.5, 0.3, 0.4, 1.0))
        drawManager.setColor(textColor)

        drawManager.text2d(OpenMaya.MPoint(500, 500), "2D SPACE TEXT", OpenMayaRender.MUIDrawManager.kLeft)

        drawManager.endDrawable()
コード例 #29
0
 def __init__(self):
     om.MUserData.__init__(self, False)
     self.fUIType = kText
     self.fColor = om.MColor((1., 0., 0., 1.))
     self.fLineWidth = 2
     self.fLineStyle = omr.MUIDrawManager.kSolid
     self.fIsFilled = False
     self.fRadius = 1.
     self.fDraw2D = False
     self.fPosition = om.MPoint(0.0, 0.0, 0.001)
     # text
     self.fText = "uiDrawManager-Text"
     self.fTextFontSize = omr.MUIDrawManager.kDefaultFontSize
     self.fFontFaceIndex = 0
     self.fTextAlignment = omr.MUIDrawManager.kLeft
     self.fTextIncline = omr.MUIDrawManager.kInclineNormal
     self.fTextWeight = omr.MUIDrawManager.kWeightBold
     self.fTextStretch = omr.MUIDrawManager.kStretchUnstretched
     self.fTextLine = 0
     self.fTextBoxSize = [0, 0]
     self.fTextBoxColor = om.MColor((0., 1., 1., 1.))
     # point
     self.fPointSize = 2.
     # line
     self.fLineStartPoint = om.MPoint(0.0, 0.0, 0.0)
     self.fLineEndPoint = om.MPoint(1.0, 1.0, 1.0)
     # rect
     self.fRectUp = om.MVector(0.0, 1.0, 0.0)
     self.fRectNormal = om.MVector(0.0, 0.0, 1.0)
     self.fRectScale = [1., 1.]
     # quad
     self.fQuadVertex = [
         om.MPoint(0.0, 0.0, 0.0),
         om.MPoint(1.0, 0.0, 0.0),
         om.MPoint(1.0, 1.0, 0.0),
         om.MPoint(0.0, 1.0, 0.0)
     ]
     # circle
     self.fCircleNormal = om.MVector(0.0, 0.0, 1.0)
     # arc
     self.fArcStart = om.MVector(1.0, 0.0, 0.0)
     self.fArcEnd = om.MVector(0.0, 1.0, 0.0)
     self.fArcNormal = om.MVector(0.0, 0.0, 1.0)
コード例 #30
0
    def _initializeTemporalyVariables(self):
        # type: () -> None

        self.xcolor = om.MColor(om.MVector(1, 0, 0))
        self.xmat = om.MMatrix(
            ((1, 0, 0, 0), (0, 1, 0, 0), (0, 0, 1, 0), (1, 0, 0, 1)))

        self.ycolor = om.MColor(om.MVector(0, 1, 0))
        self.ymat = om.MMatrix(
            ((1, 0, 0, 0), (0, 1, 0, 0), (0, 0, 1, 0), (0, 1, 0, 1)))

        self.zcolor = om.MColor(om.MVector(0, 0, 1))
        self.zmat = om.MMatrix(
            ((1, 0, 0, 0), (0, 1, 0, 0), (0, 0, 1, 0), (0, 0, 1, 1)))

        self.grayColor = om.MColor(om.MVector(0.3, 0.3, 0.3))

        self.coneRadius = 0.1
        self.coneHeight = 0.6