基于C++(QT框架)设计的网络摄像头项目(支持跨平台运行)

Widget::Widget(QWidget *parent)
    : QWidget(parent)
    , ui(new Ui::Widget)
{
    ui->setupUi(this);

    //自定义标题栏
    FramelessHelper *pHelper = new FramelessHelper(this);
    setWindowFlags(Qt::FramelessWindowHint);
    pHelper->activateOn(this);  //激活当前窗体
    //pHelper->setTitleHeight(m_pTitleBar->height());  //设置窗体的标题栏高度
    pHelper->setWidgetMovable(true);  //设置窗体可移动
    pHelper->setWidgetResizable(true);  //设置窗体可缩放
   // pHelper->setRubberBandOnMove(true);  //设置橡皮筋效果-可移动
   // pHelper->setRubberBandOnResize(true);  //设置橡皮筋效果-可缩放

    /*基本设置*/
    this->setWindowIcon(QIcon(":/log.ico")); //设置图标
    this->setWindowTitle("网络摄像头");

    //打开的窗口在屏幕中间
    QDesktopWidget *widget= QApplication::desktop();
    move((widget->width()-this->width())/2,(widget->height()-this->height())/2);

    //刷新摄像头与IP地址信息
    on_pushButton_config_clicked();

    //刷新在线人数
    timer = new QTimer(this);
    connect(timer, SIGNAL(timeout()), this, SLOT(timer_update()));
    timer->start(1000);

    tray= new QSystemTrayIcon(this);//初始化托盘对象tray
    connect(tray,SIGNAL(activated(QSystemTrayIcon::ActivationReason)),this,SLOT(show_Widget(QSystemTrayIcon::ActivationReason)));

    tray->setIcon(QIcon(QPixmap(":/log.png")));//设定托盘图标，引号内是自定义的png图片路径
    tray->setToolTip("网络摄像头"); //提示文字
    restoreAction = new QAction("打开", this);
    connect(restoreAction, SIGNAL(triggered()), this, SLOT(show()));
    quitAction = new QAction("退出", this);
    connect(quitAction, SIGNAL(triggered()), qApp, SLOT(quit()));

    trayMenu = new QMenu(this);
    trayMenu->addAction(restoreAction);
    trayMenu->addSeparator();
    trayMenu->addAction(quitAction);
    tray->setContextMenu(trayMenu);

    work_class=new VideoReadThread_0;
    work_thread=new QThread;

    //启动摄像头的信号
    connect(this,SIGNAL(StartWorkThread()),work_class,SLOT(run()));
    //释放资源-释放摄像头
    connect(this,SIGNAL(Stop_VideoAudioEncode_0()),work_class,SLOT(stop()));

    //连接摄像头采集信号,在主线程实时显示视频画面
    connect(work_class,SIGNAL(VideoDataOutput(QImage)),this,SLOT(VideoDataDisplay_0(QImage)));

    //将类移动到子线程工作
    work_class->moveToThread(work_thread);
}

void VideoReadThread_0::Camear_Init()
{
    /*创建摄像头对象，根据选择的摄像头打开*/
    camera = new QCamera(videoaudioencode_0.camera);
    m_pProbe = new QVideoProbe;
    if(m_pProbe != nullptr)
    {
        m_pProbe->setSource(camera); // Returns true, hopefully.
        connect(m_pProbe, SIGNAL(videoFrameProbed(QVideoFrame)),this, SLOT(slotOnProbeFrame(QVideoFrame)), Qt::QueuedConnection);
    }
    else
    {
           qDebug()<<"m_pProbe == nullptr";
    }

    /*配置摄像头捕    QCamera *camera;
    QVideoProbe *m_pProbe;获模式为帧捕获模式*/
     camera->setCaptureMode(QCamera::CaptureVideo);  //如果在Linux系统下运行就这样设置
     //camera->setCaptureMode(QCamera::CaptureVideo);//如果在android系统下运行就这样设置


    /*设置摄像头的采集帧率和分辨率*/
    QCameraViewfinderSettings settings;
    settings.setPixelFormat(QVideoFrame::Format_YUYV); //设置像素格式  Android上只支持NV21格式
    settings.setResolution(QSize(VIDEO_WIDTH,VIDEO_HEIGHT)); //设置摄像头的分辨率
    camera->setViewfinderSettings(settings);


    /*启动摄像头*/
    camera->start();

}

