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    }