# ##########################################################
# FlatCAM: 2D Post-processing for Manufacturing            #
# http://flatcam.org                                       #
# File Author: Matthieu Berthomé                           #
# Date: 5/26/2017                                          #
# MIT Licence                                              #
# ##########################################################

from appPreProcessor import *


class default_laser(PreProc):

    include_header = True
    coordinate_format = "%.*f"
    feedrate_format = '%.*f'

    def start_code(self, p):
        units = ' ' + str(p['units']).lower()

        gcode = '(This preprocessor is the default preprocessor when used with a laser.)\n'
        gcode += '(It is made to work with MACH3 compatible motion controllers.)\n' \
                 '(This preprocessor makes no moves on the Z axis it will only move horizontally.)\n' \
                 '(The horizontal move is done with G0 - highest possible speed set in MACH3.)\n' \
                 '(It assumes a manually focused laser.)\n' \
                 '(The laser is started with M3 command and stopped with the M5 command.)\n\n'

        xmin = '%.*f' % (p.coords_decimals, p['obj_options']['xmin'])
        xmax = '%.*f' % (p.coords_decimals, p['obj_options']['xmax'])
        ymin = '%.*f' % (p.coords_decimals, p['obj_options']['ymin'])
        ymax = '%.*f' % (p.coords_decimals, p['obj_options']['ymax'])

        gcode += '(Feedrate: ' + str(p['feedrate']) + units + '/min' + ')\n'
        gcode += '(Steps per circle: ' + str(p['steps_per_circle']) + ')\n'

        if str(p['obj_options']['type']) == 'Excellon' or str(p['obj_options']['type']) == 'Excellon Geometry':
            gcode += '(Preprocessor Excellon: ' + str(p['pp_excellon_name']) + ')\n'
        else:
            gcode += '(Preprocessor Geometry: ' + str(p['pp_geometry_name']) + ')\n' + '\n'

        gcode += '(X range: ' + '{: >9s}'.format(xmin) + ' ... ' + '{: >9s}'.format(xmax) + ' ' + units + ')\n'
        gcode += '(Y range: ' + '{: >9s}'.format(ymin) + ' ... ' + '{: >9s}'.format(ymax) + ' ' + units + ')\n\n'

        gcode += '(Laser Power - Spindle Speed: ' + str(p['spindlespeed']) + ')\n'
        gcode += '(Laser Minimum Power: ' + str(p['laser_min_power']) + ')\n\n'

        gcode += ('G20\n' if p.units.upper() == 'IN' else 'G21\n')
        gcode += 'G90\n'
        gcode += 'G94'

        return gcode

    def startz_code(self, p):
        return ''

    def lift_code(self, p):
        if float(p.laser_min_power) > 0.0:
            # the formatted text: laser OFF must always be like this else the plotting will not be done correctly
            return 'M03 S%s (laser OFF)\n' % str(p.laser_min_power)
        else:
            return 'M05'

    def down_code(self, p):
        if p.spindlespeed:
            return '%s S%s' % ('M03', str(p.spindlespeed))
        else:
            return 'M03'

    def toolchange_code(self, p):
        return ''

    def up_to_zero_code(self, p):
        return ''

    def position_code(self, p):
        # used in for the linear motion
        # formula for skewing on x for example is:
        # x_fin = x_init + y_init/slope where slope = p._bed_limit_y / p._bed_skew_x (a.k.a tangent)
        if p._bed_skew_x == 0:
            x_pos = p.x + p._bed_offset_x
        else:
            x_pos = (p.x + p._bed_offset_x) + ((p.y / p._bed_limit_y) * p._bed_skew_x)

        if p._bed_skew_y == 0:
            y_pos = p.y + p._bed_offset_y
        else:
            y_pos = (p.y + p._bed_offset_y) + ((p.x / p._bed_limit_x) * p._bed_skew_y)

        return ('X' + self.coordinate_format + ' Y' + self.coordinate_format) % \
               (p.coords_decimals, x_pos, p.coords_decimals, y_pos)

    def rapid_code(self, p):
        # a fast linear motion using the G0 command which means: "move as fast as the CNC can handle and is set in the
        # CNC controller". It is a horizontal move in the X-Y CNC plane.
        return ('G00 ' + self.position_code(p)).format(**p)

    def linear_code(self, p):
        # a linear motion using the G1 command which means: "move with a set feedrate (speed)".
        # It is a horizontal move in the X-Y CNC plane.
        return ('G01 ' + self.position_code(p)).format(**p)

    def end_code(self, p):
        # a final move at the end of the CNC job. First it moves to a safe parking Z height followed by an X-Y move
        # to the parking location.
        end_coords_xy = p['xy_end']
        gcode = ''

        if end_coords_xy and end_coords_xy != '':
            gcode += 'G00 X{x} Y{y}'.format(x=end_coords_xy[0], y=end_coords_xy[1]) + "\n"
        return gcode

    def feedrate_code(self, p):
        # set the feedrate for the linear move with G1 command on the X-Y CNC plane (horizontal move)
        return 'G01 F' + str(self.feedrate_format % (p.fr_decimals, p.feedrate))

    def z_feedrate_code(self, p):
        # set the feedrate for the linear move with G1 command on the Z CNC plane (vertical move)
        return 'G01 F' + str(self.feedrate_format % (p.fr_decimals, p.z_feedrate))

    def spindle_code(self, p):
        if p.spindlespeed:
            return '%s S%s' % ('M03', str(p.spindlespeed))
        else:
            return 'M03'

    def dwell_code(self, p):
        return ''

    def spindle_stop_code(self, p):
        return 'M05'