1*f22ef01cSRoman Divacky //===--- DeltaAlgorithm.cpp - A Set Minimization Algorithm -----*- C++ -*--===//
2*f22ef01cSRoman Divacky //
3*f22ef01cSRoman Divacky //                     The LLVM Compiler Infrastructure
4*f22ef01cSRoman Divacky //
5*f22ef01cSRoman Divacky // This file is distributed under the University of Illinois Open Source
6*f22ef01cSRoman Divacky // License. See LICENSE.TXT for details.
7*f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
8*f22ef01cSRoman Divacky 
9*f22ef01cSRoman Divacky #include "llvm/ADT/DeltaAlgorithm.h"
10*f22ef01cSRoman Divacky #include <algorithm>
11*f22ef01cSRoman Divacky #include <iterator>
12*f22ef01cSRoman Divacky using namespace llvm;
13*f22ef01cSRoman Divacky 
14*f22ef01cSRoman Divacky DeltaAlgorithm::~DeltaAlgorithm() {
15*f22ef01cSRoman Divacky }
16*f22ef01cSRoman Divacky 
17*f22ef01cSRoman Divacky bool DeltaAlgorithm::GetTestResult(const changeset_ty &Changes) {
18*f22ef01cSRoman Divacky   if (FailedTestsCache.count(Changes))
19*f22ef01cSRoman Divacky     return false;
20*f22ef01cSRoman Divacky 
21*f22ef01cSRoman Divacky   bool Result = ExecuteOneTest(Changes);
22*f22ef01cSRoman Divacky   if (!Result)
23*f22ef01cSRoman Divacky     FailedTestsCache.insert(Changes);
24*f22ef01cSRoman Divacky 
25*f22ef01cSRoman Divacky   return Result;
26*f22ef01cSRoman Divacky }
27*f22ef01cSRoman Divacky 
28*f22ef01cSRoman Divacky void DeltaAlgorithm::Split(const changeset_ty &S, changesetlist_ty &Res) {
29*f22ef01cSRoman Divacky   // FIXME: Allow clients to provide heuristics for improved splitting.
30*f22ef01cSRoman Divacky 
31*f22ef01cSRoman Divacky   // FIXME: This is really slow.
32*f22ef01cSRoman Divacky   changeset_ty LHS, RHS;
33*f22ef01cSRoman Divacky   unsigned idx = 0;
34*f22ef01cSRoman Divacky   for (changeset_ty::const_iterator it = S.begin(),
35*f22ef01cSRoman Divacky          ie = S.end(); it != ie; ++it, ++idx)
36*f22ef01cSRoman Divacky     ((idx & 1) ? LHS : RHS).insert(*it);
37*f22ef01cSRoman Divacky   if (!LHS.empty())
38*f22ef01cSRoman Divacky     Res.push_back(LHS);
39*f22ef01cSRoman Divacky   if (!RHS.empty())
40*f22ef01cSRoman Divacky     Res.push_back(RHS);
41*f22ef01cSRoman Divacky }
42*f22ef01cSRoman Divacky 
43*f22ef01cSRoman Divacky DeltaAlgorithm::changeset_ty
44*f22ef01cSRoman Divacky DeltaAlgorithm::Delta(const changeset_ty &Changes,
45*f22ef01cSRoman Divacky                       const changesetlist_ty &Sets) {
46*f22ef01cSRoman Divacky   // Invariant: union(Res) == Changes
47*f22ef01cSRoman Divacky   UpdatedSearchState(Changes, Sets);
48*f22ef01cSRoman Divacky 
49*f22ef01cSRoman Divacky   // If there is nothing left we can remove, we are done.
50*f22ef01cSRoman Divacky   if (Sets.size() <= 1)
51*f22ef01cSRoman Divacky     return Changes;
52*f22ef01cSRoman Divacky 
53*f22ef01cSRoman Divacky   // Look for a passing subset.
54*f22ef01cSRoman Divacky   changeset_ty Res;
55*f22ef01cSRoman Divacky   if (Search(Changes, Sets, Res))
56*f22ef01cSRoman Divacky     return Res;
57*f22ef01cSRoman Divacky 
58*f22ef01cSRoman Divacky   // Otherwise, partition the sets if possible; if not we are done.
59*f22ef01cSRoman Divacky   changesetlist_ty SplitSets;
60*f22ef01cSRoman Divacky   for (changesetlist_ty::const_iterator it = Sets.begin(),
61*f22ef01cSRoman Divacky          ie = Sets.end(); it != ie; ++it)
62*f22ef01cSRoman Divacky     Split(*it, SplitSets);
63*f22ef01cSRoman Divacky   if (SplitSets.size() == Sets.size())
64*f22ef01cSRoman Divacky     return Changes;
65*f22ef01cSRoman Divacky 
66*f22ef01cSRoman Divacky   return Delta(Changes, SplitSets);
67*f22ef01cSRoman Divacky }
68*f22ef01cSRoman Divacky 
69*f22ef01cSRoman Divacky bool DeltaAlgorithm::Search(const changeset_ty &Changes,
70*f22ef01cSRoman Divacky                             const changesetlist_ty &Sets,
71*f22ef01cSRoman Divacky                             changeset_ty &Res) {
72*f22ef01cSRoman Divacky   // FIXME: Parallelize.
73*f22ef01cSRoman Divacky   for (changesetlist_ty::const_iterator it = Sets.begin(),
74*f22ef01cSRoman Divacky          ie = Sets.end(); it != ie; ++it) {
75*f22ef01cSRoman Divacky     // If the test passes on this subset alone, recurse.
76*f22ef01cSRoman Divacky     if (GetTestResult(*it)) {
77*f22ef01cSRoman Divacky       changesetlist_ty Sets;
78*f22ef01cSRoman Divacky       Split(*it, Sets);
79*f22ef01cSRoman Divacky       Res = Delta(*it, Sets);
80*f22ef01cSRoman Divacky       return true;
81*f22ef01cSRoman Divacky     }
82*f22ef01cSRoman Divacky 
83*f22ef01cSRoman Divacky     // Otherwise, if we have more than two sets, see if test passes on the
84*f22ef01cSRoman Divacky     // complement.
85*f22ef01cSRoman Divacky     if (Sets.size() > 2) {
86*f22ef01cSRoman Divacky       // FIXME: This is really slow.
87*f22ef01cSRoman Divacky       changeset_ty Complement;
88*f22ef01cSRoman Divacky       std::set_difference(
89*f22ef01cSRoman Divacky         Changes.begin(), Changes.end(), it->begin(), it->end(),
90*f22ef01cSRoman Divacky         std::insert_iterator<changeset_ty>(Complement, Complement.begin()));
91*f22ef01cSRoman Divacky       if (GetTestResult(Complement)) {
92*f22ef01cSRoman Divacky         changesetlist_ty ComplementSets;
93*f22ef01cSRoman Divacky         ComplementSets.insert(ComplementSets.end(), Sets.begin(), it);
94*f22ef01cSRoman Divacky         ComplementSets.insert(ComplementSets.end(), it + 1, Sets.end());
95*f22ef01cSRoman Divacky         Res = Delta(Complement, ComplementSets);
96*f22ef01cSRoman Divacky         return true;
97*f22ef01cSRoman Divacky       }
98*f22ef01cSRoman Divacky     }
99*f22ef01cSRoman Divacky   }
100*f22ef01cSRoman Divacky 
101*f22ef01cSRoman Divacky   return false;
102*f22ef01cSRoman Divacky }
103*f22ef01cSRoman Divacky 
104*f22ef01cSRoman Divacky DeltaAlgorithm::changeset_ty DeltaAlgorithm::Run(const changeset_ty &Changes) {
105*f22ef01cSRoman Divacky   // Check empty set first to quickly find poor test functions.
106*f22ef01cSRoman Divacky   if (GetTestResult(changeset_ty()))
107*f22ef01cSRoman Divacky     return changeset_ty();
108*f22ef01cSRoman Divacky 
109*f22ef01cSRoman Divacky   // Otherwise run the real delta algorithm.
110*f22ef01cSRoman Divacky   changesetlist_ty Sets;
111*f22ef01cSRoman Divacky   Split(Changes, Sets);
112*f22ef01cSRoman Divacky 
113*f22ef01cSRoman Divacky   return Delta(Changes, Sets);
114*f22ef01cSRoman Divacky }
115