def __createLayout(self):
# Zoom items. The only place where the zoom items are specified!
# self.zitems = [0.25, 0.5, 0.75, 1, 1.25, 1.5, 1.75, 2.0, 3.0, 4.0, 8.0]
# Set up the model for the Model Editor
self.modelEditorDock = self.__createDockWindow("ModelEditor")
self.modelEditorDock.setToggleFcn(self.toggleModelEditor)
modelEditor = ModelEditor(self.modelEditorDock)
# TODO
# self.model = SimModel(QDomDocument(), self.modelEditorDock) # Do I need parent self.modelEditorDock
self.model = SimModel(
None,
self.modelEditorDock) # Do I need parent self.modelEditorDock
modelEditor.setModel(self.model) # Set the default model
modelEditor.setItemDelegate(SimDelegate(self))
modelEditor.setParams()
modelEditor.setSelectionBehavior(QAbstractItemView.SelectItems)
self.viewmanager.setModelEditor(
modelEditor) # Sets the Model Editor in the ViewManager
self.__setupDockWindow(self.modelEditorDock, Qt.LeftDockWidgetArea,
modelEditor, "Model Editor") # projectBrowser
self.latticeDataDock = self.__createDockWindow("LatticeData")
self.latticeDataDock.setToggleFcn(self.toggleLatticeData)
self.latticeDataModelTable = LatticeDataModelTable(
self.latticeDataDock, self.viewmanager)
self.latticeDataModel = LatticeDataModel()
self.latticeDataModelTable.setModel(self.latticeDataModel)
# # self.cplugins.latticeDataModelTable()
# #self.connect(self.cplugins, SIGNAL("doubleClicked(const QModelIndex &)"), self.__showPluginView)
#
self.__setupDockWindow(self.latticeDataDock, Qt.LeftDockWidgetArea,
self.latticeDataModelTable, "LatticeDataFiles")
self.setCorner(Qt.TopLeftCorner, Qt.LeftDockWidgetArea)
# Set up the console
self.consoleDock = self.__createDockWindow("Console")
self.consoleDock.setToggleFcn(self.toggleConsole)
self.console = Console(self.consoleDock)
self.consoleDock.setWidget(self.console)
# self.consoleDock.setWindowTitle("Console")
self.__setupDockWindow(self.consoleDock, Qt.BottomDockWidgetArea,
self.console, "Console")
# self.viewmanager.addWidget(self.consoleDock)
# self.viewmanager.setSizes([400, 50])
# self.consoleDock.show()
# rec = self.console.geometry()
# rec.setHeight(300)
# print rec.height()
# self.console.setGeometry(rec)
# """
# Don't know why I need dockwindows
self.dockwindows = {}
(New_Layers['drop2'], model.layers[-15]),
(model.layers[-14], New_Layers['drop2']),
(New_Layers['drop3'], model.layers[-9]),
(model.layers[-8], New_Layers['drop3']),
(New_Layers['drop4'], model.layers[-6]),
(model.layers[-5], New_Layers['drop4']),
(New_Layers['drop5'], model.layers[-2]),
(model.layers[-1], New_Layers['drop5']),
]
model_inputs = None
model_outputs = None
model = ModelEditor(model,
New_Layers=New_Layers,
IncomingLinks_2Axe=IncomingLinks_2Axe,
IncomingLinks_2Forge=IncomingLinks_2Forge,
model_inputs=model_inputs,
model_outputs=model_outputs)
models[ModelKey] = model
# Save edited model description
if SaveEditedModelDescription:
Model_Path = os.path.join(MasterPath, str(ModelKey))
SaveModelDescript(model,
save_dir=Model_Path,
save_filename=str(ModelKey + '_edited'))
print('Model descriptions saved!')
#%% Adding Top and Bottom layers to keras models instantiated above
AddTopAndBottomLayers = True
