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  }