tools/roi.py

# 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: tilegrid.json
with open(sys.argv[1]) as tb_f:
	tbj = json.load(tb_f)
tile_to_frames = {}
frame_to_tiles = {}
for tilename, tiledata in tbj.items():
	tn = tilename.split(":")[0]
	tile_offset = 0
	tile_to_frames[tn] = []
	for chunk in tiledata:
		frame, start, size = chunk
		tile_to_frames[tn].append(frame)
		if frame not in frame_to_tiles:
			frame_to_tiles[frame] = []
		frame_to_tiles[frame].append(tn)
basis_tiles = [
	"CLEM_X41Y120",
	"INT_X41Y120",
	"CLEL_R_X41Y120",
	"BRAM_X42Y120",
	"INT_INTF_L_X42Y120",
	"INT_X42Y120",
	"CLEL_R_X42Y120",
	"CLEM_X43Y120",
	"INT_X43Y120",
	"INT_INTF_R_X43Y120",
	"DSP_X43Y120",
	"CLEM_X44Y120",
	"INT_X44Y120",
	"CLEL_R_X44Y120",
	"CLEM_X45Y120",
	"INT_X45Y120",
	"INT_INTF_R_X45Y120",
	"DSP_X45Y120",
	"CLEM_X46Y120",
	"INT_X46Y120",
	"CLEL_R_X46Y120",
	"INT_X47Y120",
	"CLEL_R_X47Y120"
]

roi_frames = set()

for tile in basis_tiles:
	if tile not in tile_to_frames:
		continue
	for frame in tile_to_frames[tile]:
		roi_frames.add(frame)

roi_tiles = set()
for frame in roi_frames:
	for tile in frame_to_tiles[frame]:
		roi_tiles.add(tile)

with open(sys.argv[2], "w") as frames_f: 
	for frame in sorted(roi_frames):
		print("0x%08x" % frame, file=frames_f)

with open(sys.argv[3], "w") as tiles_f:
	for tile in sorted(roi_tiles):
		print("tile %s" % tile, file=tiles_f)