package org.dive4elements.river.artifacts.model.minfo;

import org.dive4elements.river.artifacts.model.DateRange;
import gnu.trove.TDoubleArrayList;


public class BedloadDiameterResult
extends BedQualityDiameterResult
{
    protected TDoubleArrayList diameter;

    public BedloadDiameterResult(
        String type,
        TDoubleArrayList diameter,
        TDoubleArrayList km,
        DateRange range
    ) {
        super (type, km);
        this.diameter = diameter;
    }

    public double getDiameter(int ndx) {
        if (diameter != null) {
            return this.diameter.get(ndx);
        }
        return Double.NaN;
    }

    public double getDiameter(double km) {
        if (kms.indexOf(km) >= 0) {
            return diameter.get(kms.indexOf(km));
        }
        return Double.NaN;
    }

    public double[][] getDiameterData() {
        return new double[][] {
            kms.toNativeArray(),
            diameter.toNativeArray()
        };
    }
}