emufs/external_sort/mergepool.c

   1 /* vim: set noexpandtab tabstop=4 shiftwidth=4 wrap:
   2  *----------------------------------------------------------------------------
   3  *                                  emufs
   4  *----------------------------------------------------------------------------
   5  * This file is part of emufs.
   6  *
   7  * emufs is free software; you can redistribute it and/or modify it under the
   8  * terms of the GNU General Public License as published by the Free Software
   9  * Foundation; either version 2 of the License, or (at your option) any later
  10  * version.
  11  *
  12  * emufs is distributed in the hope that it will be useful, but WITHOUT ANY
  13  * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  14  * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
  15  * details.
  16  *
  17  * You should have received a copy of the GNU General Public License along
  18  * with emufs; if not, write to the Free Software Foundation, Inc., 59 Temple
  19  * Place, Suite 330, Boston, MA  02111-1307  USA
  20  *----------------------------------------------------------------------------
  21  * Creado:  dom may 30 07:05:15 ART 2004
  22  * Autores: Leandro Lucarella <llucare@fi.uba.ar>
  23  *----------------------------------------------------------------------------
  24  *
  25  * $Id$
  26  *
  27  */
  28
  29 /** \file
  30  *
  31  * Pool de archivos temporales para hacer un ordenamiento externo.
  32  *
  33  * Implementación de un pool de archivos temporales para hacer un ordenamiento
  34  * externo.
  35  *
  36  */
  37
  38 #include "mergepool.h"
  39 #include <string.h>
  40 #include <malloc.h>
  41 #include <assert.h>
  42
  43 static int mergepool_switch_to_input(MERGEPOOL* mp);
  44
  45 MERGEPOOL* mergepool_new()
  46 {
  47         MERGEPOOL* mp = malloc(sizeof(MERGEPOOL));
  48         if (!mp) return 0; /* ERROR: no hay más memoria */
  49         mp->pool = 0;
  50         mp->size = 0;
  51         mp->mode = OUTPUT;
  52         return mp;
  53 }
  54
  55 void mergepool_delete(MERGEPOOL* mp)
  56 {
  57         int i;
  58         assert(mp);
  59         assert(mp->pool);
  60         for (i = 0; i < mp->size; i++) mergefile_delete(mp->pool[i]);
  61         free(mp->pool);
  62         free(mp);
  63 }
  64
  65 MERGEFILE* mergepool_add_file(MERGEPOOL* mp)
  66 {
  67         MERGEFILE** pool;
  68         MERGEFILE*  new_mf = mergefile_new();
  69         assert(mp);
  70         assert(mp->mode == OUTPUT);
  71         if (!new_mf) return 0;
  72         if (!(pool = realloc(mp->pool, sizeof(MERGEFILE*) * (mp->size+1)))) {
  73                 mergefile_delete(new_mf);
  74                 return 0;
  75         }
  76         mp->pool = pool;
  77         mp->pool[mp->size] = new_mf;
  78         return mp->pool[mp->size++];
  79 }
  80
  81 int mergepool_append_data(MERGEPOOL* mp, int data)
  82 {
  83         assert(mp);
  84         assert(mp->pool);
  85         assert(mp->pool[mp->size-1]);
  86         assert(mp->mode == OUTPUT);
  87         return mergefile_push(mp->pool[mp->size-1], data);
  88 }
  89
  90 int mergepool_pop_min(MERGEPOOL* mp, int* min)
  91 {
  92         int i;
  93         int assigned = -1;
  94         assert(mp);
  95         assert(mp->pool);
  96         assert(mp->size);
  97         if (mp->mode == OUTPUT) {
  98                 mergepool_switch_to_input(mp);
  99         }
 100         for (i = 0; i < mp->size; i++) {
 101                 if (mergefile_has_more(mp->pool[i])
 102                                 && ((assigned == -1) || mergefile_peek(mp->pool[i]) < *min)) {
 103                         assigned = i;
 104                         *min = mergefile_peek(mp->pool[i]);
 105                 }
 106         }
 107         if (assigned != -1) {
 108                         mergefile_pop(mp->pool[assigned]); /* lo saco del archivo */
 109                         return 1; /* OK, se obtuvo un mínimo */
 110         }
 111         return 0; /* No hay más */
 112 }
 113
 114 int mergepool_switch_to_input(MERGEPOOL* mp)
 115 {
 116         int i;
 117         assert(mp);
 118         assert(mp->pool);
 119         for (i = 0; i < mp->size; i++) {
 120                 if (!mergefile_switch_to_input(mp->pool[i])) return 0;
 121         }
 122         mp->mode = INPUT;
 123         return 1; /* OK */
 124 }
 125