/**
* 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.operator.processor;
import java.util.List;
import javax.lang.model.type.TypeMirror;
import com.asakusafw.compiler.common.Precondition;
import com.asakusafw.compiler.common.TargetOperator;
import com.asakusafw.compiler.operator.AbstractOperatorProcessor;
import com.asakusafw.compiler.operator.ExecutableAnalyzer;
import com.asakusafw.compiler.operator.ExecutableAnalyzer.TypeConstraint;
import com.asakusafw.compiler.operator.OperatorMethodDescriptor;
import com.asakusafw.compiler.operator.OperatorMethodDescriptor.Builder;
import com.asakusafw.utils.collections.Lists;
import com.asakusafw.vocabulary.flow.graph.FlowBoundary;
import com.asakusafw.vocabulary.flow.graph.ShuffleKey;
import com.asakusafw.vocabulary.operator.CoGroup;
/**
* {@link CoGroup グループ結合演算子}を処理する。
*/
@TargetOperator(CoGroup.class)
public class CoGroupOperatorProcessor extends AbstractOperatorProcessor {
@Override
public OperatorMethodDescriptor describe(Context context) {
Precondition.checkMustNotBeNull(context, "context"); //$NON-NLS-1$
ExecutableAnalyzer a = new ExecutableAnalyzer(context.environment, context.element);
if (a.isAbstract()) {
a.error("グループ結合演算子はabstractで宣言できません");
}
if (a.getReturnType().isVoid() == false) {
a.error("グループ結合演算子は戻り値にvoidを指定する必要があります");
}
int startResults = 0;
for (int i = 0, n = a.countParameters(); i < n; i++) {
TypeConstraint type = a.getParameterType(i);
if (type.isResult()) {
break;
}
if (type.isList() == false) {
a.error(i, "グループ結合演算子の先頭の引数はリスト型(java.util.List)である必要があります");
} else if (type.getTypeArgument().isModel() == false) {
a.error(i, "グループ結合演算子の先頭の引数はリストのモデルオブジェクト型である必要があります");
}
startResults++;
}
if (startResults == 0) { // リスト型がない
a.error("グループ結合演算子の引数には一つ以上のリスト型(java.util.List)を指定する必要があります");
}
int startParameters = startResults;
for (int i = startResults, n = a.countParameters(); i < n; i++) {
TypeConstraint param = a.getParameterType(i);
if (param.isResult() == false) {
break;
} else if (param.getTypeArgument().isModel() == false) {
a.error(i, "グループ結合演算子の結果は結果のモデルオブジェクト型である必要があります");
} else {
startParameters++;
}
}
if (startParameters == startResults) { // 結果型がない
a.error("グループ結合演算子の引数には一つ以上の結果(Result)型を指定する必要があります");
}
for (int i = startParameters, n = a.countParameters(); i < n; i++) {
TypeConstraint param = a.getParameterType(i);
if (param.isResult()) {
a.error(i, "ユーザー引数の後には結果型を含められません");
} else if (param.isBasic() == false) {
a.error(i, "ユーザー引数は文字列またはプリミティブ型である必要があります");
}
}
if (a.hasError()) {
return null;
}
List<ShuffleKey> keys = Lists.create();
for (int i = 0; i < startResults; i++) {
ShuffleKey key = a.getParameterKey(i);
if (key == null) {
a.error(i, "グループ結合演算子の引数には@Key注釈によってグループ化項目を指定する必要があります");
} else {
keys.add(key);
}
}
if (keys.isEmpty() == false) {
ShuffleKey first = keys.get(0);
for (int i = 1, n = keys.size(); i < n; i++) {
if (first.getGroupProperties().size() != keys.get(i).getGroupProperties().size()) {
a.error(0, "グループ結合演算子の@Key注釈ではグループ化項目の個数を全て一致させる必要があります");
}
}
}
if (a.hasError()) {
return null;
}
CoGroup annotation = context.element.getAnnotation(CoGroup.class);
if (annotation == null) {
a.error("注釈の解釈に失敗しました");
return null;
}
Builder builder = new Builder(getTargetAnnotationType(), context);
builder.addAttribute(FlowBoundary.SHUFFLE);
builder.addAttribute(a.getObservationCount());
builder.addAttribute(annotation.inputBuffer());
builder.setDocumentation(a.getExecutableDocument());
for (int i = 0; i < startResults; i++) {
builder.addInput(
a.getParameterDocument(i),
a.getParameterName(i),
a.getParameterType(i).getTypeArgument().getType(),
i,
keys.get(i));
}
for (int i = startResults; i < startParameters; i++) {
TypeConstraint outputType = a.getParameterType(i).getTypeArgument();
TypeMirror outputTypeMirror = outputType.getType();
String found = builder.findInput(outputTypeMirror);
if (found == null && outputType.isProjectiveModel()) {
a.error("出力型{0}に対する入力が見つかりません", outputTypeMirror);
}
builder.addOutput(
a.getParameterDocument(i),
a.getParameterName(i),
outputTypeMirror,
found,
i);
}
for (int i = startParameters, n = a.countParameters(); i < n; i++) {
builder.addParameter(
a.getParameterDocument(i),
a.getParameterName(i),
a.getParameterType(i).getType(),
i);
}
return builder.toDescriptor();
}
}