/*
* #%L
* data-exporter
* %%
* Copyright (C) 2012 - 2013 http://www.brsanthu.com
* %%
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* #L%
*/
package com.brsanthu.dataexporter.output.tree;
import java.io.OutputStream;
import java.io.Writer;
import java.util.List;
import com.brsanthu.dataexporter.AbstractDataWriter;
import com.brsanthu.dataexporter.model.Row;
import com.brsanthu.dataexporter.model.RowDetails;
public class TreeWriter extends AbstractDataWriter {
public TreeWriter(TreeExportOptions options) {
this(options, System.out);
}
public TreeWriter(OutputStream out) {
this(new TreeExportOptions(), out);
}
public TreeWriter(Writer out) {
this(new TreeExportOptions(), out);
}
public TreeWriter(TreeExportOptions options, OutputStream out) {
super(options, out);
}
public TreeWriter(TreeExportOptions options, Writer out) {
super(options, out);
}
public TreeExportOptions getTreeExportOptions() {
return (TreeExportOptions) getOptions();
}
@Override
public void writeRow(RowDetails rowDetails) {
Row row = rowDetails.getRow();
printTreeNode("", true, true, true, row);
}
private void printTreeNode(String prefix, boolean isRoot, boolean isHead, boolean isTail, Row row) {
String nodePrefix = null;
String nodeSuffix = "";
if (isRoot) {
nodePrefix = getStyle().getRootNodePrefix();
} else if (row.getChildren() != null && !row.getChildren().isEmpty() && getStyle().getParentNodePrefix() != null) {
nodePrefix = getStyle().getParentNodePrefix();
} else if (isHead) {
nodePrefix = getStyle().getFirstNodePrefix();
} else if (isTail) {
nodePrefix = getStyle().getLastNodePrefix();
} else {
nodePrefix = getStyle().getOtherNodesPrefix();
}
if (row.getChildren() != null && !row.getChildren().isEmpty() && getStyle().getParentNodeSuffix() != null) {
nodeSuffix = getStyle().getParentNodeSuffix();
}
println(prefix + nodePrefix + String.valueOf(row.getCellValue(0)) + nodeSuffix);
List<Row> children = row.getChildren();
if (children != null && !children.isEmpty()) {
for (int i = 0; i < children.size() - 1; i++) {
Row child = children.get(i);
printTreeNode(prefix + (isTail ? getStyle().getLastLevelSeparator() : getStyle().getLevelSeparator()), false, i==0, false, child);
}
if (children.size() >= 1) {
printTreeNode(prefix + (isTail ? getStyle().getLastLevelSeparator() : getStyle().getLevelSeparator()), false, false, true, children.get(children.size() - 1));
}
}
}
private TreeExportStyle getStyle() {
return getTreeExportOptions().getStyle();
}
}