123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138 |
- /* Copyright (C) 2021 Vladimir Sukhonosov <xornet@xornet.org>
- * Copyright (C) 2021 Martins Mozeiko <martins.mozeiko@gmail.com>
- * All Rights Reserved.
- *
- * This is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This software is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this software; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
- * USA.
- */
-
- #define MAX_H264_INSTANCES 64
-
- #include <deque>
-
- #include <rdr/MemInStream.h>
- #include <rdr/InStream.h>
- #include <rdr/OutStream.h>
- #include <rfb/LogWriter.h>
- #include <rfb/Exception.h>
- #include <rfb/H264Decoder.h>
- #include <rfb/H264DecoderContext.h>
-
- using namespace rfb;
-
- static LogWriter vlog("H264Decoder");
-
- enum rectFlags {
- resetContext = 0x1,
- resetAllContexts = 0x2,
- };
-
- H264Decoder::H264Decoder() : Decoder(DecoderOrdered)
- {
- }
-
- H264Decoder::~H264Decoder()
- {
- resetContexts();
- }
-
- void H264Decoder::resetContexts()
- {
- os::AutoMutex lock(&mutex);
- for (std::deque<H264DecoderContext*>::iterator it = contexts.begin(); it != contexts.end(); it++)
- delete *it;
- contexts.clear();
- }
-
- H264DecoderContext* H264Decoder::findContext(const Rect& r)
- {
- os::AutoMutex m(&mutex);
- for (std::deque<H264DecoderContext*>::iterator it = contexts.begin(); it != contexts.end(); it++)
- if ((*it)->isEqualRect(r))
- return *it;
- return NULL;
- }
-
- bool H264Decoder::readRect(const Rect& r, rdr::InStream* is,
- const ServerParams& server, rdr::OutStream* os)
- {
- rdr::U32 len;
-
- if (!is->hasData(8))
- return false;
-
- is->setRestorePoint();
-
- len = is->readU32();
- os->writeU32(len);
- rdr::U32 flags = is->readU32();
-
- os->writeU32(flags);
-
- if (!is->hasDataOrRestore(len))
- return false;
-
- is->clearRestorePoint();
-
- os->copyBytes(is, len);
-
- return true;
- }
-
- void H264Decoder::decodeRect(const Rect& r, const void* buffer,
- size_t buflen, const ServerParams& server,
- ModifiablePixelBuffer* pb)
- {
- rdr::MemInStream is(buffer, buflen);
- rdr::U32 len = is.readU32();
- rdr::U32 flags = is.readU32();
-
- H264DecoderContext* ctx = NULL;
- if (flags & resetAllContexts)
- {
- resetContexts();
- if (!len)
- return;
- flags &= ~(resetContext | resetAllContexts);
- } else {
- ctx = findContext(r);
- }
-
- if (!ctx)
- {
- os::AutoMutex lock(&mutex);
- if (contexts.size() >= MAX_H264_INSTANCES)
- {
- auto excecc_ctx = contexts.front();
- delete excecc_ctx;
- contexts.pop_front();
- }
- ctx = H264DecoderContext::createContext(r);
- if (!ctx)
- throw Exception("H264Decoder: Context not be created");
- contexts.push_back(ctx);
- }
-
- if (!ctx->isReady())
- throw Exception("H264Decoder: Context is not ready");
-
- if (flags & resetContext)
- ctx->reset();
-
- if (!len)
- return;
-
- ctx->decode(is.getptr(len), len, flags, pb);
- }
|