/**
* writes String value of enum to DataOutput.
*
* @param out Dataoutput stream
* @param enumVal enum value
* @throws IOException
*/
public static void writeEnum(DataOutput out, Enum<?> enumVal) throws IOException {
Text.writeString(out, enumVal.name());
}
@Override
public Collection<Writable> next() {
Text t = (Text) super.next().iterator().next();
String val = new String(t.getBytes());
Collection<Writable> ret = new ArrayList<>();
StringTokenizer tok;
int index, max;
String col;
double value;
// actual data
try {
// determine max index
max = 0;
tok = new StringTokenizer(val, " \t");
tok.nextToken(); // skip class
while (tok.hasMoreTokens()) {
col = tok.nextToken();
// finished?
if (col.startsWith("#")) break;
// qid is not supported
if (col.startsWith("qid:")) continue;
// actual value
index = Integer.parseInt(col.substring(0, col.indexOf(":")));
if (index > max) max = index;
}
// read values into array
tok = new StringTokenizer(val, " \t");
// 1. class
double classVal = Double.parseDouble(tok.nextToken());
int numRecordsAdded = 0;
// 2. attributes
while (tok.hasMoreTokens()) {
col = tok.nextToken();
// finished?
if (col.startsWith("#")) break;
// qid is not supported
if (col.startsWith("qid:")) continue;
// actual value
index = Integer.parseInt(col.substring(0, col.indexOf(":")));
if (index > numRecordsAdded) {
int totalDiff = Math.abs(numRecordsAdded - index);
for (int i = numRecordsAdded; i < index; i++) {
ret.add(new DoubleWritable(0.0));
}
numRecordsAdded += totalDiff;
}
value = Double.parseDouble(col.substring(col.indexOf(":") + 1));
ret.add(new DoubleWritable(value));
numRecordsAdded++;
}
if (numAttributes >= 1 && ret.size() < numAttributes) {
int totalDiff = Math.abs(ret.size() - numAttributes);
for (int i = 0; i < totalDiff; i++) {
ret.add(new DoubleWritable(0.0));
}
}
ret.add(new DoubleWritable(classVal));
} catch (Exception e) {
log.error("Error parsing line '" + val + "': ", e);
}
return ret;
}
/**
* Read an Enum value from DataInput, Enums are read and written using String values.
*
* @param <T> Enum type
* @param in DataInput to read from
* @param enumType Class type of Enum
* @return Enum represented by String read from DataInput
* @throws IOException
*/
public static <T extends Enum<T>> T readEnum(DataInput in, Class<T> enumType) throws IOException {
return T.valueOf(enumType, Text.readString(in));
}
