AudioProperties.java

/*
 * PS3 Media Server, for streaming any medias to your PS3.
 * Copyright (C) 2012  I. Sokolov
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; version 2
 * of the License only.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */
package net.pms.formats.v2;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.regex.Matcher;
import java.util.regex.Pattern;

import static org.apache.commons.lang3.StringUtils.isEmpty;


/**
 * Class for storing and parsing from libmediainfo output audio track's properties
 * (bitrate, channels number, etc) and meta information (title, album, etc)
 *
 * This class is not thread-safe
 *
 * @since 1.60.0
 */
public class AudioProperties {
	private static final Logger logger = LoggerFactory.getLogger(AudioProperties.class);

	private static final Pattern intPattern = Pattern.compile("([\\+-]?\\d+)([eE][\\+-]?\\d+)?");
	private static final Pattern floatPattern = Pattern.compile("([\\+-]?\\d(\\.\\d*)?|\\.\\d+)([eE][\\+-]?(\\d(\\.\\d*)?|\\.\\d+))?");

	private int numberOfChannels = 2;
	private int audioDelay = 0;
	private int sampleFrequency = 48000;


	public int getAttribute(AudioAttribute attribute) {
		switch (attribute) {
			case CHANNELS_NUMBER:
				return getNumberOfChannels();
			case DELAY:
				return getAudioDelay();
			case SAMPLE_FREQUENCY:
				return getSampleFrequency();
			default:
				throw new IllegalArgumentException("Unimplemented attribute");
		}
	}

	/**
	 * Get number of channels for this audio track.
	 * @return number of channels (default is 2)
	 */
	public int getNumberOfChannels() {
		return numberOfChannels;
	}

	/**
	 * Set number of channels for this audio track.
	 * @param numberOfChannels number of channels to set
	 */
	public void setNumberOfChannels(int numberOfChannels) {
		if (numberOfChannels < 1) {
			throw new IllegalArgumentException("Channel number can't be less than 1.");
		}
		this.numberOfChannels = numberOfChannels;
	}

	/**
	 * Set number of channels for this audio track with libmediainfo value
	 * @param mediaInfoValue libmediainfo "Channel(s)" value to parse
	 */
	public void setNumberOfChannels(String mediaInfoValue) {
		this.numberOfChannels = getChannelsNumberFromLibMediaInfo(mediaInfoValue);
	}

	/**
	 * Get delay for this audio track.
	 * @return audio delay in ms. May be negative.
	 */
	public int getAudioDelay() {
		return audioDelay;
	}

	/**
	 * Set delay for this audio track.
	 * @param audioDelay audio delay in ms to set. May be negative.
	 */
	public void setAudioDelay(int audioDelay) {
		this.audioDelay = audioDelay;
	}

	/**
	 * Set delay for this audio track with libmediainfo value
	 * @param mediaInfoValue libmediainfo "Video_Delay" value to parse
	 */
	public void setAudioDelay(String mediaInfoValue) {
		this.audioDelay = getAudioDelayFromLibMediaInfo(mediaInfoValue);
	}


	/**
	 * Get sample frequency for this audio track.
	 * @return sample frequency in Hz
	 */
	public int getSampleFrequency() {
		return sampleFrequency;
	}

	/**
	 * Set sample frequency for this audio track.
	 * @param sampleFrequency sample frequency in Hz
	 */
	public void setSampleFrequency(int sampleFrequency) {
		if (sampleFrequency < 1) {
			throw new IllegalArgumentException("Sample frequency can't be less than 1 Hz.");
		}
		this.sampleFrequency = sampleFrequency;
	}

	/**
	 * Set sample frequency for this audio track with libmediainfo value
	 * @param mediaInfoValue libmediainfo "Sampling rate" value to parse
	 */
	public void setSampleFrequency(String mediaInfoValue) {
		this.sampleFrequency = getSampleFrequencyFromLibMediaInfo(mediaInfoValue);
	}

	public static int getChannelsNumberFromLibMediaInfo(String mediaInfoValue) {
		if (isEmpty(mediaInfoValue)) {
			logger.warn("Empty value passed in. Returning default number 2.");
			return 2;
		}

		// examples of libmediainfo  (mediainfo --Full --Language=raw file):
		// Channel(s) : 2
		// Channel(s) : 6
		// Channel(s) : 2 channels / 1 channel / 1 channel

		int result = -1;
		Matcher intMatcher = intPattern.matcher(mediaInfoValue);
		while (intMatcher.find()) {
			String matchResult = intMatcher.group();
			try {
				int currentResult = Integer.parseInt(matchResult);
				if (currentResult > result) {
					result = currentResult;
				}
			} catch (NumberFormatException ex) {
				logger.warn("NumberFormatException during parsing substring {} from value {}", matchResult, mediaInfoValue);
			}
		}

		if (result <= 0) {
			logger.warn("Can't parse value {}. Returning default number 2.", mediaInfoValue);
			return 2;
		} else {
			return result;
		}
	}


	public static int getAudioDelayFromLibMediaInfo(String mediaInfoValue) {
		if (isEmpty(mediaInfoValue)) {
			logger.warn("Empty value passed in. Returning default number 0.");
			return 0;
		}

		// examples of libmediainfo output (mediainfo --Full --Language=raw file):
		// Video_Delay : 0

		int result = 0;
		Matcher intMatcher = intPattern.matcher(mediaInfoValue);
		if (intMatcher.find()) {
			String matchResult = intMatcher.group();
			try {
				result = Integer.parseInt(matchResult);
			} catch (NumberFormatException ex) {
				logger.warn("NumberFormatException during parsing substring {} from value {}", matchResult, mediaInfoValue);
			}
		}
		return result;
	}

	public static int getSampleFrequencyFromLibMediaInfo(String mediaInfoValue) {
		if (isEmpty(mediaInfoValue)) {
			logger.warn("Empty value passed in. Returning default number 48000 Hz.");
			return 48000;
		}

		// examples of libmediainfo output (mediainfo --Full --Language=raw file):
		// SamplingRate : 48000
		// SamplingRate : 44100 / 22050

		int result = -1;
		Matcher intMatcher = intPattern.matcher(mediaInfoValue);
		while (intMatcher.find()) {
			String matchResult = intMatcher.group();
			try {
				int currentResult = Integer.parseInt(matchResult);
				if (currentResult > result) {
					result = currentResult;
				}
			} catch (NumberFormatException ex) {
				logger.warn("NumberFormatException during parsing substring {} from value {}", matchResult, mediaInfoValue);
			}
		}

		if (result < 1) {
			logger.warn("Can't parse value {}. Returning default number 48000 Hz.", mediaInfoValue);
			return 48000;
		} else {
			return result;
		}
	}
}