# /**********************************************************************
# *                                                                     *
# * Copyright (c) 2021 Javier Braña <javier.branagutierrez@gmail.com>  *
# *                                                                     *
# * This program is free software; you can redistribute it and/or modify*
# * it under the terms of the GNU Lesser General Public License (LGPL)  *
# * as published by the Free Software Foundation; either version 2 of   *
# * the License, or (at your option) any later version.                 *
# * for detail see the LICENCE text file.                               *
# *                                                                     *
# * This program is distributed in the hope that it will be useful,     *
# * but WITHOUT ANY WARRANTY; without even the implied warranty of      *
# * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the       *
# * GNU Library General Public License for more details.                *
# *                                                                     *
# * You should have received a copy of the GNU Library General Public   *
# * License along with this program; if not, write to the Free Software *
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307*
# * USA                                                                 *
# *                                                                     *
# ***********************************************************************

import ArchComponent
import FreeCAD
import Part

if FreeCAD.GuiUp:
    import FreeCADGui
    from PySide import QtCore
    import draftguitools.gui_trackers as DraftTrackers
    import draftguitools.gui_tool_utils as gui_tool_utils
else:
    # \cond
    def translate(ctxt, txt):
        return txt
    def QT_TRANSLATE_NOOP(ctxt, txt):
        return txt
    # \endcond

try:
    _fromUtf8 = QtCore.QString.fromUtf8
except AttributeError:
    def _fromUtf8(s):
        return s

import os
import PVPlantResources
from PVPlantResources import DirIcons as DirIcons


def makeBuilding(name="Building"):
    obj = FreeCAD.ActiveDocument.addObject("Part::FeaturePython", "Building")
    obj.Label = name
    _Building(obj)
    _ViewProviderBuilding(obj.ViewObject)
    FreeCAD.ActiveDocument.recompute()
    return obj


