HaarWaveletCoder.java

/*
*      _______                       _____   _____ _____  
*     |__   __|                     |  __ \ / ____|  __ \ 
*        | | __ _ _ __ ___  ___  ___| |  | | (___ | |__) |
*        | |/ _` | '__/ __|/ _ \/ __| |  | |\___ \|  ___/ 
*        | | (_| | |  \__ \ (_) \__ \ |__| |____) | |     
*        |_|\__,_|_|  |___/\___/|___/_____/|_____/|_|     
*                                                         
* -------------------------------------------------------------
*
* TarsosDSP is developed by Joren Six at IPEM, University Ghent
*  
* -------------------------------------------------------------
*
*  Info: http://0110.be/tag/TarsosDSP
*  Github: https://github.com/JorenSix/TarsosDSP
*  Releases: http://0110.be/releases/TarsosDSP/
*  
*  TarsosDSP includes modified source code by various authors,
*  for credits and info, see README.
* 
*/

package be.tarsos.dsp.wavelet;

import java.util.Arrays;

import be.tarsos.dsp.AudioEvent;
import be.tarsos.dsp.AudioProcessor;

public class HaarWaveletCoder implements AudioProcessor{
	
	private final HaarWaveletTransform transform;
	
	private int compression;
	
	public HaarWaveletCoder(){
		this(16);
	}
	
	public HaarWaveletCoder(int compression){
		transform = new HaarWaveletTransform();
		this.compression = compression;
	}
	

	@Override
	public boolean process(AudioEvent audioEvent) {

		float[] audioBuffer = audioEvent.getFloatBuffer();
		float[] sortBuffer = new float[audioBuffer.length];
		transform.transform(audioEvent.getFloatBuffer());

		for (int i = 0; i < sortBuffer.length; i++) {
			sortBuffer[i] = Math.abs(audioBuffer[i]);
		}
		Arrays.sort(sortBuffer);

		double threshold = sortBuffer[compression];

		for (int i = 0; i < audioBuffer.length; i++) {
			if (Math.abs(audioBuffer[i]) <= threshold) {
				audioBuffer[i] = 0;
			}
		}

		return true;
	}

	@Override
	public void processingFinished() {
		
	}
	
	public void setCompression(int compression){
		this.compression = compression;
	}
	
	public int getCompression(){
		return this.compression;
	}

}