Skip to content

Commit

Permalink
Fixing include
Browse files Browse the repository at this point in the history
  • Loading branch information
ross-newman committed May 10, 2024
1 parent 52de031 commit 3056516
Show file tree
Hide file tree
Showing 4 changed files with 30 additions and 30 deletions.
28 changes: 14 additions & 14 deletions src/h264/gst/vaapi/rtp_h264_depayloader.cc
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@

namespace mediax::rtp::h264::gst::vaapi {

RtpH264GstOpenDepayloader::RtpH264GstOpenDepayloader() {
RtpH264GstVaapiDepayloader::RtpH264GstVaapiDepayloader() {
// Set this for empty video buffers
if (rgb_) {
SetColourSpace(mediax::rtp::ColourspaceType::kColourspaceRgb24);
Expand All @@ -42,14 +42,14 @@ RtpH264GstOpenDepayloader::RtpH264GstOpenDepayloader() {
}
}

RtpH264GstOpenDepayloader::~RtpH264GstOpenDepayloader() = default;
RtpH264GstVaapiDepayloader::~RtpH264GstVaapiDepayloader() = default;

RtpH264GstOpenDepayloader &RtpH264GstOpenDepayloader::operator=(const RtpH264GstOpenDepayloader &other
RtpH264GstVaapiDepayloader &RtpH264GstVaapiDepayloader::operator=(const RtpH264GstVaapiDepayloader &other
[[maybe_unused]]) {
return *this;
}

void RtpH264GstOpenDepayloader::SetStreamInfo(const ::mediax::rtp::StreamInformation &stream_information) {
void RtpH264GstVaapiDepayloader::SetStreamInfo(const ::mediax::rtp::StreamInformation &stream_information) {
::mediax::rtp::RtpPortType &stream = GetStream();
stream.encoding = ::mediax::rtp::ColourspaceType::kColourspaceNv12;
stream.height = stream_information.height;
Expand All @@ -62,10 +62,10 @@ void RtpH264GstOpenDepayloader::SetStreamInfo(const ::mediax::rtp::StreamInforma
GetBuffer().resize(stream.width * stream.height * mediax::BitsPerPixel(stream.encoding) / 8);
}

GstFlowReturn RtpH264GstOpenDepayloader::NewFrameCallback(GstAppSink *appsink, gpointer user_data) {
GstFlowReturn RtpH264GstVaapiDepayloader::NewFrameCallback(GstAppSink *appsink, gpointer user_data) {
gint width = 0;
gint height = 0;
auto depayloader = static_cast<RtpH264GstOpenDepayloader *>(user_data);
auto depayloader = static_cast<RtpH264GstVaapiDepayloader *>(user_data);

// Pull the sample from the appsink
GstSample *sample = gst_app_sink_pull_sample(appsink);
Expand Down Expand Up @@ -122,7 +122,7 @@ GstFlowReturn RtpH264GstOpenDepayloader::NewFrameCallback(GstAppSink *appsink, g
return GST_FLOW_OK;
}

bool RtpH264GstOpenDepayloader::Open() {
bool RtpH264GstVaapiDepayloader::Open() {
// Check that the stream info has been set
if (!GetStream().settings_valid) {
return false;
Expand Down Expand Up @@ -192,7 +192,7 @@ bool RtpH264GstOpenDepayloader::Open() {
}

// Set the callback function for the appsink
GstAppSinkCallbacks callbacks = {.new_sample = RtpH264GstOpenDepayloader::NewFrameCallback};
GstAppSinkCallbacks callbacks = {.new_sample = RtpH264GstVaapiDepayloader::NewFrameCallback};
gst_app_sink_set_callbacks(GST_APP_SINK(appsink), &callbacks, this, nullptr);

if (rgb_) {
Expand Down Expand Up @@ -223,7 +223,7 @@ bool RtpH264GstOpenDepayloader::Open() {
return true;
}

void RtpH264GstOpenDepayloader::Start() {
void RtpH264GstVaapiDepayloader::Start() {
// Return if not open
if (GetState() == ::mediax::rtp::StreamState::kClosed) {
std::cerr << "Stream not open so cant be started"
Expand All @@ -240,7 +240,7 @@ void RtpH264GstOpenDepayloader::Start() {
gst_element_set_state(pipeline_, GST_STATE_PLAYING);
}

void RtpH264GstOpenDepayloader::Stop() {
void RtpH264GstVaapiDepayloader::Stop() {
if (GetState() != ::mediax::rtp::StreamState::kStarted) {
return;
}
Expand All @@ -251,7 +251,7 @@ void RtpH264GstOpenDepayloader::Stop() {
new_rx_frame_ = false;
}

void RtpH264GstOpenDepayloader::Close() {
void RtpH264GstVaapiDepayloader::Close() {
if (GetState() != ::mediax::rtp::StreamState::kStopped) {
Stop();
}
Expand All @@ -273,7 +273,7 @@ void RtpH264GstOpenDepayloader::Close() {
gst_object_unref(pipeline_);
}

bool RtpH264GstOpenDepayloader::Receive(mediax::rtp::RtpFrameData *data, int32_t timeout) {
bool RtpH264GstVaapiDepayloader::Receive(mediax::rtp::RtpFrameData *data, int32_t timeout) {
auto start_time = std::chrono::high_resolution_clock::now();

data->cpu_buffer = GetBuffer().data();
Expand Down Expand Up @@ -301,13 +301,13 @@ bool RtpH264GstOpenDepayloader::Receive(mediax::rtp::RtpFrameData *data, int32_t
return true;
}

void RtpH264GstOpenDepayloader::Callback(::mediax::rtp::RtpFrameData frame) const {
void RtpH264GstVaapiDepayloader::Callback(::mediax::rtp::RtpFrameData frame) const {
if (GetState() == ::mediax::rtp::StreamState::kStarted) {
GetCallback()(static_cast<const RtpDepayloader &>(*this), frame);
}
}

void RtpH264GstOpenDepayloader::NewFrame() {
void RtpH264GstVaapiDepayloader::NewFrame() {
new_rx_frame_ = true;
if ((CallbackRegistered()) && (GetState() == ::mediax::rtp::StreamState::kStarted)) {
RtpFrameData arg_tx = {{GetHeight(), GetWidth()}, GetBuffer().data(), GetColourSpace()};
Expand Down
10 changes: 5 additions & 5 deletions src/h264/gst/vaapi/rtp_h264_depayloader.h
Original file line number Diff line number Diff line change
Expand Up @@ -28,34 +28,34 @@ namespace mediax::rtp::h264::gst {}
namespace mediax::rtp::h264::gst::vaapi {

/// A RTP payloader for H.264 DEF-STAN 00-82 video streams
class RtpH264GstOpenDepayloader : public mediax::rtp::RtpDepayloader {
class RtpH264GstVaapiDepayloader : public mediax::rtp::RtpDepayloader {
public:
///
/// \brief Construct a new Rtpvraw Depayloader object
///
///
RtpH264GstOpenDepayloader();
RtpH264GstVaapiDepayloader();

///
/// \brief Destroy the Rtpvraw Depayloader object
///
///
~RtpH264GstOpenDepayloader() final;
~RtpH264GstVaapiDepayloader() final;

///
/// \brief Copy operator (Deleted)
///
/// \param other
/// \return RtpH264Depayloader
///
RtpH264GstOpenDepayloader &operator=(const RtpH264GstOpenDepayloader &other);
RtpH264GstVaapiDepayloader &operator=(const RtpH264GstVaapiDepayloader &other);

///
/// \brief Construct a new RtpH264Depayloader object (Deleted)
///
/// \param other
///
RtpH264GstOpenDepayloader(RtpH264GstOpenDepayloader &&other) = delete;
RtpH264GstVaapiDepayloader(RtpH264GstVaapiDepayloader &&other) = delete;

///
/// \brief Configure at RTP input stream and dont wait for the SAP/SDP announcement
Expand Down
16 changes: 8 additions & 8 deletions src/h264/gst/vaapi/rtp_h264_payloader.cc
Original file line number Diff line number Diff line change
Expand Up @@ -26,11 +26,11 @@

namespace mediax::rtp::h264::gst::vaapi {

RtpH264GstOpenPayloader::RtpH264GstOpenPayloader() = default;
RtpH264GstVaapiPayloader::RtpH264GstVaapiPayloader() = default;

RtpH264GstOpenPayloader::~RtpH264GstOpenPayloader() = default;
RtpH264GstVaapiPayloader::~RtpH264GstVaapiPayloader() = default;

void RtpH264GstOpenPayloader::SetStreamInfo(const ::mediax::rtp::StreamInformation &stream_information) {
void RtpH264GstVaapiPayloader::SetStreamInfo(const ::mediax::rtp::StreamInformation &stream_information) {
GetEgressPort().encoding = stream_information.encoding;
GetEgressPort().height = stream_information.height;
GetEgressPort().width = stream_information.width;
Expand All @@ -41,7 +41,7 @@ void RtpH264GstOpenPayloader::SetStreamInfo(const ::mediax::rtp::StreamInformati
GetEgressPort().settings_valid = true;
}

int RtpH264GstOpenPayloader::Transmit(unsigned char *new_buffer, bool timeout) {
int RtpH264GstVaapiPayloader::Transmit(unsigned char *new_buffer, bool timeout) {
if (!started_) {
DLOG(ERROR) << "RTP H.264 payloader not started";
return -1;
Expand Down Expand Up @@ -77,7 +77,7 @@ int RtpH264GstOpenPayloader::Transmit(unsigned char *new_buffer, bool timeout) {
return 0;
}

bool RtpH264GstOpenPayloader::Open() {
bool RtpH264GstVaapiPayloader::Open() {
// Setup a gstreamer pipeline to decode H.264 with Intel VAAPI

// Create a pipeline
Expand Down Expand Up @@ -137,7 +137,7 @@ bool RtpH264GstOpenPayloader::Open() {
return true;
}

void RtpH264GstOpenPayloader::Close() {
void RtpH264GstVaapiPayloader::Close() {
// Stop the pipeline
Stop();

Expand All @@ -146,13 +146,13 @@ void RtpH264GstOpenPayloader::Close() {
gst_object_unref(pipeline_);
}

void RtpH264GstOpenPayloader::Start() {
void RtpH264GstVaapiPayloader::Start() {
started_ = true;
// Start the pipeline
gst_element_set_state(pipeline_, GST_STATE_PLAYING);
}

void RtpH264GstOpenPayloader::Stop() {
void RtpH264GstVaapiPayloader::Stop() {
started_ = false;
// Stop the pipeline
gst_element_set_state(pipeline_, GST_STATE_NULL);
Expand Down
6 changes: 3 additions & 3 deletions src/h264/gst/vaapi/rtp_h264_payloader.h
Original file line number Diff line number Diff line change
Expand Up @@ -19,19 +19,19 @@
namespace mediax::rtp::h264::gst::vaapi {

/// A RTP payloader for H.264 DEF-STAN 00-82 video streams
class RtpH264GstOpenPayloader : public mediax::rtp::RtpPayloader {
class RtpH264GstVaapiPayloader : public mediax::rtp::RtpPayloader {
public:
///
/// \brief Construct a new Rtp H.264 Payloader object
///
///
RtpH264GstOpenPayloader();
RtpH264GstVaapiPayloader();

///
/// \brief Destroy the Rtp H.264 Payloader object
///
///
~RtpH264GstOpenPayloader() final;
~RtpH264GstVaapiPayloader() final;

///
/// \brief Set the Stream Info object
Expand Down

0 comments on commit 3056516

Please sign in to comment.