Search in sources :

Example 1 with FunctionDescrBuilder

use of org.drools.drl.ast.dsl.FunctionDescrBuilder in project drools by kiegroup.

the class PackageDescrBuilderImpl method newFunction.

public FunctionDescrBuilder newFunction() {
    FunctionDescrBuilder function = new FunctionDescrBuilderImpl(this);
    descr.addFunction(initDescr(function));
    return function;
}
Also used : FunctionDescrBuilder(org.drools.drl.ast.dsl.FunctionDescrBuilder)

Example 2 with FunctionDescrBuilder

use of org.drools.drl.ast.dsl.FunctionDescrBuilder in project drools by kiegroup.

the class DRL6Parser method function.

/* ------------------------------------------------------------------------------------------------
     *                         FUNCTION STATEMENT
     * ------------------------------------------------------------------------------------------------ */
/**
 * function := FUNCTION type? ID parameters(typed) chunk_{_}
 *
 * @return
 * @throws org.antlr.runtime.RecognitionException
 */
public FunctionDescr function(PackageDescrBuilder pkg) throws RecognitionException {
    FunctionDescrBuilder function = null;
    try {
        function = helper.start(pkg, FunctionDescrBuilder.class, null);
        // 'function'
        match(input, DRL6Lexer.ID, DroolsSoftKeywords.FUNCTION, null, DroolsEditorType.KEYWORD);
        if (state.failed)
            return null;
        if (input.LA(1) != DRL6Lexer.ID || input.LA(2) != DRL6Lexer.LEFT_PAREN) {
            // type
            String type = type();
            if (state.failed)
                return null;
            if (state.backtracking == 0)
                function.returnType(type);
        }
        // name
        Token id = match(input, DRL6Lexer.ID, null, null, DroolsEditorType.IDENTIFIER);
        if (state.failed)
            return null;
        if (state.backtracking == 0) {
            function.name(id.getText());
            helper.setParaphrasesValue(DroolsParaphraseTypes.FUNCTION, "\"" + id.getText() + "\"");
        }
        // arguments
        parameters(function, true);
        if (state.failed)
            return null;
        // body
        String body = chunk(DRL6Lexer.LEFT_CURLY, DRL6Lexer.RIGHT_CURLY, -1);
        if (state.failed)
            return null;
        if (state.backtracking == 0)
            function.body(body);
    } catch (RecognitionException re) {
        reportError(re);
    } finally {
        helper.end(FunctionDescrBuilder.class, function);
    }
    return (function != null) ? function.getDescr() : null;
}
Also used : Token(org.antlr.runtime.Token) CommonToken(org.antlr.runtime.CommonToken) FunctionDescrBuilder(org.drools.drl.ast.dsl.FunctionDescrBuilder) RecognitionException(org.antlr.runtime.RecognitionException)

Example 3 with FunctionDescrBuilder

use of org.drools.drl.ast.dsl.FunctionDescrBuilder in project drools by kiegroup.

the class ParserHelper method start.

