#include "mainwindow.h"
#include "ui_mainwindow.h"
#define DEBUG_TEST qDebug()
MainWindow::MainWindow(QWidget *parent) :
QMainWindow(parent),
ui(new Ui::MainWindow)
{
ui->setupUi(this);
tableWidget_DevInfo_Init();
ScanHD();
tableWidget_DevInfo_Update();
}
MainWindow::~MainWindow()
{
delete ui;
}
char MainWindow::FirstDriveFromMask(ULONG unitmask)
{
char i;
for (i = 0; i < 26; ++i)
{
if (unitmask & 0x1)
break;
unitmask = unitmask >> 1;
}
return (i + 'A');
}
//=================================================================================
//系统启动扫描已经连接的U盘 程序只会在系统执行时执行一次
//=================================================================================
void MainWindow::ScanHD()
{
DEBUG_TEST<<"U盘扫描";
char diskPath[5] = { 0 };
DWORD allDisk = GetLogicalDrives(); //返回一个32位整数，将他转换成二进制后，表示磁盘,最低位为A盘
udisk_lists.clear();
if (allDisk!=0)
{
for (int i=0;i<26;i++) //假定最多有12个磁盘从A开始计数
{
if ((allDisk & 1)==1)
{
sprintf( diskPath, "%c", 'A'+i );
strcat( diskPath, ":" );
LPCWSTR RootPathName =(LPCWSTR)QString(diskPath).utf16();
//DEBUG_TEST<<"盘符： "<<diskPath << GetDriveType(RootPathName);
if (GetDriveType(RootPathName) == DRIVE_REMOVABLE)
{
QString path(diskPath);
//this->AddDisk(path);
udisk_list_t udisk_tmp;
udisk_tmp.udisk_path = path;
udisk_tmp.is_check = true;
if( GetVolumeInformation(RootPathName,0,0,0,0,0,0,0) ) //判断驱动是否准备就绪
{
DEBUG_TEST<<"U盘准备就绪！"<<diskPath;
append_log("U盘准备就绪！" + path);
udisk_tmp.is_ready = true;
}
else
{
DEBUG_TEST<<"U盘未就绪！"<<diskPath;
append_log("U盘未就绪！"+path);
udisk_tmp.is_ready = false;
}
udisk_lists.append(udisk_tmp);
}
}
allDisk = allDisk>>1;
}
}
}
bool MainWindow::nativeEvent(const QByteArray &eventType, void *message, long *result)
{
MSG* msg = reinterpret_cast<MSG*>(message);
int msgType = msg->message;
if(msgType==WM_DEVICECHANGE)
{
PDEV_BROADCAST_HDR lpdb = (PDEV_BROADCAST_HDR)msg->lParam;
switch (msg->wParam) {
case DBT_DEVICEARRIVAL:
if(lpdb->dbch_devicetype == DBT_DEVTYP_VOLUME)
{
PDEV_BROADCAST_VOLUME lpdbv = (PDEV_BROADCAST_VOLUME)lpdb;
if(lpdbv->dbcv_flags ==0)
{
QString USBDisk = QString(this->FirstDriveFromMask(lpdbv ->dbcv_unitmask));
DEBUG_TEST << "USB_Arrived and The USBDisk is: "<<USBDisk ;
append_log("USB_Arrived and The USBDisk is: "+USBDisk );
ScanHD();
tableWidget_DevInfo_Update();
}
}
break;
case DBT_DEVICEREMOVECOMPLETE:
if(lpdb->dbch_devicetype == DBT_DEVTYP_VOLUME)
{
PDEV_BROADCAST_VOLUME lpdbv = (PDEV_BROADCAST_VOLUME)lpdb;
if(lpdbv->dbcv_flags == 0)
{
DEBUG_TEST << "USB_Removed";
QString USBDisk;
USBDisk = QString(this->FirstDriveFromMask(lpdbv ->dbcv_unitmask));
USBDisk += ":";
append_log("USB_Removed and The USBDisk is: "+ USBDisk );
//this->DeleteDisk(USBDisk);
DEBUG_TEST<<"this-delete is ="<<USBDisk;
ScanHD();
tableWidget_DevInfo_Update();
}
}
break;
}
}
return false;
}
void MainWindow::tableWidget_DevInfo_Init()
{
ui->tableWidget_DevInfo->setRowCount(2);
ui->tableWidget_DevInfo->setColumnCount(3);
//设置表头字体加粗
QFont font = ui->tableWidget_DevInfo->horizontalHeader()->font();
font.setBold(true);
ui->tableWidget_DevInfo->horizontalHeader()->setFont(font);
QStringList header;
header <<"选择操作"<<"可移动磁盘"<<"格式化";
ui->tableWidget_DevInfo->setHorizontalHeaderLabels(header);
// ui->tableWidget_DevInfo->resizeColumnsToContents(); //自适应宽度
ui->tableWidget_DevInfo->setColumnWidth(0,50);
ui->tableWidget_DevInfo->setColumnWidth(1,80);
ui->tableWidget_DevInfo->setColumnWidth(2,50);
ui->tableWidget_DevInfo->show();
}
void MainWindow::tableWidget_DevInfo_Update()
{
int row = udisk_lists.length();
//ui->tableWidget_DevInfo->clear();//此处会清表头
ui->tableWidget_DevInfo->clearContents();
ui->tableWidget_DevInfo->setRowCount(row+1);
for(int i=0; i<row; i++)
{
if(udisk_lists[i].is_ready)
ui->tableWidget_DevInfo->setItem(i, 1, new QTableWidgetItem(udisk_lists[i].udisk_path + " 准备就绪!"));
else
ui->tableWidget_DevInfo->setItem(i, 1, new QTableWidgetItem(udisk_lists[i].udisk_path + " 未就绪!"));
QCheckBox *ischeck = new QCheckBox();
connect(ischeck,SIGNAL(clicked()),this,SLOT(tableWidget_DevIsChecked()));
if(udisk_lists[i].is_check == true)
ischeck->setCheckState (Qt::Checked);
else
ischeck->setCheckState (Qt::Unchecked);
ui->tableWidget_DevInfo->setCellWidget(i,0,ischeck);
}
}
void MainWindow::tableWidget_DevIsChecked()
{
QCheckBox *senderObj=qobject_cast<QCheckBox*>(sender());
if(senderObj == nullptr)
{
return;
}
QModelIndex idx = ui->tableWidget_DevInfo->indexAt(QPoint(senderObj->frameGeometry().x(),senderObj->frameGeometry().y()));
int row=idx.row();
int col=idx.column();
// 其他响应信息......
udisk_lists[row].is_check = senderObj->isChecked();
DEBUG_TEST << "tableWidget_DevInfo is "<< senderObj->isChecked() << row << col;
}
bool MainWindow::fomat_udisk(QString udisk_path)
{
QString program = "cmd.exe";
QStringList arguments;
arguments << "/c"<< "format " + udisk_path +" /FS:FAT32 /Q /Y";
DEBUG_TEST << program << arguments;
QProcess p;
p.start(program, arguments);
if (p.waitForStarted())
{
p.waitForFinished();
DEBUG_TEST << QString::fromLocal8Bit(p.readAllStandardOutput());
return true;
}
else
{
DEBUG_TEST << "Failed to start";
return false;
}
}
void MainWindow::append_log(QString log)
{
ui->textEdit_Log->append(log);
}
void MainWindow::on_pushButton_Update_clicked()
{
ScanHD();
tableWidget_DevInfo_Update();
}
void MainWindow::on_pushButton_Show_clicked()
{
QString program = "explorer";
QStringList arguments;
int udisk_n = udisk_lists.length();
for(int i=0; i<udisk_n; ++i)
{
if(udisk_lists[i].is_check)
{
arguments.clear();
arguments << udisk_lists[i].udisk_path;
DEBUG_TEST << program << arguments;
QProcess p;
p.start(program, arguments);
if (p.waitForStarted())
{
p.waitForFinished();
DEBUG_TEST << QString::fromLocal8Bit(p.readAllStandardOutput());
}
else
