    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.graph;
    022
    023import java.util.Arrays;
    024import java.util.Collection;
    025import java.util.HashSet;
    026import java.util.Set;
    027
    028import cascading.flow.FlowElement;
    029import cascading.flow.planner.Scope;
    030import org.jgrapht.DirectedGraph;
    031import org.jgrapht.graph.DirectedMaskSubgraph;
    032import org.jgrapht.graph.MaskFunctor;
    033
    034import static cascading.flow.planner.graph.ElementGraphs.directed;
    035import static cascading.util.Util.createIdentitySet;
    036
    037/**
    038 *
    039 */
    040public class ElementMaskSubGraph extends BaseElementGraph implements ElementGraph
    041  {
    042  private ElementGraph elementGraph;
    043  private FlowElementMaskFunctor mask;
    044
    045  private static class FlowElementMaskFunctor implements MaskFunctor<FlowElement, Scope>
    046    {
    047    Set<FlowElement> maskedElements = createIdentitySet();
    048    Set<Scope> maskedScopes = new HashSet<>();
    049
    050    public FlowElementMaskFunctor( Collection<FlowElement> flowElements, Collection<Scope> scopes )
    051      {
    052      this( flowElements );
    053
    054      if( scopes != null )
    055        maskedScopes.addAll( scopes );
    056      }
    057
    058    public FlowElementMaskFunctor( Collection<FlowElement> flowElements )
    059      {
    060      if( flowElements != null )
    061        maskedElements.addAll( flowElements );
    062      }
    063
    064    @Override
    065    public boolean isEdgeMasked( Scope scope )
    066      {
    067      return maskedScopes.contains( scope );
    068      }
    069
    070    @Override
    071    public boolean isVertexMasked( FlowElement flowElement )
    072      {
    073      return maskedElements.contains( flowElement );
    074      }
    075    }
    076
    077  public ElementMaskSubGraph( ElementGraph elementGraph, FlowElement... maskedFlowElements )
    078    {
    079    this( elementGraph, new FlowElementMaskFunctor( Arrays.asList( maskedFlowElements ) ) );
    080    }
    081
    082  public ElementMaskSubGraph( ElementGraph elementGraph, Collection<FlowElement> maskedFlowElements )
    083    {
    084    this( elementGraph, new FlowElementMaskFunctor( maskedFlowElements ) );
    085    }
    086
    087  public ElementMaskSubGraph( ElementGraph elementGraph, Collection<FlowElement> maskedFlowElements, Collection<Scope> maskedScopes )
    088    {
    089    this( elementGraph, new FlowElementMaskFunctor( maskedFlowElements, maskedScopes ) );
    090    }
    091
    092  public ElementMaskSubGraph( ElementMaskSubGraph graph )
    093    {
    094    this( graph.elementGraph, graph.mask );
    095    }
    096
    097  protected ElementMaskSubGraph( ElementGraph elementGraph, FlowElementMaskFunctor flowElementMaskFunctor )
    098    {
    099    this.graph = new DirectedMaskSubGraph( directed( elementGraph ), flowElementMaskFunctor );
    100
    101    this.elementGraph = elementGraph;
    102    this.mask = flowElementMaskFunctor;
    103    }
    104
    105  @Override
    106  public ElementGraph copyElementGraph()
    107    {
    108    return new ElementMaskSubGraph( ElementMaskSubGraph.this );
    109    }
    110
    111  private class DirectedMaskSubGraph extends DirectedMaskSubgraph<FlowElement, Scope>
    112    {
    113    public DirectedMaskSubGraph( DirectedGraph<FlowElement, Scope> base, MaskFunctor<FlowElement, Scope> mask )
    114      {
    115      super( base, mask );
    116      }
    117    }
    118  }