org.mobicents.media.server.component.audio.AudioComponent Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of components Show documentation
There is a newer version: 6.0.23
/*
 * JBoss, Home of Professional Open Source
 * Copyright 2011, Red Hat, Inc. and individual contributors
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */

package org.mobicents.media.server.component.audio;

import java.util.Iterator;

import org.mobicents.media.server.concurrent.ConcurrentMap;
import org.mobicents.media.server.spi.format.AudioFormat;
import org.mobicents.media.server.spi.format.FormatFactory;
import org.mobicents.media.server.spi.memory.Frame;
import org.mobicents.media.server.spi.memory.Memory;

/**
 * Implements compound components used by mixer and splitter.
 * 
 * @author Yulian Oifa
 */
public class AudioComponent {

	// the format of the output stream.
	private AudioFormat format = FormatFactory.createAudioFormat("LINEAR",
			8000, 16, 1);
	private long period = 20000000L;
	private int packetSize = (int) (period / 1000000) * format.getSampleRate()
			/ 1000 * format.getSampleSize() / 8;

	private ConcurrentMap inputs = new ConcurrentMap();
	private ConcurrentMap outputs = new ConcurrentMap();

	private Iterator activeInputs;
	private Iterator activeOutputs;

	protected Boolean shouldRead = false;
	protected Boolean shouldWrite = false;

	// samples storage
	private int[] data;

	private byte[] dataArray;
	private Frame inputFrame;
	private Frame outputFrame;

	int inputCount, outputCount, inputIndex, outputIndex;
	boolean first;

	private int componentId;

	/**
	 * Creates new instance with default name.
	 */
	public AudioComponent(int componentId) {
		this.componentId = componentId;
		data = new int[packetSize / 2];
	}

	public int getComponentId() {
		return componentId;
	}

	public void updateMode(Boolean shouldRead, Boolean shouldWrite) {
		this.shouldRead = shouldRead;
		this.shouldWrite = shouldWrite;
	}

	public void addInput(AudioInput input) {
		inputs.put(input.getInputId(), input);
	}

	public void addOutput(AudioOutput output) {
		outputs.put(output.getOutputId(), output);
	}

	public void remove(AudioInput input) {
		inputs.remove(input.getInputId());
	}

	public void remove(AudioOutput output) {
		outputs.remove(output.getOutputId());
	}

	public void perform() {
		first = true;
		activeInputs = inputs.valuesIterator();

		while (activeInputs.hasNext()) {
			AudioInput input = activeInputs.next();
			inputFrame = input.poll();
			if (inputFrame != null) {
				dataArray = inputFrame.getData();
				if (first) {
					inputIndex = 0;
					for (inputCount = 0; inputCount < dataArray.length; inputCount += 2) {
						data[inputIndex++] = (short) (((dataArray[inputCount + 1]) << 8) | (dataArray[inputCount] & 0xff));
					}
					first = false;
				} else {
					inputIndex = 0;
					for (inputCount = 0; inputCount < dataArray.length; inputCount += 2) {
						data[inputIndex++] += (short) (((dataArray[inputCount + 1]) << 8) | (dataArray[inputCount] & 0xff));
					}
				}
				inputFrame.recycle();
			}
		}
	}

	public int[] getData() {
		if (!this.shouldRead) {
			return null;
		}

		if (first) {
			return null;
		}

		return data;
	}

	public void offer(int[] data) {
		if (!this.shouldWrite) {
			return;
		}

		outputFrame = Memory.allocate(packetSize);
		dataArray = outputFrame.getData();

		outputIndex = 0;
		for (outputCount = 0; outputCount < data.length;) {
			dataArray[outputIndex++] = (byte) (data[outputCount]);
			dataArray[outputIndex++] = (byte) (data[outputCount++] >> 8);
		}

		outputFrame.setOffset(0);
		outputFrame.setLength(packetSize);
		outputFrame.setDuration(period);
		outputFrame.setFormat(format);

		activeOutputs = outputs.valuesIterator();
		while (activeOutputs.hasNext()) {
			AudioOutput output = activeOutputs.next();
			if (!activeOutputs.hasNext()) {
				output.offer(outputFrame);
			} else {
				output.offer(outputFrame.clone());
			}
			output.wakeup();
		}
	}
}