Connected the make root menu options.

nw/gui/doctree.py

@@ -72,7 +72,7 @@ class GuiDocTree(QTreeWidget):
             itemClass = self.theProject.getItem(pHandle).itemClass
 
         logger.verbose("Adding new item of type %s and class %s to handle %s" % (
-            itemType.name,itemClass.name,str(pHandle))
+            itemType.name, itemClass.name, str(pHandle))
         )
 
         if itemType == nwItemType.ROOT:
@@ -220,6 +220,7 @@ class GuiDocTree(QTreeWidget):
 
         if nwItem.itemType == nwItemType.ROOT:
             newItem.setIcon(self.C_NAME, QIcon.fromTheme("drive-harddisk"))
+            self.theParent.mainMenu.setAvailableRoot()
         elif nwItem.itemType == nwItemType.FOLDER:
             newItem.setIcon(self.C_NAME, QIcon.fromTheme("folder"))
         elif nwItem.itemType == nwItemType.FILE:

nw/gui/mainmenu.py

@@ -52,6 +52,14 @@ class GuiMainMenu(QMenuBar):
         self.theParent.openProject(self.mainConf.recentList[recentItem])
         return True
 
+    def setAvailableRoot(self):
+        for itemClass in nwItemClass:
+            if itemClass != nwItemClass.NO_CLASS:
+                self.rootItems[itemClass].setEnabled(
+                    self.theProject.checkRootUnique(itemClass)
+                )
+        return
+
     ##
     #  Menu Action
     ##
@@ -122,6 +130,12 @@ class GuiMainMenu(QMenuBar):
         self.rootItems[nwItemClass.TIMELINE]   = QAction(QIcon.fromTheme("folder-new"), "Timeline Root",  rootMenu)
         self.rootItems[nwItemClass.OBJECT]     = QAction(QIcon.fromTheme("folder-new"), "Object Root",    rootMenu)
         self.rootItems[nwItemClass.CUSTOM]     = QAction(QIcon.fromTheme("folder-new"), "Custom Root",    rootMenu)
+        nCount = 0
+        for itemClass in self.rootItems.keys():
+            nCount += 1 # This forces the lambdas to be unique
+            self.rootItems[itemClass].triggered.connect(
+                lambda nCount, itemClass=itemClass : self._newTreeItem(nwItemType.ROOT, itemClass)
+            )
         rootMenu.addActions(self.rootItems.values())
 
         # Project > New Folder