hep.aida.tfloat.ref.FloatHistogram2D Maven / Gradle / Ivy
Show all versions of parallelcolt Show documentation
package hep.aida.tfloat.ref;
import hep.aida.tfloat.FloatIAxis;
import hep.aida.tfloat.FloatIHistogram1D;
import hep.aida.tfloat.FloatIHistogram2D;
/**
* A reference implementation of hep.aida.IHistogram2D. The goal is to provide a
* clear implementation rather than the most efficient implementation. However,
* performance seems fine - filling 6 * 10^5 points/sec, both using FixedAxis or
* VariableAxis.
*
* @author Wolfgang Hoschek, Tony Johnson, and others.
* @version 1.0, 23/03/2000
*/
public class FloatHistogram2D extends FloatAbstractHistogram2D implements FloatIHistogram2D {
/**
*
*/
private static final long serialVersionUID = 1L;
private float[][] heights;
private float[][] errors;
private int[][] entries;
private int nEntry; // total number of times fill called
private float sumWeight; // Sum of all weights
private float sumWeightSquared; // Sum of the squares of the weights
private float meanX, rmsX;
private float meanY, rmsY;
/**
* Creates a variable-width histogram. Example:
* xEdges = (0.2, 1.0, 5.0, 6.0), yEdges = (-5, 0, 7) yields 3*2
* in-range bins.
*
* @param title
* The histogram title.
* @param xEdges
* the bin boundaries the x-axis shall have; must be sorted
* ascending and must not contain multiple identical elements.
* @param yEdges
* the bin boundaries the y-axis shall have; must be sorted
* ascending and must not contain multiple identical elements.
* @throws IllegalArgumentException
* if xEdges.length < 1 || yEdges.length < 1.
*/
public FloatHistogram2D(String title, float[] xEdges, float[] yEdges) {
this(title, new FloatVariableAxis(xEdges), new FloatVariableAxis(yEdges));
}
/**
* Creates a fixed-width histogram.
*
* @param title
* The histogram title.
* @param xBins
* The number of bins on the X axis.
* @param xMin
* The minimum value on the X axis.
* @param xMax
* The maximum value on the X axis.
* @param yBins
* The number of bins on the Y axis.
* @param yMin
* The minimum value on the Y axis.
* @param yMax
* The maximum value on the Y axis.
*/
public FloatHistogram2D(String title, int xBins, float xMin, float xMax, int yBins, float yMin, float yMax) {
this(title, new FloatFixedAxis(xBins, xMin, xMax), new FloatFixedAxis(yBins, yMin, yMax));
}
/**
* Creates a histogram with the given axis binning.
*
* @param title
* The histogram title.
* @param xAxis
* The x-axis description to be used for binning.
* @param yAxis
* The y-axis description to be used for binning.
*/
public FloatHistogram2D(String title, FloatIAxis xAxis, FloatIAxis yAxis) {
super(title);
this.xAxis = xAxis;
this.yAxis = yAxis;
int xBins = xAxis.bins();
int yBins = yAxis.bins();
entries = new int[xBins + 2][yBins + 2];
heights = new float[xBins + 2][yBins + 2];
errors = new float[xBins + 2][yBins + 2];
}
public int allEntries() {
return nEntry;
}
public int binEntries(int indexX, int indexY) {
// return entries[xAxis.map(indexX)][yAxis.map(indexY)];
return entries[mapX(indexX)][mapY(indexY)];
}
public float binError(int indexX, int indexY) {
// return Math.sqrt(errors[xAxis.map(indexX)][yAxis.map(indexY)]);
return (float) Math.sqrt(errors[mapX(indexX)][mapY(indexY)]);
}
public float binHeight(int indexX, int indexY) {
// return heights[xAxis.map(indexX)][yAxis.map(indexY)];
return heights[mapX(indexX)][mapY(indexY)];
}
public float equivalentBinEntries() {
return sumWeight * sumWeight / sumWeightSquared;
}
public void fill(float x, float y) {
// int xBin = xAxis.getBin(x);
// int yBin = xAxis.getBin(y);
int xBin = mapX(xAxis.coordToIndex(x));
int yBin = mapY(yAxis.coordToIndex(y));
entries[xBin][yBin]++;
heights[xBin][yBin]++;
errors[xBin][yBin]++;
nEntry++;
sumWeight++;
sumWeightSquared++;
meanX += x;
rmsX += x;
meanY += y;
rmsY += y;
}
public void fill(float x, float y, float weight) {
// int xBin = xAxis.getBin(x);
// int yBin = xAxis.getBin(y);
int xBin = mapX(xAxis.coordToIndex(x));
int yBin = mapY(yAxis.coordToIndex(y));
entries[xBin][yBin]++;
heights[xBin][yBin] += weight;
errors[xBin][yBin] += weight * weight;
nEntry++;
sumWeight += weight;
sumWeightSquared += weight * weight;
meanX += x * weight;
rmsX += x * weight * weight;
meanY += y * weight;
rmsY += y * weight * weight;
}
/**
* The precise meaning of the arguments to the public slice methods is
* somewhat ambiguous, so we define this internal slice method and clearly
* specify its arguments.
*
* Note 0indexY1 and indexY2 use our INTERNAL bin numbering scheme
* Note 1The slice is done between indexY1 and indexY2 INCLUSIVE
* Note 2indexY1 and indexY2 may include the use of under and over
* flow bins Note 3There is no note 3 (yet)
*/
protected FloatIHistogram1D internalSliceX(String title, int indexY1, int indexY2) {
// Attention: our internal definition of bins has been choosen
// so that this works properly even if the indeces passed in include
// the underflow or overflow bins
if (indexY2 < indexY1)
throw new IllegalArgumentException("Invalid bin range");
int sliceBins = xAxis.bins() + 2;
int[] sliceEntries = new int[sliceBins];
float[] sliceHeights = new float[sliceBins];
float[] sliceErrors = new float[sliceBins];
// for (int i=xAxis.under; i<=xAxis.over; i++)
for (int i = 0; i < sliceBins; i++) {
for (int j = indexY1; j <= indexY2; j++) {
sliceEntries[i] += entries[i][j];
sliceHeights[i] += heights[i][j];
sliceErrors[i] += errors[i][j];
}
}
FloatHistogram1D result = new FloatHistogram1D(title, xAxis);
result.setContents(sliceEntries, sliceHeights, sliceErrors);
return result;
}
/**
* The precise meaning of the arguments to the public slice methods is
* somewhat ambiguous, so we define this internal slice method and clearly
* specify its arguments.
*
* Note 0indexX1 and indexX2 use our INTERNAL bin numbering scheme
* Note 1The slice is done between indexX1 and indexX2 INCLUSIVE
* Note 2indexX1 and indexX2 may include the use of under and over
* flow bins Note 3There is no note 3 (yet)
*/
protected FloatIHistogram1D internalSliceY(String title, int indexX1, int indexX2) {
// Attention: our internal definition of bins has been choosen
// so that this works properly even if the indeces passed in include
// the underflow or overflow bins
if (indexX2 < indexX1)
throw new IllegalArgumentException("Invalid bin range");
int sliceBins = yAxis.bins() + 2;
int[] sliceEntries = new int[sliceBins];
float[] sliceHeights = new float[sliceBins];
float[] sliceErrors = new float[sliceBins];
for (int i = indexX1; i <= indexX2; i++) {
// for (int j=yAxis.under; j<=yAxis.over; j++)
for (int j = 0; j < sliceBins; j++) {
sliceEntries[j] += entries[i][j];
sliceHeights[j] += heights[i][j];
sliceErrors[j] += errors[i][j];
}
}
FloatHistogram1D result = new FloatHistogram1D(title, yAxis);
result.setContents(sliceEntries, sliceHeights, sliceErrors);
return result;
}
public float meanX() {
return meanX / sumWeight;
}
public float meanY() {
return meanY / sumWeight;
}
public void reset() {
for (int i = 0; i < entries.length; i++)
for (int j = 0; j < entries[0].length; j++) {
entries[i][j] = 0;
heights[i][j] = 0;
errors[i][j] = 0;
}
nEntry = 0;
sumWeight = 0;
sumWeightSquared = 0;
meanX = 0;
rmsX = 0;
meanY = 0;
rmsY = 0;
}
public float rmsX() {
return (float) Math.sqrt(rmsX / sumWeight - meanX * meanX / sumWeight / sumWeight);
}
public float rmsY() {
return (float) Math.sqrt(rmsY / sumWeight - meanY * meanY / sumWeight / sumWeight);
}
/**
* Used internally for creating slices and projections
*/
void setContents(int[][] entries, float[][] heights, float[][] errors) {
this.entries = entries;
this.heights = heights;
this.errors = errors;
for (int i = 0; i < entries.length; i++)
for (int j = 0; j < entries[0].length; j++) {
nEntry += entries[i][j];
sumWeight += heights[i][j];
}
// TODO: Can we do anything sensible/useful with the other statistics?
sumWeightSquared = Float.NaN;
meanX = Float.NaN;
rmsX = Float.NaN;
meanY = Float.NaN;
rmsY = Float.NaN;
}
public float sumAllBinHeights() {
return sumWeight;
}
}