rk35xx-android12/external/rkaiq_tool_server/rtspserver/RtspServer.cpp

//#define LOG_NDEBUG 0
#define LOG_TAG "RtspServer"
#include <utils/Log.h>

#include <stdlib.h>
#include <stdio.h>

#include "H264LiveVideoServerMediaSubsession.h"
#include "RtspServer.h"

RtspServer::RtspServer() : mEnv(NULL), mSms(NULL), mRtspServer(NULL), mProcThread(NULL), mCamDev(NULL), mEncoder(NULL)
{
    ALOGD("RtspServer enter");
}

RtspServer::~RtspServer()
{
    ALOGD("~RtspServer enter");
    if (mProcThread != NULL) {
        delete mProcThread;
        mProcThread = NULL;
    }

    if (mCamDev != NULL) {
        mCamDev->deinit();
        delete mCamDev;
        mCamDev = NULL;
    }
    if (mEncoder != NULL) {
        delete mEncoder;
        mEncoder = NULL;
    }
}

bool RtspServer::init(MetaInfo* meta)
{
    if (meta == NULL) {
        ALOGE("Failed to get metaData");
        return false;
    }

    if (mEnv == NULL) {
        // begin by setting up our usage environment.
        TaskScheduler* scheduler = BasicTaskScheduler::createNew();
        mEnv = BasicUsageEnvironment::createNew(*scheduler);

        OutPacketBuffer::maxSize = 1920 * 1080 * 2;
        mRtspServer = RTSPServer::createNew(*mEnv, meta->port_num);
        if (mRtspServer == NULL) {
            ALOGE("Failed to create RTSP server: %s", mEnv->getResultMsg());
            return false;
        }
        strcpy(mStreamName, meta->stream_name);
    }

    return initOther(meta);
}

bool RtspServer::initOther(MetaInfo* meta)
{
    mCamDev = new CamCaptureHelper();
    // start CamCaptureHelper.
    CamCaptureHelper::MetaInfo camInfo;

    strcat(camInfo.video_dev, meta->dev_name);
    camInfo.width = meta->width;
    camInfo.height = meta->height;
    camInfo.format = 0; // nv12

    if (!mCamDev->init(&camInfo)) {
        ALOGE("Failed to init mCamDev");
        return false;
    }

    // start RKHWEncApi.
    mEncoder = new RKHWEncApi();
    RKHWEncApi::EncCfgInfo_t encInfo;
    encInfo.width = meta->width;
    encInfo.height = meta->height;
    encInfo.format = ENC_INPUT_YUV420_SEMIPLANAR;
    encInfo.framerate = 30;    // 30fps
    encInfo.bitRate = 8000000; // 500k default
    encInfo.IDRInterval = 1;
    encInfo.rc_mode = 1;
    encInfo.qp = 20;

    if (!mEncoder->init(&encInfo)) {
        ALOGE("Failed to init mEncoder");
        return false;
    }

    return true;
}

void RtspServer::onDoGetNextFrame(QMediaBuffer* outBuf)
{
    bool ret = true;
    QMediaBuffer camBuf;

    ALOGI("onDoGetNextFrame");

    if (mCamDev->getCameraBuffer(&camBuf)) {
        ret = mEncoder->sendFrame(camBuf.getFd(), camBuf.getSize(), 0, 0);
        if (!ret) {
            ALOGD("Failed to sendFrame");
            mCamDev->putCameraBuffer(&camBuf);
            return;
        }

        EncoderOut_t encOut;
        ret = mEncoder->getOutStream(&encOut);
        if (ret) {
            // fill live source data
            outBuf->setData(encOut.data, encOut.size);
        } else {
            ALOGD("Failed to getOutStream");
        }

        mCamDev->putCameraBuffer(&camBuf);
    } else {
        ALOGD("Failed to get camera buffer");
    }
}

bool RtspServer::start()
{
    ALOGD("Start streaming...");
    resetDoneFlag();
    mProcThread = new std::thread(&RtspServer::doEventLoop, this);

    mSms = ServerMediaSession::createNew(*mEnv, mStreamName, "test", "Session streamed by \"cameraPushTest\"");
    mSms->addSubsession(H264LiveVideoServerMediaSubsession::createNew(*mEnv, True, this));
    mRtspServer->addServerMediaSession(mSms);

    char* url = mRtspServer->rtspURL(mSms);
    ALOGD("Play this stream using the URL: %s", url);
    delete[] url;

    return true;
}

bool RtspServer::stop()
{
    ALOGD("Stop streaming... Exit scheduler eventLoop");
    setDoneFlag();

    ALOGD("Exit procThread");
    if (mProcThread != NULL) {
        mProcThread->join();
        delete mProcThread;
        mProcThread = NULL;
    }

    if (mRtspServer != NULL) {
        mRtspServer->deleteServerMediaSession(mStreamName);
    }

    if (mCamDev != NULL) {
        mCamDev->deinit();
        delete mCamDev;
        mCamDev = NULL;
    }
    if (mEncoder != NULL) {
        delete mEncoder;
        mEncoder = NULL;
    }
    return true;
}

void RtspServer::doEventLoop()
{
    ALOGD("Env doEventLoop");
    mEnv->taskScheduler().doEventLoop(&fDoneFlag);
    ALOGD("Env doEventLoop exit");
}

static RtspServer _g_rtsp_server;

int init_rtsp(const char* video_dev, int width, int height)
{
    RtspServer::MetaInfo info;
    info.width = width;
    info.height = height;
    info.port_num = 1234;
    strcat(info.dev_name, video_dev);
    strcat(info.stream_name, "v");

    ALOGD("init_rtsp: video_dev %s w %d h %d", video_dev, width, height);

    if (!_g_rtsp_server.init(&info)) {
        ALOGE("Failed to init _g_rtsp_server");
        return -1;
    }
    _g_rtsp_server.start();

    return 0;
}

void deinit_rtsp()
{
    ALOGD("deinit_rtsp enter");
    _g_rtsp_server.stop();
}