be.tarsos.dsp.ui.layers.SpectrumLayer Maven / Gradle / Ivy
/*
* _______ _____ _____ _____
* |__ __| | __ \ / ____| __ \
* | | __ _ _ __ ___ ___ ___| | | | (___ | |__) |
* | |/ _` | '__/ __|/ _ \/ __| | | |\___ \| ___/
* | | (_| | | \__ \ (_) \__ \ |__| |____) | |
* |_|\__,_|_| |___/\___/|___/_____/|_____/|_|
*
* -------------------------------------------------------------
*
* 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.ui.layers;
import java.awt.Color;
import java.awt.Graphics2D;
import java.util.ArrayList;
import java.util.List;
import be.tarsos.dsp.ui.Axis;
import be.tarsos.dsp.ui.CoordinateSystem;
import be.tarsos.dsp.util.PitchConverter;
public class SpectrumLayer implements Layer {
private float[] spectrum;
private List peaksInBins;
private float multiplier = 10;
private int sampleRate;
private int fftSize;
private CoordinateSystem cs;
private final Color color;
public SpectrumLayer(CoordinateSystem cs, int fftSize, int sampleRate,Color color){
this.cs = cs;
this.sampleRate = sampleRate;
this.fftSize = fftSize;
this.color= color;
peaksInBins = new ArrayList();
}
@Override
public void draw(Graphics2D graphics) {
if(spectrum!=null){
graphics.setColor(color);
int prevFreqInCents = 0;
int prevMagnitude = 0;
for(int i = 1 ; i < spectrum.length ; i++){
float hertzValue = (i* sampleRate) / (float) fftSize;
int frequencyInCents = (int) Math.round(PitchConverter.hertzToAbsoluteCent(hertzValue));
int magintude = Math.round(spectrum[i] * multiplier);
if(cs.getMin(Axis.X) < frequencyInCents && frequencyInCents < cs.getMax(Axis.X)){
graphics.drawLine(prevFreqInCents, prevMagnitude,frequencyInCents, magintude);
prevFreqInCents = frequencyInCents;
prevMagnitude = magintude;
}
}
int markerWidth = Math.round(LayerUtilities.pixelsToUnits(graphics, 7, true));
int markerheight = Math.round(LayerUtilities.pixelsToUnits(graphics, 7, false));
graphics.setColor(Color.blue);
for(int i = 0 ; i < peaksInBins.size() ; i++){
int bin = peaksInBins.get(i) ;
float hertzValue = (bin * sampleRate) / (float) fftSize;
int frequencyInCents = (int) Math.round(PitchConverter.hertzToAbsoluteCent(hertzValue) - markerWidth/2.0f );
if(cs.getMin(Axis.X) < frequencyInCents && frequencyInCents < cs.getMax(Axis.X)){
int magintude = Math.round(spectrum[bin] * multiplier - markerheight/2.0f);
graphics.drawOval(Math.round(frequencyInCents), magintude, markerWidth, markerheight);
}
}
}
}
@Override
public String getName() {
return "Spectrum";
}
public void setSpectrum(float[] spectrum) {
this.spectrum = spectrum;
}
public void setPeak(int binIndex) {
peaksInBins.add(binIndex);
}
public void clearPeaks(){
peaksInBins.clear();
}
public void setSampleRate(int sampleRate) {
this.sampleRate = sampleRate;
}
public void setFFTSize(int fftSize) {
this.fftSize = fftSize;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy