pebble/tools/tests/test_json2commands.py
2025-01-27 11:38:16 -08:00

506 lines
24 KiB
Python

# Copyright 2024 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import os
import sys
import unittest
import array
from struct import pack
from generate_pdcs import json2commands, pebble_commands
class MyTestCase(unittest.TestCase):
def test_parse_color(self):
truncate = True
# Test valid values
color_opacity = [0, 0.333, 0.666, 1]
color = json2commands.parse_color(color_opacity, truncate)
self.assertEqual(color, pebble_commands.convert_color(0, 85, 170, 255, truncate))
# Test invalid values
color_opacity = [0, 0.333, 0.666, 2]
color = json2commands.parse_color(color_opacity, truncate)
self.assertEqual(color, 0)
color_opacity = [0, 0.333, 0.666, -2]
color = json2commands.parse_color(color_opacity, truncate)
self.assertEqual(color, 0)
def test_parse_json_line_data(self):
# Test skipping of invisible segments
json_line_data = [{
"startPoint": [json2commands.INVISIBLE_POINT_THRESHOLD + 1, 1.0],
"endPoint": [2.0, 1.0]
},{
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
}]
bidirectional_lines = json2commands.parse_json_line_data(json_line_data)
self.assertEqual(bidirectional_lines, [ ((1.0, 1.0), (2.0, 1.0)),
((2.0, 1.0), (1.0, 1.0))])
json_line_data = [{
"startPoint": [2.0, 1.0],
"endPoint": [2.0, -json2commands.INVISIBLE_POINT_THRESHOLD - 1]
}]
bidirectional_lines = json2commands.parse_json_line_data(json_line_data)
self.assertEqual(bidirectional_lines, [])
# Test skipping of duplicate segments
json_line_data = [{
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
},{
"startPoint": [2.0, 1.0],
"endPoint": [1.0, 1.0]
},{
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
}]
bidirectional_lines = json2commands.parse_json_line_data(json_line_data)
self.assertEqual(bidirectional_lines, [ ((1.0, 1.0), (2.0, 1.0)),
((2.0, 1.0), (1.0, 1.0))])
def test_determine_longest_path(self):
# Test point
json_line_data = [{
"startPoint": [1.0, 1.0],
"endPoint": [1.0, 1.0]
}]
bidirectional_lines = json2commands.parse_json_line_data(json_line_data)
longest_path = json2commands.determine_longest_path(bidirectional_lines)
self.assertEqual(longest_path, [(1.0, 1.0), (1.0, 1.0)])
self.assertEqual(len(bidirectional_lines), 0)
# Test simplest line segment
json_line_data = [{
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
}]
bidirectional_lines = json2commands.parse_json_line_data(json_line_data)
longest_path = json2commands.determine_longest_path(bidirectional_lines)
self.assertEqual(longest_path, [(1.0, 1.0), (2.0, 1.0)])
self.assertEqual(len(bidirectional_lines), 0)
# Test ordered connected line segments
json_line_data = [{
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
},{
"startPoint": [2.0, 1.0],
"endPoint": [2.0, 2.0]
}]
bidirectional_lines = json2commands.parse_json_line_data(json_line_data)
longest_path = json2commands.determine_longest_path(bidirectional_lines)
self.assertEqual(longest_path, [(1.0, 1.0), (2.0, 1.0), (2.0, 2.0)])
self.assertEqual(len(bidirectional_lines), 0)
# Test unordered connected acyclic line segments
json_line_data = [{
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
},{
"startPoint": [2.0, 2.0],
"endPoint": [2.0, 1.0]
}]
bidirectional_lines = json2commands.parse_json_line_data(json_line_data)
longest_path = json2commands.determine_longest_path(bidirectional_lines)
self.assertEqual(longest_path, [(1.0, 1.0), (2.0, 1.0), (2.0, 2.0)])
self.assertEqual(len(bidirectional_lines), 0)
# Test connected simplest cyclic segments (circle)
json_line_data = [{
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
},{
"startPoint": [2.0, 1.0],
"endPoint": [2.0, 2.0]
},{
"startPoint": [2.0, 2.0],
"endPoint": [1.0, 1.0]
}]
bidirectional_lines = json2commands.parse_json_line_data(json_line_data)
longest_path = json2commands.determine_longest_path(bidirectional_lines)
self.assertEqual(longest_path, [(1.0, 1.0), (2.0, 1.0), (2.0, 2.0), (1.0, 1.0)])
self.assertEqual(len(bidirectional_lines), 0)
# Test connected complex cyclic line segments
json_line_data = [{
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
},{
"startPoint": [2.0, 1.0],
"endPoint": [2.0, 2.0]
},{
"startPoint": [2.0, 2.0],
"endPoint": [1.0, 1.0]
},{
"startPoint": [1.0, 2.0],
"endPoint": [1.0, 1.0]
}]
bidirectional_lines = json2commands.parse_json_line_data(json_line_data)
longest_path = json2commands.determine_longest_path(bidirectional_lines)
self.assertEqual(longest_path, [(1.0, 2.0), (1.0, 1.0), (2.0, 1.0), (2.0, 2.0), (1.0, 1.0)])
self.assertEqual(len(bidirectional_lines), 0)
# Test connected segments with more than one path
json_line_data = [{
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
},{
"startPoint": [2.0, 2.0],
"endPoint": [2.0, 1.0]
},{
"startPoint": [3.0, 1.0],
"endPoint": [2.0, 1.0]
}]
bidirectional_lines = json2commands.parse_json_line_data(json_line_data)
longest_path = json2commands.determine_longest_path(bidirectional_lines)
self.assertEqual(longest_path, [(3.0, 1.0), (2.0, 1.0), (1.0, 1.0)])
self.assertEqual(bidirectional_lines, [((2.0, 2.0), (2.0, 1.0)), ((2.0, 1.0), (2.0, 2.0))])
longest_path = json2commands.determine_longest_path(bidirectional_lines)
self.assertEqual(longest_path, [(2.0, 1.0), (2.0, 2.0)])
self.assertEqual(len(bidirectional_lines), 0)
# Test (ordered) unconnected segments (implicitly more than one path)
json_line_data = [{
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
},{
"startPoint": [1.0, 2.0],
"endPoint": [2.0, 2.0]
}]
bidirectional_lines = json2commands.parse_json_line_data(json_line_data)
longest_path = json2commands.determine_longest_path(bidirectional_lines)
self.assertEqual(longest_path, [(1.0, 2.0), (2.0, 2.0)])
self.assertEqual(bidirectional_lines, [((1.0, 1.0), (2.0, 1.0)), ((2.0, 1.0), (1.0, 1.0))])
longest_path = json2commands.determine_longest_path(bidirectional_lines)
self.assertEqual(longest_path, [(1.0, 1.0), (2.0, 1.0)])
self.assertEqual(len(bidirectional_lines), 0)
def test_process_fill(self):
# Test that line style is taken from first segment
fillGroup_data = [{
"thickness": 3,
"color": [0, 0, 0, 1],
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
},{
"thickness": 3,
"color": [0, 0, 0.666, 0.333],
"startPoint": [2.0, 1.0],
"endPoint": [2.0, 2.0]
},{
"thickness": 3,
"color": [0.666, 0.333, 0, 0],
"startPoint": [2.0, 2.0],
"endPoint": [1.0, 1.0]
}]
truncate_color = True
fill_commands, error = json2commands.process_fill(
fillGroup_data,
(0, 0),
(json2commands.DISPLAY_DIM_X, json2commands.DISPLAY_DIM_Y),
False,
False,
False,
truncate_color)
self.assertFalse(error)
fill_command = fill_commands[0]
self.assertIsInstance(fill_command, pebble_commands.PathCommand)
self.assertEqual(len(fill_command.points), 4)
self.assertEqual(fill_command.fill_color, json2commands.parse_color([1, 1, 1, 1], truncate_color))
self.assertEqual(fill_command.stroke_color, json2commands.parse_color([0, 0, 0, 1], truncate_color))
self.assertEqual(fill_command.stroke_width, 3)
self.assertTrue(pebble_commands.compare_points(fill_command.points[0], (1.0, 1.0)))
self.assertTrue(pebble_commands.compare_points(fill_command.points[1], (2.0, 1.0)))
self.assertTrue(pebble_commands.compare_points(fill_command.points[2], (2.0, 2.0)))
self.assertTrue(pebble_commands.compare_points(fill_command.points[3], (1.0, 1.0)))
self.assertFalse(fill_command.open)
# Test that fill with no stroke width has no stroke color
fillGroup_data = [{
"thickness": 0,
"color": [0.666, 0, 0.666, 1],
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
},{
"thickness": 3,
"color": [0, 0, 0, 1],
"startPoint": [2.0, 1.0],
"endPoint": [2.0, 2.0]
},{
"thickness": 3,
"color": [0, 0, 0, 1],
"startPoint": [2.0, 2.0],
"endPoint": [1.0, 1.0]
}]
truncate_color = True
fill_commands, error = json2commands.process_fill(
fillGroup_data,
(0, 0),
(json2commands.DISPLAY_DIM_X, json2commands.DISPLAY_DIM_Y),
False,
False,
False,
truncate_color)
self.assertFalse(error)
fill_command = fill_commands[0]
self.assertIsInstance(fill_command, pebble_commands.PathCommand)
self.assertEqual(len(fill_command.points), 4)
self.assertEqual(fill_command.fill_color, json2commands.parse_color([1, 1, 1, 1], truncate_color))
self.assertEqual(fill_command.stroke_color, 0)
self.assertEqual(fill_command.stroke_width, 0)
self.assertTrue(pebble_commands.compare_points(fill_command.points[0], (1.0, 1.0)))
self.assertTrue(pebble_commands.compare_points(fill_command.points[1], (2.0, 1.0)))
self.assertTrue(pebble_commands.compare_points(fill_command.points[2], (2.0, 2.0)))
self.assertTrue(pebble_commands.compare_points(fill_command.points[3], (1.0, 1.0)))
self.assertFalse(fill_command.open)
# Test that fill with no stroke color has no stroke width
fillGroup_data = [{
"thickness": 3,
"color": [0, 1, 0.666, 0],
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
},{
"thickness": 3,
"color": [0, 0, 0, 1],
"startPoint": [2.0, 1.0],
"endPoint": [2.0, 2.0]
},{
"thickness": 3,
"color": [0, 0, 0, 1],
"startPoint": [2.0, 2.0],
"endPoint": [1.0, 1.0]
}]
truncate_color = True
fill_commands, error = json2commands.process_fill(
fillGroup_data,
(0, 0),
(json2commands.DISPLAY_DIM_X, json2commands.DISPLAY_DIM_Y),
False,
False,
False,
truncate_color)
self.assertFalse(error)
fill_command = fill_commands[0]
self.assertIsInstance(fill_command, pebble_commands.PathCommand)
self.assertEqual(len(fill_command.points), 4)
self.assertEqual(fill_command.fill_color, json2commands.parse_color([1, 1, 1, 1], truncate_color))
self.assertEqual(fill_command.stroke_color, 0)
self.assertEqual(fill_command.stroke_width, 0)
self.assertTrue(pebble_commands.compare_points(fill_command.points[0], (1.0, 1.0)))
self.assertTrue(pebble_commands.compare_points(fill_command.points[1], (2.0, 1.0)))
self.assertTrue(pebble_commands.compare_points(fill_command.points[2], (2.0, 2.0)))
self.assertTrue(pebble_commands.compare_points(fill_command.points[3], (1.0, 1.0)))
self.assertFalse(fill_command.open)
def test_process_open_paths(self):
# Test group of varying stroke width and stroke color.
fillGroup_data = [{
"thickness": 3,
"color": [0, 0, 0, 1], # 192
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 2.0]
},{
"thickness": 5,
"color": [0, 0, 0.333, 0.333], # 65
"startPoint": [3.0, 3.0],
"endPoint": [4.0, 4.0]
},{
"thickness": 5,
"color": [0, 0.666, 0, 0.333], # 72
"startPoint": [4.0, 4.0],
"endPoint": [4.0, 5.0]
},{
"thickness": 3,
"color": [0, 0, 0.666, 0.333], # 66
"startPoint": [2.0, 1.0],
"endPoint": [2.0, 2.0]
},{
"thickness": 2,
"color": [0.666, 0.333, 0, 1], # 228
"startPoint": [10.0, 10.0],
"endPoint": [11.0, 11.0]
},{
"thickness": 3,
"color": [0, 0, 0, 1], # 192
"startPoint": [1.0, 2.0],
"endPoint": [2.0, 2.0]
}]
truncate_color = True
open_path_commands, error = json2commands.process_open_paths(
fillGroup_data,
(0, 0),
(json2commands.DISPLAY_DIM_X, json2commands.DISPLAY_DIM_Y),
True,
False,
False,
truncate_color)
self.assertFalse(error)
self.assertEqual(len(open_path_commands), 5)
width_2_color_228_command = open_path_commands[0]
self.assertIsInstance(width_2_color_228_command, pebble_commands.PathCommand)
self.assertEqual(width_2_color_228_command.stroke_width, 2)
self.assertEqual(width_2_color_228_command.stroke_color, json2commands.parse_color([0.666, 0.333, 0, 1],
truncate_color))
self.assertEqual(len(width_2_color_228_command.points), 2)
self.assertEqual(width_2_color_228_command.fill_color, json2commands.parse_color([0, 0, 0, 0],
truncate_color))
self.assertTrue(pebble_commands.compare_points(width_2_color_228_command.points[0], (11.0, 11.0)))
self.assertTrue(pebble_commands.compare_points(width_2_color_228_command.points[1], (10.0, 10.0)))
self.assertTrue(width_2_color_228_command.open)
width_3_color_192_command = open_path_commands[1]
self.assertIsInstance(width_3_color_192_command, pebble_commands.PathCommand)
self.assertEqual(width_3_color_192_command.stroke_width, 3)
self.assertEqual(width_3_color_192_command.stroke_color, json2commands.parse_color([0, 0, 0, 1],
truncate_color))
self.assertEqual(len(width_3_color_192_command.points), 3)
self.assertEqual(width_3_color_192_command.fill_color, json2commands.parse_color([0, 0, 0, 0],
truncate_color))
self.assertTrue(pebble_commands.compare_points(width_3_color_192_command.points[0], (1.0, 2.0)))
self.assertTrue(pebble_commands.compare_points(width_3_color_192_command.points[1], (2.0, 2.0)))
self.assertTrue(pebble_commands.compare_points(width_3_color_192_command.points[2], (1.0, 1.0)))
self.assertTrue(width_3_color_192_command.open)
width_3_color_66_command = open_path_commands[2]
self.assertIsInstance(width_3_color_66_command, pebble_commands.PathCommand)
self.assertEqual(width_3_color_66_command.stroke_width, 3)
self.assertEqual(width_3_color_66_command.stroke_color, json2commands.parse_color([0, 0, 0.666, 0.333],
truncate_color))
self.assertEqual(len(width_3_color_66_command.points), 2)
self.assertEqual(width_3_color_66_command.fill_color, json2commands.parse_color([0, 0, 0, 0],
truncate_color))
self.assertTrue(pebble_commands.compare_points(width_3_color_66_command.points[0], (2.0, 1.0)))
self.assertTrue(pebble_commands.compare_points(width_3_color_66_command.points[1], (2.0, 2.0)))
self.assertTrue(width_3_color_66_command.open)
width_5_color_65_command = open_path_commands[3]
self.assertIsInstance(width_5_color_65_command, pebble_commands.PathCommand)
self.assertEqual(width_5_color_65_command.stroke_width, 5)
self.assertEqual(width_5_color_65_command.stroke_color, json2commands.parse_color([0, 0, 0.333, 0.333],
truncate_color))
self.assertEqual(len(width_5_color_65_command.points), 2)
self.assertEqual(width_5_color_65_command.fill_color, json2commands.parse_color([0, 0, 0, 0],
truncate_color))
self.assertTrue(pebble_commands.compare_points(width_5_color_65_command.points[0], (4.0, 4.0)))
self.assertTrue(pebble_commands.compare_points(width_5_color_65_command.points[1], (3.0, 3.0)))
self.assertTrue(width_5_color_65_command.open)
width_5_color_72_command = open_path_commands[4]
self.assertIsInstance(width_5_color_72_command, pebble_commands.PathCommand)
self.assertEqual(width_5_color_72_command.stroke_width, 5)
self.assertEqual(width_5_color_72_command.stroke_color, json2commands.parse_color([0, 0.666, 0, 0.333],
truncate_color))
self.assertEqual(len(width_5_color_72_command.points), 2)
self.assertEqual(width_5_color_72_command.fill_color, json2commands.parse_color([0, 0, 0, 0],
truncate_color))
self.assertTrue(pebble_commands.compare_points(width_5_color_72_command.points[0], (4.0, 5.0)))
self.assertTrue(pebble_commands.compare_points(width_5_color_72_command.points[1], (4.0, 4.0)))
self.assertTrue(width_5_color_72_command.open)
# Test that open with no stroke width has no stroke color
fillGroup_data = [{
"thickness": 0,
"color": [0.666, 0, 0.666, 1],
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
}]
truncate_color = True
open_path_commands, error = json2commands.process_open_paths(
fillGroup_data,
(0, 0),
(json2commands.DISPLAY_DIM_X, json2commands.DISPLAY_DIM_Y),
True,
False,
False,
truncate_color)
self.assertFalse(error)
open_path_command = open_path_commands[0]
self.assertIsInstance(open_path_command, pebble_commands.PathCommand)
self.assertEqual(len(open_path_command.points), 2)
self.assertEqual(open_path_command.fill_color, json2commands.parse_color([0, 0, 0, 0], truncate_color))
self.assertEqual(open_path_command.stroke_color, 0)
self.assertEqual(open_path_command.stroke_width, 0)
self.assertTrue(pebble_commands.compare_points(open_path_command.points[0], (1.0, 1.0)))
self.assertTrue(pebble_commands.compare_points(open_path_command.points[1], (2.0, 1.0)))
self.assertTrue(open_path_command.open)
# Test that open with no stroke color has no stroke width
fillGroup_data = [{
"thickness": 3,
"color": [0, 1, 0.666, 0],
"startPoint": [1.0, 1.0],
"endPoint": [2.0, 1.0]
}]
truncate_color = True
open_path_commands, error = json2commands.process_open_paths(
fillGroup_data,
(0, 0),
(json2commands.DISPLAY_DIM_X, json2commands.DISPLAY_DIM_Y),
True,
False,
False,
truncate_color)
self.assertFalse(error)
open_path_command = open_path_commands[0]
self.assertIsInstance(open_path_command, pebble_commands.PathCommand)
self.assertEqual(len(open_path_command.points), 2)
self.assertEqual(open_path_command.fill_color, json2commands.parse_color([0, 0, 0, 0], truncate_color))
self.assertEqual(open_path_command.stroke_color, 0)
self.assertEqual(open_path_command.stroke_width, 0)
self.assertTrue(pebble_commands.compare_points(open_path_command.points[0], (1.0, 1.0)))
self.assertTrue(pebble_commands.compare_points(open_path_command.points[1], (2.0, 1.0)))
self.assertTrue(open_path_command.open)
def test_parse_json_sequence(self):
# Test mix of fills and open paths with mulitple frames
current_path = os.path.dirname(os.path.realpath(__file__))
filename = current_path + '/json2commands_test.json'
frames, errors, frame_duration = json2commands.parse_json_sequence(filename, (80, 80), False, False)
self.assertEqual(frame_duration, 28)
self.assertEqual(len(frames), 2)
self.assertEqual(len(errors), 0)
frame_1 = frames[0]
self.assertEqual(len(frame_1), 4)
self.assertFalse(frame_1[0].open)
self.assertFalse(frame_1[1].open)
self.assertTrue(frame_1[2].open)
self.assertTrue(frame_1[3].open)
frame_2 = frames[1]
self.assertEqual(len(frame_2), 2)
self.assertFalse(frame_2[0].open)
self.assertTrue(frame_2[1].open)
if __name__ == '__main__':
unittest.main()