h264文件数据帧前添加帧长度

master
haungdezhi 2019-03-08 13:04:58 +08:00
parent a79b70c200
commit 8c562c19a6
1 changed files with 136 additions and 40 deletions

View File

@ -1,58 +1,104 @@
import java.io.File; import java.io.File;
import java.io.FileInputStream; import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream; import java.io.InputStream;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.List; import java.util.List;
import java.util.Formatter; import java.util.Formatter;
public class H264Parser { public class H264Parser {
//0x00000001或0x000001是一个nalu的起始标志遇到下一个此标志时为该nalu的结尾。
//起始标志的后面第一个字节type里包含有nalu的类型type & 0x1F即为该nalu的类型nalu_type
private static final int NALU_TYPE_SLICE = 1; //非IDR图像中不采用数据划分的片段为P或B帧 //0x00000001或0x000001是一个nalu的起始标志遇到下一个此标志时为该nalu的结尾。
private static final int NALU_TYPE_DPA = 2; //非IDR图像中A类数据划分片段 //起始标志的后面第一个字节type里包含有nalu的类型type & 0x1F即为该nalu的类型nalu_type
private static final int NALU_TYPE_DPB = 3; //非IDR图像中B类数据划分片段
private static final int NALU_TYPE_DPC = 4; //非IDR图像中C类数据划分片段 private static final int NALU_TYPE_SLICE = 1; //非IDR图像中不采用数据划分的片段为P或B帧
private static final int NALU_TYPE_IDR = 5; //IDR图像的片段关键帧I帧 private static final int NALU_TYPE_DPA = 2; //非IDR图像中A类数据划分片段
private static final int NALU_TYPE_SEI = 6; //补充增强信息 private static final int NALU_TYPE_DPB = 3; //非IDR图像中B类数据划分片段
private static final int NALU_TYPE_SPS = 7; //序列参数集Sequence Parameter Set private static final int NALU_TYPE_DPC = 4; //非IDR图像中C类数据划分片段
private static final int NALU_TYPE_PPS = 8; //图像参数集PPSPicture Parameter Set private static final int NALU_TYPE_IDR = 5; //IDR图像的片段关键帧I帧
private static final int NALU_TYPE_AUD = 9; //分割符 private static final int NALU_TYPE_SEI = 6; //补充增强信息
private static final int NALU_TYPE_EOSEQ = 10; //序列结束符 private static final int NALU_TYPE_SPS = 7; //序列参数集Sequence Parameter Set
private static final int NALU_TYPE_EOSTREAM = 11; //流结束符 private static final int NALU_TYPE_PPS = 8; //图像参数集PPSPicture Parameter Set
private static final int NALU_TYPE_FILL = 12; //填充数据 private static final int NALU_TYPE_AUD = 9; //分割符
private static final int NALU_TYPE_EOSEQ = 10; //序列结束符
private static final int NALU_TYPE_EOSTREAM = 11; //流结束符
private static final int NALU_TYPE_FILL = 12; //填充数据
private static final int NALU_PRIORITY_DISPOSABLE = 0; private static final int NALU_PRIORITY_DISPOSABLE = 0;
private static final int NALU_PRIORITY_LOW = 1; private static final int NALU_PRIORITY_LOW = 1;
private static final int NALU_PRIORITY_HIGH = 2; private static final int NALU_PRIORITY_HIGH = 2;
private static final int NALU_PRIORITY_HIGHEST = 3; private static final int NALU_PRIORITY_HIGHEST = 3;
private boolean mFirstFind = true; //第一次查找起始码 private boolean mFirstFind = true; //第一次查找起始码
private int mStartCodeLen = 0; //起始码长度 private int mStartCodeLen = 0; //起始码长度
private int mCurStartCodeLen = 0; private int mCurStartCodeLen = 0;
private int mFrameLen = 0; //帧长度 private int mFrameLen = 0; //帧长度
private int mFrameFirstByte = 0; //帧数据中的第一个字节 private int mFrameFirstByte = 0; //帧数据中的第一个字节
private int mCurFrameFirstByte = 0; private int mCurFrameFirstByte = 0;
private int mPos = 0; private int mPos = 0;
private int mLen = 0; private int mLen = 0;
private String mStartCode = ""; private String mStartCode = "";
private int mNaluType = 0; private int mNaluType = 0;
private int mForbiddenBit = 0; private int mForbiddenBit = 0;
private int mNalReferenceIdc = 0; private int mNalReferenceIdc = 0;
private byte[] mFrameBuffer = null;
private FileOutputStream mFileOutputStream = null;
private boolean mIsAddFrameLen = false;
public static void main(String[] args) { public static void main(String[] args) {
if (args.length > 0) { H264Parser h264Parser = null;
H264Parser h264Parser = new H264Parser(); if (args.length == 2) {
h264Parser = new H264Parser();
if ("1".equals(args[1])) {
h264Parser.mIsAddFrameLen = true;
}
h264Parser.init();
h264Parser.ParseH264(args[0]);
} else if (args.length == 1) {
h264Parser = new H264Parser();
h264Parser.init();
h264Parser.ParseH264(args[0]); h264Parser.ParseH264(args[0]);
} else { } else {
System.out.println("Missing h264 filename."); System.out.println("Missing h264 filename.");
String path = "C:\\Users\\hw\\Desktop\\test.h264";
h264Parser = new H264Parser();
h264Parser.init();
h264Parser.ParseH264(path);
}
if (h264Parser != null) {
h264Parser.clear();
} }
} }
//如果NALU对应的Slice为一帧的开始就用0x00000001(即4个字节长度)否则就用0x000001(即3个字节长度)。 public H264Parser() {
}
public void init() {
mFrameBuffer = new byte[2 * 1024 * 1024];
if (mIsAddFrameLen) {
File file = new File("test_2.h264");
try {
mFileOutputStream = new FileOutputStream(file);
} catch (Exception e) {
e.printStackTrace();
}
}
}
public void clear() {
if (mFileOutputStream != null) {
try {
mFileOutputStream.close();
} catch (Exception e) {
e.printStackTrace();
}
}
}
//如果NALU对应的Slice为一帧的开始就用0x00000001(即4个字节长度)否则就用0x000001(即3个字节长度)。
public int FindStartCode(InputStream inputStream) { public int FindStartCode(InputStream inputStream) {
byte[] startCode = new byte[4]; byte[] startCode = new byte[4];
@ -102,7 +148,7 @@ public class H264Parser {
public int GetNalu(InputStream inputStream) { public int GetNalu(InputStream inputStream) {
mPos = mPos + mLen; mPos = mPos + mLen;
//第一次查找起始码,只是为了移动文件指针 //第一次查找起始码,只是为了移动文件指针
if (mFirstFind) { if (mFirstFind) {
mFirstFind = false; mFirstFind = false;
if (-1 == (mStartCodeLen = FindStartCode(inputStream))) { if (-1 == (mStartCodeLen = FindStartCode(inputStream))) {
@ -112,40 +158,55 @@ public class H264Parser {
if (mStartCodeLen == 3) { if (mStartCodeLen == 3) {
mStartCode = "001"; mStartCode = "001";
// 添加起始码
mFrameBuffer[0] = 0;
mFrameBuffer[1] = 0;
mFrameBuffer[2] = 1;
} else { } else {
mStartCode = "0001"; mStartCode = "0001";
// 添加起始码
mFrameBuffer[0] = 0;
mFrameBuffer[1] = 0;
mFrameBuffer[2] = 0;
mFrameBuffer[3] = 1;
} }
List<Integer> startCodeList = new ArrayList<>(); List<Integer> startCodeList = new ArrayList<>();
int frameLen = 0; int frameLen = 0;
while(true) { while(true) {
int tmp = -1; int tmp = -1;
//一个字节一个字节从文件流中读取 //一个字节一个字节从文件流中读取
try { try {
tmp = inputStream.read(); tmp = inputStream.read();
} catch (Exception e) { } catch (Exception e) {
e.printStackTrace(); e.printStackTrace();
} }
//读取到文件尾 //读取到文件尾
if (tmp == -1) { if (tmp == -1) {
mFrameLen = frameLen; mFrameLen = frameLen;
// 另存为
saveAs(frameLen);
return -1; return -1;
} }
//帧数据长度累加 //读取到的数据保存到buffer
mFrameBuffer[mStartCodeLen+frameLen] = (byte)(0xff & tmp);
//帧数据长度累加
frameLen++; frameLen++;
//将从文件中读取的数据保存到list中 //将从文件中读取的数据保存到list中
startCodeList.add(tmp); startCodeList.add(tmp);
//从一帧的第一个字节中获取NALU信息 //从一帧的第一个字节中获取NALU信息
if (frameLen == 1) { if (frameLen == 1) {
//如果上一次查找到的起始码长度是3则上次已经保存了第一个字节的数据 //如果上一次查找到的起始码长度是3则上次已经保存了第一个字节的数据
if (mStartCodeLen == 3) { if (mStartCodeLen == 3) {
tmp = mFrameFirstByte; tmp = mFrameFirstByte;
frameLen++; //帧长度加上第一个字节 //读取到的数据保存到buffer
mFrameBuffer[mStartCodeLen+frameLen] = (byte)(0xff & tmp);
frameLen++; //帧长度加上第一个字节
} }
mCurFrameFirstByte = tmp; mCurFrameFirstByte = tmp;
mForbiddenBit = tmp & 0x80; mForbiddenBit = tmp & 0x80;
@ -153,12 +214,12 @@ public class H264Parser {
mNaluType = tmp & 0x1f; mNaluType = tmp & 0x1f;
} }
//保存四个字节到list中 //保存四个字节到list中
if (startCodeList.size() < 4) { if (startCodeList.size() < 4) {
continue; continue;
} }
//查找起始码如果返回值大于0就表示找到了 //查找起始码如果返回值大于0就表示找到了
int startCodeLen = FindStartCode(startCodeList); int startCodeLen = FindStartCode(startCodeList);
if (startCodeLen > 0) { if (startCodeLen > 0) {
if (startCodeLen == 3) { if (startCodeLen == 3) {
@ -172,9 +233,29 @@ public class H264Parser {
startCodeList.remove(0); startCodeList.remove(0);
} }
mFrameLen = frameLen; mFrameLen = frameLen;
// 另存为
saveAs(frameLen);
return 0; return 0;
} }
/**
*
*
* @param frameLen
*/
public void saveAs(int frameLen) {
if (mFileOutputStream != null) {
try {
//System.out.println("len = " + mStartCodeLen+frameLen);
mFileOutputStream.write(intToBytes(mStartCodeLen+frameLen), 0, 4);
mFileOutputStream.write(mFrameBuffer, 0, mStartCodeLen+frameLen);
} catch (Exception e) {
e.printStackTrace();
}
mFrameBuffer[0] = '\0';
}
}
public boolean ParseH264(String fileName) { public boolean ParseH264(String fileName) {
if (fileName.equals("")) { if (fileName.equals("")) {
return false; return false;
@ -227,6 +308,9 @@ public class H264Parser {
if (dataLen == -1) { if (dataLen == -1) {
break; break;
} }
if (iNaluNum == 3) {
//break;
}
} }
try { try {
@ -236,4 +320,16 @@ public class H264Parser {
} }
return true; return true;
} }
/**
* intbyte
*/
public static byte[] intToBytes(int i){
byte[] result=new byte[4];
result[3]=(byte)((i >> 24)& 0xFF);
result[2]=(byte)((i >> 16)& 0xFF);
result[1]=(byte)((i >> 8)& 0xFF);
result[0]=(byte)(i & 0xFF);
return result;
}
} }