tools/llvm-cov/SourceCoverageDataManager.cpp

   1 //===- SourceCoverageDataManager.cpp - Manager for source file coverage
   2 // data-===//
   3 //
   4 //                     The LLVM Compiler Infrastructure
   5 //
   6 // This file is distributed under the University of Illinois Open Source
   7 // License. See LICENSE.TXT for details.
   8 //
   9 //===----------------------------------------------------------------------===//
  10 //
  11 // This class separates and merges mapping regions for a specific source file.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "SourceCoverageDataManager.h"
  16
  17 using namespace llvm;
  18 using namespace coverage;
  19
  20 void SourceCoverageDataManager::insert(const CountedRegion &CR) {
  21   Regions.push_back(CR);
  22   Uniqued = false;
  23 }
  24
  25 ArrayRef<CountedRegion> SourceCoverageDataManager::getSourceRegions() {
  26   if (Uniqued || Regions.size() <= 1)
  27     return Regions;
  28
  29   // Sort.
  30   std::sort(Regions.begin(), Regions.end());
  31
  32   // Merge duplicate source ranges and sum their execution counts.
  33   auto Prev = Regions.begin();
  34   for (auto I = Prev + 1, E = Regions.end(); I != E; ++I) {
  35     if (I->coversSameSource(*Prev)) {
  36       Prev->ExecutionCount += I->ExecutionCount;
  37       continue;
  38     }
  39     ++Prev;
  40     *Prev = *I;
  41   }
  42   ++Prev;
  43   Regions.erase(Prev, Regions.end());
  44
  45   Uniqued = true;
  46   return Regions;
  47 }