content: /*******************************************************************************
   * Copyright (c) 2009, 2019 Mountainminds GmbH & Co. KG and Contributors
   * All rights reserved. This program and the accompanying materials
   * are made available under the terms of the Eclipse Public License v1.0
   * which accompanies this distribution, and is available at
   * http://www.eclipse.org/legal/epl-v10.html
   *
   * Contributors:
   *    Brock Janiczak -initial API and implementation
   *    
   *******************************************************************************/
  package org.jacoco.report.xml;
  
  import java.io.IOException;
  import java.io.OutputStream;
  import java.util.Collection;
  import java.util.List;
  
  import org.jacoco.core.analysis.IBundleCoverage;
  import org.jacoco.core.data.ExecutionData;
  import org.jacoco.core.data.SessionInfo;
  import org.jacoco.report.IReportGroupVisitor;
  import org.jacoco.report.IReportVisitor;
  import org.jacoco.report.ISourceFileLocator;
  import org.jacoco.report.internal.xml.ReportElement;
  import org.jacoco.report.internal.xml.XMLCoverageWriter;
  import org.jacoco.report.internal.xml.XMLGroupVisitor;
  
  /**
   * Report formatter that creates a single XML file for a coverage session
   */
  public class XMLFormatter {
  
  	private String outputEncoding = "UTF-8";
  
  	/**
  	 * Sets the encoding used for generated XML document. Default is UTF-8.
  	 * 
  	 * @param outputEncoding
  	 *            XML output encoding
  	 */
  	public void setOutputEncoding(final String outputEncoding) {
  		this.outputEncoding = outputEncoding;
  	}
  
  	/**
  	 * Creates a new visitor to write a report to the given stream.
  	 * 
  	 * @param output
  	 *            output stream to write the report to
  	 * @return visitor to emit the report data to
  	 * @throws IOException
  	 *             in case of problems with the output stream
  	 */
  	public IReportVisitor createVisitor(final OutputStream output)
  			throws IOException {
  		class RootVisitor implements IReportVisitor {
  
  			private ReportElement report;
  			private List<SessionInfo> sessionInfos;
  			private XMLGroupVisitor groupVisitor;
  
  			public void visitInfo(final List<SessionInfo> sessionInfos,
  					final Collection<ExecutionData> executionData)
  					throws IOException {
  				this.sessionInfos = sessionInfos;
  			}
  
  			public void visitBundle(final IBundleCoverage bundle,
  					final ISourceFileLocator locator) throws IOException {
  				createRootElement(bundle.getName());
  				XMLCoverageWriter.writeBundle(bundle, report);
  			}
  
  			public IReportGroupVisitor visitGroup(final String name)
  					throws IOException {
  				createRootElement(name);
  				groupVisitor = new XMLGroupVisitor(report, name);
  				return groupVisitor;
  			}
  
  			private void createRootElement(final String name)
  					throws IOException {
  				report = new ReportElement(name, output, outputEncoding);
  				for (final SessionInfo i : sessionInfos) {
  					report.sessioninfo(i);
  				}
  			}
  
  			public void visitEnd() throws IOException {
  				if (groupVisitor != null) {
  					groupVisitor.visitEnd();
  				}
  				report.close();
  			}
  		}
  		return new RootVisitor();
  	}
  
  }
  