001 /* 002 * Copyright (C) 2008-2010 by Holger Arndt 003 * 004 * This file is part of the Universal Java Matrix Package (UJMP). 005 * See the NOTICE file distributed with this work for additional 006 * information regarding copyright ownership and licensing. 007 * 008 * UJMP is free software; you can redistribute it and/or modify 009 * it under the terms of the GNU Lesser General Public License as 010 * published by the Free Software Foundation; either version 2 011 * of the License, or (at your option) any later version. 012 * 013 * UJMP is distributed in the hope that it will be useful, 014 * but WITHOUT ANY WARRANTY; without even the implied warranty of 015 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 016 * GNU Lesser General Public License for more details. 017 * 018 * You should have received a copy of the GNU Lesser General Public 019 * License along with UJMP; if not, write to the 020 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, 021 * Boston, MA 02110-1301 USA 022 */ 023 024 package org.ujmp.core.io; 025 026 import java.io.File; 027 import java.io.IOException; 028 import java.io.OutputStream; 029 import java.io.OutputStreamWriter; 030 import java.io.Writer; 031 032 import org.ujmp.core.Matrix; 033 import org.ujmp.core.exceptions.MatrixException; 034 import org.ujmp.core.util.UJMPFormat; 035 import org.ujmp.core.util.io.IntelligentFileWriter; 036 037 public class ExportMatrixTEX { 038 039 public static final void toFile(File file, Matrix m, Object... parameters) throws IOException { 040 IntelligentFileWriter w = new IntelligentFileWriter(file); 041 toWriter(w, m, parameters); 042 w.close(); 043 } 044 045 public static final void toStream(OutputStream out, Matrix m, Object... parameters) 046 throws IOException { 047 OutputStreamWriter w = new OutputStreamWriter(out); 048 toWriter(w, m, parameters); 049 w.close(); 050 } 051 052 public static void toWriter(Writer w, Matrix m, Object... parameters) throws IOException, 053 MatrixException { 054 String EOL = System.getProperty("line.separator"); 055 056 long rowCount = m.getRowCount(); 057 long colCount = m.getColumnCount(); 058 059 w.write("\\begin{table}[!ht]" + EOL); 060 w.write("\\centering" + EOL); 061 062 if (m.getMatrixAnnotation() != null) { 063 w.write("\\caption{" 064 + UJMPFormat.getSingleLineInstance().format(m.getMatrixAnnotation()) + "}" 065 + EOL); 066 } 067 068 StringBuilder buf = new StringBuilder(); 069 for (long i = m.getColumnCount() - 1; i != -1; i--) { 070 buf.append('c'); 071 } 072 String alignment = buf.toString(); 073 074 w.write("\\begin{tabular}{" + alignment + "}" + EOL); 075 w.write("\\toprule" + EOL); 076 077 for (int row = 0; row < rowCount; row++) { 078 for (int col = 0; col < colCount; col++) { 079 w.write(UJMPFormat.getSingleLineInstance().format(m.getAsObject(row, col))); 080 081 if (col < colCount - 1) { 082 w.write(" & "); 083 } 084 } 085 w.write(" \\\\" + EOL); 086 } 087 088 w.write("\\bottomrule" + EOL); 089 w.write("\\end{tabular}" + EOL); 090 w.write("\\end{table}" + EOL); 091 } 092 }