appc-dg/appc-dg-shared/appc-dg-dependency-model/src/main/java/org/openecomp/appc/dg/flowbuilder/helper/Graph.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * openECOMP : APP-C
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights
   6  *                                              reserved.
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  *      http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END=========================================================
  20  */
  21
  22 package org.openecomp.appc.dg.flowbuilder.helper;
  23
  24 import java.util.*;
  25
  26 import org.openecomp.appc.dg.flowbuilder.exception.InvalidDependencyModel;
  27
  28
  29 public class Graph<T> {
  30     private int size;
  31     private List<T> vertexList;
  32
  33     private int[][] dependencyMatrix;
  34
  35     public Graph(int size){
  36         this.size =size;
  37         vertexList = new ArrayList<>();
  38         dependencyMatrix = new int[size][size];
  39     }
  40
  41     public void addVertex(T vertex){
  42         vertexList.add(vertex);
  43     }
  44
  45     public int getIndex(T vertex){
  46         return vertexList.indexOf(vertex);
  47     }
  48
  49     public void addEdge(T vertex1,T vertex2){
  50         dependencyMatrix[vertexList.indexOf(vertex1)][vertexList.indexOf(vertex2)] = 1;
  51     }
  52
  53     public int[][] getDependencyMatrix() {
  54         return dependencyMatrix;
  55     }
  56
  57     public int getSize() {
  58         return size;
  59     }
  60
  61     public List<T> getVertexList() {
  62         return vertexList;
  63     }
  64 }