class _Building(ArchComponent.Component):
    "A Building Obcject"

    def __init__(self, obj):
        # Definición de Variables:
        ArchComponent.Component.__init__(self, obj)
        self.rooftype = ""
        self.setProperties(obj)

    def setProperties(self, obj):
        pl = obj.PropertiesList
        # Dimensions: --------------------------------------------------------------------------------------------------
        if not "RoofWall" in pl:
            obj.addProperty("App::PropertyLength",
                            "RoofWall",
                            "Building",
                            "The height of this object"
                            ).RoofWall = 0

        if not "RoofWallWidth" in pl:
            obj.addProperty("App::PropertyLength",
                            "RoofWallWidth",
                            "Building",
                            "The height of this object"
                            ).RoofWallWidth = 300

        if not "Height" in pl:
            obj.addProperty("App::PropertyLength",
                            "Height",
                            "Building",
                            "The height of this object"
                            ).Height = 4000

        if not "Width" in pl:
            obj.addProperty("App::PropertyLength",
                            "Width",
                            "Building",
                            "The width of this object"
                            ).Width = 7000

        if not "Length" in pl:
            obj.addProperty("App::PropertyLength",
                            "Length",
                            "Building",
                            "The height of this object"
                            ).Length = 14000

        if not "RoofHeight" in pl:
            obj.addProperty("App::PropertyLength",
                            "RoofHeight",
                            "Building",
                            "The height of this object"
                            ).RoofHeight = 1500

        if not "TopCenter" in pl:
            obj.addProperty("App::PropertyPercent",
                            "TopCenter",
                            "Building",
                            "The height of this object"
                            ).TopCenter = 50

        if not "TopLength" in pl:
            obj.addProperty("App::PropertyPercent",
                            "TopLength",
                            "Building",
                            "The height of this object"
                            ).TopLength = 0

        # outputs:
        if not "InternalVolume" in pl:
            obj.addProperty("App::PropertyVolume",
                            "InternalVolume",
                            "Outputs",
                            "The height of this object"
                            )

        if not "ExternalVolume" in pl:
            obj.addProperty("App::PropertyVolume",
                            "ExternalVolume",
                            "Outputs",
                            "The height of this object"
                            )

        self.Type = "Building"
        obj.Proxy = self

    def onDocumentRestored(self, obj):
        """Method run when the document is restored.
        Re-adds the Arch component, and Arch wall properties."""
        ArchComponent.Component.onDocumentRestored(self, obj)
        self.setProperties(obj)

    def onChanged(self, obj, prop):
        '''Do something when a property has changed'''


    def execute(self, obj):
        w = obj.Width.Value
        w_med = w / 2
        l_med = obj.Length.Value / 2

        p1 = FreeCAD.Vector(-l_med, -w_med, 0)
        p2 = FreeCAD.Vector(-l_med, w_med, 0)
        p3 = FreeCAD.Vector(-l_med, w_med, obj.Height.Value)
        p4 = FreeCAD.Vector(-l_med, w * (obj.TopCenter - 50) / 100, obj.Height.Value + obj.RoofHeight.Value)
        p5 = FreeCAD.Vector(-l_med, -w_med, obj.Height.Value)
        profile = Part.Face(Part.makePolygon([p1, p2, p3, p4, p5, p1, ]))
        shape = profile.extrude(FreeCAD.Vector(obj.Length.Value, 0, 0))

        if obj.TopLength > 0:
            p1 = FreeCAD.Vector(-l_med, w_med, obj.Height.Value)
            p2 = FreeCAD.Vector(-l_med, w * (obj.TopCenter - 50) / 100, obj.Height.Value + obj.RoofHeight.Value)
            p3 = FreeCAD.Vector(-l_med, -w_med, obj.Height.Value)
            p4 = FreeCAD.Vector(-l_med + (obj.TopLength * obj.Length.Value / 200),
                                w * (obj.TopCenter - 50) / 100, obj.Height.Value + obj.RoofHeight.Value)
            f1 = Part.Face(Part.makePolygon([p1, p2, p3, p1, ]))
            f2 = Part.Face(Part.makePolygon([p1, p4, p3, p1, ]))
            f3 = Part.Face(Part.makePolygon([p1, p2, p4, p1, ]))
            f4 = Part.Face(Part.makePolygon([p3, p4, p2, p3, ]))
            tool = Part.makeSolid(Part.Shell([f1, f4, f2, f3]))
            shape = shape.cut(tool)

            p1 = FreeCAD.Vector(l_med, w_med, obj.Height.Value)
            p2 = FreeCAD.Vector(l_med,
                                w * (obj.TopCenter - 50) / 100,
                                obj.Height.Value + obj.RoofHeight.Value)
            p3 = FreeCAD.Vector(l_med, -w_med, obj.Height.Value)
            p4 = FreeCAD.Vector(l_med - (obj.TopLength * obj.Length.Value / 200),
                                w * (obj.TopCenter - 50) / 100,
                                obj.Height.Value + obj.RoofHeight.Value)

            f1 = Part.Face(Part.makePolygon([p1, p2, p3, p1, ]))
            f2 = Part.Face(Part.makePolygon([p1, p4, p3, p1, ]))
            f3 = Part.Face(Part.makePolygon([p1, p2, p4, p1, ]))
            f4 = Part.Face(Part.makePolygon([p3, p4, p2, p3, ]))
            tool = Part.makeSolid(Part.Shell([f1, f4, f2, f3]))
            shape = shape.cut(tool)

        if obj.RoofWall.Value > 0 and obj.RoofWallWidth.Value > 0:
            offset = 2 * obj.RoofWallWidth.Value
            box1 = Part.makeBox(obj.Length.Value, obj.Width.Value, obj.RoofWall.Value)
            box2 = Part.makeBox(obj.Length.Value - offset, obj.Width.Value - offset, obj.RoofWall.Value)
            box2.Placement.Base.x += obj.RoofWallWidth.Value
            box2.Placement.Base.y += obj.RoofWallWidth.Value
            box1 = box1.cut(box2)
            box1.Placement.Base.x = -l_med
            box1.Placement.Base.y = -w_med
            box1.Placement.Base.z = obj.Height.Value
            shape = shape.fuse(box1).removeSplitter()

        obj.Shape = shape


