UIMediumSelector.cpp@ 91341

Last change on this file since 91341 was 91341, checked in by vboxsync, 4 years ago
FE/Qt: bugref:9340. Make sure newly created VISO is selected even if UICommon::sigMediumCreated is not emitted.
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 24.9 KB

Line
1	/* $Id: UIMediumSelector.cpp 91341 2021-09-23 07:18:29Z vboxsync $ */
2	/** @file
3	* VBox Qt GUI - UIMediumSelector class implementation.
4	*/
5
6	/*
7	* Copyright (C) 2006-2020 Oracle Corporation
8	*
9	* This file is part of VirtualBox Open Source Edition (OSE), as
10	* available from http://www.virtualbox.org. This file is free software;
11	* you can redistribute it and/or modify it under the terms of the GNU
12	* General Public License (GPL) as published by the Free Software
13	* Foundation, in version 2 as it comes in the "COPYING" file of the
14	* VirtualBox OSE distribution. VirtualBox OSE is distributed in the
15	* hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
16	*/
17
18	/* Qt includes: */
19	#include <QAction>
20	#include <QHeaderView>
21	#include <QMenuBar>
22	#include <QVBoxLayout>
23	#include <QPushButton>
24
25	/* GUI includes: */
26	#include "QIDialogButtonBox.h"
27	#include "QIFileDialog.h"
28	#include "QIMessageBox.h"
29	#include "QITabWidget.h"
30	#include "QIToolButton.h"
31	#include "UICommon.h"
32	#include "UIDesktopWidgetWatchdog.h"
33	#include "UIExtraDataManager.h"
34	#include "UIMediumSearchWidget.h"
35	#include "UIMediumSelector.h"
36	#include "UIMessageCenter.h"
37	#include "UIIconPool.h"
38	#include "UIMedium.h"
39	#include "UIMediumItem.h"
40	#include "QIToolBar.h"
41
42	/* COM includes: */
43	#include "COMEnums.h"
44	#include "CMachine.h"
45	#include "CMediumAttachment.h"
46	#include "CMediumFormat.h"
47	#include "CStorageController.h"
48	#include "CSystemProperties.h"
49
50	#ifdef VBOX_WS_MAC
51	# include "UIWindowMenuManager.h"
52	#endif /* VBOX_WS_MAC */
53
54
55	UIMediumSelector::UIMediumSelector(UIMediumDeviceType enmMediumType, const QString &machineName,
56	const QString &machineSettingsFilePath, const QString &strMachineGuestOSTypeId,
57	const QUuid &uMachineID, QWidget *pParent)
58	:QIWithRetranslateUI<QIMainDialog>(pParent)
59	, m_pCentralWidget(0)
60	, m_pMainLayout(0)
61	, m_pTreeWidget(0)
62	, m_enmMediumType(enmMediumType)
63	, m_pButtonBox(0)
64	, m_pCancelButton(0)
65	, m_pChooseButton(0)
66	, m_pLeaveEmptyButton(0)
67	, m_pMainMenu(0)
68	, m_pToolBar(0)
69	, m_pActionAdd(0)
70	, m_pActionCreate(0)
71	, m_pActionRefresh(0)
72	, m_pAttachedSubTreeRoot(0)
73	, m_pNotAttachedSubTreeRoot(0)
74	, m_pParent(pParent)
75	, m_pSearchWidget(0)
76	, m_iCurrentShownIndex(0)
77	, m_strMachineFolder(machineSettingsFilePath)
78	, m_strMachineName(machineName)
79	, m_strMachineGuestOSTypeId(strMachineGuestOSTypeId)
80	, m_uMachineID(uMachineID)
81	{
82	/* Start full medium-enumeration (if necessary): */
83	if (!uiCommon().isFullMediumEnumerationRequested())
84	uiCommon().enumerateMedia();
85	configure();
86	finalize();
87	}
88
89	void UIMediumSelector::setEnableCreateAction(bool fEnable)
90	{
91	if (!m_pActionCreate)
92	return;
93	m_pActionCreate->setEnabled(fEnable);
94	m_pActionCreate->setVisible(fEnable);
95	}
96
97	QList<QUuid> UIMediumSelector::selectedMediumIds() const
98	{
99	QList<QUuid> selectedIds;
100	if (!m_pTreeWidget)
101	return selectedIds;
102	QList<QTreeWidgetItem*> selectedItems = m_pTreeWidget->selectedItems();
103	for (int i = 0; i < selectedItems.size(); ++i)
104	{
105	UIMediumItem item = dynamic_cast<UIMediumItem>(selectedItems.at(i));
106	if (item)
107	selectedIds.push_back(item->medium().id());
108	}
109	return selectedIds;
110	}
111
112	void UIMediumSelector::retranslateUi()
113	{
114	if (m_pMainMenu)
115	m_pMainMenu->setTitle(tr("Medium"));
116
117	if (m_pActionAdd)
118	{
119	m_pActionAdd->setText(tr("&Add..."));
120	m_pActionAdd->setToolTip(tr("Add Disk Image"));
121	m_pActionAdd->setStatusTip(tr("Add existing disk image file"));
122	}
123
124	if (m_pActionCreate)
125	{
126	m_pActionCreate->setText(tr("&Create..."));
127	m_pActionCreate->setToolTip(tr("Create Disk Image"));
128	m_pActionCreate->setStatusTip(tr("Create new disk image file"));
129	}
130
131	if (m_pActionRefresh)
132	{
133	m_pActionRefresh->setText(tr("&Refresh"));
134	m_pActionRefresh->setToolTip(tr("Refresh Disk Image Files (%1)").arg(m_pActionRefresh->shortcut().toString()));
135	m_pActionRefresh->setStatusTip(tr("Refresh the list of disk image files"));
136	}
137
138	if (m_pCancelButton)
139	m_pCancelButton->setText(tr("Cancel"));
140	if (m_pLeaveEmptyButton)
141	m_pLeaveEmptyButton->setText(tr("Leave Empty"));
142	if (m_pChooseButton)
143	m_pChooseButton->setText(tr("Choose"));
144
145	if (m_pTreeWidget)
146	{
147	m_pTreeWidget->headerItem()->setText(0, tr("Name"));
148	m_pTreeWidget->headerItem()->setText(1, tr("Virtual Size"));
149	m_pTreeWidget->headerItem()->setText(2, tr("Actual Size"));
150	}
151	}
152
153	void UIMediumSelector::configure()
154	{
155	/* Apply window icons: */
156	setWindowIcon(UIIconPool::iconSetFull(":/media_manager_32px.png", ":/media_manager_16px.png"));
157	setTitle();
158	prepareWidgets();
159	prepareActions();
160	prepareMenuAndToolBar();
161	prepareConnections();
162	}
163
164	void UIMediumSelector::prepareActions()
165	{
166	QString strPrefix("hd");
167	switch (m_enmMediumType)
168	{
169	case UIMediumDeviceType_DVD:
170	strPrefix = "cd";
171	break;
172	case UIMediumDeviceType_Floppy:
173	strPrefix = "fd";
174	break;
175	case UIMediumDeviceType_HardDisk:
176	case UIMediumDeviceType_All:
177	case UIMediumDeviceType_Invalid:
178	default:
179	strPrefix = "hd";
180	break;
181	}
182
183	m_pActionAdd = new QAction(this);
184	if (m_pActionAdd)
185	{
186	/* Configure add-action: */
187	m_pActionAdd->setShortcut(QKeySequence(""));
188
189	m_pActionAdd->setIcon(UIIconPool::iconSetFull(QString(":/%1_add_32px.png").arg(strPrefix),
190	QString(":/%1_add_16px.png").arg(strPrefix),
191	QString(":/%1_add_disabled_32px.png").arg(strPrefix),
192	QString(":/%1_add_disabled_16px.png").arg(strPrefix)));
193	}
194
195	m_pActionCreate = new QAction(this);
196	if (m_pActionCreate)
197	{
198	m_pActionCreate->setShortcut(QKeySequence(""));
199	m_pActionCreate->setIcon(UIIconPool::iconSetFull(QString(":/%1_create_32px.png").arg(strPrefix),
200	QString(":/%1_create_16px.png").arg(strPrefix),
201	QString(":/%1_create_disabled_32px.png").arg(strPrefix),
202	QString(":/%1_create_disabled_16px.png").arg(strPrefix)));
203	}
204
205	m_pActionRefresh = new QAction(this);
206	if (m_pActionRefresh)
207	{
208	m_pActionRefresh->setShortcut(QKeySequence());
209	if (m_pActionRefresh && m_pActionRefresh->icon().isNull())
210	m_pActionRefresh->setIcon(UIIconPool::iconSetFull(":/refresh_32px.png", ":/refresh_16px.png",
211	":/refresh_disabled_32px.png", ":/refresh_disabled_16px.png"));
212	}
213	}
214
215	void UIMediumSelector::prepareMenuAndToolBar()
216	{
217	if (!m_pMainMenu \|\| !m_pToolBar)
218	return;
219
220	m_pMainMenu->addAction(m_pActionAdd);
221	m_pMainMenu->addAction(m_pActionCreate);
222	m_pMainMenu->addSeparator();
223	m_pMainMenu->addAction(m_pActionRefresh);
224
225	m_pToolBar->addAction(m_pActionAdd);
226	if (!(gEDataManager->restrictedDialogTypes(m_uMachineID) & UIExtraDataMetaDefs::DialogType_VISOCreator))
227	m_pToolBar->addAction(m_pActionCreate);
228	m_pToolBar->addSeparator();
229	m_pToolBar->addAction(m_pActionRefresh);
230	}
231
232	void UIMediumSelector::prepareConnections()
233	{
234	/* Configure medium-enumeration connections: */
235	connect(&uiCommon(), &UICommon::sigMediumCreated,
236	this, &UIMediumSelector::sltHandleMediumCreated);
237	connect(&uiCommon(), &UICommon::sigMediumEnumerationStarted,
238	this, &UIMediumSelector::sltHandleMediumEnumerationStart);
239	connect(&uiCommon(), &UICommon::sigMediumEnumerated,
240	this, &UIMediumSelector::sltHandleMediumEnumerated);
241	connect(&uiCommon(), &UICommon::sigMediumEnumerationFinished,
242	this, &UIMediumSelector::sltHandleMediumEnumerationFinish);
243	if (m_pActionAdd)
244	connect(m_pActionAdd, &QAction::triggered, this, &UIMediumSelector::sltAddMedium);
245	if (m_pActionCreate)
246	connect(m_pActionCreate, &QAction::triggered, this, &UIMediumSelector::sltCreateMedium);
247	if (m_pActionRefresh)
248	connect(m_pActionRefresh, &QAction::triggered, this, &UIMediumSelector::sltHandleRefresh);
249
250	if (m_pTreeWidget)
251	{
252	connect(m_pTreeWidget, &QITreeWidget::itemSelectionChanged, this, &UIMediumSelector::sltHandleItemSelectionChanged);
253	connect(m_pTreeWidget, &QITreeWidget::itemDoubleClicked, this, &UIMediumSelector::sltHandleTreeWidgetDoubleClick);
254	connect(m_pTreeWidget, &QITreeWidget::customContextMenuRequested, this, &UIMediumSelector::sltHandleTreeContextMenuRequest);
255	}
256
257	if (m_pCancelButton)
258	connect(m_pCancelButton, &QPushButton::clicked, this, &UIMediumSelector::sltButtonCancel);
259	if (m_pChooseButton)
260	connect(m_pChooseButton, &QPushButton::clicked, this, &UIMediumSelector::sltButtonChoose);
261	if (m_pLeaveEmptyButton)
262	connect(m_pLeaveEmptyButton, &QPushButton::clicked, this, &UIMediumSelector::sltButtonLeaveEmpty);
263
264	if (m_pSearchWidget)
265	{
266	connect(m_pSearchWidget, &UIMediumSearchWidget::sigPerformSearch,
267	this, &UIMediumSelector::sltHandlePerformSearch);
268	}
269	}
270
271	UIMediumItem* UIMediumSelector::addTreeItem(const UIMedium &medium, QITreeWidgetItem *pParent)
272	{
273	if (!pParent)
274	return 0;
275	switch (m_enmMediumType)
276	{
277	case UIMediumDeviceType_DVD:
278	return new UIMediumItemCD(medium, pParent);
279	break;
280	case UIMediumDeviceType_Floppy:
281	return new UIMediumItemFD(medium, pParent);
282	break;
283	case UIMediumDeviceType_HardDisk:
284	case UIMediumDeviceType_All:
285	case UIMediumDeviceType_Invalid:
286	default:
287	return createHardDiskItem(medium, pParent);
288	break;
289	}
290	}
291
292	UIMediumItem* UIMediumSelector::createHardDiskItem(const UIMedium &medium, QITreeWidgetItem *pParent)
293	{
294	if (medium.medium().isNull())
295	return 0;
296	if (!m_pTreeWidget)
297	return 0;
298	/* Search the tree to see if we already have the item: */
299	UIMediumItem *pMediumItem = searchItem(0, medium.id());
300	if (pMediumItem)
301	return pMediumItem;
302	/* Check if the corresponding medium has a parent */
303	if (medium.parentID() != UIMedium::nullID())
304	{
305	UIMediumItem *pParentMediumItem = searchItem(0, medium.parentID());
306	/* If parent medium-item was not found we create it: */
307	if (!pParentMediumItem)
308	{
309	/* Make sure corresponding parent medium is already cached! */
310	UIMedium parentMedium = uiCommon().medium(medium.parentID());
311	if (parentMedium.isNull())
312	AssertMsgFailed(("Parent medium with ID={%s} was not found!\n", medium.parentID().toString().toUtf8().constData()));
313	/* Try to create parent medium-item: */
314	else
315	pParentMediumItem = createHardDiskItem(parentMedium, pParent);
316	}
317	if (pParentMediumItem)
318	{
319	pMediumItem = new UIMediumItemHD(medium, pParentMediumItem);
320	LogRel2(("UIMediumManager: Child hard-disk medium-item with ID={%s} created.\n", medium.id().toString().toUtf8().constData()));
321	}
322	else
323	AssertMsgFailed(("Parent medium with ID={%s} could not be created!\n", medium.parentID().toString().toUtf8().constData()));
324	}
325
326	/* No parents, thus just create item as top-level one: */
327	else
328	{
329	pMediumItem = new UIMediumItemHD(medium, pParent);
330	LogRel2(("UIMediumManager: Root hard-disk medium-item with ID={%s} created.\n", medium.id().toString().toUtf8().constData()));
331	}
332	return pMediumItem;
333	}
334
335	void UIMediumSelector::restoreSelection(const QList<QUuid> &selectedMediums, QVector<UIMediumItem*> &mediumList)
336	{
337	if (!m_pTreeWidget)
338	return;
339	if (selectedMediums.isEmpty())
340	{
341	m_pTreeWidget->setCurrentItem(0);
342	return;
343	}
344	bool selected = false;
345	for (int i = 0; i < mediumList.size(); ++i)
346	{
347	if (!mediumList[i])
348	continue;
349	if (selectedMediums.contains(mediumList[i]->medium().id()))
350	{
351	mediumList[i]->setSelected(true);
352	selected = true;
353	}
354	}
355
356	if (!selected)
357	m_pTreeWidget->setCurrentItem(0);
358	return;
359	}
360
361	void UIMediumSelector::prepareWidgets()
362	{
363	m_pCentralWidget = new QWidget;
364	if (!m_pCentralWidget)
365	return;
366	setCentralWidget(m_pCentralWidget);
367
368	m_pMainLayout = new QVBoxLayout;
369	m_pCentralWidget->setLayout(m_pMainLayout);
370
371	if (!m_pMainLayout \|\| !menuBar())
372	return;
373
374	m_pMainMenu = menuBar()->addMenu(tr("Medium"));
375
376	m_pToolBar = new QIToolBar;
377	if (m_pToolBar)
378	{
379	/* Configure toolbar: */
380	const int iIconMetric = (int)(QApplication::style()->pixelMetric(QStyle::PM_LargeIconSize));
381	m_pToolBar->setIconSize(QSize(iIconMetric, iIconMetric));
382	m_pToolBar->setToolButtonStyle(Qt::ToolButtonTextUnderIcon);
383	m_pMainLayout->addWidget(m_pToolBar);
384	}
385
386	m_pTreeWidget = new QITreeWidget;
387	if (m_pTreeWidget)
388	{
389	m_pTreeWidget->setSelectionMode(QAbstractItemView::SingleSelection);
390	m_pMainLayout->addWidget(m_pTreeWidget);
391	m_pTreeWidget->setAlternatingRowColors(true);
392	int iColumnCount = (m_enmMediumType == UIMediumDeviceType_HardDisk) ? 3 : 2;
393	m_pTreeWidget->setColumnCount(iColumnCount);
394	m_pTreeWidget->setSortingEnabled(true);
395	m_pTreeWidget->sortItems(0, Qt::AscendingOrder);
396	m_pTreeWidget->setContextMenuPolicy(Qt::CustomContextMenu);
397	}
398
399	m_pSearchWidget = new UIMediumSearchWidget;
400	if (m_pSearchWidget)
401	{
402	m_pMainLayout->addWidget(m_pSearchWidget);
403	}
404
405	m_pButtonBox = new QIDialogButtonBox;
406	if (m_pButtonBox)
407	{
408	/* Configure button-box: */
409	m_pCancelButton = m_pButtonBox->addButton(tr("Cancel"), QDialogButtonBox::RejectRole);
410
411	/* Only DVDs and Floppies can be left empty: */
412	if (m_enmMediumType == UIMediumDeviceType_DVD \|\| m_enmMediumType == UIMediumDeviceType_Floppy)
413	m_pLeaveEmptyButton = m_pButtonBox->addButton(tr("Leave Empty"), QDialogButtonBox::ActionRole);
414
415	m_pChooseButton = m_pButtonBox->addButton(tr("Choose"), QDialogButtonBox::AcceptRole);
416	m_pCancelButton->setShortcut(Qt::Key_Escape);
417
418	/* Add button-box into main layout: */
419	m_pMainLayout->addWidget(m_pButtonBox);
420	}
421
422	repopulateTreeWidget();
423	}
424
425	void UIMediumSelector::sltButtonChoose()
426	{
427	done(static_cast<int>(ReturnCode_Accepted));
428	}
429
430	void UIMediumSelector::sltButtonCancel()
431	{
432	done(static_cast<int>(ReturnCode_Rejected));
433	}
434
435	void UIMediumSelector::sltButtonLeaveEmpty()
436	{
437	done(static_cast<int>(ReturnCode_LeftEmpty));
438	}
439
440	void UIMediumSelector::sltAddMedium()
441	{
442	QUuid uMediumID = uiCommon().openMediumWithFileOpenDialog(m_enmMediumType, this, m_strMachineFolder);
443	if (uMediumID.isNull())
444	return;
445	repopulateTreeWidget();
446	selectMedium(uMediumID);
447	}
448
449	void UIMediumSelector::sltCreateMedium()
450	{
451	QUuid uMediumId = uiCommon().openMediumCreatorDialog(this, m_enmMediumType, m_strMachineFolder,
452	m_strMachineName, m_strMachineGuestOSTypeId);
453	/* Make sure that the data structure is updated and newly created medium is selected and visible: */
454	sltHandleMediumCreated(uMediumId);
455	}
456
457	void UIMediumSelector::sltHandleItemSelectionChanged()
458	{
459	updateChooseButton();
460	}
461
462	void UIMediumSelector::sltHandleTreeWidgetDoubleClick(QTreeWidgetItem * item, int column)
463	{
464	Q_UNUSED(column);
465	if (!dynamic_cast<UIMediumItem*>(item))
466	return;
467	accept();
468	}
469
470	void UIMediumSelector::sltHandleMediumCreated(const QUuid &uMediumId)
471	{
472	if (uMediumId.isNull())
473	return;
474	/* Update the tree widget making sure we show the new item: */
475	repopulateTreeWidget();
476	/* Select the new item: */
477	selectMedium(uMediumId);
478	/* Update the search: */
479	m_pSearchWidget->search(m_pTreeWidget);
480	}
481
482	void UIMediumSelector::sltHandleMediumEnumerationStart()
483	{
484	/* Disable controls. Left Alone button box 'Ok' button. it is handle by tree population: */
485	if (m_pActionRefresh)
486	m_pActionRefresh->setEnabled(false);
487	}
488
489	void UIMediumSelector::sltHandleMediumEnumerated()
490	{
491	}
492
493	void UIMediumSelector::sltHandleMediumEnumerationFinish()
494	{
495	repopulateTreeWidget();
496	if (m_pActionRefresh)
497	m_pActionRefresh->setEnabled(true);
498	}
499
500	void UIMediumSelector::sltHandleRefresh()
501	{
502	/* Restart full medium-enumeration: */
503	uiCommon().enumerateMedia();
504	/* Update the search: */
505	m_pSearchWidget->search(m_pTreeWidget);
506	}
507
508	void UIMediumSelector::sltHandlePerformSearch()
509	{
510	//performMediumSearch();
511	if (!m_pSearchWidget)
512	return;
513	m_pSearchWidget->search(m_pTreeWidget);
514	}
515
516	void UIMediumSelector::sltHandleTreeContextMenuRequest(const QPoint &point)
517	{
518	QWidget pSender = qobject_cast<QWidget>(sender());
519	if (!pSender)
520	return;
521
522	QMenu menu;
523	QAction *pExpandAll = menu.addAction(tr("Expand All"));
524	QAction *pCollapseAll = menu.addAction(tr("Collapse All"));
525	if (!pExpandAll \|\| !pCollapseAll)
526	return;
527
528	pExpandAll->setIcon(UIIconPool::iconSet(":/expand_all_16px.png"));
529	pCollapseAll->setIcon(UIIconPool::iconSet(":/collapse_all_16px.png"));
530
531	connect(pExpandAll, &QAction::triggered, this, &UIMediumSelector::sltHandleTreeExpandAllSignal);
532	connect(pCollapseAll, &QAction::triggered, this, &UIMediumSelector::sltHandleTreeCollapseAllSignal);
533
534	menu.exec(pSender->mapToGlobal(point));
535	}
536
537	void UIMediumSelector::sltHandleTreeExpandAllSignal()
538	{
539	if (m_pTreeWidget)
540	m_pTreeWidget->expandAll();
541	}
542
543	void UIMediumSelector::sltHandleTreeCollapseAllSignal()
544	{
545	if (m_pTreeWidget)
546	m_pTreeWidget->collapseAll();
547
548	if (m_pAttachedSubTreeRoot)
549	m_pTreeWidget->setExpanded(m_pTreeWidget->itemIndex(m_pAttachedSubTreeRoot), true);
550	if (m_pNotAttachedSubTreeRoot)
551	m_pTreeWidget->setExpanded(m_pTreeWidget->itemIndex(m_pNotAttachedSubTreeRoot), true);
552	}
553
554	void UIMediumSelector::selectMedium(const QUuid &uMediumID)
555	{
556	if (!m_pTreeWidget)
557	return;
558	UIMediumItem *pMediumItem = searchItem(0, uMediumID);
559	if (pMediumItem)
560	m_pTreeWidget->setCurrentItem(pMediumItem);
561	}
562
563	void UIMediumSelector::updateChooseButton()
564	{
565
566	if (!m_pTreeWidget \|\| !m_pChooseButton)
567	return;
568	QList<QTreeWidgetItem*> selectedItems = m_pTreeWidget->selectedItems();
569	if (selectedItems.isEmpty())
570	{
571	m_pChooseButton->setEnabled(false);
572	return;
573	}
574
575	/* check if at least one of the selected items is a UIMediumItem */
576	bool mediumItemSelected = false;
577	for (int i = 0; i < selectedItems.size() && !mediumItemSelected; ++i)
578	{
579	if (dynamic_cast<UIMediumItem*>(selectedItems.at(i)))
580	mediumItemSelected = true;
581	}
582	if (mediumItemSelected)
583	m_pChooseButton->setEnabled(true);
584	else
585	m_pChooseButton->setEnabled(false);
586	}
587
588	void UIMediumSelector::finalize()
589	{
590	/* Apply language settings: */
591	retranslateUi();
592	}
593
594	void UIMediumSelector::showEvent(QShowEvent *pEvent)
595	{
596	Q_UNUSED(pEvent);
597
598	/* Try to determine the initial size: */
599	QSize proposedSize;
600	int iHostScreen = 0;
601	if (m_pParent)
602	iHostScreen = gpDesktop->screenNumber(m_pParent);
603	else
604	iHostScreen = gpDesktop->screenNumber(this);
605	if (iHostScreen >= 0 && iHostScreen < gpDesktop->screenCount())
606	{
607	/* On the basis of current host-screen geometry if possible: */
608	const QRect screenGeometry = gpDesktop->screenGeometry(iHostScreen);
609	if (screenGeometry.isValid())
610	proposedSize = screenGeometry.size() * 5 / 15;
611	}
612	/* Fallback to default size if we failed: */
613	if (proposedSize.isNull())
614	proposedSize = QSize(800, 600);
615	/* Resize to initial size: */
616	resize(proposedSize);
617
618	if (m_pParent)
619	UIDesktopWidgetWatchdog::centerWidget(this, m_pParent, false);
620	}
621
622	void UIMediumSelector::repopulateTreeWidget()
623	{
624	if (!m_pTreeWidget)
625	return;
626	/* Cache the currently selected items: */
627	QList<QTreeWidgetItem*> selectedItems = m_pTreeWidget->selectedItems();
628	QList<QUuid> selectedMedia = selectedMediumIds();
629	/* uuid list of selected items: */
630	/* Reset the related data structure: */
631	m_mediumItemList.clear();
632	m_pTreeWidget->clear();
633	m_pAttachedSubTreeRoot = 0;
634	m_pNotAttachedSubTreeRoot = 0;
635	QVector<UIMediumItem*> menuItemVector;
636	foreach (const QUuid &uMediumID, uiCommon().mediumIDs())
637	{
638	UIMedium medium = uiCommon().medium(uMediumID);
639	if (medium.type() == m_enmMediumType)
640	{
641	bool isMediumAttached = !(medium.medium().GetMachineIds().isEmpty());
642	QITreeWidgetItem *pParent = 0;
643	if (isMediumAttached)
644	{
645	if (!m_pAttachedSubTreeRoot)
646	{
647	QStringList strList;
648	strList << "Attached";
649	m_pAttachedSubTreeRoot = new QITreeWidgetItem(m_pTreeWidget, strList);
650	}
651	pParent = m_pAttachedSubTreeRoot;
652
653	}
654	else
655	{
656	if (!m_pNotAttachedSubTreeRoot)
657	{
658	QStringList strList;
659	strList << "Not Attached";
660	m_pNotAttachedSubTreeRoot = new QITreeWidgetItem(m_pTreeWidget, strList);
661	}
662	pParent = m_pNotAttachedSubTreeRoot;
663	}
664	UIMediumItem *treeItem = addTreeItem(medium, pParent);
665	m_mediumItemList.append(treeItem);
666	menuItemVector.push_back(treeItem);
667	}
668	}
669	restoreSelection(selectedMedia, menuItemVector);
670	saveDefaultForeground();
671	updateChooseButton();
672	if (m_pAttachedSubTreeRoot)
673	m_pTreeWidget->expandItem(m_pAttachedSubTreeRoot);
674	if (m_pNotAttachedSubTreeRoot)
675	m_pTreeWidget->expandItem(m_pNotAttachedSubTreeRoot);
676	m_pTreeWidget->resizeColumnToContents(0);
677	}
678
679	void UIMediumSelector::saveDefaultForeground()
680	{
681	if (!m_pTreeWidget)
682	return;
683	if (m_defaultItemForeground == QBrush() && m_pTreeWidget->topLevelItemCount() >= 1)
684	{
685	QTreeWidgetItem *item = m_pTreeWidget->topLevelItem(0);
686	if (item)
687	{
688	QVariant data = item->data(0, Qt::ForegroundRole);
689	if (data.canConvert<QBrush>())
690	{
691	m_defaultItemForeground = data.value<QBrush>();
692	}
693	}
694	}
695	}
696
697	UIMediumItem* UIMediumSelector::searchItem(const QTreeWidgetItem *pParent, const QUuid &mediumId)
698	{
699	if (!m_pTreeWidget)
700	return 0;
701	if (!pParent)
702	pParent = m_pTreeWidget->invisibleRootItem();
703	if (!pParent)
704	return 0;
705
706	for (int i = 0; i < pParent->childCount(); ++i)
707	{
708	QTreeWidgetItem *pChild = pParent->child(i);
709	if (!pChild)
710	continue;
711	UIMediumItem mediumItem = dynamic_cast<UIMediumItem>(pChild);
712	if (mediumItem)
713	{
714	if (mediumItem->id() == mediumId)
715	{
716	return mediumItem;
717	}
718	}
719	UIMediumItem *pResult = searchItem(pChild, mediumId);
720	if (pResult)
721	return pResult;
722	}
723	return 0;
724	}
725
726	void UIMediumSelector::scrollToItem(UIMediumItem* pItem)
727	{
728	if (!pItem)
729	return;
730
731	QModelIndex itemIndex = m_pTreeWidget->itemIndex(pItem);
732	for (int i = 0; i < m_mediumItemList.size(); ++i)
733	{
734	QFont font = m_mediumItemList[i]->font(0);
735	font.setBold(false);
736	m_mediumItemList[i]->setFont(0, font);
737	}
738	QFont font = pItem->font(0);
739	font.setBold(true);
740	pItem->setFont(0, font);
741
742	m_pTreeWidget->scrollTo(itemIndex);
743	}
744
745	void UIMediumSelector::setTitle()
746	{
747	switch (m_enmMediumType)
748	{
749	case UIMediumDeviceType_DVD:
750	if (!m_strMachineName.isEmpty())
751	setWindowTitle(QString("%1 - %2").arg(m_strMachineName).arg(tr("Optical Disk Selector")));
752	else
753	setWindowTitle(QString("%1").arg(tr("Optical Disk Selector")));
754	break;
755	case UIMediumDeviceType_Floppy:
756	if (!m_strMachineName.isEmpty())
757	setWindowTitle(QString("%1 - %2").arg(m_strMachineName).arg(tr("Floppy Disk Selector")));
758	else
759	setWindowTitle(QString("%1").arg(tr("Floppy Disk Selector")));
760	break;
761	case UIMediumDeviceType_HardDisk:
762	if (!m_strMachineName.isEmpty())
763	setWindowTitle(QString("%1 - %2").arg(m_strMachineName).arg(tr("Hard Disk Selector")));
764	else
765	setWindowTitle(QString("%1").arg(tr("Hard Disk Selector")));
766	break;
767	case UIMediumDeviceType_All:
768	case UIMediumDeviceType_Invalid:
769	default:
770	if (!m_strMachineName.isEmpty())
771	setWindowTitle(QString("%1 - %2").arg(m_strMachineName).arg(tr("Virtual Medium Selector")));
772	else
773	setWindowTitle(QString("%1").arg(tr("Virtual Medium Selector")));
774	break;
775	}
776	}

Note: See TracBrowser for help on using the repository browser.

source: vbox/trunk/src/VBox/Frontends/VirtualBox/src/medium/UIMediumSelector.cpp@ 91341

Download in other formats: