More TS-related improvements, to ADTS support specifically.

This commit is contained in:
Thulinma 2016-07-07 13:20:12 +02:00
parent f7473bd26a
commit bd38fd20c3
4 changed files with 55 additions and 28 deletions

View file

@ -20,8 +20,8 @@ namespace aac {
bool adts::sameHeader(const adts & rhs) const { bool adts::sameHeader(const adts & rhs) const {
if (len < 7 || rhs.len < 7){return false;} if (!rhs || !*this){return false;}
return (memcmp(data, rhs.data, 7) == 0); return (getAACProfile() == rhs.getAACProfile() && getFrequencyIndex() == rhs.getFrequencyIndex() && getChannelConfig() == rhs.getChannelConfig());
} }
adts::adts(const adts & rhs){ adts::adts(const adts & rhs){
@ -46,14 +46,14 @@ namespace aac {
} }
} }
unsigned long adts::getAACProfile(){ unsigned long adts::getAACProfile() const{
if (!data || !len){ if (!data || !len){
return 0; return 0;
} }
return ((data[2] >> 6) & 0x03) + 1; return ((data[2] >> 6) & 0x03) + 1;
} }
unsigned long adts::getFrequencyIndex(){ unsigned long adts::getFrequencyIndex() const{
if (!data || !len){ if (!data || !len){
return 0; return 0;
} }
@ -61,7 +61,7 @@ namespace aac {
} }
unsigned long adts::getFrequency(){ unsigned long adts::getFrequency() const{
if (!data || len < 3){ if (!data || len < 3){
return 0; return 0;
} }
@ -83,28 +83,28 @@ namespace aac {
} }
} }
unsigned long adts::getChannelConfig(){ unsigned long adts::getChannelConfig() const{
if (!data || !len){ if (!data || !len){
return 0; return 0;
} }
return ((data[2] & 0x01) << 2) | ((data[3] >> 6) & 0x03); return ((data[2] & 0x01) << 2) | ((data[3] >> 6) & 0x03);
} }
unsigned long adts::getChannelCount(){ unsigned long adts::getChannelCount() const{
if (!data || !len){ if (!data || !len){
return 0; return 0;
} }
return (getChannelConfig() == 7 ? 8 : getChannelConfig()); return (getChannelConfig() == 7 ? 8 : getChannelConfig());
} }
unsigned long adts::getHeaderSize(){ unsigned long adts::getHeaderSize() const{
if (!data || !len){ if (!data || !len){
return 0; return 0;
} }
return (data[1] & 0x01 ? 7 : 9); return (data[1] & 0x01 ? 7 : 9);
} }
unsigned long adts::getPayloadSize(){ unsigned long adts::getPayloadSize() const{
if (!data || len < 6){ if (!data || len < 6){
return 0; return 0;
} }
@ -122,7 +122,7 @@ namespace aac {
return ret; return ret;
} }
unsigned long adts::getSampleCount(){ unsigned long adts::getSampleCount() const{
if (!data || len < 7){ if (!data || len < 7){
return 0; return 0;
} }
@ -135,11 +135,32 @@ namespace aac {
} }
return data + getHeaderSize(); return data + getHeaderSize();
} }
std::string adts::toPrettyString(){ std::string adts::toPrettyString() const{
std::stringstream res; std::stringstream res;
res << "SyncWord: " << std::hex << (((int)data[0] << 4) | ((data[1] >> 4) & 0x0F)) << std::endl; res << "ADTS packet (payload size: " << getPayloadSize() << ")" << std::endl;
res << "HeaderSize: " << std::dec << getHeaderSize() << std::endl; int syncWord = (((int)data[0] << 4) | ((data[1] >> 4) & 0x0F));
res << "PayloadSize: " << std::dec << getPayloadSize() << std::endl; if (syncWord != 0xfff){
res << " Sync word " << std::hex << syncWord << " != fff!" << std::endl;
}
if (data[1] & 0x8 == 0x8){
res << " MPEG-2" << std::endl;
}else{
res << " MPEG-4" << std::endl;
}
if (data[1] & 0x6 != 0){
res << " Non-zero layer!" << std::endl;
}
if (data[1] & 0x1 == 0x0){
res << " CRC present" << std::endl;
}
res << " MPEG-4 audio object type: " << getAACProfile() << std::endl;
res << " Frequency: " << getFrequency() << "Hz" << std::endl;
res << " Channels: " << getChannelCount() << std::endl;
res << " Samples: " << getSampleCount() << std::endl;
return res.str(); return res.str();
} }
adts::operator bool() const{
return (((int)data[0] << 4) | ((data[1] >> 4) & 0x0F)) == 0xfff && getFrequency();
}
} }

View file

