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