Examples with WriterMatrixMarket org.apache.sysml.runtime.io.WriterMatrixMarket used on opensource projects

Example 1 with WriterMatrixMarket

use of org.apache.sysml.runtime.io.WriterMatrixMarket in project incubator-systemml by apache.

the class VariableCPInstruction method writeMMFile.

/**
	 * Helper function to write MM files to HDFS.
	 * 
	 * @param ec execution context
	 * @param fname file name
	 * @throws DMLRuntimeException if DMLRuntimeException occurs
	 */
private void writeMMFile(ExecutionContext ec, String fname) throws DMLRuntimeException {
    MatrixObject mo = ec.getMatrixObject(input1.getName());
    String outFmt = "matrixmarket";
    if (mo.isDirty()) {
        // there exist data computed in CP that is not backed up on HDFS
        // i.e., it is either in-memory or in evicted space
        mo.exportData(fname, outFmt);
    } else {
        OutputInfo oi = ((MatrixFormatMetaData) mo.getMetaData()).getOutputInfo();
        MatrixCharacteristics mc = mo.getMatrixCharacteristics();
        if (oi == OutputInfo.TextCellOutputInfo) {
            try {
                WriterMatrixMarket writer = new WriterMatrixMarket();
                writer.mergeTextcellToMatrixMarket(mo.getFileName(), fname, mc.getRows(), mc.getCols(), mc.getNonZeros());
            } catch (IOException e) {
                throw new DMLRuntimeException(e);
            }
        } else if (oi == OutputInfo.BinaryBlockOutputInfo) {
            mo.exportData(fname, outFmt);
        } else {
            throw new DMLRuntimeException("Unexpected data format (" + OutputInfo.outputInfoToString(oi) + "): can not export into MatrixMarket format.");
        }
    }
}