mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-24 03:30:52 +08:00
116 lines
2.9 KiB
C++
116 lines
2.9 KiB
C++
#ifndef ZLMEDIAKIT_SRT_COMMON_H
|
||
#define ZLMEDIAKIT_SRT_COMMON_H
|
||
#if defined(_WIN32)
|
||
#include <winsock2.h>
|
||
#include <ws2tcpip.h>
|
||
|
||
#include <Iphlpapi.h>
|
||
#pragma comment(lib, "Ws2_32.lib")
|
||
#pragma comment(lib, "Iphlpapi.lib")
|
||
#else
|
||
#include <netdb.h>
|
||
#include <sys/socket.h>
|
||
#endif // defined(_WIN32)
|
||
|
||
#include <chrono>
|
||
#define MAX_SEQ 0x7fffffff
|
||
#define SEQ_NONE 0xffffffff
|
||
#define MAX_TS 0xffffffff
|
||
|
||
namespace SRT {
|
||
using SteadyClock = std::chrono::steady_clock;
|
||
using TimePoint = std::chrono::time_point<SteadyClock>;
|
||
|
||
using Microseconds = std::chrono::microseconds;
|
||
using Milliseconds = std::chrono::milliseconds;
|
||
|
||
static inline int64_t DurationCountMicroseconds(SteadyClock::duration dur) {
|
||
return std::chrono::duration_cast<std::chrono::microseconds>(dur).count();
|
||
}
|
||
|
||
static inline uint32_t loadUint32(uint8_t *ptr) {
|
||
return ptr[0] << 24 | ptr[1] << 16 | ptr[2] << 8 | ptr[3];
|
||
}
|
||
|
||
static inline uint16_t loadUint16(uint8_t *ptr) {
|
||
return ptr[0] << 8 | ptr[1];
|
||
}
|
||
|
||
inline static int64_t seqCmp(uint32_t seq1, uint32_t seq2) {
|
||
if(seq1 > seq2){
|
||
if((seq1 - seq2) >(MAX_SEQ>>1)){
|
||
return (int64_t)seq1 - (int64_t)(seq2+MAX_SEQ);
|
||
}else{
|
||
return (int64_t)seq1 - (int64_t)seq2;
|
||
}
|
||
}else{
|
||
if((seq2-seq1) >(MAX_SEQ>>1)){
|
||
return (int64_t)(seq1+MAX_SEQ) - (int64_t)seq2;
|
||
}else{
|
||
return (int64_t)seq1 - (int64_t)seq2;
|
||
}
|
||
}
|
||
}
|
||
|
||
inline static uint32_t incSeq(int32_t seq) {
|
||
return (seq == MAX_SEQ) ? 0 : seq + 1;
|
||
}
|
||
static inline void storeUint32(uint8_t *buf, uint32_t val) {
|
||
buf[0] = val >> 24;
|
||
buf[1] = (val >> 16) & 0xff;
|
||
buf[2] = (val >> 8) & 0xff;
|
||
buf[3] = val & 0xff;
|
||
}
|
||
|
||
static inline void storeUint16(uint8_t *buf, uint16_t val) {
|
||
buf[0] = (val >> 8) & 0xff;
|
||
buf[1] = val & 0xff;
|
||
}
|
||
|
||
static inline void storeUint32LE(uint8_t *buf, uint32_t val) {
|
||
buf[0] = val & 0xff;
|
||
buf[1] = (val >> 8) & 0xff;
|
||
buf[2] = (val >> 16) & 0xff;
|
||
buf[3] = (val >> 24) & 0xff;
|
||
}
|
||
|
||
static inline void storeUint16LE(uint8_t *buf, uint16_t val) {
|
||
buf[0] = val & 0xff;
|
||
buf[1] = (val >> 8) & 0xff;
|
||
}
|
||
|
||
static inline uint32_t srtVersion(int major, int minor, int patch) {
|
||
return patch + minor * 0x100 + major * 0x10000;
|
||
}
|
||
static inline uint32_t genExpectedSeq(uint32_t seq) {
|
||
return MAX_SEQ & seq;
|
||
}
|
||
|
||
class UTicker {
|
||
public:
|
||
UTicker() { _created = _begin = SteadyClock::now(); }
|
||
~UTicker() = default;
|
||
|
||
/**
|
||
* 获取创建时间，单位微妙
|
||
*/
|
||
int64_t elapsedTime(TimePoint now) const { return DurationCountMicroseconds(now - _begin); }
|
||
|
||
/**
|
||
* 获取上次resetTime后至今的时间，单位毫秒
|
||
*/
|
||
int64_t createdTime(TimePoint now) const { return DurationCountMicroseconds(now - _created); }
|
||
|
||
/**
|
||
* 重置计时器
|
||
*/
|
||
void resetTime(TimePoint now) { _begin = now; }
|
||
|
||
private:
|
||
TimePoint _begin;
|
||
TimePoint _created;
|
||
};
|
||
|
||
} // namespace SRT
|
||
|
||
#endif // ZLMEDIAKIT_SRT_COMMON_H
|