Pycad 构建一个简易的PolylineJig

枫叶棋语

1. def AddPolyline(pts):
   
2.     polyline = Polyline()
   
3.     polyline.SetDatabaseDefaults()
   
4.     plane = Plane()
   
5.     for index, pt in enumerate(pts):
   
6.         polyline.AddVertexAt(index, pt.Convert2d(plane), 0, 0, 0)
   
7. 
   
8.     return polyline
   
9. 
   
10. class PolylineJig(DrawJig):
    
11.     def __init__(self, basept=None):
    
12.         self.basept = basept
    
13.         self.location = basept
    
14.         self.pts = [basept, basept]
    
15. 
    
16.     def Sampler(self, prompts:JigPrompts):
    
17.         opts = JigPromptPointOptions('\n请输入终点:')
    
18.         opts.UserInputControls = (
    
19.             UserInputControls.Accept3dCoordinates |
    
20.             UserInputControls.NoZeroResponseAccepted |
    
21.             UserInputControls.NoNegativeResponseAccepted)
    
22.         res = prompts.AcquirePoint(opts)
    
23. 
    
24.         if res.Value != self.location:
    
25.             self.location = res.Value
    
26.         else:
    
27.             return SamplerStatus.NoChange
    
28. 
    
29.         if res.Status == PromptStatus.Cancel:
    
30.             return SamplerStatus.Cancel
    
31.         else:
    
32.             return SamplerStatus.OK
    
33.             
    
34.     def Update(self):
    
35.         self.pts[-1] = self.location
    
36.         self.pline = AddPolyline(self.pts)
    
37.         
    
38.     def GetEntity(self):
    
39.         self.Update()
    
40.         return self.pline
    
41. 
    
42.     def WorldDraw(self, draw:WorldDraw):
    
43.         draw.Geometry.Draw(self.GetEntity())
    
44.         return True
    
45. 
    
46. 
    
47. @Command()
    
48. def ts01(doc: Document):
    
49.     with dbtrans(doc) as tr:
    
50.         ed = doc.Editor
    
51.         btr = tr.opencurrspace()
    
52.         res = getpoint()
    
53.         if not res.ok:
    
54.             return
    
55. 
    
56.         basept = res.value
    
57.         jig = PolylineJig(basept)
    
58.         res = ed.Drag(jig)
    
59. 
    
60.         while res.Status == PromptStatus.OK:
    
61.             jig.pts.append(jig.location)
    
62.             basept = jig.location
    
63.             jig.basept = basept
    
64.             res = ed.Drag(jig)
    
65. 
    
66.             if res.Status != PromptStatus.OK:
    
67.                 if len(jig.pts) > 2:
    
68.                     jig.pts.pop()
    
69.                     pline = AddPolyline(jig.pts)
    
70.                     tr.addentity(btr, pline)

复制代码