@SuppressWarnings("unchecked")
public <T extends DescrBuilder<?, ?>> T start(DescrBuilder<?, ?> ctxBuilder, Class<T> clazz, String param) {
    if (state.backtracking == 0) {
        if (PackageDescrBuilder.class.isAssignableFrom(clazz)) {
            pushParaphrases(DroolsParaphraseTypes.PACKAGE);
            beginSentence(DroolsSentenceType.PACKAGE);
            setStart(ctxBuilder);
        } else if (ImportDescrBuilder.class.isAssignableFrom(clazz)) {
            ImportDescrBuilder imp;
            if (validateLT(2, DroolsSoftKeywords.FUNCTION) || validateLT(2, DroolsSoftKeywords.STATIC)) {
                imp = ctxBuilder == null ? DescrFactory.newPackage().newFunctionImport() : ((PackageDescrBuilder) ctxBuilder).newFunctionImport();
            } else {
                imp = ctxBuilder == null ? DescrFactory.newPackage().newImport() : ((PackageDescrBuilder) ctxBuilder).newImport();
            }
            pushParaphrases(DroolsParaphraseTypes.IMPORT);
            beginSentence(DroolsSentenceType.IMPORT_STATEMENT);
            setStart(imp);
            return (T) imp;
        } else if (UnitDescrBuilder.class.isAssignableFrom(clazz)) {
            UnitDescrBuilder imp = ctxBuilder == null ? DescrFactory.newPackage().newUnit() : ((PackageDescrBuilder) ctxBuilder).newUnit();
            pushParaphrases(DroolsParaphraseTypes.UNIT);
            beginSentence(DroolsSentenceType.UNIT);
            setStart(imp);
            return (T) imp;
        } else if (AccumulateImportDescrBuilder.class.isAssignableFrom(clazz)) {
            AccumulateImportDescrBuilder imp = ctxBuilder == null ? DescrFactory.newPackage().newAccumulateImport() : ((PackageDescrBuilder) ctxBuilder).newAccumulateImport();
            pushParaphrases(DroolsParaphraseTypes.ACCUMULATE_IMPORT);
            beginSentence(DroolsSentenceType.ACCUMULATE_IMPORT_STATEMENT);
            setStart(imp);
            return (T) imp;
        } else if (GlobalDescrBuilder.class.isAssignableFrom(clazz)) {
            GlobalDescrBuilder global = ctxBuilder == null ? DescrFactory.newPackage().newGlobal() : ((PackageDescrBuilder) ctxBuilder).newGlobal();
            pushParaphrases(DroolsParaphraseTypes.GLOBAL);
            beginSentence(DroolsSentenceType.GLOBAL);
            setStart(global);
            return (T) global;
        } else if (DeclareDescrBuilder.class.isAssignableFrom(clazz)) {
            DeclareDescrBuilder declare = ctxBuilder == null ? DescrFactory.newPackage().newDeclare() : ((PackageDescrBuilder) ctxBuilder).newDeclare();
            return (T) declare;
        } else if (TypeDeclarationDescrBuilder.class.isAssignableFrom(clazz)) {
            TypeDeclarationDescrBuilder declare = ctxBuilder == null ? DescrFactory.newPackage().newDeclare().type() : ((DeclareDescrBuilder) ctxBuilder).type();
            pushParaphrases(DroolsParaphraseTypes.TYPE_DECLARE);
            beginSentence(DroolsSentenceType.TYPE_DECLARATION);
            setStart(declare);
            return (T) declare;
        } else if (EnumDeclarationDescrBuilder.class.isAssignableFrom(clazz)) {
            EnumDeclarationDescrBuilder declare = ctxBuilder == null ? DescrFactory.newPackage().newDeclare().enumerative() : ((DeclareDescrBuilder) ctxBuilder).enumerative();
            pushParaphrases(DroolsParaphraseTypes.ENUM_DECLARE);
            beginSentence(DroolsSentenceType.ENUM_DECLARATION);
            setStart(declare);
            return (T) declare;
        } else if (EntryPointDeclarationDescrBuilder.class.isAssignableFrom(clazz)) {
            EntryPointDeclarationDescrBuilder declare = ctxBuilder == null ? DescrFactory.newPackage().newDeclare().entryPoint() : ((DeclareDescrBuilder) ctxBuilder).entryPoint();
            pushParaphrases(DroolsParaphraseTypes.ENTRYPOINT_DECLARE);
            beginSentence(DroolsSentenceType.ENTRYPOINT_DECLARATION);
            setStart(declare);
            return (T) declare;
        } else if (WindowDeclarationDescrBuilder.class.isAssignableFrom(clazz)) {
            WindowDeclarationDescrBuilder declare = ctxBuilder == null ? DescrFactory.newPackage().newDeclare().window() : ((DeclareDescrBuilder) ctxBuilder).window();
            pushParaphrases(DroolsParaphraseTypes.WINDOW_DECLARE);
            beginSentence(DroolsSentenceType.WINDOW_DECLARATION);
            setStart(declare);
            return (T) declare;
        } else if (FieldDescrBuilder.class.isAssignableFrom(clazz)) {
            FieldDescrBuilder field = ((AbstractClassTypeDeclarationBuilder) ctxBuilder).newField(param);
            setStart(field);
            return (T) field;
        } else if (EnumLiteralDescrBuilder.class.isAssignableFrom(clazz)) {
            EnumLiteralDescrBuilder literal = ((EnumDeclarationDescrBuilder) ctxBuilder).newEnumLiteral(param);
            setStart(literal);
            return (T) literal;
        } else if (FunctionDescrBuilder.class.isAssignableFrom(clazz)) {
            FunctionDescrBuilder function;
            if (ctxBuilder == null) {
                function = DescrFactory.newPackage().newFunction();
            } else {
                PackageDescrBuilder pkg = (PackageDescrBuilder) ctxBuilder;
                function = pkg.newFunction().namespace(pkg.getDescr().getName());
                AttributeDescr attribute = pkg.getDescr().getAttribute("dialect");
                if (attribute != null) {
                    function.dialect(attribute.getValue());
                }
            }
            pushParaphrases(DroolsParaphraseTypes.FUNCTION);
            beginSentence(DroolsSentenceType.FUNCTION);
            setStart(function);
            return (T) function;
        } else if (RuleDescrBuilder.class.isAssignableFrom(clazz)) {
            RuleDescrBuilder rule = ctxBuilder == null ? DescrFactory.newPackage().newRule() : ((PackageDescrBuilder) ctxBuilder).newRule();
            pushParaphrases(DroolsParaphraseTypes.RULE);
            beginSentence(DroolsSentenceType.RULE);
            setStart(rule);
            return (T) rule;
        } else if (QueryDescrBuilder.class.isAssignableFrom(clazz)) {
            QueryDescrBuilder query = ctxBuilder == null ? DescrFactory.newPackage().newQuery() : ((PackageDescrBuilder) ctxBuilder).newQuery();
            pushParaphrases(DroolsParaphraseTypes.QUERY);
            beginSentence(DroolsSentenceType.QUERY);
            setStart(query);
            return (T) query;
        } else if (AttributeDescrBuilder.class.isAssignableFrom(clazz)) {
            AttributeDescrBuilder<?> attribute = ((AttributeSupportBuilder<?>) ctxBuilder).attribute(param);
            setStart(attribute);
            return (T) attribute;
        } else if (EvalDescrBuilder.class.isAssignableFrom(clazz)) {
            EvalDescrBuilder<?> eval = ((CEDescrBuilder<?, ?>) ctxBuilder).eval();
            pushParaphrases(DroolsParaphraseTypes.EVAL);
            beginSentence(DroolsSentenceType.EVAL);
            setStart(eval);
            return (T) eval;
        } else if (ForallDescrBuilder.class.isAssignableFrom(clazz)) {
            ForallDescrBuilder<?> forall = ((CEDescrBuilder<?, ?>) ctxBuilder).forall();
            setStart(forall);
            return (T) forall;
        } else if (CEDescrBuilder.class.isAssignableFrom(clazz)) {
            setStart(ctxBuilder);
            return (T) ctxBuilder;
        } else if (PatternDescrBuilder.class.isAssignableFrom(clazz)) {
            PatternDescrBuilder<?> pattern = ((PatternContainerDescrBuilder<?, ?>) ctxBuilder).pattern();
            pushParaphrases(DroolsParaphraseTypes.PATTERN);
            setStart(pattern);
            return (T) pattern;
        } else if (CollectDescrBuilder.class.isAssignableFrom(clazz)) {
            CollectDescrBuilder<?> collect = ((PatternDescrBuilder<?>) ctxBuilder).from().collect();
            setStart(collect);
            return (T) collect;
        } else if (AccumulateDescrBuilder.class.isAssignableFrom(clazz)) {
            AccumulateDescrBuilder<?> accumulate = ((PatternDescrBuilder<?>) ctxBuilder).from().accumulate();
            setStart(accumulate);
            return (T) accumulate;
        } else if (BehaviorDescrBuilder.class.isAssignableFrom(clazz)) {
            BehaviorDescrBuilder<?> behavior = ((PatternDescrBuilder<?>) ctxBuilder).behavior();
            setStart(behavior);
            return (T) behavior;
        } else if (NamedConsequenceDescrBuilder.class.isAssignableFrom(clazz)) {
            NamedConsequenceDescrBuilder<?> namedConsequence = ((CEDescrBuilder<?, ?>) ctxBuilder).namedConsequence();
            setStart(namedConsequence);
            return (T) namedConsequence;
        } else if (ConditionalBranchDescrBuilder.class.isAssignableFrom(clazz)) {
            ConditionalBranchDescrBuilder<?> conditionalBranch = ((CEDescrBuilder<?, ?>) ctxBuilder).conditionalBranch();
            setStart(conditionalBranch);
            return (T) conditionalBranch;
        }
    }
    return null;
}
Also used : PackageDescrBuilder(org.drools.drl.ast.dsl.PackageDescrBuilder) DeclareDescrBuilder(org.drools.drl.ast.dsl.DeclareDescrBuilder) AttributeDescrBuilder(org.drools.drl.ast.dsl.AttributeDescrBuilder) ForallDescrBuilder(org.drools.drl.ast.dsl.ForallDescrBuilder) NamedConsequenceDescrBuilder(org.drools.drl.ast.dsl.NamedConsequenceDescrBuilder) AccumulateImportDescrBuilder(org.drools.drl.ast.dsl.AccumulateImportDescrBuilder) ImportDescrBuilder(org.drools.drl.ast.dsl.ImportDescrBuilder) FunctionDescrBuilder(org.drools.drl.ast.dsl.FunctionDescrBuilder) CEDescrBuilder(org.drools.drl.ast.dsl.CEDescrBuilder) FieldDescrBuilder(org.drools.drl.ast.dsl.FieldDescrBuilder) QueryDescrBuilder(org.drools.drl.ast.dsl.QueryDescrBuilder) UnitDescrBuilder(org.drools.drl.ast.dsl.UnitDescrBuilder) GlobalDescrBuilder(org.drools.drl.ast.dsl.GlobalDescrBuilder) EntryPointDeclarationDescrBuilder(org.drools.drl.ast.dsl.EntryPointDeclarationDescrBuilder) WindowDeclarationDescrBuilder(org.drools.drl.ast.dsl.WindowDeclarationDescrBuilder) RuleDescrBuilder(org.drools.drl.ast.dsl.RuleDescrBuilder) AccumulateImportDescrBuilder(org.drools.drl.ast.dsl.AccumulateImportDescrBuilder) PatternDescrBuilder(org.drools.drl.ast.dsl.PatternDescrBuilder) AccumulateDescrBuilder(org.drools.drl.ast.dsl.AccumulateDescrBuilder) EnumDeclarationDescrBuilder(org.drools.drl.ast.dsl.EnumDeclarationDescrBuilder) EnumLiteralDescrBuilder(org.drools.drl.ast.dsl.EnumLiteralDescrBuilder) AbstractClassTypeDeclarationBuilder(org.drools.drl.ast.dsl.AbstractClassTypeDeclarationBuilder) TypeDeclarationDescrBuilder(org.drools.drl.ast.dsl.TypeDeclarationDescrBuilder) AttributeDescr(org.drools.drl.ast.descr.AttributeDescr)

