# Copyright 2020 Project U-Ray Authors
#
# 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 re
import sys
import json
# Usage: frames.txt tiles.txt tilegrid.json

frame_line_re = re.compile(r'0x([0-9A-Fa-f]+).*')

frame_rc_height = {}

with open(sys.argv[1], 'r') as f:
    for line in f:
        m = frame_line_re.match(line)
        if not m:
            continue
        frame = int(m.group(1), 16)
        bus = (frame >> 24) & 0x7
        half = (frame >> 23) & 0x1
        row = (frame >> 18) & 0x1F
        col = (frame >> 8) & 0x3FF
        minor = frame & 0xFF
        if bus != 0 or half != 0:
            continue
        if (row, col) not in frame_rc_height:
            frame_rc_height[(row, col, frame & ~0xFF)] = minor + 1
        else:
            frame_rc_height[(row, col, frame & ~0xFF)] = max(
                frame_rc_height[(row, col)], minor + 1)

tiles_to_xy = {}
with open(sys.argv[2], 'r') as tilef:
    for line in tilef:
        sl = line.strip().split(",")
        if len(sl) < 4:
            continue
        x = int(sl[0])
        y = int(sl[1])
        name = sl[2]
        tiles_to_xy[name] = (x, y)

with open(sys.argv[3]) as tb_f:
    tbj = json.load(tb_f)

frames_to_tiles = {}
for tilename, tiledata in tbj.items():
    tile_offset = 0
    for chunk in tiledata:
        frame, start, size = chunk
        if frame not in frames_to_tiles:
            frames_to_tiles[frame] = []
        name = tilename.split(":")[0]
        frames_to_tiles[frame].append((start, tiles_to_xy[name][1],
                                       tiles_to_xy[name][0], name))
        tile_offset += size

for frame, tiles in frames_to_tiles.items():
    tiles.sort()

for rc, height in sorted(frame_rc_height.items()):
    row, col, frame = rc
    line = "%08x %6d %6d %6d" % (frame, row, col, height)
    print(line)
    frame = (row << 18) | (col << 8)
    last_start = 0
    if frame in frames_to_tiles and len(frames_to_tiles[frame]) > 0:
        for tile in frames_to_tiles[frame]:
            start, ty, tx, tname = tile
            print("                            %6d (%4d) %6d %6d %s" %
                  (start, start - last_start, tx, ty, tname))
            last_start = start