multimedia/client/qt_gl_/yuvgl/mainwindow.cpp

#include "mainwindow.h"
#include "ui_mainwindow.h"

MainWindow::MainWindow(QWidget *parent) :
    QssMainWindow(parent),
    ui(new Ui::MainWindow),
    m_bCameraOpen(false),
    mCamera(nullptr),
    m_bRtmpPushing(false),
    mPlayerWidget(nullptr),
    mVideoCoder(nullptr),
    mPusher(nullptr)
{
    ui->setupUi(this);
    std::vector<std::wstring>  cameras = Camera::EnumAllCamera();
    for(std::wstring x : cameras){
        ui->comboBox->addItem(QString::fromWCharArray(x.c_str(),x.size()),
                              QString::fromWCharArray(x.c_str(),x.size()));
    }
    mPusher = new H264RtmpPuser();
}

MainWindow::~MainWindow(){
    delete ui;
}

void MainWindow::on_pushButton_clicked(){
    if(nullptr == mPlayerWidget){
        mPlayerWidget = new CPlayWidget(nullptr);
    }
    if(!m_bCameraOpen){
        mPlayerWidget->SetDataType(CPlayWidget::IMG_TYPE::TYPE_RGB32);
        mPlayerWidget->SetImgSize(640,480);

        qDebug()<<ui->comboBox->currentText().size()<<ui->comboBox->currentText();
        wchar_t *opencamera = new wchar_t[ui->comboBox->currentText().size()];
        ui->comboBox->currentText().toWCharArray(opencamera);
        wstring ss = wstring(opencamera,ui->comboBox->currentText().size());
        if(nullptr == mCamera){
            this->mCamera = new Camera(ss);
        }
        this->mCamera->SetObserver(mPlayerWidget);
        qDebug()<<ui->comboBox->currentText();
        ui->pushButton->setText("关闭摄像头");
        m_bCameraOpen = true;
        mPlayerWidget->show();
        ui->verticalLayout->addWidget(mPlayerWidget);
        ui->verticalLayout->setStretch(0,1);
        ui->verticalLayout->setStretch(1,0);
        ui->verticalLayout->setStretch(2,9);

    } else{
        m_bCameraOpen = false;
        ui->pushButton->setText("打开摄像头");
    }
}

void MainWindow::on_pushButton_2_clicked()
{
    if(!m_bRtmpPushing){
        if(!m_bCameraOpen){
            ToastWidget::showTip("请打开摄像头",this);
            return;
        }else{
            //
            if(nullptr == mVideoCoder){
                mVideoCoder = new VideoCoder(mCamera->GetWidth(),
                                             mCamera->GetHeight(),
                                             GUIDToAvFormat(mCamera->MediaType()));
            }
            mCamera->SetObserver(mVideoCoder);

            // todo 根据返回结果判断是否推流
                qDebug()<<"连接RTMP服务器"<<ui->lineEdit->text();
                if (!mPusher->IfConnect()) {
                    const char* address = ui->lineEdit->text().toLocal8Bit().data();
                    if (0 == mPusher->RTMP264_Connect("rtmp://127.0.0.1:1935/live/1")) {
                        ToastWidget::showTip("已经连接上RTMP服务器",this);
                        mVideoCoder->SetOberver(mPusher);
                        mPusher->StartPush();
                        ui->pushButton_2->setText("关闭推流");
                        /*
                        if (nullptr != this->mAudioCoder) {
                            this->mAudioCoder->SetObserver(mPusher);
                            //音频流先不推流
                        }*/
                    }
                    else {
                        ToastWidget::showTip("连接RTMP服务器失败，请检查服务器地址",this);
                    }
                }else{
                    ToastWidget::showTip("正在推流，请先关闭",this);

                }
        }
    }
}