FlvAudioOutputEntry.java

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

import java.io.File;
import java.nio.channels.Channels;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

import org.apache.log4j.Level;
import org.apache.log4j.Logger;

import com.ttProject.container.flv.FlvHeaderTag;
import com.ttProject.container.flv.FlvTagWriter;
import com.ttProject.convertprocess.frame.IShareFrameListener;
import com.ttProject.convertprocess.server.ProcessClient;
import com.ttProject.frame.IAudioFrame;
import com.ttProject.frame.IFrame;

/**
 * flvの音声の部分のみ取り出したデータとして、ffmpegにデータを渡すentry
 * @author taktod
 */
public class FlvAudioOutputEntry implements IShareFrameListener {
	/** ロガー */
	@SuppressWarnings("unused")
	private Logger logger = Logger.getLogger(FlvAudioOutputEntry.class);
	/** 動作クライアント */
	private ProcessClient client = null;
	/** flvの出力モジュール */
	private FlvTagWriter writer = null;
	private FlvTagWriter writer2 = null;
	/** 標準出力がnettyから紐づいたthreadで実施するとフリーズする問題があったので、調整した */
	private ExecutorService exec = Executors.newCachedThreadPool();
	/**
	 * エントリーポイント
	 * @param args
	 */
	public static void main(String args[]) {
		Logger.getRootLogger().setLevel(Level.OFF);
		if(args == null || args.length != 1) {
			System.err.println("引数の数がおかしいです。");
			System.exit(-1);
			return;
		}
		int port = 0;
		try {
			port = Integer.parseInt(args[0]);
		}
		catch(Exception e) {
			System.err.println("入力ポート番号の数値解釈できませんでした。");
			System.exit(-1);
			return;
		}
		FlvAudioOutputEntry entry = new FlvAudioOutputEntry();
		entry.start(port);
	}
	/**
	 * コンストラクタ
	 */
	public FlvAudioOutputEntry() {
		client = new ProcessClient(this);
		// 通常のflvの出力としてデータを出したい。
		try {
			writer = new FlvTagWriter(Channels.newChannel(System.out));
			new File("audioOnly.flv").delete();
			writer2 = new FlvTagWriter("audioOnly.flv");
			FlvHeaderTag headerTag = new FlvHeaderTag();
			headerTag.setAudioFlag(true);
			headerTag.setVideoFlag(false);
			writer.addContainer(headerTag);
			writer2.addContainer(headerTag);
			Runtime.getRuntime().addShutdownHook(new Thread(){
				@Override
				public void run() {
					try {
						exec.shutdownNow();
						if(writer != null) {
							writer.prepareTailer();
							writer2.prepareTailer();
						}
					}
					catch(Exception e) {
					}
				}
			});
		}
		catch(Exception e) {
			writer = null;
		}
	}
	/**
	 * クライアントアクセス開始
	 * @param port
	 */
	public void start(int port) {
		// 動作開始
		client.connect("localhost", port);
		client.waitForClose();
	}
	/**
	 * {@inheritDoc}
	 */
	@Override
	public void pushFrame(final IFrame frame, final int id) {
		if(writer != null && frame instanceof IAudioFrame) {
			exec.execute(new Runnable() {
				@Override
				public void run() {
					try {
						writer.addFrame(id, frame);
						writer2.addFrame(id, frame);
					}
					catch(Exception e) {
						e.printStackTrace();
					}
				}
			});
		}
	}
}