import sys
import Object3DQt as qt
from Object3DIcons import IconDict
import ObjectTree
import weakref

DEBUG = 0 

class ObjectTreeWidget(qt.QTreeWidget):
    def __init__(self, parent=None, tree=None, labels=None):
        qt.QTreeWidget.__init__(self, parent)
        if labels is None:
            labels = ['Name', 'Type'] #, 'Vertices']
        if tree is None:
            self.tree = ObjectTree.ObjectTree('__Scene__', 'Scene')
        else:
            self.tree = tree
        ncols = len(labels)
        self.setColumnCount(ncols)
        self.setHeaderLabels(labels)

    def focusInEvent(self, event):
        event.accept()

    def addObject(self, item, name=None, parent=None, update=True):
        if name is None:
            name = item.name()
        if parent is None:
            self.tree.addChild(item, name)
        else:
            self[parent].addChild(item, name)
        if update:
            self.updateView()

    def removeObject(self, name):
        #An object has to correspond to an entry in the tree
        treeObject = self.tree.find(name)
        if treeObject is None:
            # do nothing?
            return
        treeObject.erase()
        self.updateView()

    def updateView(self):
        self.clear()
        self.showInView(self.tree)

    def showInView(self, tree, parent=None):
        """
        Represent a tree in the QTreeWidget
        """
        if parent is None:
            widgetItem = Object3DTreeWidgetItem(0, tree)
            self.addTopLevelItem(widgetItem)
        else:
            #find the parent item
            itemList = self.findItems(parent.name(),
              qt.Qt.MatchExactly|qt.Qt.MatchCaseSensitive|qt.Qt.MatchRecursive,
                        0)
            if len(itemList):
                widgetItemParent = itemList[0]
                widgetItem = Object3DTreeWidgetItem(1, tree)
                widgetItemParent.addChild(widgetItem)
            else:
                return
        ob = tree.root[0]
        if hasattr(ob,'selected'):
            if ob.selected():
                widgetItem.setSelected(True)
                self.scrollToItem(widgetItem,
                    qt.QAbstractItemView.EnsureVisible)

        for subTree in tree.childList():
            self.showInView(subTree, tree.parent(subTree.name()))

    def setSelected(self, name):
        itemList = self.findItems(name,
              qt.Qt.MatchExactly|qt.Qt.MatchCaseSensitive|qt.Qt.MatchRecursive,
                        0)
        if len(itemList):
            itemList[0].setSelected(True)

class Object3DTreeWidgetItem(qt.QTreeWidgetItem):
    def __init__(self, wtype, object3D):
        if type(wtype) != type(1):
            raise TypeError, "First argument must be an integer"
        #if (wtype != 0) and (wtype < qt.QTreeWidgetItem.UserType):
        #    raise TypeError, "First argument must be 0 or an integer >= 1000"
        actualType = wtype
        qt.QTreeWidgetItem.__init__(self, wtype + qt.QTreeWidgetItem.UserType)
        self.__object3D = object3D
        self.setText(0, object3D.name())
        if wtype == 0:
            text = "Scene"
        else:
            text = "3D Object"
        self.setText(1, text)

