Qt编写安防视频监控系统44-视频上传

8.jpg

五、核心代码

void frmVideoUpload::addItem(const QString &name, const QString &data)
{
    int row = fileNames.count();
    QString fileName = data + "/" + name;
    fileNames << fileName;

    //复选框,如果勾选了全选则自动选中当前设备
    QCheckBox *itemCk = new QCheckBox(this);
    itemCk->setChecked(ckAll->isChecked());

    //添加文件名称
    QTableWidgetItem *itemFile = new QTableWidgetItem;
    itemFile->setText(name);
    itemFile->setData(Qt::UserRole, QString(data));
    itemFile->setTextAlignment(Qt::AlignCenter);

    //添加文件大小
    QTableWidgetItem *itemSize = new QTableWidgetItem;
    quint64 size = QFile(fileName).size();
    itemSize->setText(TcpFileHelper::getFileSizeMB(size));
    itemSize->setTextAlignment(Qt::AlignCenter);

    //添加单位
    QTableWidgetItem *itemUnit = new QTableWidgetItem;
    itemUnit->setText("MB");
    itemUnit->setTextAlignment(Qt::AlignCenter);

    //增加进度条
    QProgressBar *itemBar = new QProgressBar(this);
    itemBar->setRange(0, size);
    itemBar->setValue(0);

    ui->tableWidget->setCellWidget(row, 0, itemCk);
    ui->tableWidget->setItem(row, 1, itemFile);
    ui->tableWidget->setItem(row, 2, itemSize);
    ui->tableWidget->setItem(row, 3, itemUnit);
    ui->tableWidget->setCellWidget(row, 4, itemBar);
}

void frmVideoUpload::on_btnSelect_clicked()
{
    QDate dateStart = ui->dateStart->date();
    QDate dateEnd = ui->dateEnd->date();
    if (dateStart > dateEnd) {
        QUIHelper::showMessageBoxError("开始时间不能大于结束时间!", 3);
        return;
    }

    //将日期转换为日期时间计算相差的天数,超过最大天数则提示不用继续
    if (dateStart.daysTo(dateEnd) >= 60) {
        QUIHelper::showMessageBoxError("每次最大只能查询60天内!", 3);
        return;
    }

    fileNames.clear();
    ui->tableWidget->clearContents();
    QString filePath;
    if (ui->cboxType->currentText() == "存储视频") {
        filePath = AppData::VideoNormalPath;
    } else {
        filePath = AppData::VideoAlarmPath;
    }

    //获取所有文件夹名称,根据时间查询对应通道对应类型视频
    //如果开始时间小于或者等于结束时间,则将开始时间对应文件夹下的视频文件添加到列表
    //然后将开始时间加一天,直到大于结束时间
    while (dateStart <= dateEnd) {
        //生成对应日期的文件夹
        QString savePath = QString("%1/%2").arg(filePath).arg(dateStart.toString("yyyy-MM-dd"));
        QDir saveDir(savePath);
        //判断文件夹是否存在
        if (saveDir.exists()) {
            //指定文件拓展名过滤,按照时间升序排序
            QStringList filter;
            filter << "*.mp4" << "*.h264";
            QStringList names = saveDir.entryList(filter, QDir::NoFilter, QDir::Time | QDir::Reversed);

            foreach (QString name, names) {
                //如果是选择的所有通道,则不过滤视频文件
                if (ui->cboxCh->currentText() == "所有通道") {
                    addItem(name, savePath);
                } else {
                    //对应通道的视频文件添加进来
                    QString videoCh = name.split("_").first();
                    QString chName = QString("Ch%1").arg(ui->cboxCh->currentIndex());
                    if (videoCh == chName) {
                        addItem(name, savePath);
                    }
                }
            }
        }

        dateStart = dateStart.addDays(1);
    }
}

void frmVideoUpload::on_btnUpload_clicked()
{
    //过滤选中的文件
    int fileCount = fileNames.count();
    QList<QString> files;
    QList<QProgressBar *> bars;
    for (int row = 0; row < fileCount; row++) {
        QString fileName = fileNames.at(row);
        QCheckBox *itemCk = (QCheckBox *)ui->tableWidget->cellWidget(row, 0);
        QProgressBar *bar = (QProgressBar *)ui->tableWidget->cellWidget(row, 4);
        bar->setValue(0);
        if (itemCk && itemCk->isChecked()) {
            files << fileName;
            bars << bar;
        }
    }

    int count = files.count();
    if (count == 0) {
        QUIHelper::showMessageBoxError("没有选中设备!", 3);
        return;
    }

    //每个上传都一个线程
    for (int i = 0; i < count; i++) {
        TcpSendFileThread *sendFileThread = new TcpSendFileThread(this);
        sendFileThread->setFileName(fileNames.at(i));
        //网络不好可以自行调整每次发的包的大小
        sendFileThread->setMaxPackageSize(0xFFFF);
        sendFileThread->setReceiveIP(AppConfig::UploadHost);
        sendFileThread->setReceivePort(AppConfig::UploadPort);
        sendFileThread->setProgressBar(bars.at(i));
        sendFileThread->start();
    }
}