# SPDX-License-Identifier: Apache-2.0
#
# Copyright (C) 2015, ARM Limited and contributors.
#
# Licensed 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.
#

import logging

""" Helper module for Analysis classes """


class AnalysisModule(object):
    """
    Base class for Analysis modules.

    :param trace: input Trace object
    :type trace: :mod:`libs.utils.Trace`
    """

    def __init__(self, trace):

        self._log = logging.getLogger('Analysis')

        self._trace = trace
        self._platform = trace.platform
        self._tasks = trace.tasks
        self._data_dir = trace.data_dir

        self._dfg_trace_event = trace._dfg_trace_event

        trace._registerDataFrameGetters(self)

        # Further initialization not possible if platform info is missing
        if not self._platform:
            return

        # By default assume SMP system
        self._big_cap = 1024
        self._little_cap = 1024
        nrg_model = self._platform.get('nrg_model', None)
        if nrg_model:
            try:
                self._big_cap = nrg_model['big']['cpu']['cap_max']
                self._little_cap = nrg_model['little']['cpu']['cap_max']
            except TypeError:
                self._log.debug('Failed parsing EM from platform data')
        else:
            self._log.debug('Platform data without Energy Model info')

        # By default assume SMP system
        self._big_cpus = []
        self._little_cpus = []
        if 'big' in self._platform['clusters']:
            self._log.debug('Parsing big.LITTLE system clusters')
            self._big_cpus = self._platform['clusters']['big']
            self._little_cpus = self._platform['clusters']['little']
        else:
            self._log.debug('Parsing SMP clusters')
            for cid in self._platform['clusters']:
                self._big_cpus.append(self._platform['clusters'][cid])

# vim :set tabstop=4 shiftwidth=4 expandtab