| | |
| | | p_this->ic->pb = p_this->avio; |
| | | |
| | | int err = av_probe_input_buffer(p_this->ic->pb, &p_this->ic->iformat, nullptr, nullptr, 0, p_this->m_buf_size); |
| | | int err1 = avformat_open_input(&p_this->ic, "", NULL, NULL); |
| | | if(err){ |
| | | ERR("av_probe_input_buffer" << err); |
| | | } |
| | | err = avformat_open_input(&p_this->ic, "", NULL, NULL); |
| | | if(err){ |
| | | ERR("avformat_open_input" << err); |
| | | } |
| | | // int err2 = avformat_find_stream_info(ic, nullptr); |
| | | int err2 = avformat_find_stream_info(p_this->ic, NULL); |
| | | err = avformat_find_stream_info(p_this->ic, NULL); |
| | | if(err){ |
| | | ERR("avformat_find_stream_info" << err); |
| | | } |
| | | int vi = -1; |
| | | for (int i = 0; i < p_this->ic->nb_streams; ++i) { |
| | | if (p_this->ic->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) { |
| | |
| | | delete iter->m_packet.data; |
| | | iter = m_packetsVec.erase(iter); |
| | | while (!(iter->m_packet.flags & AV_PKT_FLAG_KEY)) { |
| | | // INFO("DropFrame: " << iter->m_frameId); |
| | | INFO("DropFrame: " << iter->m_frameId); |
| | | delete iter->m_packet.data; |
| | | iter = m_packetsVec.erase(iter); |
| | | } |
| | |
| | | |
| | | bool BASICGB28181::FFmpegDecoderJPG::SaveVideo(std::string path, int64_t lastFrameId) { |
| | | std::lock_guard<std::mutex> lock(g_mutex); |
| | | INFO("SaveVideo: " << path); |
| | | INFO("SaveVideo: " << path << "m_packetsVec.size : " <<m_packetsVec.size()); |
| | | if (!m_packetsVec.empty()) { |
| | | startWrite(path.c_str()); |
| | | int64_t firstKeyFramePts = m_packetsVec[0].m_packet.pts; |
| | | int64_t firstKeyFrameDts = m_packetsVec[0].m_packet.dts; |
| | | unsigned long int frame_index = 0; |
| | | for (const auto &item:m_packetsVec) { |
| | | if (item.m_frameId < lastFrameId) { |
| | | DBG("item.m_frameId < lastFrameId " << item.m_frameId << " " << lastFrameId); |
| | | conversion(const_cast<AVPacket *> (&item.m_packet), firstKeyFramePts, firstKeyFrameDts, video_st); |
| | | conversion(const_cast<AVPacket *> (&item.m_packet), firstKeyFramePts, firstKeyFrameDts, video_st, frame_index); |
| | | frame_index++; |
| | | av_write_frame(m_pOutFmtCtx, &item.m_packet); |
| | | } else { |
| | | DBG("item.m_frameId > lastFrameId " << item.m_frameId << " " << lastFrameId); |
| | |
| | | int BASICGB28181::FFmpegDecoderJPG::stopWrite() { |
| | | if (m_pOutFmtCtx == nullptr) return -1; |
| | | av_write_trailer(m_pOutFmtCtx); |
| | | avio_close(m_pOutFmtCtx->pb); |
| | | |
| | | avcodec_close(m_pOutFmtCtx->streams[0]->codec); |
| | | av_freep(&m_pOutFmtCtx->streams[0]->codec); |
| | | av_freep(&m_pOutFmtCtx->streams[0]); |
| | | |
| | | avio_close(m_pOutFmtCtx->pb); |
| | | av_free(m_pOutFmtCtx); |
| | | m_pOutFmtCtx = nullptr; |
| | | m_bstartWrite = false; |
| | |
| | | |
| | | |
| | | void BASICGB28181::FFmpegDecoderJPG::conversion(void *packet, const long int &firstKeyPts, const long int &firstKeyDts, |
| | | void *inVideoStream) { |
| | | void *inVideoStream, unsigned long int frame_index) { |
| | | if ((packet != nullptr) && (inVideoStream != nullptr)) { |
| | | AVStream *inStream = (AVStream *) inVideoStream; |
| | | AVPacket *pkg = static_cast<AVPacket *>(packet); |
| | | // static int a = 0; |
| | | // pkg->dts = a++; |
| | | // pkg->pts = a; |
| | | pkg->pts -= firstKeyPts; |
| | | pkg->dts -= firstKeyDts; |
| | | pkg->pts = av_rescale_q_rnd(pkg->pts, inStream->time_base, |
| | | m_pOutVideo_stream->time_base, |
| | | (AVRounding) (AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX)); |
| | | pkg->dts = av_rescale_q_rnd(pkg->dts, inStream->time_base, |
| | | m_pOutVideo_stream->time_base, |
| | | (AVRounding) (AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX)); |
| | | pkg->duration = av_rescale_q(pkg->duration, inStream->time_base, |
| | | m_pOutVideo_stream->time_base); |
| | | pkg->pos = -1; |
| | | // static unsigned long int frame_index = 0 |
| | | // pkg->dts = ++frame_index; |
| | | // pkg->pts = frame_index; |
| | | |
| | | //todo |
| | | // if(pkg->pts==AV_NOPTS_VALUE){ |
| | | DBG("frame_index==%d\n" << frame_index); |
| | | //Write PTS |
| | | AVRational time_base1=inStream->time_base; |
| | | //Duration between 2 frames (us) |
| | | int64_t calc_duration=(double)AV_TIME_BASE/av_q2d(inStream->r_frame_rate); |
| | | //Parameters |
| | | pkg->pts=(double)(frame_index*calc_duration)/(double)(av_q2d(time_base1)*AV_TIME_BASE); |
| | | pkg->dts=pkg->pts; |
| | | pkg->duration=(double)calc_duration/(double)(av_q2d(time_base1)*AV_TIME_BASE); |
| | | // frame_index++; |
| | | // } |
| | | // else{ |
| | | // DBG("pts" << pkg->pts << "firstKeyPts" << firstKeyPts); |
| | | // DBG("dts" << pkg->dts << "firstKeyDts" << firstKeyDts); |
| | | // pkg->pts -= firstKeyPts; |
| | | // pkg->dts -= firstKeyDts; |
| | | // pkg->pts = av_rescale_q_rnd(pkg->pts, inStream->time_base, |
| | | // m_pOutVideo_stream->time_base, |
| | | // (AVRounding) (AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX)); |
| | | // pkg->dts = av_rescale_q_rnd(pkg->dts, inStream->time_base, |
| | | // m_pOutVideo_stream->time_base, |
| | | // (AVRounding) (AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX)); |
| | | // pkg->duration = av_rescale_q(pkg->duration, inStream->time_base, |
| | | // m_pOutVideo_stream->time_base); |
| | | // pkg->pos = -1; |
| | | // } |
| | | |
| | | DBG("pts:" << pkg->pts); |
| | | DBG("dts:" << pkg->dts); |
| | | } |
| | | } |