Module pygw.statistics.index.field_visibility_count_statistic

#
# Copyright (c) 2013-2022 Contributors to the Eclipse Foundation

#
# See the NOTICE file distributed with this work for additional information regarding copyright
# ownership. All rights reserved. This program and the accompanying materials are made available
# under the terms of the Apache License, Version 2.0 which accompanies this distribution and is
# available at http://www.apache.org/licenses/LICENSE-2.0.txt
# ===============================================================================================
from pygw.config import geowave_pkg
from ..statistic import IndexStatistic
from ..statistic_type import IndexStatisticType
from ...base.java_transformer import JavaTransformer
from ...base.type_conversions import PrimitiveByteArrayType


class FieldVisibilityCountStatistic(IndexStatistic):
    """
    Maintains a count of entries for every visibility.
    """
    STATS_TYPE = IndexStatisticType(geowave_pkg.core.store.statistics.index.FieldVisibilityCountStatistic.STATS_TYPE)

    def __init__(self, index_name=None, java_ref=None):
        if java_ref is None:
            if index_name is None:
                java_ref = geowave_pkg.core.store.statistics.index.FieldVisibilityCountStatistic()
            else:
                java_ref = geowave_pkg.core.store.statistics.index.FieldVisibilityCountStatistic(index_name)
        super().__init__(java_ref, FieldVisibilityCountTransformer())


class FieldVisibilityCountTransformer(JavaTransformer):
    def transform(self, j_object):
        field_visibilities = {}
        byte_array_type = PrimitiveByteArrayType()
        for key, value in j_object.items():
            field_visibilities[byte_array_type.from_java(key.getBytes())] = value
        return field_visibilities