j2js-compiler/src/main/java/com/j2js/assembly/ProcedureUnit.java

package com.j2js.assembly;

import java.io.IOException;
import java.io.Writer;
import java.util.Collection;
import java.util.HashSet;
import java.util.Iterator;

import com.j2js.Log;

public abstract class ProcedureUnit extends MemberUnit {
    
    // Set of all member signatures targeted by this method.
    private Collection<Signature> targetSignatures = new HashSet<Signature>();

    public ProcedureUnit(Signature theSignature, ClassUnit theDeclaringClazz) {
        super(theSignature, theDeclaringClazz);
    }
    
    public void addTarget(Signature targetSignature) {
        if (!targetSignature.toString().contains("#")) {
            throw new IllegalArgumentException("Signature must be field or method: " + targetSignature);
        }
        //Logger.getLogger().info("Adding " + this + " -> " + targetSignature);
        targetSignatures.add(targetSignature);
    }
    
    public void removeTargets() {
      Iterator iter = targetSignatures.iterator();
      while (iter.hasNext()) {
          iter.next();
          iter.remove();
      }
    }
    
    void write(int depth, Writer writer) throws IOException {
        if (getData() == null) return;
        Log.getLogger().debug(getIndent(depth) + getSignature());
        writer.write(getData());
    }
    
    public String getData() {
        if (!declaringClass.isResolved()) throw new RuntimeException("Class must be resolved");
        return super.getData();
    }

    public Collection<Signature> getTargetSignatures() {
        return targetSignatures;
    }
    
}