qt 显示插入U盘的信息

墨蓝 2024-04-19 12:50 32阅读 0赞

参考自 [https://blog.csdn.net/u014597198/article/details/72820737][https_blog.csdn.net_u014597198_article_details_72820737]

效果  
![在这里插入图片描述][20190918152512662.png]

udisk.h

#ifndef UDISK_H
    #define UDISK_H
    #include <QWidget>
    #include <windows.h>
    #include <QAbstractNativeEventFilter>
    #include <dbt.h>
    
    #define GB (1024 * 1024 * 1024)
    #define MB (1024 * 1024)
    #define KB (1024)
    
    
    
    class uDisk : public QWidget, public QAbstractNativeEventFilter
    { 
        Q_OBJECT
    protected:
        bool nativeEventFilter(const QByteArray &eventType, void *message, long *result);
    
    signals:
        void sigUDiskCome(QString uDiskName);
        void sigUDiskRemove();
    private:
        char FirstDriveFromMask(ULONG unitmask);
    public:
         ~uDisk();
        QStringList getDiskInfo(QString driver);
        quint64  getDiskFreeSpace(QString driver);
    
    };
    
    
    #endif // UDISK_H

udisk.cpp

#include <QApplication>
    #include <QDebug>
    #include "udisk.h"
    #include "global_api.h"
    
    
    char uDisk::FirstDriveFromMask( ULONG unitmask )
    { 
        char i;
        for ( i = 0; i < 26; ++i )
        { 
            if ( unitmask & 0x1 )
            { 
                break;
            }
            unitmask = unitmask >> 1;
        }
        return(i + 'A');
    }
    
    uDisk::~uDisk()
    { 
    
    }
    
    
    bool uDisk::nativeEventFilter(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));
                            qDebug() << __FUNCTION__ << __LINE__ << USBDisk;
                            DiskUrl = USBDisk;
                            qDebug() << __FUNCTION__ << __LINE__ << DiskUrl;
                            emit sigUDiskCome(USBDisk);
                        }
                    }
                    break;
    
                case DBT_DEVICEREMOVECOMPLETE:
                    if(lpdb->dbch_devicetype == DBT_DEVTYP_VOLUME)
                    { 
                        PDEV_BROADCAST_VOLUME lpdbv = (PDEV_BROADCAST_VOLUME)lpdb;
                        if(lpdbv->dbcv_flags == 0)
                        { 
                            emit sigUDiskRemove();
                        }
                    }
                    break;
    
                case DBT_DEVNODES_CHANGED:
                    break;
                default:
                    break;
                }
        }
    
        return QWidget::nativeEvent(eventType, message, result);
    }
    
    QStringList uDisk::getDiskInfo( QString driver )
    { 
        QStringList	diskmsg;
        LPCWSTR		lpcwstrDriver = (LPCWSTR) driver.utf16();
    
        ULARGE_INTEGER liFreeBytesAvailable, liTotalBytes, liTotalFreeBytes;
    
        if ( !GetDiskFreeSpaceEx( lpcwstrDriver, &liFreeBytesAvailable, &liTotalBytes, &liTotalFreeBytes ) )
        { 
            qDebug() << "ERROR: Call to GetDiskFreeSpaceEx() failed.";
            diskmsg << "Error";
        }
    
    // qDebug() << "liFreeBytesAvailable" <<(quint64) liFreeBytesAvailable.QuadPart/1024/1024/1024 ;
    // qDebug() << "liTotalBytes" << (quint64) liTotalBytes.QuadPart/1024/1024/1024 ;
    // quint64 liTotal = ((quint64) liTotalBytes.QuadPart/1024/1024/1024);
    // QString liTotal_msg = tr("%1").arg(liTotal);
    // qDebug() << "liTotal_msg" <<liTotal_msg;
    
    // qDebug() << "liTotalFree" << ((quint64) liTotalFreeBytes.QuadPart/1024/1024/1024) ;
    // quint64 liFreeTotal = ((quint64) liTotalFreeBytes.QuadPart/1024/1024/1024);
    // QString liFreeTotal_msg = tr("%1").arg(liFreeTotal);
    // qDebug() << "liFreeTotal_msg" <<liTotal_msg;
    // diskmsg << liTotal_msg << liFreeTotal_msg;
    // qDebug() << diskmsg;
    
    
    
        QString use = QString::number( (double) (liTotalBytes.QuadPart - liTotalFreeBytes.QuadPart) / GB, 'f', 2 );
        use = use + "G";
        QString free = QString::number( (double) liTotalFreeBytes.QuadPart / GB, 'f', 2 );
        free = free + "G";
        QString all = QString::number( (double) liTotalBytes.QuadPart / GB, 'f', 2 );
        all = all + "G";
        int	percent	= 100 - ( (double) liTotalFreeBytes.QuadPart / liTotalBytes.QuadPart) * 100; /* 剩余百分比 */
        QString percent_str	= QString::number( percent, 10 );
    // qDebug() << "--------------------------------------------";
    // myDebug( "use", use );
    // myDebug( "free", free );
    // myDebug( "all", all );
    // myDebug( "percent_str", percent_str );
        diskmsg << use << free << all << percent_str;
        myDebug( "diskmsg", diskmsg );
    
        return(diskmsg);
    }
    
    quint64 uDisk::getDiskFreeSpace(QString driver)
    { 
        LPCWSTR lpcwstrDriver=(LPCWSTR)driver.utf16();
    
        ULARGE_INTEGER liFreeBytesAvailable, liTotalBytes, liTotalFreeBytes;
    
        if( !GetDiskFreeSpaceEx( lpcwstrDriver, &liFreeBytesAvailable, &liTotalBytes, &liTotalFreeBytes) )
        { 
         qDebug() << "ERROR: Call to GetDiskFreeSpaceEx() failed.";
         return 0;
        }
        return (quint64) liTotalFreeBytes.QuadPart/1024/1024/1024;
    }

