PyCAM/SketchOp.py at master · danheeks/PyCAM · GitHub

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
from DepthOp import DepthOp
import cad
from Object import PyProperty

class SketchOp(DepthOp):
    def __init__(self, sketch):
        DepthOp.__init__(self)
        self.sketch = sketch

    def CopyFrom(self, object):
        DepthOp.CopyFrom(self, object)
        self.sketch = object.sketch

    def WriteXml(self):
        cad.SetXmlValue('sketch', self.sketch)
        DepthOp.WriteXml(self)

    def ReadXml(self):
        self.sketch = cad.GetXmlInt('sketch')
        DepthOp.ReadXml(self)

    def GetProperties(self):
        properties = []

        properties.append(PyProperty("Sketch", 'sketch', self))

        properties += DepthOp.GetProperties(self)

        return properties

    def OnGlCommands(self, select, marked, no_color):
        if marked and not select:
            object = cad.GetObjectFromId(cad.OBJECT_TYPE_SKETCH, self.sketch)
            if object != None:
                cad.Material(cad.Color(255, 255, 0)).glMaterial(1.0)
                cad.DrawEnableLighting()
                #cad.DrawDisableDepthTesting()
                cad.DrawEnableCullFace()
                cad.Sketch.RenderAsExtrusion(object, self.start_depth, self.final_depth)
                cad.DrawDisableCullFace()
                #cad.DrawEnableDepthTesting()
                cad.DrawDisableLighting()

    def DoEachSketch(self, callback, data):
        # data is a tuple, whatever you want to pass on
        # callback gets called for each separate sketch

        object = cad.GetObjectFromId(cad.OBJECT_TYPE_SKETCH, self.sketch)

        if object != None:
            if object.GetType() == cad.OBJECT_TYPE_SKETCH:
                re_ordered_sketch = None
                sketch_order = object.GetSketchOrder()
                if sketch_order == cad.SketchOrderType.SketchOrderTypeBad:
                    re_ordered_sketch = object.MakeACopy()
                    re_ordered_sketch.ReOrderSketch(cad.SketchOrderType.SketchOrderTypeReOrder)
                    object = re_ordered_sketch
                    sketch_order = object.GetSketchOrder()
                if (sketch_order == cad.SketchOrderType.SketchOrderTypeMultipleCurves) or (sketch_order == cad.SketchOrderType.SketchOrderHasCircles):
                    new_separate_sketches = object.Split()
                    for one_curve_sketch in new_separate_sketches:
                        callback(one_curve_sketch, data)
                else:
                    callback(object, data)