MkvWriteTest.java

/*
 * myLib - https://github.com/taktod/myLib
 * Copyright (c) 2014 ttProject. All rights reserved.
 * 
 * Licensed under The MIT license.
 */
package com.ttProject.container.mkv.test;

import org.apache.log4j.Logger;
import org.junit.Test;

import com.ttProject.container.IContainer;
import com.ttProject.container.IReader;
import com.ttProject.container.IWriter;
import com.ttProject.container.mkv.MkvBlockTag;
import com.ttProject.container.mkv.MkvTagReader;
import com.ttProject.container.mkv.MkvTagWriter;
import com.ttProject.frame.CodecType;
import com.ttProject.frame.IFrame;
import com.ttProject.nio.channels.FileReadChannel;
import com.ttProject.nio.channels.IReadChannel;

/**
 * mkv data write test.
 * this doesn't work properly.
 * @author taktod
 */
public class MkvWriteTest {
	/** logger */
	private Logger logger = Logger.getLogger(MkvWriteTest.class);
	/**
	 * work test
	 * @throws Exception
	 */
	@Test
	public void test() throws Exception {
		logger.info("start write test.");
		IReadChannel source = FileReadChannel.openFileReadChannel(
				Thread.currentThread().getContextClassLoader().getResource("test.h264aac.mkv")
		);
		IReader reader = new MkvTagReader();
		IWriter writer = new MkvTagWriter("output.mkv");
		writer.prepareHeader(CodecType.H264, CodecType.AAC);
		IContainer container = null;
		while((container = reader.read(source)) != null) {
			if(container instanceof MkvBlockTag) {
				MkvBlockTag blockTag = (MkvBlockTag) container;
				IFrame frame = blockTag.getFrame();
				writer.addFrame(blockTag.getTrackId().get(), frame);
			}
		}
		writer.prepareTailer();
		logger.info("end");
	}
}