CorrelationLambdaTest.java Source code

Introduction

Here is the source code for CorrelationLambdaTest.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */

import java.awt.BorderLayout;
import java.awt.Label;
import java.util.Arrays;
import java.util.function.Function;
import java.util.function.UnaryOperator;
import javax.swing.JFrame;
import javax.swing.JPanel;
import jwave.Transform;
import jwave.transforms.FastWaveletTransform;
import jwave.transforms.WaveletTransform;
import jwave.transforms.wavelets.haar.Haar1;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.plot.XYPlot;
import org.jfree.data.xy.CategoryTableXYDataset;
import org.junit.After;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import static org.junit.Assert.*;
import ru.spbspu.model.IMathOperations;
import ru.spbspu.model.ISpectrogramPanel;
import ru.spbspu.model.MathOperations;
import ru.spbspu.model.TestData;

/**
 *
 * @author float
 */
public class CorrelationLambdaTest {

    private double calculateCoefficient(double[] wavelet, double[] signal, int index, int window) {
        double coeff = 0;
        for (int i = index; i < index + window; i++) {
            //            if (index + i > signal.length - 1) {
            //                return coeff;
            //            } else {
            coeff += wavelet[i] * signal[i];
            //            }
        }
        return coeff;
    }

    public double[] crossCorrelationCoefficient(double[] signal, int window) {
        Transform t = new Transform(new FastWaveletTransform(new Haar1()));
        double[] wavelet = t.forward(signal);
        System.out.println("Wavelet's length is " + wavelet.length + " Signal length is " + signal.length);
        double[] coeffs = new double[signal.length / window];
        int index = 0;
        for (int i = 0; i < coeffs.length; i++, index += window) {
            coeffs[i] = calculateCoefficient(wavelet, signal, index, window);
        }
        return coeffs;
    }

    @After
    public void tearDown() throws InterruptedException {
        while (true) {
            Thread.sleep(1000);
        }
    }

    public void crossCorrelationTest() {
        Transform t = new Transform(new FastWaveletTransform(new Haar1()));
        double[] signal = testData.get1DSimpleSignal(1.5, 5.0, 32768, 5000);
        double[] data = crossCorrelationCoefficient(signal, window);
        for (int i = 0; i < data.length; i++) {
            System.out.println(data[i]);
        }
    }

    @Test

    public void crossCorrelationGraphicTest() {

        CategoryTableXYDataset serie = new CategoryTableXYDataset();
        serie.setNotify(false);
        double step = 1.0 / discretization;
        double startPosition = step * framePosition;
        //100 ? - 100 , 50 ? - 50 , 25 ?- 25 
        Transform t = new Transform(new FastWaveletTransform(new Haar1()));
        double[] signal = testData.get1DSimpleSignal(1.5, 500, 32768, 5000);
        double[] data = crossCorrelationCoefficient(signal, window);

        //        double[] data = math.convolve(testData.get1DSignal(100, 200, frameWidth, discretization), math.lpf(70, step, 128));

        //        double[] data = math.convolve(testData.get1DSignal(100, 200, 32768, 10000), math.lpf(70, 1./10000, 32));
        //        double[] data = testData.get1DSignal(100, 200, frameWidth, discretization);
        //        double[] data = math.lpf(70, step,128);
        for (int i = 0; i < data.length; i++) {
            serie.add(startPosition, data[i], "");
            startPosition += step;
        }
        JFreeChart chart = ChartFactory.createXYLineChart("", "t,c", "wave", serie);
        chart.removeLegend();
        chart.setAntiAlias(false);

        XYPlot plot = chart.getXYPlot();
        //plot.setRangeGridlinePaint(Color.BLACK);
        org.jfree.chart.axis.ValueAxis yAxis = plot.getRangeAxis();
        org.jfree.chart.axis.ValueAxis xAxis = plot.getDomainAxis();
        double start = framePosition * 1.0 / discretization;
        double max = start + frameWidth * 1.0 / discretization;
        xAxis.setRange(start, max);
        ChartPanel chartPanel = new ChartPanel(chart);

        JPanel p = new JPanel(new BorderLayout());

        p.removeAll();
        p.add(chartPanel);
        p.validate();
        //1. Create the frame.
        JFrame frame = new JFrame("FrameDemo");

        //2. Optional: What happens when the frame closes?
        frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);

        //3. Create components and put them in the frame.
        //...create emptyLabel...
        frame.getContentPane().add(new Label("olol"), BorderLayout.CENTER);
        frame.getContentPane().add(p, BorderLayout.CENTER);

        //4. Size the frame.
        frame.pack();

        //5. Show it.
        frame.setVisible(true);
    }

    TestData testData = new TestData();
    int discretization = 1024;
    int framePosition = 0;
    int frameWidth = 131072;
    int window = 256;
    IMathOperations math = new MathOperations();
    ISpectrogramPanel specPanel;

    public double[][] getWaveletSpectrogram() {
        Transform t = new Transform(new FastWaveletTransform(new Haar1()));
        int position = 0;
        double[][] dataForSpectrogram = new double[frameWidth / window * 2][];
        double[] dataSource = testData.get1DSimpleSignal(1, 50, frameWidth, 5000);
        double[] wavelet = t.forward(math.HammingWindow(dataSource, dataSource.length));
        for (int j = 0; j < dataForSpectrogram.length; j++) {
            if (dataSource.length % 2 != 0) {
                break;
            }
            //            dataForSpectrogram[j] = crossCorrelationCoefficient(wavelet, dataSource);
            position += window / 2;
        }
        return dataForSpectrogram;
    }
}