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:
   *    Marc R. Hoffmann - initial API and implementation
   *    
   *******************************************************************************/
  package org.jacoco.ant;
  
  import java.io.File;
  import java.io.IOException;
  import java.io.InputStream;
  import java.util.HashMap;
  import java.util.Map;
  
  import org.apache.tools.ant.types.Resource;
  import org.jacoco.report.InputStreamSourceFileLocator;
  
  /**
   * Source locator based on Ant file resources.
   */
  class AntFilesLocator extends InputStreamSourceFileLocator {
  
  	private final Map<String, Resource> resources;
  
  	public AntFilesLocator(final String encoding,
  			final int tabWidth) {
  		super(encoding, tabWidth);
  		this.resources = new HashMap<String, Resource>();
  	}
  
  	/**
  	 * Adds the given file resource as a potential source file.
  	 * 
  	 * @param file
  	 *            file resource to add
  	 */
  	void add(final Resource file) {
  		resources.put(file.getName().replace(File.separatorChar, '/'), file);
  	}
  
  	@Override
  	protected InputStream getSourceStream(final String path) throws IOException {
  		final Resource file = resources.get(path);
  		if (file == null) {
  			return null;
  		} else {
  			return file.getInputStream();
  		}
  	}
  
  }
  