# Copyright (c) Facebook, Inc. and its affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.
#


import json
import pickle

if __name__ == '__main__':
    val_annotation_file="v2_mscoco_val2014_annotations.json"
    minival_id_file = "data/vqa_v2.0/minival_ids.pkl"
    minival_annotation_file="v2_mscoco_minival2014_annotations.json"

    with open(minival_id_file, 'rb') as f:
        q_im_ids = pickle.load(f)

    minival_ids = [x[1] for x in q_im_ids]

    with open(val_annotation_file, 'r') as f:
        file_info = json.load(f)
        annotations = file_info['annotations']
        info = file_info['info']
        data_subtype = file_info['data_subtype']
        license_info = file_info['license']

    minival_annotations = [
        a for a in annotations if a['question_id'] in minival_ids]

    minival_info = {
        'data_subtype': data_subtype,
        'license': license_info,
        'info': info,
        'annotations': minival_annotations
    }

    with open(minival_annotation_file, 'w') as w:
        json.dump(minival_info, w)