org.codehaus.groovy.control

Class CompilationUnit

java.lang.Object

org.codehaus.groovy.control.ProcessingUnit

org.codehaus.groovy.control.CompilationUnit

Direct Known Subclasses:

JavaAwareCompilationUnit, JavaStubCompilationUnit

public class CompilationUnit
extends ProcessingUnit

The CompilationUnit collects all compilation data as it is generated by the compiler system. You can use this object to add additional source units to the compilation, or force the compilation to be run again (to affect only the deltas).

You can also add PhaseOperations to this compilation using the addPhaseOperation method. This is commonly used when you want to wire a new AST Transformation into the compilation.

Author:

Chris Poirier, Jochen Theodorou, Roshan Dawrani

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	CompilationUnit.ClassgenCallback A callback interface you can use to "accompany" the classgen() code as it traverses the ClassNode tree.
static class	CompilationUnit.GroovyClassOperation
static class	CompilationUnit.PrimaryClassNodeOperation An callback interface for use in the applyToPrimaryClassNodes loop driver.
static class	CompilationUnit.ProgressCallback A callback interface you can use to get a callback after every unit of the compile process.
static class	CompilationUnit.SourceUnitOperation An callback interface for use in the applyToSourceUnits loop driver.

Field Summary

Fields

Modifier and Type	Field and Description
protected CompileUnit	ast
protected ASTTransformationsContext	astTransformationsContext
protected CompilationUnit.ClassgenCallback	classgenCallback
protected ClassNodeResolver	classNodeResolver
protected Map	classSourcesByPublicClassName
protected boolean	configured
protected boolean	debug
protected List<GroovyClass>	generatedClasses
protected List<String>	names
protected OptimizerVisitor	optimizer
protected CompilationUnit.ProgressCallback	progressCallback
protected LinkedList<SourceUnit>	queuedSources
protected ResolveVisitor	resolveVisitor
protected Map<String,SourceUnit>	sources
protected StaticImportVisitor	staticImportVisitor
protected Map	summariesByPublicClassName
protected Map	summariesBySourceName
protected Verifier	verifier

Fields inherited from class org.codehaus.groovy.control.ProcessingUnit

classLoader, configuration, errorCollector, phase, phaseComplete

Constructor Summary

Constructors

Constructor and Description
CompilationUnit() Initializes the CompilationUnit with defaults.
CompilationUnit(CompilerConfiguration configuration) Initializes the CompilationUnit with no security considerations.
CompilationUnit(CompilerConfiguration configuration, CodeSource security, GroovyClassLoader loader) Initializes the CompilationUnit with a CodeSource for controlling security stuff and a class loader for loading classes.
CompilationUnit(CompilerConfiguration configuration, CodeSource security, GroovyClassLoader loader, GroovyClassLoader transformLoader) Initializes the CompilationUnit with a CodeSource for controlling security stuff, a class loader for loading classes, and a class loader for loading AST transformations.
CompilationUnit(GroovyClassLoader loader) Initializes the CompilationUnit with defaults except for class loader.

Method Summary

