    001/*
    002 * Copyright (c) 2007-2017 Xplenty, Inc. All Rights Reserved.
    003 *
    004 * Project and contact information: http://www.cascading.org/
    005 *
    006 * This file is part of the Cascading project.
    007 *
    008 * Licensed under the Apache License, Version 2.0 (the "License");
    009 * you may not use this file except in compliance with the License.
    010 * You may obtain a copy of the License at
    011 *
    012 *     http://www.apache.org/licenses/LICENSE-2.0
    013 *
    014 * Unless required by applicable law or agreed to in writing, software
    015 * distributed under the License is distributed on an "AS IS" BASIS,
    016 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    017 * See the License for the specific language governing permissions and
    018 * limitations under the License.
    019 */
    020
    021package cascading.flow.planner.iso.expression;
    022
    023import cascading.flow.FlowElement;
    024import cascading.flow.planner.PlannerContext;
    025import cascading.flow.planner.graph.AnnotatedGraph;
    026import cascading.flow.planner.graph.ElementGraph;
    027
    028/**
    029 *
    030 */
    031public class AnnotationExpression extends ElementExpression
    032  {
    033  Enum annotation = null;
    034
    035  public AnnotationExpression( Enum annotation )
    036    {
    037    this.annotation = annotation;
    038    }
    039
    040  public AnnotationExpression( ElementCapture capture, Enum annotation )
    041    {
    042    super( capture );
    043    this.annotation = annotation;
    044    }
    045
    046  @Override
    047  public boolean applies( PlannerContext plannerContext, ElementGraph elementGraph, FlowElement flowElement )
    048    {
    049    if( elementGraph instanceof AnnotatedGraph && ( (AnnotatedGraph) elementGraph ).hasAnnotations() )
    050      return ( (AnnotatedGraph) elementGraph ).getAnnotations().getValues( annotation ).contains( flowElement );
    051
    052    return false;
    053    }
    054
    055  @Override
    056  public String toString()
    057    {
    058    final StringBuilder sb = new StringBuilder( "AnnotationExpression{" );
    059    sb.append( "annotation=" ).append( annotation );
    060    sb.append( '}' );
    061    return sb.toString();
    062    }
    063  }