001/*
002 * The Apache Software License, Version 1.1
003 *
004 * Copyright (c) 2000-2002 The Apache Software Foundation.  All rights
005 * reserved.
006 *
007 * Redistribution and use in source and binary forms, with or without
008 * modification, are permitted provided that the following conditions
009 * are met:
010 *
011 * 1. Redistributions of source code must retain the above copyright
012 *    notice, this list of conditions and the following disclaimer.
013 *
014 * 2. Redistributions in binary form must reproduce the above copyright
015 *    notice, this list of conditions and the following disclaimer in
016 *    the documentation and/or other materials provided with the
017 *    distribution.
018 *
019 * 3. The end-user documentation included with the redistribution, if
020 *    any, must include the following acknowlegement:
021 *       "This product includes software developed by the
022 *        Apache Software Foundation (http://www.apache.org/)."
023 *    Alternately, this acknowlegement may appear in the software itself,
024 *    if and wherever such third-party acknowlegements normally appear.
025 *
026 * 4. The names "Ant" and "Apache Software
027 *    Foundation" must not be used to endorse or promote products derived
028 *    from this software without prior written permission. For written
029 *    permission, please contact apache@apache.org.
030 *
031 * 5. Products derived from this software may not be called "Apache"
032 *    nor may "Apache" appear in their names without prior written
033 *    permission of the Apache Group.
034 *
035 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
036 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
037 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
038 * DISCLAIMED.  IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
039 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
040 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
041 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
042 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
043 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
044 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
045 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
046 * SUCH DAMAGE.
047 * ====================================================================
048 *
049 * This software consists of voluntary contributions made by many
050 * individuals on behalf of the Apache Software Foundation.  For more
051 * information on the Apache Software Foundation, please see
052 * <http://www.apache.org/>.
053 */
054package com.jcoverage.ant;
055
056import com.jcoverage.coverage.Version;
057
058import java.io.File;
059import java.io.FileInputStream;
060import java.io.FileOutputStream;
061import java.io.InputStream;
062import java.io.IOException;
063import java.io.OutputStream;
064
065import java.util.Arrays;
066import java.util.HashSet;
067import java.util.Iterator;
068import java.util.LinkedList;
069import java.util.List;
070import java.util.Set;
071
072import org.apache.tools.ant.AntClassLoader;
073import org.apache.tools.ant.BuildException;
074import org.apache.tools.ant.Project;
075import org.apache.tools.ant.taskdefs.Copy;
076import org.apache.tools.ant.taskdefs.Java;
077import org.apache.tools.ant.taskdefs.MatchingTask;
078import org.apache.tools.ant.types.FileSet;
079import org.apache.tools.ant.types.Path;
080import org.apache.tools.ant.types.Reference;
081import org.apache.tools.ant.util.IdentityMapper;
082import org.apache.tools.ant.util.SourceFileScanner;
083
084public class InstrumentTask extends CommonMatchingTask {
085  Ignore ignoreRegex=null;
086
087  public InstrumentTask() {
088    super("com.jcoverage.coverage.Instrument");
089  }
090
091  public Ignore createIgnore() {
092    ignoreRegex=new Ignore();
093    return ignoreRegex;
094  }
095
096  void coverage(String[] filenames) {
097    if(filenames.length==0) {
098      return;
099    }
100      
101    for(int i=0;i<filenames.length;i++) {
102      getProject().log("Adding "+filenames[i]+" to list",Project.MSG_VERBOSE);
103      getJava().createArg().setValue(filenames[i]);
104    }
105      
106    System.out.println("jcoverage "+Version.VERSION_STRING+" copyright (c)2003 jcoverage ltd. http://jcoverage.com/");
107    System.out.println("jcoverage is licensed under the GNU General Public License");
108    System.out.println("jcoverage comes with ABSOLUTELY NO WARRANTY");
109    System.out.println("instrumenting "+filenames.length+" "+(filenames.length==1?"class":"classes")+(toDir!=null?" to "+toDir:""));
110    
111    if(getJava().executeJava()!=0) {
112      throw new BuildException();
113    }
114  }
115
116  public void execute() throws BuildException {
117    if(toDir!=null) {
118      getJava().createArg().setValue("-d");
119      getJava().createArg().setValue(toDir.toString());
120    }
121
122    if(ignoreRegex!=null) {
123      getJava().createArg().setValue("-ignore");
124      getJava().createArg().setValue(ignoreRegex.getRegex());
125    }
126
127
128    Set filenames=new HashSet();
129    Iterator i=fileSets.iterator();
130
131    while(i.hasNext()) {
132      FileSet fileSet=(FileSet)i.next();
133
134      getJava().createArg().setValue("-basedir");
135      getJava().createArg().setValue(baseDir(fileSet));
136
137      filenames.addAll(Arrays.asList(getFilenames(fileSet)));
138    }
139
140    coverage((String[])filenames.toArray(new String[filenames.size()]));
141  }
142}