src/revwalk.c

   1 /*
   2  * This file is free software; you can redistribute it and/or modify
   3  * it under the terms of the GNU General Public License, version 2,
   4  * as published by the Free Software Foundation.
   5  *
   6  * In addition to the permissions in the GNU General Public License,
   7  * the authors give you unlimited permission to link the compiled
   8  * version of this file into combinations with other programs,
   9  * and to distribute those combinations without any restriction
  10  * coming from the use of this file.  (The General Public License
  11  * restrictions do apply in other respects; for example, they cover
  12  * modification of the file, and distribution when not linked into
  13  * a combined executable.)
  14  *
  15  * This file is distributed in the hope that it will be useful, but
  16  * WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18  * General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; see the file COPYING.  If not, write to
  22  * the Free Software Foundation, 51 Franklin Street, Fifth Floor,
  23  * Boston, MA 02110-1301, USA.
  24  */
  25
  26 #include "common.h"
  27 #include "commit.h"
  28 #include "revwalk.h"
  29
  30 git_revpool *gitrp_alloc(git_odb *db)
  31 {
  32         git_revpool *walk = git__malloc(sizeof(*walk));
  33         if (!walk)
  34                 return NULL;
  35
  36     memset(walk, 0x0, sizeof(git_revpool));
  37
  38         walk->db = db;
  39         return walk;
  40 }
  41
  42 void gitrp_free(git_revpool *walk)
  43 {
  44     git_commit_list_clear(&(walk->iterator), 0);
  45     git_commit_list_clear(&(walk->roots), 1);
  46         free(walk);
  47 }
  48
  49 void gitrp_push(git_revpool *pool, git_commit *commit)
  50 {
  51     if (commit->object.pool != pool)
  52         return;
  53
  54     if (commit->seen)
  55         return;
  56
  57     if (!commit->parsed)
  58     {
  59         if (git_commit_parse_existing(commit) < 0)
  60             return;
  61     }
  62
  63     // Sanity check: make sure that if the commit
  64     // has been manually marked as uninteresting,
  65     // all the parent commits are too.
  66     if (commit->uninteresting)
  67         git_commit__mark_uninteresting(commit);
  68
  69     commit->seen = 1;
  70
  71     git_commit_list_append(&pool->roots, commit);
  72     git_commit_list_append(&pool->iterator, commit);
  73 }
  74
  75 void gitrp_hide(git_revpool *pool, git_commit *commit)
  76 {
  77     git_commit__mark_uninteresting(commit);
  78     gitrp_push(pool, commit);
  79 }
  80
  81 void gitrp_prepare_walk(git_revpool *pool)
  82 {
  83     git_commit_node *roots;
  84
  85     roots = pool->roots.head;
  86     while (roots)
  87     {
  88         git_commit_list_append(&pool->iterator, roots->commit);
  89         roots = roots->next;
  90     }
  91
  92     pool->walking = 1;
  93 }
  94
  95 git_commit *gitrp_next(git_revpool *pool)
  96 {
  97     git_commit *next;
  98
  99     if (!pool->walking)
 100         gitrp_prepare_walk(pool);
 101
 102     while ((next = git_commit_list_pop_front(&pool->iterator)) != NULL)
 103     {
 104         git_commit_node *parents;
 105
 106         parents = next->parents.head;
 107         while (parents)
 108         {
 109             git_commit *parent = parents->commit;
 110             parents = parents->next;
 111
 112             if (parent->seen)
 113                 continue;
 114
 115             if (parent->parsed == 0)
 116                 git_commit_parse_existing(parent);
 117
 118             git_commit_list_append(&pool->iterator, parent);
 119         }
 120
 121         if (next->uninteresting == 0)
 122             return next;
 123     }
 124
 125     // No commits left to iterate
 126     gitrp_reset(pool);
 127     return NULL;
 128 }
 129
 130 void gitrp_reset(git_revpool *pool)
 131 {
 132     git_commit_list_clear(&pool->iterator, 0);
 133     pool->walking = 0;
 134 }
 135