class Object3DObjectTree(qt.QGroupBox):
    def __init__(self, parent = None, tree=None):
        qt.QGroupBox.__init__(self, parent)
        self.setTitle('Objects Tree')
        self.mainLayout = qt.QVBoxLayout(self)
        self.mainLayout.setMargin(0)
        self.mainLayout.setMargin(0)
        self.treeWidget = ObjectTreeWidget(self, tree=tree)
        self.tree = weakref.proxy(self.treeWidget.tree)
        self.actions = ObjectActions(self)
        self.mainLayout.addWidget(self.treeWidget)
        self.mainLayout.addWidget(self.actions)
        self.addObject = self.treeWidget.addObject
        self.__current = 'Scene'
        self.__previous= None
        self.__cutObject = None
        self.__replacing = False
        self.connect(self.actions.cutButton,
                     qt.SIGNAL('clicked()'),
                     self.cutObject)

        self.connect(self.actions.pasteButton,
                     qt.SIGNAL('clicked()'),
                     self.pasteObject)

        self.connect(self.actions.deleteButton,
                     qt.SIGNAL('clicked()'),
                     self.deleteObject)

        self.connect(self.actions.replaceButton,
                     qt.SIGNAL('clicked()'),
                     self.replaceWithObject)

        self.connect(self.treeWidget,
            qt.SIGNAL('currentItemChanged( QTreeWidgetItem *, QTreeWidgetItem *)'),
            self.itemChanged)

    def updateView(self, expand=False):
        self.treeWidget.updateView()
        if expand:
            if qt.qVersion() >= '4.2.0':
                self.treeWidget.expandAll()
        objectList = self.getSelectedObjectList()
        if len(objectList):
            self.__current = objectList[0]

    def getSelectedObjectList(self):
        selected = []
        for item in self.tree.childList():
            ob = item.root[0]
            if hasattr(ob, 'selected'):
                if ob.selected():
                    selected.append(item.name())
        return selected

    def setSelectedObject(self, name=None):
        if name is None:
            name = self.__current
        else:
            self.__current = name

        #reset all the children
        for item in self.tree.childList():
            ob = item.root[0]
            if hasattr(ob, 'selected'):
                ob.setSelected(False)

        #but do not forget the scene itself
        if hasattr(self.tree.root[0], "setSelected"):
            self.tree.root[0].setSelected(False)

        #now select the proper one
        if self.tree.name() == name:
            self.tree.root[0].setSelected(True)
        else:
            child = self.tree.find(name)
            if child is not None:
                ob = child.root[0]
                if hasattr(ob, 'selected'):
                    ob.setSelected(True)
                    self.treeWidget.setSelected(child.name())

    def cutObject(self):
        if self.__current == 'Scene':
            self.__cutObject = None
            qt.QMessageBox.critical(self, "Error on cut",
                "You cannot cut the Scene itself.",
                qt.QMessageBox.Ok | qt.QMessageBox.Default,
                            qt.QMessageBox.NoButton)
        elif self.__current is None:
            qt.QMessageBox.critical(self, "Error on cut",
                "Please select an object.",
                qt.QMessageBox.Ok | qt.QMessageBox.Default,
                            qt.QMessageBox.NoButton)
        else:
            self.__cutObject = self.__current

    def pasteObject(self):
        if self.__cutObject is None:
            qt.QMessageBox.critical(self, "Error on paste",
                "Please cut an object first.",
                qt.QMessageBox.Ok | qt.QMessageBox.Default,
                            qt.QMessageBox.NoButton)
            return

        if self.__cutObject == self.__current:
            #do nothing
            if DEBUG:
                print "Doing nothing"
            self.__cutObject = None
            self.treeWidget.resizeColumnToContents(0)
            return

        child = self.tree.find(self.__cutObject)
        self.tree.delChild(self.__cutObject)
        destination = self.tree.find(self.__current)
        destination.addChildTree(child)
        if DEBUG:
            print "TREE after addition = ", self.tree
        self.updateView()

        if 1:
            #this works
            itemList = self.treeWidget.findItems(self.__cutObject,
                      qt.Qt.MatchExactly|qt.Qt.MatchCaseSensitive|qt.Qt.MatchRecursive,
                        0)
            if len(itemList):
                self.treeWidget.scrollToItem(itemList[0],
                    qt.QAbstractItemView.EnsureVisible)
            else:
                if DEBUG:
                    print "Is this a problem?"

        else:
            #this too
            name = self.__cutObject
            while name != 'Scene':
                name = self.tree.parent(name).name()
                itemList = self.treeWidget.findItems(name,
                      qt.Qt.MatchExactly|qt.Qt.MatchCaseSensitive|qt.Qt.MatchRecursive,
                        0)
                if len(itemList):
                    self.treeWidget.expandItem(itemList[0])
                else:
                    if DEBUG:
                        print "Is this a problem?"
                    break
        self.treeWidget.resizeColumnToContents(0)
        self.__cutObject = None
        self.emitSignal('treeChanged')
        
    def deleteObject(self):
        if self.__current == 'Scene':
            qt.QMessageBox.critical(self, "Error on deletion",
                "You cannot delete the Scene itself.",
                qt.QMessageBox.Ok | qt.QMessageBox.Default,
                            qt.QMessageBox.NoButton)
        self.tree.delChild(self.__current)
        self.__previous = str(self.__current)
        self.setSelectedObject(self.tree.name())
        self.emitSignal('objectDeleted')
        self.updateView()

    def replaceWithObject(self):            
        if self.__current in [None, 'None']:
            return
        self.__replacing = True

        if self.__current == 'Scene':
            itemList = self.tree.childList()
            for item in itemList:
                self.tree.delChild(item.name())
            self.updateView()
            self.__cutObject = self.__current * 1            
        else:
            self.__cutObject = self.__current * 1
            self.__current = 'Scene'
            child = self.tree.find(self.__cutObject)
            self.tree.delChild(self.__cutObject)
            itemList = self.tree.childList()
            for item in itemList:
                self.tree.delChild(item.name())
            self.tree.addChildTree(child)
            self.updateView()

        itemList = self.treeWidget.findItems(self.__cutObject,
                  qt.Qt.MatchExactly|qt.Qt.MatchCaseSensitive|qt.Qt.MatchRecursive,
                    0)
        if len(itemList):
            self.treeWidget.scrollToItem(itemList[0],
                qt.QAbstractItemView.EnsureVisible)
        self.treeWidget.resizeColumnToContents(0)
        self.__current = self.__cutObject * 1
        self.__cutObject = None
        self.__replacing = False        
        self.emitSignal('objectReplaced')
        
    def itemChanged(self, current, previous):
        if current is None:
            #This happens when updating because I clear the tree
            return
            #This was giving a lot of problems:
            self.__current = 'Scene'
        else:
            self.__current = current.text(0)
        if previous is None:
            self.__previous = None
        else:
            self.__previous = previous.text(0)
        if DEBUG:
            print "current = ", self.__current
            print "previous = ", self.__previous
        if self.__current != self.__previous:
            self.setSelectedObject(str(self.__current))
            self.emitSignal('objectSelected')

    def emitSignal(self, event):
        if self.__replacing:
            if DEBUG:
                print "EVENT = ", event, "NOT SENT"
        ddict = {}
        ddict['event'] = event
        ddict['current'] = str(self.__current)
        ddict['previous'] = str(self.__previous)
        qt.QObject.emit(self,
                        qt.SIGNAL('ObjectTreeSignal'),
                        ddict)