void VideoReadThread_0::slotOnProbeFrame(const QVideoFrame &frame)
{
//    qDebug()<<"开始采集.";
   QVideoFrame cloneFrame(frame);
   cloneFrame.map(QAbstractVideoBuffer::ReadOnly);

//    qDebug()<<"height:"<<cloneFrame.height();
//    qDebug()<<"width:"<<cloneFrame.width();
//    qDebug()<<"bytesPerLine:"<<cloneFrame.bytesPerLine();
//    qDebug()<<"mappedBytes:"<<cloneFrame.mappedBytes();
//    qDebug()<<"pixelFormat:"<<cloneFrame.pixelFormat();

   unsigned char rgb_buffer[VIDEO_WIDTH*VIDEO_HEIGHT*3];
   if(cloneFrame.pixelFormat()==QVideoFrame::Format_YUYV)
   {
       yuyv_to_rgb(cloneFrame.bits(),rgb_buffer,cloneFrame.width(),cloneFrame.height());
   }
   else if(cloneFrame.pixelFormat()==QVideoFrame::Format_NV21)
   {
        NV21_TO_RGB24(cloneFrame.bits(),rgb_buffer,cloneFrame.width(),cloneFrame.height());
   }
   else
   {
       qDebug()<<"当前格式编码为%1,暂时不支持转换.\n";
       return;
   }
    cloneFrame.unmap();

    //加载图片数据
    QImage image(rgb_buffer,
                       cloneFrame.width(),
                       cloneFrame.height(),
                       QImage::Format_RGB888);

    //绘制图片水印
    QDateTime dateTime(QDateTime::currentDateTime());
    //时间效果: 2020-03-05 16:25::04 周一
    QString qStr="";
    qStr+=dateTime.toString("yyyy-MM-dd hh:mm:ss ddd");
    QPainter pp(&image);
    QPen pen = QPen(Qt::white);
    pp.setPen(pen);
    pp.setFont(QFont("宋体",20));
    pp.drawText(QPointF(0,40),qStr);

    mutex.lock();
    QBuffer buff;
    image.save(&buff,"jpg",80); // 30表示压宿率，值从0 – 100, 值越小表示编码出来的图像文件就越小，当然也就越不清晰
    lcd_image_data=buff.data();
    cond_wait.wakeAll();
    mutex.unlock();

    emit VideoDataOutput(image); //发送信号
}

/*
函数功能: 将YUV数据转为RGB格式
函数参数:
unsigned char *yuv_buffer: YUV源数据
unsigned char *rgb_buffer: 转换之后的RGB数据
int iWidth,int iHeight   : 图像的宽度和高度
*/
void yuyv_to_rgb(unsigned char *yuv_buffer,unsigned char *rgb_buffer,int iWidth,int iHeight)
{
    int x;
    int z=0;
    unsigned char *ptr = rgb_buffer;
    unsigned char *yuyv= yuv_buffer;
    for (x = 0; x < iWidth*iHeight; x++)
    {
        int r, g, b;
        int y, u, v;

        if (!z)
        y = yuyv[0] << 8;
        else
        y = yuyv[2] << 8;
        u = yuyv[1] - 128;
        v = yuyv[3] - 128;

        r = (y + (359 * v)) >> 8;
        g = (y - (88 * u) - (183 * v)) >> 8;
        b = (y + (454 * u)) >> 8;

        *(ptr++) = (r > 255) ? 255 : ((r < 0) ? 0 : r);
        *(ptr++) = (g > 255) ? 255 : ((g < 0) ? 0 : g);
        *(ptr++) = (b > 255) ? 255 : ((b < 0) ? 0 : b);

        if(z++)
        {
            z = 0;
            yuyv += 4;
        }
    }
}

