Package com.asakusafw.compiler.flow.processor

Source Code of com.asakusafw.compiler.flow.processor.CoGroupFlowProcessor

/**
* Copyright 2011-2014 Asakusa Framework Team.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.asakusafw.compiler.flow.processor;

import java.util.List;

import com.asakusafw.compiler.common.TargetOperator;
import com.asakusafw.compiler.flow.RendezvousProcessor;
import com.asakusafw.utils.collections.Lists;
import com.asakusafw.utils.java.model.syntax.Expression;
import com.asakusafw.utils.java.model.syntax.ModelFactory;
import com.asakusafw.utils.java.model.util.ExpressionBuilder;
import com.asakusafw.utils.java.model.util.Models;
import com.asakusafw.vocabulary.flow.graph.FlowElementPortDescription;
import com.asakusafw.vocabulary.flow.graph.OperatorDescription;
import com.asakusafw.vocabulary.flow.processor.InputBuffer;
import com.asakusafw.vocabulary.operator.CoGroup;

/**
* {@link CoGroup グループ化結合演算子}を処理する。
*/
@TargetOperator(CoGroup.class)
public class CoGroupFlowProcessor extends RendezvousProcessor {

    @Override
    public void emitRendezvous(Context context) {
        ModelFactory f = context.getModelFactory();
        OperatorDescription desc = context.getOperatorDescription();
        InputBuffer bufferKind = desc.getAttribute(InputBuffer.class);
        assert bufferKind != null;

        List<Expression> arguments = Lists.create();
        List<ListBufferMirror> buffers = Lists.create();
        for (FlowElementPortDescription input : desc.getInputPorts()) {
            ListBufferMirror list = context.createListBuffer(input.getDataType(), bufferKind);
            buffers.add(list);

            context.addBegin(list.createBegin());

            Expression proc = context.getProcessInput(input);
            context.addProcess(input, list.createAdvance(proc));

            context.addEnd(list.createEnd());

            arguments.add(list.get());
        }
        for (FlowElementPortDescription output : desc.getOutputPorts()) {
            arguments.add(context.getOutput(output).get());
        }
        for (OperatorDescription.Parameter param : desc.getParameters()) {
            arguments.add(Models.toLiteral(f, param.getValue()));
        }

        Expression impl = context.createImplementation();
        context.addEnd(new ExpressionBuilder(f, impl)
            .method(desc.getDeclaration().getName(), arguments)
            .toStatement());

        for (ListBufferMirror list : buffers) {
            context.addEnd(list.createShrink());
        }
    }
}
TOP

Related Classes of com.asakusafw.compiler.flow.processor.CoGroupFlowProcessor

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.