Method writeTrees

src/figtree/application/JSONTreeExporter.java:89–172

(Collection<? extends Tree> trees)

    }

    private void writeTrees(Collection<? extends Tree> trees) throws IOException {
        int indent = 0;
        int treeCount = 0;

        writer.println("{");

        if (trees.size() > 1) {
            writer.println("\t\"trees\": [");
            indent ++;
        }

        Map<String, Set<String>> attributeMap = new LinkedHashMap<String, Set<String>>();

        for( Tree t : trees ) {
            final boolean isRooted = t instanceof RootedTree;
            final RootedTree rtree = isRooted ? (RootedTree)t : Utils.rootTheTree(t);

            for (Node node : rtree.getNodes()) {
                for (String name : node.getAttributeNames()) {
                    if (ATTRIBUTE_NAMES.contains(name)) {
                        Object valueObject = node.getAttribute(name);
                        if (valueObject instanceof String) {
                            Set<String> values = attributeMap.get(name);
                            if (values == null) {
                                values = new TreeSet<String>();
                                attributeMap.put(name, values);
                            }
                            String value = (String)valueObject;
                            if (value.contains("+")) {
                                values.add(value.split("\\+")[0]);
                            } else {
                                values.add(valueObject.toString());
                            }
                        }
                    }
                }
            }

            StringBuilder builder = new StringBuilder();

            if (trees.size() > 1) {
                appendIndent(builder, indent);
                builder.append("\"tree\": {\n");
            }

            appendIndent(builder, indent + 1);
            builder.append("\"root\": ");
            appendTree(rtree, rtree.getRootNode(), builder, indent + 1);

            appendAttributes(rtree, builder, indent + 1);
            builder.append((trees.size() == 1 || treeCount < trees.size() - 1 ? ",\n" : "\n"));

            if (trees.size() > 1) {
                appendIndent(builder, indent);
                builder.append((treeCount < trees.size() - 1 ? "},\n" : "}\n"));
            }

            writer.println(builder);