import glob
import os
import json

# Data
keypoints = [
    "Point 1",
    "Point 2"
]

keypoints_style = [
    "#00FF00",
    "#00FFFF"
]

ellipses = [
	"Ellipse 1",
	"Ellipse 2"
]

ellipses_style = [
	"#FF00FF",
	"#FFFF00",
]

categories = [{
    "id" : "0",
    "name" : "black mouse",
    "supercategory" : "mouse",
    "keypoints" : keypoints,
    "keypoints_style" : keypoints_style,
    "ellipses": ellipses,
    "ellipses_style": ellipses_style,
},
{
    "id" : "1",
    "name" : "white mouse",
    "supercategory" : "mouse",
    "keypoints" : keypoints,
    "keypoints_style" : keypoints_style,
    "ellipses": ellipses,
    "ellipses_style": ellipses_style,
}]


image_dir_regx = '/home/lunarian/dev/personal/annotation_tools/local/mice/*.jpg'
images = []
for image_path in glob.glob(image_dir_regx):
    image_file_name = os.path.basename(image_path)
    image_id = os.path.splitext(image_file_name)[0]
    image_url = "http://localhost:6008/" + image_file_name
    images.append({
        "id" : image_id,
        "file_name" : image_file_name,
        "url" : image_url
    })


dataset = {
    "categories" : categories,
    "images" : images,
    "annotations" : [],
    "licenses" : []
}

dataset_file_path = "local/datasets/mouse_dataset.json"
with open(dataset_file_path, 'w') as f:
    json.dump(dataset, f)