src/main/java/org/onap/clamp/clds/util/drawing/SvgLoopGenerator.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP CLAMP
   4  * ================================================================================
   5  * Copyright (C) 2019 Nokia. 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  * Modifications copyright (c) 2019 AT&T
  21  * ===================================================================
  22  *
  23  */
  24
  25 package org.onap.clamp.clds.util.drawing;
  26
  27 import java.util.HashSet;
  28 import java.util.Set;
  29 import org.apache.batik.svggen.SVGGraphics2D;
  30 import org.onap.clamp.clds.util.XmlTools;
  31 import org.onap.clamp.loop.Loop;
  32 import org.onap.clamp.loop.template.LoopElementModel;
  33 import org.onap.clamp.loop.template.LoopTemplate;
  34 import org.onap.clamp.loop.template.LoopTemplateLoopElementModel;
  35 import org.w3c.dom.Document;
  36
  37 public class SvgLoopGenerator {
  38     /**
  39      * Generate the SVG images from the loop.
  40      *
  41      * @param loop The loop object, so it won't use the loop template
  42      * @return A String containing the SVG
  43      */
  44     public static String getSvgImage(Loop loop) {
  45         SVGGraphics2D svgGraphics2D = new SVGGraphics2D(XmlTools.createEmptySvgDocument());
  46         Document document = XmlTools.createEmptySvgDocument();
  47         DocumentBuilder dp = new DocumentBuilder(document, svgGraphics2D.getDOMFactory());
  48         Painter painter = new Painter(svgGraphics2D, dp);
  49         ClampGraphBuilder cgp = new ClampGraphBuilder(painter).collector("VES");
  50         cgp.addAllMicroServices(loop.getMicroServicePolicies());
  51         ClampGraph cg = cgp.addAllPolicies(loop.getOperationalPolicies()).build();
  52         return cg.getAsSvg();
  53     }
  54
  55     /**
  56      * Generate the SVG images from the loop template.
  57      *
  58      * @param loopTemplate The loop template
  59      * @return A String containing the SVG
  60      */
  61     public static String getSvgImage(LoopTemplate loopTemplate) {
  62         SVGGraphics2D svgGraphics2D = new SVGGraphics2D(XmlTools.createEmptySvgDocument());
  63         Document document = XmlTools.createEmptySvgDocument();
  64         DocumentBuilder dp = new DocumentBuilder(document, svgGraphics2D.getDOMFactory());
  65         Painter painter = new Painter(svgGraphics2D, dp);
  66         ClampGraphBuilder cgp = new ClampGraphBuilder(painter).collector("VES");
  67         Set<LoopElementModel> elementModelsSet = new HashSet<>();
  68         for (LoopTemplateLoopElementModel elementModelLink:loopTemplate.getLoopElementModelsUsed()) {
  69             elementModelsSet.add(elementModelLink.getLoopElementModel());
  70         }
  71         ClampGraph cg = cgp.addAllLoopElementModels(elementModelsSet).build();
  72         return cg.getAsSvg();
  73     }
  74
  75 }