mockit.coverage.reporting.lineCoverage.LineSegmentsFormatter Maven / Gradle / Ivy

Go to download

/*
 * Copyright (c) 2006-2013 Rogério Liesenfeld
 * This file is subject to the terms of the MIT license (see LICENSE.txt).
 */
package mockit.coverage.reporting.lineCoverage;

import java.util.*;

import org.jetbrains.annotations.*;

import mockit.coverage.*;
import mockit.coverage.lines.*;
import mockit.coverage.reporting.*;
import mockit.coverage.reporting.parsing.*;

final class LineSegmentsFormatter
{
   @Nullable private final ListOfCallPoints listOfCallPoints;
   @NotNull private final StringBuilder line;
   private int lineNumber;

   // Helper fields:
   @Nullable private LineElement element;
   private int segmentIndex;

   LineSegmentsFormatter(boolean withCallPoints, @NotNull StringBuilder line)
   {
      listOfCallPoints = withCallPoints ? new ListOfCallPoints() : null;
      this.line = line;
   }

   void formatSegments(@NotNull LineParser lineParser, @NotNull LineCoverageData lineData)
   {
      lineNumber = lineParser.getNumber();

      List branchData = lineData.getBranches();
      int numSegments = 1 + branchData.size();

      element = lineParser.getInitialElement().appendUntilNextCodeElement(line);

      segmentIndex = 0;
      appendUntilFirstElementAfterNextBranchingPoint(lineData);

      for (segmentIndex = 1; element != null && segmentIndex < numSegments; segmentIndex++) {
         LineSegmentData segmentData = branchData.get(segmentIndex - 1);

         element = element.appendUntilNextCodeElement(line);
         appendUntilFirstElementAfterNextBranchingPoint(segmentData);
      }

      line.append("

");

      if (listOfCallPoints != null && lineData.containsCallPoints()) {
         line.append(listOfCallPoints.getContents());
      }
   }

   private void appendUntilFirstElementAfterNextBranchingPoint(@NotNull LineSegmentData segmentData)
   {
      if (element != null) {
         LineElement firstElement = element;
         element = element.findNextBranchingPoint();

         appendToFormattedLine(firstElement, segmentData);

         if (element != null && element.isBranchingElement()) {
            line.append(element.getText());
            element = element.getNext();
         }
      }
   }

   private void appendToFormattedLine(@NotNull LineElement firstElement, @NotNull LineSegmentData segmentData)
   {
      if (firstElement == element) {
         return;
      }

      appendStartTag(segmentData);
      firstElement.appendAllBefore(line, element);
      appendEndTag(segmentData);
   }

   private void appendStartTag(@NotNull LineSegmentData segmentData)
   {
      line.append("");
      }
      else {
         line.append("class='uncovered'>");
      }
   }

   private void appendTooltipWithExecutionCounts(@NotNull LineSegmentData segmentData)
   {
      line.append("title='Executions: ").append(segmentData.getExecutionCount()).append("' ");
   }

   private void appendEndTag(@NotNull LineSegmentData segmentData)
   {
      int i = line.length() - 1;

      while (Character.isWhitespace(line.charAt(i))) {
         i--;
      }
      
      line.insert(i + 1, "");

      if (listOfCallPoints != null) {
         List callPoints = segmentData.getCallPoints();

         if (callPoints != null) {
            listOfCallPoints.insertListOfCallPoints(callPoints);
         }
      }
   }
}