class ObjectActions(qt.QGroupBox):
    def __init__(self, parent = None):
        qt.QGroupBox.__init__(self, parent)
        self.setTitle('Object Actions')
        self.mainLayout = qt.QVBoxLayout(self)
        self.mainLayout.setSpacing(0)
        self.mainLayout.setMargin(0)
        self.cutButtonIcon = qt.QIcon(qt.QPixmap(IconDict['cut']))
        self.cutButton = qt.QPushButton(self)
        self.cutButton.setIcon(self.cutButtonIcon)
        self.cutButton.setText('Cut')
        self.pasteButtonIcon = qt.QIcon(qt.QPixmap(IconDict['paste']))
        self.pasteButton = qt.QPushButton(self)
        self.pasteButton.setIcon(self.pasteButtonIcon)
        self.pasteButton.setText('Paste')
        self.deleteButtonIcon = qt.QIcon(qt.QPixmap(IconDict['delete']))
        self.deleteButton = qt.QPushButton(self)
        self.deleteButton.setIcon(self.deleteButtonIcon)
        self.deleteButton.setText('Delete')
        self.replaceButton = qt.QPushButton(self)
        self.replaceButton.setText('Replace')

        self.mainLayout.addWidget(self.cutButton)
        self.mainLayout.addWidget(self.pasteButton)
        self.mainLayout.addWidget(self.deleteButton)
        self.mainLayout.addWidget(self.replaceButton)


if __name__ == "__main__":
    import Object3DBase
    app = qt.QApplication([])
    qt.QObject.connect(app, qt.SIGNAL("lastWindowClosed()"),
                       app, qt.SLOT("quit()"))
    o0 = Object3DBase.Object3D("DummyObject0")
    o1 = Object3DBase.Object3D("DummyObject1")
    o01 = Object3DBase.Object3D("DummyObject01")
    w = Object3DObjectTree()
    w.addObject(o0, update=False)
    w.addObject(o1, update=False)
    w.addObject(o01, update=True)
    tree = w.tree.find("DummyObject0")
    w.tree.delChild("DummyObject01")
    tree.addChild(o01)
    w.updateView()

    w.show()
    app.exec_()