void NV21_TO_RGB24(unsigned char *yuyv, unsigned char *rgb, int width, int height)
{
    const int nv_start = width * height ;
    int  index = 0, rgb_index = 0;
    uint8_t y, u, v;
    int r, g, b, nv_index = 0,i, j;

    for(i = 0; i < height; i++){
        for(j = 0; j < width; j ++){
            //nv_index = (rgb_index / 2 - width / 2 * ((i + 1) / 2)) * 2;
            nv_index = i / 2  * width + j - j % 2;

            y = yuyv[rgb_index];
            u = yuyv[nv_start + nv_index ];
            v = yuyv[nv_start + nv_index + 1];

            r = y + (140 * (v-128))/100;  //r
            g = y - (34 * (u-128))/100 - (71 * (v-128))/100; //g
            b = y + (177 * (u-128))/100; //b

            if(r > 255)   r = 255;
            if(g > 255)   g = 255;
            if(b > 255)   b = 255;
            if(r < 0)     r = 0;
            if(g < 0)     g = 0;
            if(b < 0)     b = 0;

            index = rgb_index % width + (height - i - 1) * width;
            //rgb[index * 3+0] = b;
            //rgb[index * 3+1] = g;
            //rgb[index * 3+2] = r;

            //颠倒图像
            //rgb[height * width * 3 - i * width * 3 - 3 * j - 1] = b;
            //rgb[height * width * 3 - i * width * 3 - 3 * j - 2] = g;
            //rgb[height * width * 3 - i * width * 3 - 3 * j - 3] = r;

            //正面图像
            rgb[i * width * 3 + 3 * j + 0] = b;
            rgb[i * width * 3 + 3 * j + 1] = g;
            rgb[i * width * 3 + 3 * j + 2] = r;

            rgb_index++;
        }
    }
}

void TcpServerThread::run()
{
    qDebug()<<"TcpServerThread_子线程run槽函数的ID:"<<QThread::currentThreadId();
   QTcpSocket *tcpSocket;
    tcpSocket=new QTcpSocket;
    /*使用socketDescriptor套接字初始化QAbstractSocket*/
    if(!tcpSocket->setSocketDescriptor(socketDescriptor))
    {
        return;
    }

    //读取接收的数据
    QString text;
    if(tcpSocket->waitForReadyRead())
    {
        text=tcpSocket->readAll();
    }

    //处理浏览器的请求
    if(text.contains("GET / HTTP/1.1", Qt::CaseInsensitive))
    {
        //如果是Android系统
#ifdef Q_OS_ANDROID
        text="assets:/login.html";
#else
         text=QCoreApplication::applicationDirPath();
         text+="/www/login.html";
#endif
        SendFileData(buff,text,"text/html",tcpSocket);
    }
    else if(text.contains("GET /image.jpg HTTP/1.1", Qt::CaseInsensitive))
    {
        SendImageData(buff,tcpSocket);
    }
    else if(text.contains("GET /favicon.ico HTTP/1.1", Qt::CaseInsensitive))
    {

#ifdef Q_OS_ANDROID
        text="assets:/logo.ico";
#else
        text=QCoreApplication::applicationDirPath();
        text+="/www/logo.ico";
#endif
        SendFileData(buff,text,"image/x-icon",tcpSocket);
    }
    else if(text.contains("GET /three.min.js", Qt::CaseInsensitive))
    {

#ifdef Q_OS_ANDROID
        text="assets:/three.min.js";
#else
        text=QCoreApplication::applicationDirPath();
        text+="/www/three.min.js";
#endif
        SendFileData(buff,text,"application/x-javascript",tcpSocket);
    }
    else if(text.contains("GET /style.css", Qt::CaseInsensitive))
    {

#ifdef Q_OS_ANDROID
        text="assets:/style.css";
#else
        text=QCoreApplication::applicationDirPath();
        text+="/www/style.css";
#endif
        SendFileData(buff,text,"text/css",tcpSocket);
    }
    else if(text.contains("GET /Stats.min.js", Qt::CaseInsensitive))
    {

#ifdef Q_OS_ANDROID
        text="assets:/Stats.min.js";
#else
        text=QCoreApplication::applicationDirPath();
        text+="/www/Stats.min.js";
#endif
        SendFileData(buff,text,"application/x-javascript",tcpSocket);
    }
    else if(text.contains("GET /logo.png HTTP/1.1", Qt::CaseInsensitive))
    {

#ifdef Q_OS_ANDROID
        text="assets:/logo.png";
#else
        text=QCoreApplication::applicationDirPath();
        text+="/www/logo.png";
#endif
        SendFileData(buff,text,"image/png",tcpSocket);
    }
    else if(text.contains("userName=admin&passWord=12345678", Qt::CaseInsensitive))
    {   

#ifdef Q_OS_ANDROID
        text="assets:/index.html";
#else
        text=QCoreApplication::applicationDirPath();
        text+="/www/index.html";
#endif
        SendFileData(buff,text,"text/html",tcpSocket);
    }

    //卸载连接的套接字
    fd_list.Del_fd(socketDescriptor);
    //关闭连接
    tcpSocket->close();
    delete  tcpSocket;
}

