org.dawnsci.spectrum.ui.processing.AverageProcess.java Source code

Introduction

Here is the source code for org.dawnsci.spectrum.ui.processing.AverageProcess.java

Source

/*
 * Copyright (c) 2012 Diamond Light Source Ltd.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 */
package org.dawnsci.spectrum.ui.processing;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import org.apache.commons.math3.stat.descriptive.MultivariateSummaryStatistics;
import org.dawnsci.spectrum.ui.file.IContain1DData;
import org.dawnsci.spectrum.ui.utils.Contain1DDataImpl;
import org.dawnsci.spectrum.ui.utils.SpectrumUtils;
import org.eclipse.january.dataset.Dataset;
import org.eclipse.january.dataset.DatasetFactory;
import org.eclipse.january.dataset.DatasetUtils;
import org.eclipse.january.dataset.DoubleDataset;
import org.eclipse.january.dataset.IDataset;

public class AverageProcess extends AbstractProcess {

    @Override
    public List<IContain1DData> process(List<IContain1DData> list) {

        list = SpectrumUtils.getCompatibleDatasets(list);

        if (list == null)
            return null;

        IDataset x0 = list.get(0).getxDataset();

        StringBuilder sb = new StringBuilder();
        sb.append("Average of: ");
        sb.append("\n");
        MultivariateSummaryStatistics ms = new MultivariateSummaryStatistics(x0.getSize(), false);
        for (IContain1DData file : list) {

            sb.append(file.getName() + ":");

            for (IDataset ds : file.getyDatasets()) {

                sb.append(ds.getName() + ":");

                DoubleDataset dd;
                if (ds instanceof DoubleDataset)
                    dd = (DoubleDataset) ds;
                else {
                    dd = (DoubleDataset) DatasetUtils.cast(ds, Dataset.FLOAT64);
                }
                double[] raw = dd.getData();
                ms.addValue(raw);
            }

            sb.deleteCharAt(sb.length() - 1);
            sb.deleteCharAt(sb.length() - 1);
            sb.append("\n");
        }
        List<IDataset> sets = new ArrayList<IDataset>();
        Dataset dd = DatasetFactory.createFromObject(ms.getMean());

        dd.setName("Average");
        sets.add(dd);

        String shortName = "Average: " + list.get(0).getName() + " to " + list.get(list.size() - 1).getName();

        return Arrays.asList(new IContain1DData[] {
                new Contain1DDataImpl(x0, sets, shortName, sb.toString() + "[" + sets.hashCode() + "]") });

    }

    @Override
    protected Dataset process(Dataset x, Dataset y) {
        return null;
    }

    @Override
    protected String getAppendingName() {
        //Should not be called
        return "_average";
    }

}