/**
* Set the definitions of the parameters in the compiled function, as an array.
*
* @param fn the compiled object representing the user-written function
*/
public void setParameterDefinitions(UserFunction fn) {
UserFunctionParameter[] params = new UserFunctionParameter[getNumberOfArguments()];
fn.setParameterDefinitions(params);
int count = 0;
AxisIterator kids = iterateAxis(Axis.CHILD);
while (true) {
NodeInfo node = kids.next();
if (node == null) {
return;
}
if (node instanceof XSLParam) {
UserFunctionParameter param = new UserFunctionParameter();
params[count++] = param;
param.setRequiredType(((XSLParam) node).getRequiredType());
param.setVariableQName(((XSLParam) node).getVariableQName());
param.setSlotNumber(((XSLParam) node).getSlotNumber());
((XSLParam) node).fixupBinding(param);
int refs = ExpressionTool.getReferenceCount(fn.getBody(), param, false);
param.setReferenceCount(refs);
}
}
}
public void checkArguments(StaticContext env) throws XPathException {
if (checked) return;
checked = true;
super.checkArguments(env);
Optimizer opt = env.getConfiguration().getOptimizer();
argument[1] = ExpressionTool.unsorted(opt, argument[1], false);
if (argument[0] instanceof StringValue) {
// common case, key name is supplied as a constant
try {
keyFingerprint =
((ExpressionContext) env)
.getFingerprint(((StringValue) argument[0]).getStringValue(), false);
} catch (XPathException e) {
StaticError err =
new StaticError(
"Error in key name "
+ ((StringValue) argument[0]).getStringValue()
+ ": "
+ e.getMessage());
err.setLocator(this);
err.setErrorCode("XTDE1260");
throw err;
}
if (keyFingerprint == -1) {
StaticError err =
new StaticError(
"Key " + ((StringValue) argument[0]).getStringValue() + " has not been defined");
err.setLocator(this);
err.setErrorCode("XTDE1260");
throw err;
}
} else {
// we need to save the namespace context
nsContext = env.getNamespaceResolver();
}
}
public void optimize(Declaration declaration) throws XPathException {
Expression exp = compiledFunction.getBody();
ExpressionVisitor visitor = makeExpressionVisitor();
Expression exp2 = exp;
Optimizer opt = getConfiguration().obtainOptimizer();
try {
if (opt.getOptimizationLevel() != Optimizer.NO_OPTIMIZATION) {
exp2 = exp.optimize(visitor, null);
}
} catch (XPathException err) {
err.maybeSetLocation(this);
compileError(err);
}
// Try to extract new global variables from the body of the function
if (opt.getOptimizationLevel() != Optimizer.NO_OPTIMIZATION) {
Expression exp3 = opt.promoteExpressionsToGlobal(exp2, visitor);
if (exp3 != null) {
exp2 = visitor.optimize(exp3, null);
}
}
// Add trace wrapper code if required
exp2 = makeTraceInstruction(this, exp2);
allocateSlots(exp2);
if (exp2 != exp) {
compiledFunction.setBody(exp2);
}
int tailCalls =
ExpressionTool.markTailFunctionCalls(exp2, getObjectName(), getNumberOfArguments());
if (tailCalls != 0) {
compiledFunction.setTailRecursive(tailCalls > 0, tailCalls > 1);
compiledFunction.setBody(new TailCallLoop(compiledFunction));
}
// Generate byte code if appropriate
if (getConfiguration().isGenerateByteCode(Configuration.XSLT)) {
try {
Expression cbody =
opt.compileToByteCode(
compiledFunction.getBody(),
nameAtt,
Expression.PROCESS_METHOD | Expression.ITERATE_METHOD);
if (cbody != null) {
compiledFunction.setBody(cbody);
}
} catch (Exception e) {
System.err.println("Failed while compiling function " + nameAtt);
e.printStackTrace();
throw new XPathException(e);
}
}
compiledFunction.computeEvaluationMode();
if (isExplaining()) {
exp2.explain(getConfiguration().getStandardErrorOutput());
}
}