/**
 * 向浏览器响应请求数据
 */
int TcpServerThread::SendFileData(char *buff, QString file_path, const char *type, QTcpSocket *tcpSocket)
{
    qint64 size=0;
    /*1. 读取文件*/
    QFile file(file_path);
    if(file.open(QIODevice::ReadOnly)!=true)
    {
        qDebug()<<"文件不存在:"<<file_path<<endl;
        return -1;
    }

    size=file.size(); //得到文件大小
    QByteArray byte=file.readAll(); //读取所有数据
    file.close();//关闭文件
    /*2. 构建响应格式字符串*/
    sprintf(buff,"HTTP/1.1 200 OK\r\n"
                 "Content-type:%s\r\n"
                 "Content-Length:%lld\r\n"
                 "\r\n",type,size);
    /*3. 向客户端发送响应请求*/
    if(tcpSocket->write(buff,strlen(buff))<=0)return -2;
    tcpSocket->waitForBytesWritten();  //等待写
    /*4. 向客户端发送响应实体数据*/
    if(tcpSocket->write(byte)<=0)return -3;
    tcpSocket->waitForBytesWritten();  //等待写
    return 0;
}

/*
向客户端循环发送图片数据流
*/
int TcpServerThread::SendImageData(char *buff, QTcpSocket *tcpSocket)
{
    /*1. 构建响应格式字符串*/
    sprintf(buff,"HTTP/1.0 200 OK\r\n"
                "Server: wbyq\r\n"
                "Content-Type:multipart/x-mixed-replace;boundary=boundarydonotcross\r\n"
                "\r\n"
                "--boundarydonotcross\r\n");
    /*2. 向客户端发送响应请求*/
    if(tcpSocket->write(buff,strlen(buff))<=0)return -2;
    tcpSocket->waitForBytesWritten();  //等待写
    /*3. 循环发送数据*/
    while(thread_run_flag)
    {
        //从全局缓冲区取数据
        mutex.lock();
        cond_wait.wait(&mutex);
        QByteArray image_data; //保存一帧图像数据
        image_data=lcd_image_data;
        mutex.unlock();

//        QBuffer data_buff;
//        QPixmap pixmap;
//        QScreen *screen = QGuiApplication::primaryScreen();
//        pixmap=screen->grabWindow(0); //获取当前屏幕的图像
//        //缩放图像
//        pixmap = pixmap.scaled(lcd_image_w,lcd_image_h, Qt::KeepAspectRatio);
//        pixmap.save(&data_buff,"jpg",image_val);

//        QByteArray image_data1=data_buff.data();


        /*4. 向浏览器发送响应头*/
        sprintf(buff,"Content-type:image/jpeg\r\n"
                     "Content-Length:%d\r\n"
                     "\r\n",image_data.size());
        if(tcpSocket->write(buff,strlen(buff))<=0)break;
        tcpSocket->waitForBytesWritten();  //等待写

        /*5. 发送实体数据*/
         if(tcpSocket->write(image_data)<=0)break;
         tcpSocket->waitForBytesWritten();  //等待写

        /*6. 发送边界符*/
        sprintf(buff,"\r\n--boundarydonotcross\r\n");
        if(tcpSocket->write(buff,strlen(buff))<=0)break;
        tcpSocket->waitForBytesWritten();  //等待写

        //等待一段时间
        msleep(5);
    }
    return 0;
}