class AnnotationContainer:
    def __init__(self):
        self.clear()

    def filename(self):
        return self.filename_

    def clear(self):
        self.annotations_ = []
        self.filename_    = None

    def load(self, filename):
        self.loadRectFormat(filename)

    def loadRectFormat(self, filename):
        self.annotations_ = []
        self.filename_ = filename
        filename_helper = {}

        try:
            f = open(filename)
            for line in f:
                s = line.split()
                image_fname = s[0]
                num_rects = int(s[1])
                if image_fname in filename_helper:
                    rects = filename_helper[image_fname]['annotations']
                else:
                    rects = []

                for i in range(0, 4*num_rects, 4):
                    rects.append({
                        'type': 'rect',
                        'x': s[i+2],
                        'y': s[i+3],
                        'width':  s[i+4],
                        'height': s[i+5]
                        })

                if image_fname not in filename_helper:
                    annotation = {
                            'type':     'image',
                            'filename': image_fname,
                            'annotations': rects
                            }
                    self.annotations_.append(annotation)
                    filename_helper[image_fname] = annotation
        except Exception as e:
            self.annotations = []
            raise e


    def saveRectFormat(self, filename):
        f = open(filname, "w")

        for file in self.annotations_:
            if file['type'] == 'image':
                rect_anns = [ann for ann in file['annotations'] if ann['type'] == 'rect']
                f.write("%s %d", file['filename'], len(rect_anns))
                for ann in rect_anns:
                    f.write(' %s %s %s %s', str(ann['x']), str(ann['y']), str(ann['width']), str(ann['height']))
                f.write('\n')

    def asDict(self):
        return self.annotations_

    def numFiles(self):
        return len(self.annotations_)

    def numAnnotations(self):
        if self.annotations_ is None:
            return 0
        num = 0
        for file in self.annotations_:
            if file['type'] == 'image':
                num += len(file['annotations'])
            elif file['type'] == video:
                for frame in file['frames']:
                    num += len(frame['annotations'])
        return num