Module sverchok.utils.macros.gp_macros

# This file is part of project Sverchok. It's copyrighted by the contributors
# recorded in the version control history of the file, available from
# its original location https://github.com/nortikin/sverchok/commit/master
#
# SPDX-License-Identifier: GPL3
# License-Filename: LICENSE

class sv_sock(object):
    def __init__(self, socket):
        self.socket = socket
        self.links = socket.id_data.links

    def __gt__(self, other):
        self.links.new(self.socket, other.socket)

def gp_macro_one(context, operator, term, nodes, links):
    needed_nodes = [
        ['SvGetAssetPropertiesMK2', (0.00, 0.00)],
        ['SvPathLengthMk2Node', (250, 55)],
        ['SvScalarMathNodeMK4', (430, 115)],
        ['Float2IntNode', (600, 50)],
        ['SvGenNumberRange', (720, 90)],
        ['SvInterpolationNodeMK3', (880, 40)],
        ['LineConnectNodeMK2', (1060, -40)],
        ['SvViewerDrawMk4', (1245, 50)],
    ]

    made_nodes = []
    x, y = context.space_data.cursor_location[:]
    for node_bl_idname, node_location in needed_nodes:
        n = nodes.new(node_bl_idname)
        n.location = node_location[0] + x, node_location[1] + y
        made_nodes.append(n)

    # ID Selector
    made_nodes[0].Mode = 'grease_pencils'  # the rest must be user driven
    links.new(made_nodes[0].outputs[0], made_nodes[1].inputs[0])

    # Scalar Math node
    made_nodes[2].current_op = 'MUL'
    made_nodes[2].y_ = 2.5
    links.new(made_nodes[1].outputs[1], made_nodes[2].inputs[0])   # snlite-> math
    links.new(made_nodes[2].outputs[0], made_nodes[3].inputs[0])   # math -> float

    # Float2Int node
    # made_nodes[3]
    links.new(made_nodes[3].outputs[0], made_nodes[4].inputs[2])

    # Float range
    made_nodes[4].range_mode = 'RANGE_COUNT'
    made_nodes[4].stop_float = 1.0
    links.new(made_nodes[4].outputs[0], made_nodes[5].inputs[1])

    # Vector Interpolate
    made_nodes[5]
    links.new(made_nodes[0].outputs[0], made_nodes[5].inputs[0])
    links.new(made_nodes[5].outputs[0], made_nodes[6].inputs[0])

    # UV connect
    made_nodes[6].polygons = 'Edges'
    made_nodes[6].slice_check = False
    made_nodes[6].dir_check = 'U_dir'

    # Viewer Draw
    made_nodes[7]
    links.new(made_nodes[6].outputs[0], made_nodes[7].inputs[0])
    links.new(made_nodes[6].outputs[1], made_nodes[7].inputs[1])


def gp_macro_two(context, operator, term, nodes, links):
    needed_nodes = [
        ['SvGetAssetPropertiesMK2', (0.00, 0.00)],
        ['SvPathLengthMk2Node', (250, 55)],
        ['SvScalarMathNodeMK4', (430, 115)],
        # needs a list join lev2 from MULX to VECINTRANGE
        ['SvInterpolationNodeMK3', (680, 40)],
        ['LineConnectNodeMK2', (860, -40)],
        ['SvViewerDrawMk4', (1045, 50)],
        ['ListJoinNode', (430, -20)]
    ]

    made_nodes = []
    x, y = context.space_data.cursor_location[:]
    for node_bl_idname, node_location in needed_nodes:
        asset_node = "SvGetAssetPropertiesMK2"
        if node_bl_idname == asset_node and context.active_node and context.active_node.bl_idname == asset_node:
            # in this case pick up the active object id selector node.
            n = context.active_node
        else:
            n = nodes.new(node_bl_idname)

        n.location = node_location[0] + x, node_location[1] + y
        made_nodes.append(n)

    # nodes!
    obj_id, path_len, scalar_math, vec_int, uv_con, drawnode, listjoin = made_nodes

    # -- node settings --
    obj_id.Mode = 'grease_pencils'
    vec_int.infer_from_integer_input = True
    scalar_math.current_op = 'MUL'
    scalar_math.y_ = 25
    uv_con.polygons = 'Edges'
    uv_con.slice_check = False
    uv_con.dir_check = 'U_dir'
    listjoin.JoinLevel = 2
    listjoin.hide = True


    sv_sock(obj_id.outputs[0]) > sv_sock(path_len.inputs[0])
    sv_sock(obj_id.outputs[0]) > sv_sock(vec_int.inputs[0])
    sv_sock(path_len.outputs[1]) > sv_sock(scalar_math.inputs[0])

    sv_sock(scalar_math.outputs[0]) > sv_sock(listjoin.inputs[0])
    sv_sock(listjoin.outputs[0]) > sv_sock(vec_int.inputs[1])

    sv_sock(vec_int.outputs[0]) > sv_sock(uv_con.inputs[0])
    sv_sock(uv_con.outputs[0]) > sv_sock(drawnode.inputs[0])
    sv_sock(uv_con.outputs[1]) > sv_sock(drawnode.inputs[1])