#-------------------------------------------------------------
#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you 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.
#
#-------------------------------------------------------------

class Topk:
    k: int
    min_score: float
    slices: []
    keys: []

    def __init__(self, k):
        self.k = k
        self.slices = []
        self.keys = []
        self.min_score = 1

    def top_k_min_score(self):
        self.slices.sort(key=lambda x: x.score, reverse=True)
        self.min_score = self.slices[len(self.slices) - 1].score
        return self.min_score

    def add_new_top_slice(self, new_top_slice):
        self.min_score = new_top_slice.score
        if len(self.slices) < self.k:
            self.slices.append(new_top_slice)
            self.keys.append(new_top_slice.key)
            return self.top_k_min_score()
        else:
            self.slices[len(self.slices) - 1] = new_top_slice
            self.keys[len(self.slices) - 1] = new_top_slice.key
            return self.top_k_min_score()

    def print_topk(self):
        for candidate in self.slices:
            print(candidate.name + ": " + "score = " + str(candidate.score) + "; size = " + str(candidate.size))

    def buckets_top_k(self, cur_lvl_slices, x_size, alpha):
        for bucket in cur_lvl_slices:
            if bucket.check_constraint(self, x_size, alpha):
                self.add_new_top_slice(bucket)
        return self