如何使用：

1、引用头文件\#include “uDisk.h”

2、声明uDisk \*m\_uDisk = Q\_NULLPTR;

3、调用：

m_uDisk = new uDisk;
        qApp->installNativeEventFilter(m_uDisk);
        connect(m_uDisk, &uDisk::sigUDiskCome, [=](QString uDiskName)
        { 
            ui->label->setText(tr("U Disk Come!").append(tr("The Disk is ") + uDiskName));
        });
        connect(m_uDisk, &uDisk::sigUDiskRemove, [=]()
        { 
            ui->label->setText(tr("U Disk Remove!"));
        });

4、在调用类析构时记得析构掉：

if(m_uDisk) 
    {
        delete m_uDisk;
        m_uDisk = Q_NULLPTR;
    }

--------------------

例子

{ 
        ui->setupUi(this);
    
        ui->progressBar->hide();
        ui->label_mem_info->hide();
    
        uDisk *m_uDisk = Q_NULLPTR;
        m_uDisk = new uDisk;
        qApp->installNativeEventFilter(m_uDisk);
        QString current_uDiskName;
        connect(m_uDisk, &uDisk::sigUDiskCome, [=](QString uDiskName)
        { 
            ui->label_sd_info->setText(tr("U盘已插入!").append(tr(" 盘符为：") + uDiskName));
            QStringList DiskInfo =  m_uDisk->getDiskInfo(QString(uDiskName + ":/"));
            myDebug("DiskInfo",DiskInfo);
    
            //显示内存信息
            ui->progressBar->show();
            ui->label_mem_info->show();
    
            //设置进度条的显示
            int progressBarValue = DiskInfo.last().toInt();
            ui->progressBar->setRange(0,100);
            ui->progressBar->setValue(progressBarValue);
    
            //显示存储状态信息
            QString use = DiskInfo.at(0); //use
            QString free = DiskInfo.at(1); //free
            QString total = DiskInfo.at(2); //total
            qDebug() << use << free << total;
            QString disk_info_show = free + "可用, "  + "共" + total + ".";
            qDebug() << disk_info_show;
            ui->label_mem_info->setText(disk_info_show);
    
        });
        connect(m_uDisk, &uDisk::sigUDiskRemove, [=]()
        { 
            ui->label_sd_info->setText(tr("U Disk Remove!"));
            ui->progressBar->hide();
            ui->label_mem_info->hide();
        });
    
    }

[https_blog.csdn.net_u014597198_article_details_72820737]: https://blog.csdn.net/u014597198/article/details/72820737
[20190918152512662.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/18/7dcb2cd3459e43a3a40d9fe1e0f0bf84.png