# -*- tab-width: 4; indent-tabs-mode: nil; py-indent-offset: 4 -*-
#
# This file is part of the LibreOffice project.
#
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
#
from uitest.framework import UITestCase
from uitest.uihelper.common import get_state_as_dict, get_url_for_data_file

from libreoffice.calc.document import get_cell_by_position
from libreoffice.uno.propertyvalue import mkPropertyValues


class regression(UITestCase):
    def test_regression_row(self):
        self._regression_check(data_groupedby_column = False)

    def test_regression_column(self):
        self._regression_check(data_groupedby_column = True)

    def _regression_check(self, data_groupedby_column = True):
        with self.ui_test.load_file(get_url_for_data_file("regression.ods")) as calc_doc:
            # Initially the final check status is "FALSE" (failed).
            self.assertEqual(get_cell_by_position(calc_doc, 14, 1, 6).getString(), "FALSE",
                             "Check status must be FALSE before the test")
            self._do_regression(regression_type = "LINEAR", data_groupedby_column = data_groupedby_column, calc_intercept = True)
            self._do_regression(regression_type = "LINEAR", data_groupedby_column = data_groupedby_column, calc_intercept = False)
            self._do_regression(regression_type = "LOG", data_groupedby_column = data_groupedby_column)
            self._do_regression(regression_type = "POWER", data_groupedby_column = data_groupedby_column)
            self.assertEqual(get_cell_by_position(calc_doc, 14, 1, 6).getString(), "TRUE",
                             "One of more of the checks failed for data_groupedby_column = {}, manually try with the calc_doc".
                             format(data_groupedby_column))

    def _do_regression(self, regression_type, data_groupedby_column = True, calc_intercept = True):
        assert(regression_type == "LINEAR" or regression_type == "LOG" or regression_type == "POWER")
        with self.ui_test.execute_modeless_dialog_through_command(".uno:RegressionDialog") as xDialog:
            xvariable1rangeedit = xDialog.getChild("variable1-range-edit")
            xvariable2rangeedit = xDialog.getChild("variable2-range-edit")
            xoutputrangeedit = xDialog.getChild("output-range-edit")
            xwithlabelscheck = xDialog.getChild("withlabels-check")
            xgroupedbyrowsradio = xDialog.getChild("groupedby-rows-radio")
            xgroupedbycolumnsradio = xDialog.getChild("groupedby-columns-radio")
            xlinearradio = xDialog.getChild("linear-radio")
            xlogarithmicradio = xDialog.getChild("logarithmic-radio")
            xpowerradio = xDialog.getChild("power-radio")
            xnointerceptcheck = xDialog.getChild("nointercept-check")

            ## Set the X, Y and output ranges
            xvariable1rangeedit.executeAction("FOCUS", tuple()) # Without this the range parser does not kick in somehow
            xvariable1rangeedit.executeAction("TYPE", mkPropertyValues({"KEYCODE":"CTRL+A"}))
            xvariable1rangeedit.executeAction("TYPE", mkPropertyValues({"KEYCODE":"BACKSPACE"}))
            if data_groupedby_column:
                xvariable1rangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$DataInColumns.$A$1:$C$11"}))
            else:
                xvariable1rangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$DataInRows.$A$1:$K$3"}))

            xvariable2rangeedit.executeAction("FOCUS", tuple()) # Without this the range parser does not kick in somehow
            xvariable2rangeedit.executeAction("TYPE", mkPropertyValues({"KEYCODE":"CTRL+A"}))
            xvariable2rangeedit.executeAction("TYPE", mkPropertyValues({"KEYCODE":"BACKSPACE"}))
            if data_groupedby_column:
                xvariable2rangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$DataInColumns.$D$1:$D$11"}))
            else:
                xvariable2rangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$DataInRows.$A$4:$K$4"}))
            # The data ranges have labels in them
            if (get_state_as_dict(xwithlabelscheck)["Selected"]) == "false":
                xwithlabelscheck.executeAction("CLICK", tuple())

            xoutputrangeedit.executeAction("FOCUS", tuple()) # Without this the range parser does not kick in somehow
            xoutputrangeedit.executeAction("TYPE", mkPropertyValues({"KEYCODE":"CTRL+A"}))
            xoutputrangeedit.executeAction("TYPE", mkPropertyValues({"KEYCODE":"BACKSPACE"}))
            if regression_type == "LINEAR":
                if calc_intercept:
                    xoutputrangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$ActualLinear.$A$1"}))
                else:
                    xoutputrangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$ActualLinearNoIntercept.$A$1"}))
            elif regression_type == "LOG":
                xoutputrangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$ActualLog.$A$1"}))
            else:
                xoutputrangeedit.executeAction("TYPE", mkPropertyValues({"TEXT":"$ActualPower.$A$1"}))

            if data_groupedby_column:
                xgroupedbycolumnsradio.executeAction("CLICK", tuple())
            else:
                xgroupedbyrowsradio.executeAction("CLICK", tuple())

            if regression_type == "LINEAR":
                xlinearradio.executeAction("CLICK", tuple())
            elif regression_type == "LOG":
                xlogarithmicradio.executeAction("CLICK", tuple())
            else:
                xpowerradio.executeAction("CLICK", tuple())

            if not calc_intercept:
                xnointerceptcheck.executeAction("CLICK", tuple())


    def test_regression_cancel(self):
        with self.ui_test.create_doc_in_start_center("calc"):
            with self.ui_test.execute_modeless_dialog_through_command(".uno:RegressionDialog", close_button="cancel"):
                pass

# vim: set shiftwidth=4 softtabstop=4 expandtab: