plugins/org.eclipse.osee.orcs.core/src/org/eclipse/osee/orcs/core/internal/graph/impl/GraphDataImpl.java

/*******************************************************************************
 * Copyright (c) 2013 Boeing.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Boeing - initial API and implementation
 *******************************************************************************/
package org.eclipse.osee.orcs.core.internal.graph.impl;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import org.eclipse.osee.framework.core.data.HasLocalId;
import org.eclipse.osee.framework.jdk.core.type.OseeArgumentException;
import org.eclipse.osee.framework.jdk.core.type.OseeCoreException;
import org.eclipse.osee.orcs.OrcsSession;
import org.eclipse.osee.orcs.core.internal.graph.GraphAdjacencies;
import org.eclipse.osee.orcs.core.internal.graph.GraphData;
import org.eclipse.osee.orcs.core.internal.graph.GraphNode;

/**
 * @author Roberto E. Escobar
 */
public class GraphDataImpl implements GraphData {

   private final Map<Integer, GraphNode> nodesById = new ConcurrentHashMap<>();
   private final Map<Integer, GraphAdjacencies> adjacenciesById = new ConcurrentHashMap<>();

   private final Long branchId;
   private final int txId;
   private final OrcsSession orcsSession;

   public GraphDataImpl(OrcsSession session, Long branchId, int txId) {
      super();
      this.orcsSession = session;
      this.branchId = branchId;
      this.txId = txId;
   }

   @Override
   public int getTransaction() {
      return txId;
   }

   @Override
   public <T extends GraphNode> T getNode(HasLocalId<Integer> data) {
      return getNode(data.getLocalId());
   }

   @SuppressWarnings("unchecked")
   @Override
   public <T extends GraphNode> T getNode(Integer id) {
      return (T) nodesById.get(id);
   }

   @Override
   public void addNode(GraphNode node, boolean useBackingData) throws OseeCoreException {
      boolean sameBranches = getBranchUuid().equals(node.getBranchUuid());
      if (!sameBranches) {
         throw new OseeArgumentException("Invalid node added to graph. Graph[%s] Node[%s]", this,
            node.getExceptionString());
      }

      GraphData oldGraph = node.getGraph();
      if (!this.equals(oldGraph) || useBackingData) {
         if (oldGraph != null) {
            oldGraph.removeNode(node);
         }
         nodesById.put(node.getLocalId(), node);
         node.setGraph(this);
      }
   }

   @Override
   public <T extends GraphNode> T removeNode(HasLocalId<Integer> node) {
      return removeNode(node.getLocalId());
   }

   @SuppressWarnings("unchecked")
   @Override
   public <T extends GraphNode> T removeNode(Integer id) {
      T removed = (T) nodesById.remove(id);
      removeAdjacencies(id);
      return removed;
   }

   @Override
   public <T extends GraphAdjacencies> T getAdjacencies(HasLocalId<Integer> node) {
      return getAdjacencies(node.getLocalId());
   }

   @SuppressWarnings("unchecked")
   @Override
   public <T extends GraphAdjacencies> T getAdjacencies(Integer id) {
      return (T) adjacenciesById.get(id);
   }

   @Override
   public void addAdjacencies(HasLocalId<Integer> node, GraphAdjacencies adjacencies) {
      addAdjacencies(node.getLocalId(), adjacencies);
   }

   @Override
   public void addAdjacencies(Integer id, GraphAdjacencies adjacencies) {
      adjacenciesById.put(id, adjacencies);
   }

   @Override
   public <T extends GraphAdjacencies> T removeAdjacencies(HasLocalId<Integer> node) {
      return removeAdjacencies(node.getLocalId());
   }

   @SuppressWarnings("unchecked")
   @Override
   public <T extends GraphAdjacencies> T removeAdjacencies(Integer id) {
      return (T) adjacenciesById.remove(id);
   }

   @Override
   public String toString() {
      return String.format("Graph - branch[%s] txId[%s] nodes[%s] adjacencies[%s]", getBranchUuid(), getTransaction(),
         nodesById.size(), adjacenciesById.size());
   }

   @Override
   public int hashCode() {
      final int prime = 31;
      int result = getBranchUuid().hashCode();
      result = prime * result + getTransaction();
      return result;
   }

   @Override
   public boolean equals(Object obj) {
      if (this == obj) {
         return true;
      }
      if (!(obj instanceof GraphData)) {
         return false;
      }
      GraphData other = (GraphData) obj;
      if (!getBranchUuid().equals(other.getBranchUuid())) {
         return false;
      }
      if (getTransaction() != other.getTransaction()) {
         return false;
      }
      return true;
   }

   @Override
   public OrcsSession getSession() {
      return orcsSession;
   }

   @Override
   public Long getBranchUuid() {
      return branchId;
   }
}