class _ViewProviderBuilding(ArchComponent.ViewProviderComponent):
    "A View Provider for the Pipe object"

    def __init__(self, vobj):
        ArchComponent.ViewProviderComponent.__init__(self, vobj)

    def getIcon(self):
        return str(os.path.join(DirIcons, "house.svg"))

    def setEdit(self, vobj, mode):
        """Method called when the document requests the object to enter edit mode.

        Edit mode is entered when a user double clicks on an object in the tree
        view, or when they use the menu option [Edit -> Toggle Edit Mode].

        Just display the standard Arch component task panel.

        Parameters
        ----------
        mode: int or str
            The edit mode the document has requested. Set to 0 when requested via
            a double click or [Edit -> Toggle Edit Mode].

        Returns
        -------
        bool
            If edit mode was entered.
        """

        if (mode == 0) and hasattr(self, "Object"):
            taskd = _BuildingTaskPanel(self.Object)
            taskd.obj = self.Object
            # taskd.update()
            FreeCADGui.Control.showDialog(taskd)
            return True

        return False


class _BuildingTaskPanel:
    def __init__(self, obj=None):
        self.new = False
        if obj is None:
            self.new = True
            obj = makeBuilding()

        self.obj = obj

        self.form = FreeCADGui.PySideUic.loadUi(PVPlantResources.__dir__ + "/PVPlantBuilding.ui")

        self.node = None
        self.view = FreeCADGui.ActiveDocument.ActiveView
        self.tracker = DraftTrackers.ghostTracker(obj)
        self.tracker.on()
        self.call = self.view.addEventCallback("SoEvent", self.action)

    def action(self, arg):
        """Handle the 3D scene events.

        This is installed as an EventCallback in the Inventor view.

        Parameters
        ----------
        arg: dict
            Dictionary with strings that indicates the type of event received
            from the 3D view.
        """

        if arg["Type"] == "SoKeyboardEvent" and arg["Key"] == "ESCAPE":
            self.finish()

        elif arg["Type"] == "SoLocation2Event":
            point, ctrlPoint, info = gui_tool_utils.getPoint(self, arg)
            if info:
                self.tracker.move(FreeCAD.Vector(info["x"], info["y"], info["z"]))
            else:
                self.tracker.move(point)

        elif (arg["Type"] == "SoMouseButtonEvent" and
              arg["State"] == "DOWN" and
              arg["Button"] == "BUTTON1"):

            point, ctrlPoint, info = gui_tool_utils.getPoint(self, arg)
            if info:
                self.obj.Placement.Base = FreeCAD.Vector(info["x"], info["y"], info["z"])
            else:
                self.obj.Placement.Base = point
            self.finish()

    def finish(self):
        self.accept()

    def accept(self):
        self.closeForm()
        return True

    def reject(self):
        if self.new:
            FreeCAD.ActiveDocument.removeObject(self.obj.Name)
        self.closeForm()
        return True

    def closeForm(self):
        self.tracker.finalize()
        FreeCADGui.Control.closeDialog()
        self.view.removeEventCallback("SoEvent", self.call)


class _CommandBuilding:
    "the Arch Building command definition"
    def GetResources(self):
        return {'Pixmap': str(os.path.join(DirIcons, "house.svg")),
                'MenuText': "Building",
                'Accel': "C, M",
                'ToolTip': "Creates a Building object from setup dialog."}

    def IsActive(self):
        return not FreeCAD.ActiveDocument is None

        if FreeCAD.ActiveDocument is not None:
            if FreeCADGui.Selection.getCompleteSelection():
                for ob in FreeCAD.ActiveDocument.Objects:
                    if ob.Name[:4] == "Site":
                        return True

    def Activated(self):
        TaskPanel = _BuildingTaskPanel()
        FreeCADGui.Control.showDialog(TaskPanel)
        return


if FreeCAD.GuiUp:
    FreeCADGui.addCommand('PVPlantBuilding', _CommandBuilding())