def selectFile(self, fileLink=None):
if fileLink == False:
linetext = QFileDialog.getOpenFileName(
self, self.tr("Select File"),
QStandardPaths.displayName(QStandardPaths.HomeLocation),
self.tr("Files (*.*)"))[0]
else:
linetext = fileLink
if linetext != "":
self.setWindowTitle(linetext)
self.ui.dataFileLineEdit.setText(linetext)
self.ui.dataTypeText.setText(linetext.split(".")[1].upper())
if self.ui.fileParserCombo.findText(
linetext.split(".")[1].upper() + " Parser") != -1:
headerDict = {}
self.ui.fileParserCombo.setCurrentIndex(
self.ui.fileParserCombo.findText(
linetext.split(".")[1].upper() + " Parser"))
if linetext.split(".")[1].upper() == "CTF":
headerDict = ctf.parse_header_as_dict(linetext)
elif linetext.split(".")[1].upper() == "ANG":
headerDict = ang.parse_header_as_dict(linetext)
elif linetext.split(".")[1].upper() == "XML":
print("XML Parser used")
self.createtablemodel = TableModelC(headerDict, self)
self.filterModel.displayed = []
self.filterModel.setSourceModel(self.createtablemodel)
self.filterModel.fileType.append(linetext)
self.createTableSearchFilterModel = QSortFilterProxyModel(self)
self.createTableSearchFilterModel.setSourceModel(self.filterModel)
self.createTableSearchFilterModel.setFilterKeyColumn(1)
self.createTableSearchFilterModel.setDynamicSortFilter(True)
self.ui.metadataTableView.setModel(
self.createTableSearchFilterModel)
self.treeModel = TreeModel(["Available File Metadata"], headerDict,
self.createtablemodel)
if len(self.customInputs) != 0:
for i in range(len(self.customInputs)):
self.createtablemodel.beginInsertRows(
self.createtablemodel.index(
len(self.createtablemodel.metadataList), 0), i, i)
self.createtablemodel.metadataList.append(
self.customInputs[i])
self.createtablemodel.endInsertRows()
self.createTreeSearchFilterModel = QSortFilterProxyModel(self)
self.createTreeSearchFilterModel.setSourceModel(self.treeModel)
self.createTreeSearchFilterModel.setFilterKeyColumn(0)
self.createTreeSearchFilterModel.setDynamicSortFilter(True)
self.createTreeSearchFilterModel.setRecursiveFilteringEnabled(True)
self.ui.metadataTreeView.setModel(self.createTreeSearchFilterModel)
self.treeModel.checkChanged.connect(self.filterModel.checkList)
self.createtrashDelegate.pressed.connect(self.handleRemoveCreate)
self.toggleButtons()
return True
def __init__(self):
QMainWindow.__init__(self)
items = []
items.append(TreeNode("Root"))
items[0].addChild(TreeNode(["Leaf A", "a", "b"]))
items[0].addChild(TreeNode(["Leaf B", "c", "d"]))
items[0].child(1).addChild(TreeNode(["Sub Leaf", "e", "f"]))
self.setWindowTitle("Qt Demo")
tree = QTreeView()
tree.setModel(TreeModel(items))
tree.setHeaderHidden(True)
tree.setColumnWidth(0, 150)
self.setCentralWidget(tree)
def _openWorkDir(self, wd=None, reopen=False):
'''打开工作目录'''
if wd is None:
wd = fs.FS.getRecentDirs()[-1]
if wd == self.basedir:
return
print("--openWorkDir:" + wd)
tree = self.tree
self.basedir = wd #
self.fs = fs.FS(self.basedir)
model = TreeModel(None, self.fs.loadDir())
tree.setModel(model)
for i in range(1, model.columnCount()):
tree.setColumnHidden(i, True) # 隐藏其它列
fs.FS.addRecentDir(wd)
if not reopen:
has_pass = self.fs.getConf("has_pass")
if has_pass == "True":
self.pswd = de.inputPassword(self)
self.has_pass = True
else:
self.pswd = ""
self.has_pass = False
last_open = self.fs.getConf("last_open")
if last_open != "":
self.__openNote(self.fs.apath(last_open))
win_geo = self.fs.getConf("window_geometry")
if win_geo != "":
v = [int(i) for i in win_geo.split(",")]
self.win_geo = QtCore.QRect(v[0], v[1], v[2], v[3])
self.last_sel = self._expand_opened_tree(last_open, tree)
if not hasattr(self, 'selchangedevent'):
self.tree.selectionModel().selectionChanged.connect(
self.treeSelChg) #这个放在上个语句前, 有打不开的情况
self.selchangedevent = True
def read_file(self):
"""Build tree = treelib.Tree() to facilitate displaying the CAD model and
constructing the tree view showing the assembly/component relationships.
Each node of self.tree contains the following:
(Name, UID, ParentUID, {Data}) where the Data keys are:
'a' (isAssy?), 'l' (TopLoc_Location), 'c' (Quantity_Color), 's' (TopoDS_Shape)
"""
logger.info("Reading STEP file")
tmodel = TreeModel("STEP")
self.shape_tool = tmodel.shape_tool
self.color_tool = tmodel.color_tool
step_reader = STEPCAFControl_Reader()
step_reader.SetColorMode(True)
step_reader.SetLayerMode(True)
step_reader.SetNameMode(True)
step_reader.SetMatMode(True)
status = step_reader.ReadFile(self.filename)
if status == IFSelect_RetDone:
logger.info("Transfer doc to STEPCAFControl_Reader")
step_reader.Transfer(tmodel.doc)
labels = TDF_LabelSequence()
self.shape_tool.GetShapes(labels)
logger.info('Number of labels at root : %i', labels.Length())
try:
rootlabel = labels.Value(1) # First label at root
except RuntimeError:
return
name = self.getName(rootlabel)
logger.info('Name of root label: %s', name)
isAssy = self.shape_tool.IsAssembly(rootlabel)
logger.info("First label at root holds an assembly? %s", isAssy)
if isAssy:
# If first label at root holds an assembly, it is the Top Assembly.
# Through this label, the entire assembly is accessible.
# there is no need to examine other labels at root explicitly.
topLoc = TopLoc_Location()
topLoc = self.shape_tool.GetLocation(rootlabel)
self.assyLocStack.append(topLoc)
entry = rootlabel.EntryDumpToString()
logger.debug("Entry: %s", entry)
logger.debug("Top assy name: %s", name)
# Create root node for top assy
newAssyUID = self.getNewUID()
self.tree.create_node(name, newAssyUID, None,
{'a': True, 'l': None, 'c': None, 's': None})
self.assyUidStack.append(newAssyUID)
topComps = TDF_LabelSequence() # Components of Top Assy
subchilds = False
isAssy = self.shape_tool.GetComponents(rootlabel, topComps, subchilds)
logger.debug("Is Assembly? %s", isAssy)
logger.debug("Number of components: %s", topComps.Length())
logger.debug("Is Reference? %s", self.shape_tool.IsReference(rootlabel))
if topComps.Length():
self.findComponents(rootlabel, topComps)
else:
# Labels at root can hold solids or compounds (which are 'crude' assemblies)
# Either way, we will need to create a root node in self.tree
newAssyUID = self.getNewUID()
self.tree.create_node(os.path.basename(self.filename),
newAssyUID, None,
{'a': True, 'l': None, 'c': None, 's': None})
self.assyUidStack = [newAssyUID]
for j in range(labels.Length()):
label = labels.Value(j+1)
name = self.getName(label)
isAssy = self.shape_tool.IsAssembly(label)
logger.debug("Label %i is assembly?: %s", j+1, isAssy)
shape = self.shape_tool.GetShape(label)
color = self.getColor(shape)
isSimpleShape = self.shape_tool.IsSimpleShape(label)
logger.debug("Is Simple Shape? %s", isSimpleShape)
shapeType = shape.ShapeType()
logger.debug("The shape type is: %i", shapeType)
if shapeType == 0:
logger.debug("The shape type is OCC.Core.TopAbs.TopAbs_COMPOUND")
topo = TopologyExplorer(shape)
#topo = aocutils.topology.Topo(shape)
logger.debug("Nb of compounds : %i", topo.number_of_compounds())
logger.debug("Nb of solids : %i", topo.number_of_solids())
logger.debug("Nb of shells : %i", topo.number_of_shells())
newAssyUID = self.getNewUID()
for i, solid in enumerate(topo.solids()):
name = "P%s" % str(i+1)
self.tree.create_node(name, self.getNewUID(),
self.assyUidStack[-1],
{'a': False, 'l': None,
'c': color, 's': solid})
elif shapeType == 2:
logger.debug("The shape type is OCC.Core.TopAbs.TopAbs_SOLID")
self.tree.create_node(name, self.getNewUID(),
self.assyUidStack[-1],
{'a': False, 'l': None,
'c': color, 's': shape})
elif shapeType == 3:
logger.debug("The shape type is OCC.Core.TopAbs.TopAbs_SHELL")
self.tree.create_node(name, self.getNewUID(),
self.assyUidStack[-1],
{'a': False, 'l': None,
'c': color, 's': shape})
return tmodel.doc # <class 'OCC.Core.TDocStd.TDocStd_Document'>
def __init__(self):
super(MainWindow, self).__init__()
self.hyperthoughtui = HyperthoughtDialogImpl()
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
self.ui.TabWidget.setCurrentWidget(self.ui.CreateTemplateTab)
self.ui.actionHelp.triggered.connect(self.help)
self.ui.actionOpenPackage.triggered.connect(self.openPackage)
self.ui.actionSave_Package.triggered.connect(self.savePackageAs)
self.ui.actionClose.triggered.connect(self.close)
self.ui.actionSave_Template.triggered.connect(self.saveTemplate)
self.ui.actionOpen_Template.triggered.connect(self.restoreTemplate)
self.ui.dataFileSelect.clicked.connect(self.selectFile)
self.ui.hyperthoughtTemplateSelect.clicked.connect(self.selectTemplate)
self.ui.saveTemplateButton.clicked.connect(self.saveTemplate)
self.ui.otherDataFileSelect.clicked.connect(self.extractFile)
self.ui.hyperthoughtUploadButton.clicked.connect(
self.uploadToHyperthought)
self.setAcceptDrops(True)
self.numCustoms = 0
self.customInputs = []
self.templateFilePath = ""
aTree = {}
self.createtablemodel = TableModelC(aTree, self)
self.filterModel = FilterModel(self)
self.filterModel.setSourceModel(self.createtablemodel)
self.ui.metadataTableView.setModel(self.filterModel)
self.ui.metadataTableView.horizontalHeader().setSectionResizeMode(
self.createtablemodel.K_SOURCE_COL_INDEX,
QHeaderView.ResizeToContents)
self.ui.metadataTableView.horizontalHeader().setSectionResizeMode(
self.createtablemodel.K_HTNAME_COL_INDEX,
QHeaderView.ResizeToContents)
self.ui.metadataTableView.horizontalHeader().setSectionResizeMode(
self.createtablemodel.K_SOURCEVAL_COL_INDEX,
QHeaderView.ResizeToContents)
self.ui.metadataTableView.horizontalHeader().setSectionResizeMode(
self.createtablemodel.K_HTVALUE_COL_INDEX,
QHeaderView.ResizeToContents)
self.ui.metadataTableView.setColumnWidth(
self.createtablemodel.K_USESOURCE_COL_INDEX,
self.width() * .1)
self.checkboxDelegate = CheckBoxDelegate()
self.createtrashDelegate = TrashDelegate()
self.ui.metadataTableView.setItemDelegateForColumn(
self.createtablemodel.K_USESOURCE_COL_INDEX, self.checkboxDelegate)
self.ui.metadataTableView.setItemDelegateForColumn(
self.createtablemodel.K_EDITABLE_COL_INDEX, self.checkboxDelegate)
self.ui.metadataTableView.setItemDelegateForColumn(
self.createtablemodel.K_REMOVE_COL_INDEX, self.createtrashDelegate)
self.ui.metadataTableView.setColumnWidth(
self.createtablemodel.K_REMOVE_COL_INDEX,
self.width() * .05)
self.treeModel = TreeModel(["Available File Metadata"], aTree,
self.createtablemodel)
self.ui.metadataTreeView.setModel(self.treeModel)
self.treeModel.checkChanged.connect(self.filterModel.checkList)
self.createtrashDelegate.pressed.connect(self.handleRemoveCreate)
self.editableKeys = []
self.usetablemodel = TableModelU(self, [], self.editableKeys)
self.usefilterModel = FilterModelU(self)
self.usefilterModel.setSourceModel(self.usetablemodel)
self.ui.useTemplateTableView.setModel(self.usefilterModel)
self.ui.useTemplateTableView.setColumnWidth(
self.usetablemodel.K_HTKEY_COL_INDEX,
self.width() * .25)
self.ui.useTemplateTableView.setColumnWidth(
self.usetablemodel.K_HTVALUE_COL_INDEX,
self.width() * .25)
self.ui.useTemplateTableView.setColumnWidth(
self.usetablemodel.K_SOURCE_COL_INDEX,
self.width() * .25)
self.ui.useTemplateTableView.setColumnWidth(
self.usetablemodel.K_USESOURCE_COL_INDEX,
self.width() * .1)
self.ui.useTemplateTableView.setColumnWidth(
self.usetablemodel.K_HTANNOTATION_COL_INDEX,
self.width() * .1)
self.usetrashDelegate = TrashDelegate()
self.ui.useTemplateTableView.setItemDelegateForColumn(
self.usetablemodel.K_USESOURCE_COL_INDEX, self.checkboxDelegate)
self.ui.useTemplateTableView.setItemDelegateForColumn(
self.usetablemodel.K_REMOVE_COL_INDEX, self.usetrashDelegate)
self.ui.useTemplateTableView.setColumnWidth(
self.usetablemodel.K_REMOVE_COL_INDEX,
self.width() * .075)
self.usetrashDelegate.pressed.connect(self.handleRemoveUse)
self.uselistmodel = ListModel(self, self.usetablemodel, [])
self.ui.useTemplateListView.setModel(self.uselistmodel)
self.uselistmodel.rowRemoved.connect(self.toggleButtons)
self.uselistmodel.rowAdded.connect(self.toggleButtons)
self.useFileDelegate = UseFileDelegate(self)
self.ui.useTemplateListView.setItemDelegate(self.useFileDelegate)
self.ui.useTemplateListView.clicked.connect(
self.removeRowfromUsefileType)
self.addAppendButton()
self.ui.TabWidget.currentChanged.connect(self.movethedamnbutton)
self.appendSourceFilesButton.clicked.connect(self.addFile)
self.ui.appendCreateTableRowButton.clicked.connect(
self.addCreateTableRow)
self.ui.appendUseTableRowButton.clicked.connect(self.addUseTableRow)
self.ui.hyperthoughtLocationButton.clicked.connect(
self.hyperthoughtui.exec)
self.ui.usetableSearchBar.textChanged.connect(self.filterUseTable)
self.ui.createTemplateListSearchBar.textChanged.connect(
self.filterCreateTable)
self.ui.createTemplateTreeSearchBar.textChanged.connect(
self.filterCreateTree)
self.ui.addMetadataFileCheckBox.stateChanged.connect(
self.checkFileList)
self.fileType = ""
self.accessKey = ""
self.folderuuid = ""
self.mThread = QThread()
self.uploader = Uploader()
self.uploader.moveToThread(self.mThread)
self.mThread.start()
self.createUpload.connect(self.uploader.performUpload)
self.hyperthoughtui.apiSubmitted.connect(self.acceptKey)
self.hyperthoughtui.finished.connect(self.getLocation)
self.ui.hyperthoughtTemplateLineEdit.installEventFilter(self)
self.ui.otherDataFileLineEdit.installEventFilter(self)
self.ui.dataFileLineEdit.installEventFilter(self)
import sys
from treeitem import TreeItem
from treemodel import TreeModel
from PyQt5.QtCore import Qt
from PyQt5.QtWidgets import QTreeView
from PyQt5.QtWidgets import QApplication
app = QApplication(sys.argv)
with open('view/default.txt', 'r', encoding='utf8') as f:
model = TreeModel(f.read())
view = QTreeView()
view.setModel(model)
view.setWindowTitle("Simple Tree Model")
view.show()
sys.exit(app.exec())
