com.xxdb.data.BasicDoubleMatrix Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of api-java Show documentation
Show all versions of api-java Show documentation
The messaging and data conversion protocol between Java and DolphinDB server
package com.xxdb.data;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.ByteOrder;
import java.util.List;
import com.xxdb.io.ExtendedDataInput;
import com.xxdb.io.ExtendedDataOutput;
/**
*
* Corresponds to DolphinDB double matrix
*
*/
public class BasicDoubleMatrix extends AbstractMatrix{
private double[] values;
public BasicDoubleMatrix(int rows, int columns){
super(rows, columns);
values = new double[rows * columns];
}
public BasicDoubleMatrix(int rows, int columns, List listOfArrays) throws Exception {
super(rows,columns);
values = new double[rows*columns];
if (listOfArrays == null || listOfArrays.size() != columns)
throw new Exception("input list of arrays does not have " + columns + " columns");
for (int i=0; i getElementClass(){
return BasicDouble.class;
}
@Override
protected void readMatrixFromInputStream(int rows, int columns, ExtendedDataInput in) throws IOException{
int size = rows * columns;
values =new double[size];
int totalBytes = size * 8, off = 0;
ByteOrder bo = in.isLittleEndian() ? ByteOrder.LITTLE_ENDIAN : ByteOrder.BIG_ENDIAN;
while (off < totalBytes) {
int len = Math.min(BUF_SIZE, totalBytes - off);
in.readFully(buf, 0, len);
int start = off / 8, end = len / 8;
ByteBuffer byteBuffer = ByteBuffer.wrap(buf, 0, len).order(bo);
for (int i = 0; i < end; i++)
values[i + start] = byteBuffer.getDouble(i * 8);
off += len;
}
}
protected void writeVectorToOutputStream(ExtendedDataOutput out) throws IOException{
for(double value : values)
out.writeDouble(value);
}
}