org.dflib.row.ValueHolderRowProxy Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of dflib Show documentation
Show all versions of dflib Show documentation
dflib: DataFrame library
package org.dflib.row;
import org.dflib.Extractor;
import org.dflib.Index;
import org.dflib.builder.ValueHolder;
/**
* A {@link RowProxy} with a resettable internal buffer to store a single row.
*
* @since 0.16
*/
public class ValueHolderRowProxy implements RowProxy {
private final Index index;
private final ValueExtractor[] row;
public ValueHolderRowProxy(Index index, Extractor[] extractors) {
this.index = index;
this.row = createRow(extractors);
}
private ValueExtractor[] createRow(Extractor[] extractors) {
int w = extractors.length;
ValueExtractor[] row = new ValueExtractor[w];
for (int i = 0; i < w; i++) {
row[i] = new ValueExtractor<>(extractors[i]);
}
return row;
}
public void reset(S source) {
int w = row.length;
for (int i = 0; i < w; i++) {
row[i].reset(source);
}
}
@Override
public Index getIndex() {
return index;
}
// TODO: ValueHolder should have primitive-aware getters (kinda like ValueStore primitive-aware "push)
@Override
public Object get(int columnPos) {
return row[columnPos].holder.get();
}
@Override
public Object get(String columnName) {
return row[index.position(columnName)].holder.get();
}
@Override
public int getInt(int columnPos) {
// TODO: this code is unaware of primitive columns, and hence will do extra boxing/unboxing
return (int) row[columnPos].holder.get();
}
@Override
public int getInt(String columnName) {
return getInt(index.position(columnName));
}
@Override
public long getLong(int columnPos) {
// TODO: this code is unaware of primitive columns, and hence will do extra boxing/unboxing
return (long) row[columnPos].holder.get();
}
@Override
public long getLong(String columnName) {
return getLong(index.position(columnName));
}
@Override
public double getDouble(int columnPos) {
// TODO: this code is unaware of primitive columns, and hence will do extra boxing/unboxing
return (double) row[columnPos].holder.get();
}
@Override
public double getDouble(String columnName) {
return getDouble(index.position(columnName));
}
@Override
public boolean getBool(int columnPos) {
// TODO: this code is unaware of primitive columns, and hence will do extra boxing/unboxing
return (boolean) row[columnPos].holder.get();
}
@Override
public boolean getBool(String columnName) {
return getBool(index.position(columnName));
}
@Override
public void copyRange(RowBuilder to, int fromOffset, int toOffset, int len) {
// TODO: do we need this one for any reason?
throw new UnsupportedOperationException("Not supported");
}
static class ValueExtractor {
private final Extractor extractor;
private final ValueHolder holder;
ValueExtractor(Extractor extractor) {
this.extractor = extractor;
this.holder = extractor.createHolder();
}
void reset(S s) {
extractor.extractAndStore(s, holder);
}
}
}