Example 4 with FunctionDescrBuilder

use of org.drools.drl.ast.dsl.FunctionDescrBuilder in project drools by kiegroup.

the class DRL5Parser method function.

/* ------------------------------------------------------------------------------------------------
     *                         FUNCTION STATEMENT
     * ------------------------------------------------------------------------------------------------ */
/**
 * function := FUNCTION type? ID parameters(typed) chunk_{_}
 *
 * @return
 * @throws RecognitionException
 */
public FunctionDescr function(PackageDescrBuilder pkg) throws RecognitionException {
    FunctionDescrBuilder function = null;
    try {
        function = helper.start(pkg, FunctionDescrBuilder.class, null);
        // 'function'
        match(input, DRL5Lexer.ID, DroolsSoftKeywords.FUNCTION, null, DroolsEditorType.KEYWORD);
        if (state.failed)
            return null;
        if (input.LA(1) != DRL5Lexer.ID || input.LA(2) != DRL5Lexer.LEFT_PAREN) {
            // type
            String type = type();
            if (state.failed)
                return null;
            if (state.backtracking == 0)
                function.returnType(type);
        }
        // name
        Token id = match(input, DRL5Lexer.ID, null, null, DroolsEditorType.IDENTIFIER);
        if (state.failed)
            return null;
        if (state.backtracking == 0) {
            function.name(id.getText());
            helper.setParaphrasesValue(DroolsParaphraseTypes.FUNCTION, "\"" + id.getText() + "\"");
        }
        // arguments
        parameters(function, true);
        if (state.failed)
            return null;
        // body
        String body = chunk(DRL5Lexer.LEFT_CURLY, DRL5Lexer.RIGHT_CURLY, -1);
        if (state.failed)
            return null;
        if (state.backtracking == 0)
            function.body(body);
    } catch (RecognitionException re) {
        reportError(re);
    } finally {
        helper.end(FunctionDescrBuilder.class, function);
    }
    return (function != null) ? function.getDescr() : null;
}
Also used : Token(org.antlr.runtime.Token) CommonToken(org.antlr.runtime.CommonToken) FunctionDescrBuilder(org.drools.drl.ast.dsl.FunctionDescrBuilder) RecognitionException(org.antlr.runtime.RecognitionException)

