"""Tests IR interpreter handling of basic floating point operations (fadd, fsub, etc)."""

import lldb
from lldbsuite.test.decorators import *
from lldbsuite.test.lldbtest import *
from lldbsuite.test import lldbutil

class FPEvalTestCase(TestBase):

    def setUp(self):
        # Call super's setUp().
        TestBase.setUp(self)
        self.jit_opts = lldb.SBExpressionOptions()
        self.jit_opts.SetAllowJIT(True)
        self.no_jit_opts = lldb.SBExpressionOptions()
        self.no_jit_opts.SetAllowJIT(False)
        # Find the line number to break inside main().
        self.line = line_number('main.c', '// Set break point at this line.')

    def test(self):
        """Test floating point expressions while jitter is disabled."""
        self.build()
        exe = self.getBuildArtifact("a.out")
        self.runCmd("file " + exe, CURRENT_EXECUTABLE_SET)

        # Break inside the main.
        lldbutil.run_break_set_by_file_and_line(
            self, "main.c", self.line, num_expected_locations=1, loc_exact=True)

        
        value = self.frame().EvaluateExpression("a + b", self.no_jit_opts)

        self.runCmd("run", RUN_SUCCEEDED)
        # test double
        self.expect("expr --allow-jit false  -- a + b", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['double', '44'])
        self.expect("expr --allow-jit false  -- a - b", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['double', '40'])
        self.expect("expr --allow-jit false  -- a / b", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['double', '21'])
        self.expect("expr --allow-jit false  -- a * b", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['double', '84'])
        self.expect("expr --allow-jit false  -- a + 2", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['double', '44'])
        self.expect("expr --allow-jit false  -- a > b", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['true'])
        self.expect("expr --allow-jit false  -- a >= b", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['true'])
        self.expect("expr --allow-jit false  -- a < b", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['false'])
        self.expect("expr --allow-jit false  -- a <= b", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['false'])
        self.expect("expr --allow-jit false  -- a == b", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['false'])
        self.expect("expr --allow-jit false  -- a != b", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['true'])
        
        # test single
        self.expect("expr --allow-jit false  -- f + q", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['float', '44'])
        self.expect("expr --allow-jit false  -- f - q", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['float', '40'])
        self.expect("expr --allow-jit false  -- f / q", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['float', '21'])
        self.expect("expr --allow-jit false  -- f * q", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['float', '84'])
        self.expect("expr --allow-jit false  -- f + 2", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['float', '44'])
        self.expect("expr --allow-jit false  -- f > q", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['true'])
        self.expect("expr --allow-jit false  -- f >= q", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['true'])
        self.expect("expr --allow-jit false  -- f < q", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['false'])
        self.expect("expr --allow-jit false  -- f <= q", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['false'])
        self.expect("expr --allow-jit false  -- f == q", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['false'])
        self.expect("expr --allow-jit false  -- f != q", VARIABLES_DISPLAYED_CORRECTLY,
                    substrs=['true'])

        # compare jit and interpreter output
        self.assertTrue(self.process().IsValid())
        thread = lldbutil.get_stopped_thread(self.process(), lldb.eStopReasonBreakpoint)
        self.assertTrue(thread.IsValid())
        frame = thread.GetSelectedFrame()
        self.assertTrue(frame.IsValid())
       
        dividents = [42, 79, 666]
        divisors = [1.618, 2.718281828, 3.1415926535, 6.62607015]

        for x in dividents:
            for y in divisors:
                vardef = "double x = {0}, y = {1};".format(x, y) 
                v1 = frame.EvaluateExpression("{0}; eval(x, y, 2)".format(vardef), self.jit_opts)
                v2 = frame.EvaluateExpression("{0}; x / y".format(vardef), self.no_jit_opts)
                self.assertTrue(v1.IsValid() and v2.IsValid())
                self.assertEqual(str(v1.GetData()), str(v2.GetData()))