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 */
054 package com.jcoverage.ant;
055
056 import com.jcoverage.coverage.Version;
057
058 import java.io.File;
059 import java.io.FileInputStream;
060 import java.io.FileOutputStream;
061 import java.io.InputStream;
062 import java.io.IOException;
063 import java.io.OutputStream;
064
065 import java.util.Arrays;
066 import java.util.HashSet;
067 import java.util.Iterator;
068 import java.util.LinkedList;
069 import java.util.List;
070 import java.util.Set;
071
072 import org.apache.tools.ant.AntClassLoader;
073 import org.apache.tools.ant.BuildException;
074 import org.apache.tools.ant.Project;
075 import org.apache.tools.ant.taskdefs.Copy;
076 import org.apache.tools.ant.taskdefs.Java;
077 import org.apache.tools.ant.taskdefs.MatchingTask;
078 import org.apache.tools.ant.types.FileSet;
079 import org.apache.tools.ant.types.Path;
080 import org.apache.tools.ant.types.Reference;
081 import org.apache.tools.ant.util.IdentityMapper;
082 import org.apache.tools.ant.util.SourceFileScanner;
083
084 public 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 }