Added scaling plugin

bCNC/plugins/scaling.py

@@ -0,0 +1,190 @@
+#!/usr/bin/python
+# -*- coding: ascii -*-
+# $Id$
+#
+# Author: Mario Basz
+#mariob_1960@yahoo.com.ar
+# Date: 20 may 2018
+# A special thanks to Vasilis Viachoudis, Filippo Rivato and Buschhardt 
+#This plugin is based on a variation
+# of yours Driller plugin and My_Plugin example.
+
+__author__ = "Mario S Basz"
+__email__  = "mariob_1960@yaho.com.ar"
+
+__name__ = _("Scaling")
+__version__= "0.1"
+
+import os.path
+import re
+import math
+from collections import OrderedDict
+from CNC import CNC,Block
+from ToolsPage import Plugin
+from bmath import pi
+from math import pi, sqrt, sin, cos, asin, acos, atan2, hypot, degrees, radians, copysign, fmod, tan
+
+#==============================================================================
+# Scaling selected block
+#==============================================================================
+class Tool(Plugin):
+	__doc__ = _("Scaling the selected block")
+	def __init__(self, master):
+		Plugin.__init__(self, master, "Scaling")
+		self.icon  = "scale"
+		self.group = "CAM"
+
+		self.variables = [
+			("name",          "db",    "",    _("Name")),
+            ("xscale" ,"float", "", _("X Scale")),
+            ("yscale" ,"float", "", _("Y Scale")),
+            ("zscale" ,"float", "", _("Z Scale")),
+			("feed"		, "int", 1200	, _("Feed")),
+			("zfeed"		, "int", ""	, _("Plunge Feed")),
+			("rpm",          "int" ,    12000, _("RPM"))
+		]
+		self.buttons.append("exe")
+	# -----------------------------------------------------
+	def scaling(self,xyz):
+		xscale= self["xscale"]
+		if xscale=="":xscale=1
+		yscale= self["yscale"]
+		if yscale=="":yscale=1
+		zscale= self["zscale"]
+		if zscale=="":zscale=1
+		A = xyz[0]
+		B = xyz[1]
+#		xlength = B[0]-A[0]
+#		ylength = B[1]-A[1]
+#		zlength = B[2]-A[2]
+		xnew=B[0]*xscale
+		ynew=B[1]*yscale
+		znew=B[2]*zscale
+
+		return xnew,ynew,znew
+
+	# Calc subsegments -----------------------------------------------------
+	def calcSegmentLength(self, xyz):
+		if xyz:
+			A = xyz[0]
+			B = xyz[1]
+			seglength_x = B[0]-A[0]
+			seglength_y = B[1]-A[1]
+			seglength_z = B[2]-A[2]
+			return math.sqrt(seglength_x**2 + seglength_y**2 + seglength_z**2)
+		else:
+			return 0
+	# ----------------------------------------------------------------------
+
+	#Extract all segments from commands ------------ -----------------------
+	def extractAllSegments(self, app,selectedBlock):
+		allSegments = []
+		allBlocks = app.gcode.blocks
+
+		for bid in selectedBlock:
+			bidSegments = []
+			block = allBlocks[bid]
+			if block.name() in ("Header", "Footer"): continue
+			#if not block.enable : continue
+			app.gcode.initPath(bid)
+			for line in block:
+				try:
+					cmd = app.cnc.breakLine(app.gcode.evaluate(app.cnc.compileLine(line)))
+				except:
+					cmd = None
+
+				if cmd:
+					app.cnc.motionStart(cmd)
+					xyz = app.cnc.motionPath()
+					app.cnc.motionEnd()
+
+					if xyz:
+						#exclude if fast move or z only movement
+						G0 =('g0' in cmd) or ('G0' in cmd)
+						Zonly = (xyz[0][0] == xyz[1][0] and xyz[0][1] == xyz[1][1])
+						exclude = Zonly
+
+						#save length for later use
+						segLength = self.calcSegmentLength(xyz)
+
+						if len(xyz) < 3:
+							bidSegments.append([xyz[0],xyz[1],exclude,segLength])
+						else:
+							for i in range(len(xyz)-1):
+								bidSegments.append([xyz[i],xyz[i+1],exclude,segLength])
+			#append bidSegmentes to allSegmentes
+			allSegments.append(bidSegments)
+
+		#Disabled used block
+		for bid in selectedBlock:
+			block = allBlocks[bid]
+			if block.name() in ("Header", "Footer"): continue
+			block.enable = False
+
+		return allSegments
+
+	# ----------------------------------------------------------------------
+	def execute(self, app):
+		feed = self["feed"]
+		zfeed = CNC.vars["cutfeedz"]
+		rpm = self["rpm"]
+		if self["zfeed"]:
+				zfeed = self["zfeed"]
+
+		zup = self["zup"]
+
+		surface = CNC.vars["surface"]
+#		zbeforecontact=surface+CNC.vars["zretract"]
+#		hardcrust = surface - CNC.vars["hardcrust"]
+#		feedbeforecontact = CNC.vars["feedbeforecontact"]/100.0
+#		hardcrustfeed = CNC.vars["hardcrustfeed"]/100.0
+
+		# Get selected blocks from editor
+		selBlocks = app.editor.getSelectedBlocks()
+
+		if not selBlocks:
+			app.setStatus(_("Scaling abort: Please select some path"))
+			return 
+
+		#Get all segments from gcode
+		allSegments = self.extractAllSegments(app,selBlocks)
+
+		#Create holes locations
+#		allHoles=[]
+		for bidSegment in allSegments:
+			if len(bidSegment)==0:
+				continue
+		blocks = []
+		n = self["name"]
+#		if not n or n=="default": n="Trochoidal_3D"
+		n="Scaling"
+		scal_block = Block(n)
+
+		for idx, segm in enumerate(bidSegment):
+			if idx >= 0:
+				if idx == 0:
+					scal_block.append("M03")
+					scal_block.append("S "+str(rpm))
+					scal_block.append(CNC.zsafe())
+					scal_block.append("F "+str(feed))
+					scal_block.append("(--------------------------------------------------)")
+
+				B=self.scaling(segm)
+
+				if segm[0][2]> surface and segm[1][2]>=surface:
+					scal_block.append("g0 x "+str(B[0])+" y "+str(B[1])+ " z "+str(B[2]))
+				else:
+					scal_block.append("g1 x "+str(B[0])+" y "+str(B[1])+ " z "+str(B[2]))
+		scal_block.append(CNC.zsafe()) 			#<<< Move rapid Z axis to the safe height in Stock Material
+		blocks.append(scal_block)
+		self.finish_blocks(app, blocks)
+	#--------------------------------------------------------------
+
+
+	#Insert created blocks
+	def finish_blocks(self, app, blocks):
+		active = app.activeBlock()
+		if active==0: active=1
+		app.gcode.insBlocks(active, blocks, "scale")
+		app.refresh()
+		app.setStatus(_("Scaling Generated"))