helperfunctions.py

import datastructures as ds
import config as conf

def generateJSON(data: [], rot: [], mirr: [], zigzag: []):
    message = {}
    message['brightnessFactor'] = 0.1
    message['panels'] = []
    for i in range(len(data)):
        panel = {}
        panel["rot"] = rot[i]
        panel["mirr"] = mirr[i]
        panel["zigzag"] = zigzag[i]
        panel["dots"] = data[i]
        message['panels'].append(panel)
    return message

def clamp(x):
  return max(0, min(x, 255))

def clampTuple(x: tuple):
  return tuple([clamp(val) for val in x])

#point order in x1, y2, x2, y2...
def setPointsAbsolute(*argv):
    points = []
    i = 0
    while i < len(argv):

        tmp = ds.Vector()
        tmp.x = argv[i]
        tmp.y = argv[i + 1]

        if i == 0:
            tmp.x = 0
        elif i is len(argv) - 2:
            tmp.x = conf._MatrixSizeX - 1
        i += 2

        if tmp.x < 0:
            tmp.x = 0
        if tmp.y < 0:
            tmp.y = 0
        if tmp.x > conf._MatrixSizeX - 1:
            tmp.x = conf._MatrixSizeX - 1
        if tmp.y > conf._MatrixSizeY - 1:
            tmp.y = conf._MatrixSizeY - 1

        points.append(tmp)
    return points

#point order in x1, y1, x2, y2...
def setPointsRelative(*argv):
    points = []
    i = 0
    while i < len(argv):

        minX = 1 / conf._MatrixSizeX
        minY = 1 / conf._MatrixSizeY

        tmp = ds.Vector()
        tmp.x = argv[i]
        tmp.y = argv[i + 1]

        if i == 0:
            tmp.x = 0
        elif i is len(argv) - 2:
            tmp.x = 1
        i += 2

        if tmp.x < 0:
            tmp.x = 0
        if tmp.y < 0:
            tmp.y = 0
        if tmp.x > 1 - minX:
            tmp.x = 1 - minX
        if tmp.y > 1 - minY:
            tmp.y = 1 - minY

        scalingVector = ds.Vector(conf._MatrixSizeX, conf._MatrixSizeY)
        tmp *= scalingVector

        #convert to int bc dealing only with full pixels
        tmp.x = round(tmp.x)
        tmp.y = round(tmp.y)

        points.append(tmp)
    return points