Modifier and Type	Method and Description
void	addClassNode(ClassNode node) Adds a ClassNode directly to the unit (ie.
void	addFirstPhaseOperation(CompilationUnit.PrimaryClassNodeOperation op, int phase)
void	addNewPhaseOperation(CompilationUnit.SourceUnitOperation op, int phase)
void	addPhaseOperation(CompilationUnit.GroovyClassOperation op)
void	addPhaseOperation(CompilationUnit.PrimaryClassNodeOperation op, int phase)
void	addPhaseOperation(CompilationUnit.SourceUnitOperation op, int phase)
SourceUnit	addSource(File file) Adds a source file to the unit.
SourceUnit	addSource(SourceUnit source) Adds a SourceUnit to the unit.
SourceUnit	addSource(String name, InputStream stream) Adds a InputStream source to the unit.
SourceUnit	addSource(String name, String scriptText)
SourceUnit	addSource(URL url) Adds a source file to the unit.
void	addSources(File[] files) Adds a set of source files to the unit.
void	addSources(String[] paths) Adds a set of file paths to the unit.
void	applyToGeneratedGroovyClasses(CompilationUnit.GroovyClassOperation body)
void	applyToPrimaryClassNodes(CompilationUnit.PrimaryClassNodeOperation body) A loop driver for applying operations to all primary ClassNodes in our AST.
void	applyToSourceUnits(CompilationUnit.SourceUnitOperation body) A loop driver for applying operations to all SourceUnits.
void	compile() Synonym for compile(Phases.ALL).
void	compile(int throughPhase) Compiles the compilation unit from sources.
void	configure(CompilerConfiguration configuration) Configures its debugging mode and classloader classpath from a given compiler configuration.
protected org.objectweb.asm.ClassVisitor	createClassVisitor()
protected boolean	dequeued() Dequeues any source units add through addSource and resets the compiler phase to initialization.
CompileUnit	getAST() Returns the CompileUnit that roots our AST.
ASTTransformationsContext	getASTTransformationsContext()
List	getClasses() Get the GroovyClasses generated by compile().
CompilationUnit.ClassgenCallback	getClassgenCallback()
ClassNode	getClassNode(String name) Convenience routine to get the named ClassNode.
ClassNodeResolver	getClassNodeResolver()
Map	getClassSourcesByPublicClassName()
ClassNode	getFirstClassNode() Convenience routine to get the first ClassNode, for when you are sure there is only one.
CompilationUnit.ProgressCallback	getProgressCallback()
Map	getSummariesByPublicClassName()
Map	getSummariesBySourceName() Get the source summaries
GroovyClassLoader	getTransformLoader() Returns the class loader for loading AST transformations.
boolean	isPublicClass(String className)
Iterator<SourceUnit>	iterator() Returns an iterator on the unit's SourceUnits.
protected void	mark() Updates the phase marker on all sources.
void	setClassgenCallback(CompilationUnit.ClassgenCallback visitor) Sets a ClassgenCallback.
void	setClassNodeResolver(ClassNodeResolver classNodeResolver)
void	setProgressCallback(CompilationUnit.ProgressCallback callback) Sets a ProgressCallback.

Methods inherited from class org.codehaus.groovy.control.ProcessingUnit

completePhase, getClassLoader, getConfiguration, getErrorCollector, getPhase, getPhaseDescription, gotoPhase, nextPhase, setClassLoader, setConfiguration

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

astTransformationsContext

protected ASTTransformationsContext astTransformationsContext

sources

protected Map<String,SourceUnit> sources

summariesBySourceName

protected Map summariesBySourceName

summariesByPublicClassName

protected Map summariesByPublicClassName

classSourcesByPublicClassName

protected Map classSourcesByPublicClassName

names

protected List<String> names

queuedSources

protected LinkedList<SourceUnit> queuedSources

ast

protected CompileUnit ast

generatedClasses

protected List<GroovyClass> generatedClasses

verifier

protected Verifier verifier

debug

protected boolean debug

configured

protected boolean configured

classgenCallback

protected CompilationUnit.ClassgenCallback classgenCallback

progressCallback

protected CompilationUnit.ProgressCallback progressCallback

resolveVisitor

protected ResolveVisitor resolveVisitor

staticImportVisitor

protected StaticImportVisitor staticImportVisitor

optimizer

protected OptimizerVisitor optimizer

classNodeResolver

protected ClassNodeResolver classNodeResolver

Constructor Detail

CompilationUnit

public CompilationUnit()

Initializes the CompilationUnit with defaults.

CompilationUnit

public CompilationUnit(GroovyClassLoader loader)

Initializes the CompilationUnit with defaults except for class loader.

CompilationUnit

public CompilationUnit(CompilerConfiguration configuration)

Initializes the CompilationUnit with no security considerations.

CompilationUnit

public CompilationUnit(CompilerConfiguration configuration,
                       CodeSource security,
                       GroovyClassLoader loader)

Initializes the CompilationUnit with a CodeSource for controlling security stuff and a class loader for loading classes.

CompilationUnit

public CompilationUnit(CompilerConfiguration configuration,
                       CodeSource security,
                       GroovyClassLoader loader,
                       GroovyClassLoader transformLoader)

Initializes the CompilationUnit with a CodeSource for controlling security stuff, a class loader for loading classes, and a class loader for loading AST transformations. Note The transform loader must be able to load compiler classes. That means CompilationUnit.class.classLoader must be at last a parent to transformLoader. The other loader has no such constraint.

Parameters:

transformLoader - - the loader for transforms

loader - - loader used to resolve classes against during compilation

security - - security setting for the compilation

configuration - - compilation configuration

Method Detail

getTransformLoader

public GroovyClassLoader getTransformLoader()

Returns the class loader for loading AST transformations.

Returns:

- the transform class loader

addPhaseOperation

public void addPhaseOperation(CompilationUnit.SourceUnitOperation op,
                              int phase)

addPhaseOperation

public void addPhaseOperation(CompilationUnit.PrimaryClassNodeOperation op,
                              int phase)

addFirstPhaseOperation

public void addFirstPhaseOperation(CompilationUnit.PrimaryClassNodeOperation op,
                                   int phase)

addPhaseOperation

public void addPhaseOperation(CompilationUnit.GroovyClassOperation op)

addNewPhaseOperation

public void addNewPhaseOperation(CompilationUnit.SourceUnitOperation op,
                                 int phase)

configure

public void configure(CompilerConfiguration configuration)

Configures its debugging mode and classloader classpath from a given compiler configuration. This cannot be done more than once due to limitations in URLClassLoader.

Overrides:

configure in class ProcessingUnit

getAST

public CompileUnit getAST()

Returns the CompileUnit that roots our AST.

getSummariesBySourceName

public Map getSummariesBySourceName()

Get the source summaries

getSummariesByPublicClassName

public Map getSummariesByPublicClassName()

getClassSourcesByPublicClassName

public Map getClassSourcesByPublicClassName()

isPublicClass

public boolean isPublicClass(String className)

getClasses

public List getClasses()

Get the GroovyClasses generated by compile().

getFirstClassNode

public ClassNode getFirstClassNode()

Convenience routine to get the first ClassNode, for when you are sure there is only one.

getClassNode

public ClassNode getClassNode(String name)

Convenience routine to get the named ClassNode.

getASTTransformationsContext

public ASTTransformationsContext getASTTransformationsContext()

Returns:

the AST transformations current context

addSources

public void addSources(String[] paths)

Adds a set of file paths to the unit.

addSources

public void addSources(File[] files)

Adds a set of source files to the unit.

addSource

public SourceUnit addSource(File file)

Adds a source file to the unit.

addSource

public SourceUnit addSource(URL url)

Adds a source file to the unit.

addSource

public SourceUnit addSource(String name,
                            InputStream stream)

Adds a InputStream source to the unit.

addSource

public SourceUnit addSource(String name,
                            String scriptText)

addSource

public SourceUnit addSource(SourceUnit source)

Adds a SourceUnit to the unit.

iterator

public Iterator<SourceUnit> iterator()

Returns an iterator on the unit's SourceUnits.

addClassNode

public void addClassNode(ClassNode node)

Adds a ClassNode directly to the unit (ie. without source). WARNING: the source is needed for error reporting, using this method without setting a SourceUnit will cause NullPinterExceptions

setClassgenCallback

public void setClassgenCallback(CompilationUnit.ClassgenCallback visitor)

Sets a ClassgenCallback. You can have only one, and setting it to null removes any existing setting.

setProgressCallback

public void setProgressCallback(CompilationUnit.ProgressCallback callback)

Sets a ProgressCallback. You can have only one, and setting it to null removes any existing setting.

getClassgenCallback

public CompilationUnit.ClassgenCallback getClassgenCallback()

getProgressCallback

public CompilationUnit.ProgressCallback getProgressCallback()

compile

public void compile()
             throws CompilationFailedException

Synonym for compile(Phases.ALL).

Throws:

CompilationFailedException

compile

public void compile(int throughPhase)
             throws CompilationFailedException

Compiles the compilation unit from sources.

Throws:

CompilationFailedException

dequeued

protected boolean dequeued()
                    throws CompilationFailedException

Dequeues any source units add through addSource and resets the compiler phase to initialization.

Note: this does not mean a file is recompiled. If a SourceUnit has already passed a phase it is skipped until a higher phase is reached.

Returns:

true if there was a queued source

Throws:

CompilationFailedException

createClassVisitor

protected org.objectweb.asm.ClassVisitor createClassVisitor()

mark

protected void mark()
             throws CompilationFailedException

Updates the phase marker on all sources.

Throws:

CompilationFailedException

applyToSourceUnits

public void applyToSourceUnits(CompilationUnit.SourceUnitOperation body)
                        throws CompilationFailedException

A loop driver for applying operations to all SourceUnits. Automatically skips units that have already been processed through the current phase.

Throws:

CompilationFailedException

applyToPrimaryClassNodes

public void applyToPrimaryClassNodes(CompilationUnit.PrimaryClassNodeOperation body)
                              throws CompilationFailedException

A loop driver for applying operations to all primary ClassNodes in our AST. Automatically skips units that have already been processed through the current phase.

Throws:

CompilationFailedException

applyToGeneratedGroovyClasses

public void applyToGeneratedGroovyClasses(CompilationUnit.GroovyClassOperation body)
                                   throws CompilationFailedException

Throws:

CompilationFailedException

getClassNodeResolver

public ClassNodeResolver getClassNodeResolver()

setClassNodeResolver

public void setClassNodeResolver(ClassNodeResolver classNodeResolver)