Welcome to the SeeMeCNC Community

Code: Select all

#Script to analyse GCode and saturate Extruder commands to SATURATION value
#Guillaume Maugein - Jan 2014

import sys
import re
import os
import Tkinter as tkinter
import tkMessageBox as mbox

#if launched from Slic3r, the filename is passed in parameter automatically
#else, if launched manually, it will parse the file indicated thereafter
if len(sys.argv)>1 :
    filename = sys.argv[1]
else :
    filename = "C:/Users/Guillaume/AppData/Local/RepetierHost/composition.gcode"

#While parsed, modifications are saved to a temporary file in the same folder
#as parsed gcode file
tmpname = os.path.dirname(filename) + "tmp.gcode"

#SATURATION value (in mm)
SATURATION = 10.0

#Variables Initialisations
count = 0 #counting lines in GCode
MaxLine = 0 #Line with Max Extrusion value - for display only
MaxDist = 0.0 #Max Extrusion value - for display only
E = 0.0 #Current Line extrusion value
OldE= 0.0 #Changed Extrusion value of previous line
PreviousE = 0.0 #Actual Extrusion value of previous line
SupSat = 0 #Number of saturated lines - for display only

outputfile = open(tmpname,'w')
inputfile = open(filename,'r')
for line in inputfile:
    count = count + 1
    if line.count(";") > 0 : #Removing comments that could be troublesome
        tmpline = line[0:line.find(";")]
    else :
        tmpline = line

    Out = tmpline.split(" ")
    GCommand = (tmpline.count("G92") != -1) #Searching for G92 commands reseting Extruding distances
    if GCommand :
        for Command in Out:

            Command = Command.upper()
            if Command.count("E0") >= 1:
                E = 0.0
                OldE = 0.0
                PreviousE = 0.0

    GCommand = (tmpline.count("G0") != -1) or (tmpline.count("G1") != -1) #Searching for Extruding commands
    if GCommand :
        for Command in Out:

            CommandU = Command.upper()
            if CommandU.count("E") == 1 : #if there is an extruding command
               OldE = E 
               E = float(CommandU[1:])
               Dist = E - OldE

               if Dist > SATURATION : #if Extruding more than SATURATION mm
                   if E - PreviousE > 0 : 
                       print "Modified Line " + str(count)
                       print "   - Was Extruding " + str(E - PreviousE) + "mm"
                       E = SATURATION+OldE
                       line = re.sub(Command,"E{:.5f}".format(E),line)
                       line = line[0:line.find("\n")-1] + "; Modified by script \n"
                       SupSat = SupSat+1
                   else : 
                       E = OldE + E - PreviousE
                       line = re.sub(Command,"E{:.5f}".format(E),line)
                       line = line[0:line.find("\n")-1] + "; Modified by script \n"
                       print "Modified Line " + str(count)
                       SupSat = SupSat+1
                   
               PreviousE = float(CommandU[1:])
               
               if (Dist > MaxDist) : #Saving Maximum Extruding command (For display only)
                   MaxDist = float(Dist)
                   MaxLine = count



    outputfile.write(line) #Writting to temporary file



outputfile.close()
inputfile.close()



if SupSat > 0 : #If no command is saturated, there is no message
    window = tkinter.Tk()
    window.wm_withdraw()
    #Message to ask if you want to replace GCode or not
    if mbox.askyesno ('GCode Extruder Check',"Extruder Limite Reached : " + str(MaxDist) + " mm in Ligne nb : " + str(MaxLine) + ". \nReplace GCode ?") :
        os.remove(filename)
        os.rename(tmpname, filename)
        print str(SupSat) + " modified lines." #Message Nb of modified lines.
    else :
        os.remove(tmpname) #removing Temporary file
else :
    os.remove(tmpname) #removing Temporary file
    print "Clean Code."