Example 5 with FunctionDescrBuilder

use of org.drools.drl.ast.dsl.FunctionDescrBuilder in project drools by kiegroup.

the class DRL6StrictParser method function.

/* ------------------------------------------------------------------------------------------------
     *                         FUNCTION STATEMENT
     * ------------------------------------------------------------------------------------------------ */
/**
 * function := FUNCTION type? ID parameters(typed) chunk_{_}
 *
 * @return
 * @throws org.antlr.runtime.RecognitionException
 */
public FunctionDescr function(PackageDescrBuilder pkg) throws RecognitionException {
    FunctionDescrBuilder function = null;
    try {
        function = helper.start(pkg, FunctionDescrBuilder.class, null);
        // 'function'
        match(input, DRL6Lexer.ID, DroolsSoftKeywords.FUNCTION, null, DroolsEditorType.KEYWORD);
        if (state.failed)
            return null;
        if (input.LA(1) != DRL6Lexer.ID || input.LA(2) != DRL6Lexer.LEFT_PAREN) {
            // type
            String type = type();
            if (state.failed)
                return null;
            if (state.backtracking == 0)
                function.returnType(type);
        }
        // name
        Token id = match(input, DRL6Lexer.ID, null, null, DroolsEditorType.IDENTIFIER);
        if (state.failed)
            return null;
        if (state.backtracking == 0) {
            function.name(id.getText());
            helper.setParaphrasesValue(DroolsParaphraseTypes.FUNCTION, "\"" + id.getText() + "\"");
        }
        // arguments
        parameters(function, true);
        if (state.failed)
            return null;
        // body
        String body = chunk(DRL6Lexer.LEFT_CURLY, DRL6Lexer.RIGHT_CURLY, -1);
        if (state.failed)
            return null;
        if (state.backtracking == 0)
            function.body(body);
    } catch (RecognitionException re) {
        reportError(re);
    } finally {
        helper.end(FunctionDescrBuilder.class, function);
    }
    return (function != null) ? function.getDescr() : null;
}
Also used : Token(org.antlr.runtime.Token) CommonToken(org.antlr.runtime.CommonToken) FunctionDescrBuilder(org.drools.drl.ast.dsl.FunctionDescrBuilder) RecognitionException(org.antlr.runtime.RecognitionException)

