#!/usr/bin/env python3
#
# Copyright (c) 2021 ARM Limited
# All rights reserved
#
# The license below extends only to copyright in the software and shall
# not be construed as granting a license to any other intellectual
# property including but not limited to intellectual property relating
# to a hardware implementation of the functionality of the software
# licensed hereunder.  You may use the software subject to the license
# terms below provided that you ensure that this notice is replicated
# unmodified and in its entirety in all distributions of the software,
# modified or unmodified, in source code or in binary form.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met: redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer;
# redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution;
# neither the name of the copyright holders nor the names of its
# contributors may be used to endorse or promote products derived from
# this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

import unittest

from m5.util import convert


def _ip(*args):
    return (args[0] << 24) | (args[1] << 16) | (args[2] << 8) | args[3]


class ConvertTestSuite(unittest.TestCase):
    """Test cases for unit conversion"""

    def test_toMetricFloat(self):
        def conv(x):
            return convert.toMetricFloat(x, "value", "X")

        self.assertEqual(conv("42"), 42e0)
        self.assertEqual(conv("42.5"), 42.5e0)
        self.assertEqual(conv("42kX"), 42e3)
        self.assertEqual(conv("42.5kX"), 42.5e3)
        self.assertEqual(conv("42MX"), 42e6)
        self.assertEqual(conv("42GX"), 42e9)
        self.assertEqual(conv("42TX"), 42e12)
        self.assertEqual(conv("42PX"), 42e15)
        self.assertEqual(conv("42EX"), 42e18)

        self.assertEqual(conv("42KiX"), 42 * 2**10)
        self.assertEqual(conv("42MiX"), 42 * 2**20)
        self.assertEqual(conv("42GiX"), 42 * 2**30)
        self.assertEqual(conv("42TiX"), 42 * 2**40)
        self.assertEqual(conv("42PiX"), 42 * 2**50)
        self.assertEqual(conv("42EiX"), 42 * 2**60)

        self.assertRaises(ValueError, conv, "42k")
        self.assertRaises(ValueError, conv, "42KX")
        self.assertRaises(ValueError, conv, "42kiX")

        self.assertEqual(convert.toMetricFloat("42"), 42)
        # Prefixes not allowed without a unit
        self.assertRaises(ValueError, convert.toMetricFloat, "42k")

    def test_toMetricInteger(self):
        def conv(x):
            return convert.toMetricInteger(x, "value", "X")

        self.assertEqual(conv("42"), 42 * 10**0)
        self.assertEqual(conv("42kX"), 42 * 10**3)
        self.assertEqual(conv("42MX"), 42 * 10**6)
        self.assertEqual(conv("42GX"), 42 * 10**9)
        self.assertEqual(conv("42TX"), 42 * 10**12)
        self.assertEqual(conv("42PX"), 42 * 10**15)
        self.assertEqual(conv("42EX"), 42 * 10**18)

        self.assertEqual(conv("42KiX"), 42 * 2**10)
        self.assertEqual(conv("42MiX"), 42 * 2**20)
        self.assertEqual(conv("42GiX"), 42 * 2**30)
        self.assertEqual(conv("42TiX"), 42 * 2**40)
        self.assertEqual(conv("42PiX"), 42 * 2**50)
        self.assertEqual(conv("42EiX"), 42 * 2**60)

        self.assertRaises(ValueError, conv, "42.1")
        self.assertRaises(ValueError, conv, "42.1kX")

        self.assertRaises(ValueError, conv, "42k")
        self.assertRaises(ValueError, conv, "42KX")
        self.assertRaises(ValueError, conv, "42kiX")

        self.assertEqual(convert.toMetricInteger("42"), 42)

        # Prefixes not allowed without a unit
        self.assertRaises(ValueError, convert.toMetricInteger, "42k")

    def test_toBool(self):
        conv = convert.toBool

        self.assertEqual(conv("TRUE"), True)
        self.assertEqual(conv("true"), True)
        self.assertEqual(conv("t"), True)
        self.assertEqual(conv("yes"), True)
        self.assertEqual(conv("y"), True)
        self.assertEqual(conv("1"), True)

        self.assertEqual(conv("FALSE"), False)
        self.assertEqual(conv("false"), False)
        self.assertEqual(conv("f"), False)
        self.assertEqual(conv("no"), False)
        self.assertEqual(conv("n"), False)
        self.assertEqual(conv("0"), False)

        self.assertRaises(ValueError, conv, "not a bool")
        self.assertRaises(ValueError, conv, "2")

    def test_toFrequency(self):
        conv = convert.toFrequency

        self.assertEqual(conv("42"), 42.0)
        self.assertEqual(conv("42Hz"), 42)
        self.assertEqual(conv("42kHz"), 42e3)

        # Prefixes need a unit
        self.assertRaises(ValueError, conv, "42k")
        # Seconds isn't a valid unit unless using anyToFrequency.
        self.assertRaises(ValueError, conv, "42s")

    def test_toLatency(self):
        conv = convert.toLatency

        self.assertEqual(conv("42"), 42.0)
        self.assertEqual(conv("42s"), 42.0)

        # We allow prefixes for seconds.
        self.assertEqual(conv("42ks"), 42e3)

        # Prefixe need a unit
        self.assertRaises(ValueError, conv, "42k")
        # Hz shouldn't be converted unless using anyToLatency
        self.assertRaises(ValueError, conv, "42Hz")

    def test_anyToLatency(self):
        conv = convert.anyToLatency

        self.assertEqual(conv("42s"), 42.0)

        # We currently allow prefixes for seconds.
        self.assertEqual(conv("42ks"), 42e3)

        self.assertEqual(conv("10Hz"), 0.1)
        self.assertEqual(conv("1kHz"), 1e-3)

        self.assertRaises(ValueError, conv, "42k")
        self.assertRaises(ValueError, conv, "42")

    def test_anyToFrequency(self):
        conv = convert.anyToFrequency

        self.assertEqual(conv("42kHz"), 42e3)

        self.assertEqual(conv("0.1s"), 10.0)
        self.assertEqual(conv("1ms"), 1000.0)

        self.assertRaises(ValueError, conv, "42k")
        self.assertRaises(ValueError, conv, "42")

    def test_toNetworkBandwidth(self):
        conv = convert.toNetworkBandwidth

        self.assertEqual(conv("42"), 42.0)
        self.assertEqual(conv("42bps"), 42.0)
        self.assertEqual(conv("42kbps"), 42e3)

        self.assertRaises(ValueError, conv, "42Kbps")

    def test_toMemoryBandwidth(self):
        conv = convert.toMemoryBandwidth

        self.assertEqual(conv("42"), 42.0)
        self.assertEqual(conv("42B/s"), 42.0)

        self.assertEqual(conv("42MB/s"), 42 * 2**20)
        self.assertEqual(conv("42MiB/s"), 42 * 2**20)

        self.assertRaises(ValueError, conv, "42KB/s")
        self.assertRaises(ValueError, conv, "42Mi")

    def test_toMemorySize(self):
        conv = convert.toMemorySize

        self.assertEqual(conv("42"), 42.0)
        self.assertEqual(conv("42B"), 42.0)

        self.assertEqual(conv("42kB"), 42 * 2**10)
        self.assertEqual(conv("42MB"), 42 * 2**20)

        self.assertEqual(conv("42KiB"), 42 * 2**10)
        self.assertEqual(conv("42MiB"), 42 * 2**20)

    def test_toIpAddress(self):
        conv = convert.toIpAddress

        self.assertEqual(conv("255.255.255.255"), _ip(255, 255, 255, 255))
        self.assertEqual(conv("1.2.3.4"), _ip(1, 2, 3, 4))

        self.assertRaises(TypeError, conv, 0)
        self.assertRaises(ValueError, conv, "0.0.0")
        self.assertRaises(ValueError, conv, "0.0.0.300")
        self.assertRaises(ValueError, conv, "0.0.0.0.0")

    def test_toIpNetmask(self):
        conv = convert.toIpNetmask

        self.assertEqual(conv("1.2.3.4/24"), (_ip(1, 2, 3, 4), 24))
        self.assertEqual(conv("1.2.3.4/255.255.255.0"), (_ip(1, 2, 3, 4), 24))

        self.assertEqual(conv("1.2.3.4/0"), (_ip(1, 2, 3, 4), 0))
        self.assertEqual(conv("1.2.3.4/0.0.0.0"), (_ip(1, 2, 3, 4), 0))

        self.assertRaises(ValueError, conv, "0.0.0.0")
        self.assertRaises(ValueError, conv, "0.0.0.0/")
        self.assertRaises(ValueError, conv, "0.0.0.0/64")

    def test_toIpWithPort(self):
        conv = convert.toIpWithPort

        self.assertEqual(conv("1.2.3.4:42"), (_ip(1, 2, 3, 4), 42))

        self.assertRaises(ValueError, conv, "0.0.0.0")
        self.assertRaises(ValueError, conv, "0.0.0.0:")
        self.assertRaises(ValueError, conv, "0.0.0.0:65536")

    def test_toVoltage(self):
        conv = convert.toVoltage

        self.assertEqual(conv("42"), 42)
        self.assertEqual(conv("42V"), 42)
        self.assertEqual(conv("42kV"), 42e3)

    def test_toCurrent(self):
        conv = convert.toCurrent

        self.assertEqual(conv("42"), 42)
        self.assertEqual(conv("42A"), 42)
        self.assertEqual(conv("42kA"), 42e3)

    def test_toEnergy(self):
        conv = convert.toEnergy

        self.assertEqual(conv("42"), 42)
        self.assertEqual(conv("42J"), 42)
        self.assertEqual(conv("42kJ"), 42e3)

    def test_temperature(self):
        conv = convert.toTemperature

        self.assertEqual(conv("1.0K"), 1.0)
        self.assertEqual(conv("1.0mK"), 1.0e-3)

        self.assertEqual(conv("0C"), 273.15)
        self.assertEqual(conv("-1C"), 272.15)
        self.assertRaises(ValueError, conv, "1.0")
        self.assertRaises(ValueError, conv, "-1K")

        self.assertEqual(conv("32F"), 273.15)