emufs/external_sort/mergefile.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 04:54:53 ART 2004
  22  * Autores: Leandro Lucarella <llucare@fi.uba.ar>
  23  *----------------------------------------------------------------------------
  24  *
  25  * $Id: tipo1.c 548 2004-05-28 22:44:27Z llucare $
  26  *
  27  */
  28
  29 /** \file
  30  *
  31  * Archivo temporal con un fragmento ordenado del archivo original.
  32  *
  33  * Implementación de un archivo temporal con un fragmento ordenado del archivo
  34  * original utilizado para hacer el ordenamiento por fusión (merge sort).
  35  *
  36  */
  37
  38 #include "mergefile.h"
  39 #include <malloc.h>
  40 #include <stdio.h>
  41 #include <string.h>
  42 #include <assert.h>
  43
  44 char* mergefile_makefilename(int i)
  45 {
  46         size_t size = sizeof(MERGEFILE_TEMPLATE) + 8; /* 8 más para el int */
  47         char* filename = malloc(size);
  48         if (!filename) return 0;
  49         /* Hasta que me de la memoria para crear el nombre del archivo */
  50         while (snprintf(filename, size, MERGEFILE_TEMPLATE, i) >= size) {
  51                 char* new;
  52                 size += 8;
  53                 new = realloc(filename, size);
  54                 if (new) {
  55                         filename = new;
  56                 } else {
  57                         free(filename);
  58                         return 0;
  59                 }
  60         }
  61         return filename;
  62 }
  63
  64 MERGEFILE* mergefile_new(int i)
  65 {
  66         MERGEFILE* mf = malloc(sizeof(MERGEFILE));
  67         if (!mf) {
  68                 return 0;
  69         }
  70         /* asigno el nombre de archivo */
  71         if (!(mf->filename = mergefile_makefilename(i))) {
  72                 free(mf);
  73                 return 0;
  74         }
  75         /* abre archivo */
  76         if (!(mf->fp = fopen(mf->filename, "rb"))) {
  77                 free(mf->filename);
  78                 free(mf);
  79                 return 0;
  80         }
  81         /* obtiene dato */
  82         if (fscanf(mf->fp, "%i", &(mf->next)) <= 0) {
  83                 fclose(mf->fp);
  84                 free(mf->filename);
  85                 free(mf);
  86                 return 0;
  87         }
  88         mf->more = 1;
  89         return mf;
  90 }
  91
  92 void mergefile_delete(MERGEFILE* mf)
  93 {
  94         assert(mf);
  95         assert(mf->fp);
  96         assert(mf->filename);
  97         fclose(mf->fp);
  98         remove(mf->filename);
  99         free(mf->filename);
 100         free(mf);
 101 }
 102
 103 int mergefile_peek_next(MERGEFILE* mf)
 104 {
 105         assert(mf);
 106         assert(mf->fp);
 107         assert(mf->more);
 108         return mf->next;
 109 }
 110
 111 int mergefile_pop_next(MERGEFILE* mf)
 112 {
 113         int ret;
 114         assert(mf);
 115         assert(mf->fp);
 116         assert(mf->more);
 117         ret = mf->next;
 118         /* obtiene dato, si no hay más, se activa el flag */
 119         if (fscanf(mf->fp, "%i", &(mf->next)) <= 0) mf->more = 0;
 120         return ret;
 121 }
 122
 123 int mergefile_has_more(MERGEFILE* mf)
 124 {
 125         assert(mf);
 126         assert(mf->fp);
 127         return mf->more;
 128 }
 129