File: TestStringToPath.py

package info (click to toggle)
paraview 5.13.2%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 544,220 kB
  • sloc: cpp: 3,374,605; ansic: 1,332,409; python: 150,381; xml: 122,166; sql: 65,887; sh: 7,317; javascript: 5,262; yacc: 4,417; java: 3,977; perl: 2,363; lex: 1,929; f90: 1,397; makefile: 170; objc: 153; tcl: 59; pascal: 50; fortran: 29
file content (110 lines) | stat: -rwxr-xr-x 3,039 bytes parent folder | download | duplicates (6)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
#!/usr/bin/env python

from vtkmodules.vtkCommonCore import (
    vtkIntArray,
    vtkPoints,
)

from vtkmodules.vtkCommonDataModel import (
    vtkPath,
)

from vtkmodules.vtkViewsContext2D import (
    vtkContextView,
)

from vtkmodules.vtkRenderingContext2D import (
    vtkContext2D,
    vtkContextItem,
    vtkContextScene,
    vtkPen,
)

from vtkmodules.vtkRenderingCore import (
    vtkRenderer,
    vtkRenderWindow,
    vtkRenderWindowInteractor,
    vtkTextProperty,
)

from vtkmodules.vtkRenderingFreeType import (
    vtkMathTextUtilities,
)

from vtkmodules.vtkPythonContext2D import (
    vtkPythonItem,
)

import vtkmodules.vtkRenderingContextOpenGL2
import vtkmodules.vtkRenderingMatplotlib
import vtkmodules.vtkRenderingOpenGL2
import vtkmodules.vtkRenderingUI

#------------------------------------------------------------------------------
class StringToPathContextTest(object):
    def __init__(self):
        self.Path = None

    def SetPath(self, path):
        self.Path = path

    def Initialize(self, vtkSelf):
        return True

    def Paint(self, vtkSelf, painter):
        # Paint event for the chart, called whenever the chart needs to be drawn
        # This function aims to test the primitives provided by the 2D API.
        # RGB color lookup table by path point code:
        color = [(), (), (), ()]
        color[vtkPath.MOVE_TO] = (1.0, 0.0, 0.0)
        color[vtkPath.LINE_TO] = (0.0, 1.0, 0.0)
        color[vtkPath.CONIC_CURVE] = (0.0, 0.0, 1.0)
        color[vtkPath.CUBIC_CURVE] = (1.0, 0.0, 1.0)

        points = self.Path.GetPoints()
        codes = self.Path.GetCodes()

        if points.GetNumberOfPoints() != codes.GetNumberOfTuples():
            return False

        # scaling factor and offset to ensure that the points will fit the view:
        scale = 5.16591
        offset = 20.0

        # Draw the control points, colored by codes:
        point = [0.0, 0.0, 0.0]
        painter.GetPen().SetWidth(2)
        for i in range(points.GetNumberOfPoints()):
            points.GetPoint(i, point)
            code = codes.GetValue(i)

            painter.GetPen().SetColorF(color[code])
            painter.DrawPoint(point[0] * scale + offset, point[1] * scale + offset)

        return True

#------------------------------------------------------------------------------
# Set up a 2D context view, context test object and add it to the scene
view = vtkContextView()
ren = view.GetRenderer()
ren.SetBackground(1.0, 1.0, 1.0)
renWin = view.GetRenderWindow()
renWin.SetSize(325, 150)
test = vtkPythonItem()
testObject = StringToPathContextTest()
test.SetPythonObject(testObject)
view.GetScene().AddItem(test)

path = vtkPath()
tprop = vtkTextProperty()

vtkMathTextUtilities.GetInstance().StringToPath(
    "$\\frac{-b\\pm\\sqrt{b^2-4ac}}{2a}$", path, tprop, view.GetRenderWindow().GetDPI())

testObject.SetPath(path)

renWin.SetMultiSamples(0)
iren = view.GetInteractor()
iren.Initialize()
if 'rtTester' not in locals() or rtTester.IsInteractiveModeSpecified():
    iren.Start()