/*

  * Licensed to the Apache Software Foundation (ASF) under one or more

  * contributor license agreements.  See the NOTICE file distributed with

  * this work for additional information regarding copyright ownership.

  * The ASF licenses this file to You 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 org.apache.commons.nabla.automatic.analysis;

 import java.util.HashSet;

 import java.util.Set;

 import org.objectweb.asm.tree.AbstractInsnNode;

 import org.objectweb.asm.tree.analysis.BasicValue;

 import org.objectweb.asm.tree.analysis.Value;

 /** A value that keep track of both instructions producing and consuming it.

  */

 public class TrackingValue implements Value {

     /** Special value for uninitialized values. */

     public static final TrackingValue UNINITIALIZED_VALUE =

         new TrackingValue((BasicValue) BasicValue.UNINITIALIZED_VALUE);

     /** Underlying value. */

     private final BasicValue value;

     /** Instructions that consume this value. */

     private Set<AbstractInsnNode> consumers;

     /** Instructions that produce this value. */

     private Set<AbstractInsnNode> producers;

     /** Values that are merged with this value. */

     private Set<TrackingValue> merged;

     /** Build a new value without any link to instructions.

      * @param value wrapped {@link BasicValue} value

      */

     public TrackingValue(final BasicValue value) {

         this.value = value;

         consumers  = new HashSet<AbstractInsnNode>();

         producers  = new HashSet<AbstractInsnNode>();

         merged     = new HashSet<TrackingValue>();

     }

     /** Get the wrapped {@link BasicValue}.

      * @return wrapped {@link BasicValue}

      */

     public BasicValue getValue() {

         return value;

     }

     /** {@inheritDoc} */

     public int getSize() {

         return value.getSize();

     }

     /** Add a consumer for this value.

      * @param consumer consumer for this value

      */

     public void addConsumer(final AbstractInsnNode consumer) {

         consumers.add(consumer);

     }

     /** Get the consumers for this value and all values it is merged with.

      * @return the instructions consuming either this value or the values

      * it has been merged with

      */

     public Set<AbstractInsnNode> getConsumers() {

         return consumers;

     }

     /** Add a producer for this value.

      * @param producer producer for this value

      */

     public void addProducer(final AbstractInsnNode producer) {

         producers.add(producer);

     }

     /** Get the producers for this value and all values it is merged with.

      * @return the instructions producing either this value or the values

      * it has been merged with

      */

     public Set<AbstractInsnNode> getProducers() {

         return producers;

     }

     /** Merge two instances.

      * <p>Once merged, values share the same producers and consumers sets.</p>

      * @param value1 first value to merge

      * @param value2 second value to merge

      */

     public static void merge(final TrackingValue value1,

                              final TrackingValue value2) {

         // merge the sets

         value1.consumers.addAll(value2.consumers);

         value1.producers.addAll(value2.producers);

         value1.merged.addAll(value2.merged);

         // share the merged sets

         for (TrackingValue value : value2.merged) {

             value.consumers = value1.consumers;

             value.producers = value1.producers;

             value.merged    = value1.merged;

         }

         value2.consumers = value1.consumers;

         value2.producers = value1.producers;

         value2.merged    = value1.merged;

     }

 }