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