Aggregations

FunctionDescrBuilder (org.drools.drl.ast.dsl.FunctionDescrBuilder)5 CommonToken (org.antlr.runtime.CommonToken)3 RecognitionException (org.antlr.runtime.RecognitionException)3 Token (org.antlr.runtime.Token)3 AttributeDescr (org.drools.drl.ast.descr.AttributeDescr)1 AbstractClassTypeDeclarationBuilder (org.drools.drl.ast.dsl.AbstractClassTypeDeclarationBuilder)1 AccumulateDescrBuilder (org.drools.drl.ast.dsl.AccumulateDescrBuilder)1 AccumulateImportDescrBuilder (org.drools.drl.ast.dsl.AccumulateImportDescrBuilder)1 AttributeDescrBuilder (org.drools.drl.ast.dsl.AttributeDescrBuilder)1 CEDescrBuilder (org.drools.drl.ast.dsl.CEDescrBuilder)1 DeclareDescrBuilder (org.drools.drl.ast.dsl.DeclareDescrBuilder)1 EntryPointDeclarationDescrBuilder (org.drools.drl.ast.dsl.EntryPointDeclarationDescrBuilder)1 EnumDeclarationDescrBuilder (org.drools.drl.ast.dsl.EnumDeclarationDescrBuilder)1 EnumLiteralDescrBuilder (org.drools.drl.ast.dsl.EnumLiteralDescrBuilder)1 FieldDescrBuilder (org.drools.drl.ast.dsl.FieldDescrBuilder)1 ForallDescrBuilder (org.drools.drl.ast.dsl.ForallDescrBuilder)1 GlobalDescrBuilder (org.drools.drl.ast.dsl.GlobalDescrBuilder)1 ImportDescrBuilder (org.drools.drl.ast.dsl.ImportDescrBuilder)1 NamedConsequenceDescrBuilder (org.drools.drl.ast.dsl.NamedConsequenceDescrBuilder)1 PackageDescrBuilder (org.drools.drl.ast.dsl.PackageDescrBuilder)1