@ -9,16 +9,17 @@ namespace aac {
~adts(); ~adts();
adts& operator = (const adts & rhs); adts& operator = (const adts & rhs);
bool sameHeader(const adts & rhs) const; bool sameHeader(const adts & rhs) const;
unsigned long getAACProfile(); unsigned long getAACProfile() const;
unsigned long getFrequencyIndex(); unsigned long getFrequencyIndex() const;
unsigned long getFrequency(); unsigned long getFrequency() const;
unsigned long getChannelConfig(); unsigned long getChannelConfig() const;
unsigned long getChannelCount(); unsigned long getChannelCount() const;
unsigned long getHeaderSize(); unsigned long getHeaderSize() const;
unsigned long getPayloadSize(); unsigned long getPayloadSize() const;
unsigned long getSampleCount(); unsigned long getSampleCount() const;
char * getPayload(); char * getPayload();
std::string toPrettyString(); std::string toPrettyString() const;
operator bool() const;
private: private:
char * data; char * data;
unsigned long len; unsigned long len;

View file

@ -418,15 +418,18 @@ namespace TS {
} }
while (offsetInPes < realPayloadSize){ while (offsetInPes < realPayloadSize){
aac::adts adtsPack(pesPayload + offsetInPes, realPayloadSize - offsetInPes); aac::adts adtsPack(pesPayload + offsetInPes, realPayloadSize - offsetInPes);
if (adtsPack.getFrequency() && adtsPack.getPayloadSize()){ if (adtsPack){
if (!adtsInfo.count(tid) || !adtsInfo[tid].sameHeader(adtsPack)){ if (!adtsInfo.count(tid) || !adtsInfo[tid].sameHeader(adtsPack)){
MEDIUM_MSG("Setting new ADTS header: %s", adtsPack.toPrettyString().c_str());
adtsInfo[tid] = adtsPack; adtsInfo[tid] = adtsPack;
} }
outPackets[tid].push_back(DTSC::Packet()); outPackets[tid].push_back(DTSC::Packet());
outPackets[tid].back().genericFill(timeStamp + ((samplesRead * 1000) / adtsPack.getFrequency()), timeOffset, tid, adtsPack.getPayload(), adtsPack.getPayloadSize(), bPos, 0); outPackets[tid].back().genericFill(timeStamp + ((samplesRead * 1000) / adtsPack.getFrequency()), timeOffset, tid, adtsPack.getPayload(), adtsPack.getPayloadSize(), bPos, 0);
}
samplesRead += adtsPack.getSampleCount(); samplesRead += adtsPack.getSampleCount();
offsetInPes += adtsPack.getHeaderSize() + adtsPack.getPayloadSize(); offsetInPes += adtsPack.getHeaderSize() + adtsPack.getPayloadSize();
}else{
offsetInPes++;
}
} }
if (threaded){ if (threaded){
globalSem.post(); globalSem.post();

View file

@ -35,6 +35,7 @@ namespace Analysers {
if (d[0] != 0 || d[1] != 0 || d[2] != 1){ if (d[0] != 0 || d[1] != 0 || d[2] != 1){
res << " [!!! INVALID START CODE: " << (int)d[0] << " " << (int)d[1] << " " << (int)d[2] << " ]"; res << " [!!! INVALID START CODE: " << (int)d[0] << " " << (int)d[1] << " " << (int)d[2] << " ]";
} }
unsigned int padding = 0;
if (known){ if (known){
if ((d[6] & 0xC0) != 0x80){ if ((d[6] & 0xC0) != 0x80){
res << " [!INVALID FIRST BITS!]"; res << " [!INVALID FIRST BITS!]";
@ -90,7 +91,8 @@ namespace Analysers {
headSize += 0; /// \todo Implement this. Complicated field, bah. headSize += 0; /// \todo Implement this. Complicated field, bah.
} }
if (d[8] != headSize){ if (d[8] != headSize){
res << " [Padding: " << ((int)d[8] - headSize) << "b]"; padding = d[8] - headSize;
res << " [Padding: " << padding << "b]";
} }
if (timeFlags & 0x02){ if (timeFlags & 0x02){
long long unsigned int time = (((unsigned int)d[9] & 0xE) >> 1); long long unsigned int time = (((unsigned int)d[9] & 0xE) >> 1);
@ -116,7 +118,7 @@ namespace Analysers {
if(detailLevel==1){ if(detailLevel==1){
unsigned int counter = 0; unsigned int counter = 0;
for (unsigned int i = 9+headSize; i<d.size(); ++i){ for (unsigned int i = 9+headSize+padding; i<d.size(); ++i){
if ((i < d.size() - 4) && d[i] == 0 && d[i+1] == 0 && d[i+2] == 0 && d[i+3] == 1){res << std::endl; counter = 0;} if ((i < d.size() - 4) && d[i] == 0 && d[i+1] == 0 && d[i+2] == 0 && d[i+3] == 1){res << std::endl; counter = 0;}
res << std::hex << std::setw(2) << std::setfill('0') << (int)(d[i]&0xff) << " "; res << std::hex << std::setw(2) << std::setfill('0') << (int)(d[i]&0xff) << " ";
counter++; counter++;