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 MERGEFILE* mergefile_new()
  45 {
  46         MERGEFILE* mf = malloc(sizeof(MERGEFILE));
  47         if (!mf) {
  48                 return 0;
  49         }
  50         /* abre archivo */
  51         if (!(mf->fp = tmpfile())) {
  52                 free(mf);
  53                 return 0;
  54         }
  55         mf->more = 0;
  56         return mf;
  57 }
  58
  59 void mergefile_delete(MERGEFILE* mf)
  60 {
  61         assert(mf);
  62         assert(mf->fp);
  63         fclose(mf->fp);
  64         free(mf);
  65 }
  66
  67 int mergefile_switch_to_input(MERGEFILE* mf)
  68 {
  69         /* obtiene dato, debe tener al menos uno para ser un mergefile */
  70         if (fseek(mf->fp, 0L, SEEK_SET)) return 0;
  71         if (fscanf(mf->fp, "%i", &(mf->next)) <= 0) return 0;
  72         mf->more = 1;
  73         return 1; /* OK */
  74 }
  75
  76 int mergefile_push(MERGEFILE* mf, int data)
  77 {
  78         assert(mf);
  79         assert(mf->fp);
  80         return fprintf(mf->fp, "%i\n", data);
  81 }
  82
  83 int mergefile_pop(MERGEFILE* mf)
  84 {
  85         int ret;
  86         assert(mf);
  87         assert(mf->fp);
  88         assert(mf->more);
  89         ret = mf->next;
  90         /* obtiene dato, si no hay más, se activa el flag */
  91         if (fscanf(mf->fp, "%i", &(mf->next)) <= 0) mf->more = 0;
  92         return ret;
  93 }
  94
  95 int mergefile_peek(MERGEFILE* mf)
  96 {
  97         assert(mf);
  98         assert(mf->fp);
  99         assert(mf->more);
 100         return mf->next;
 101 }
 102
 103 int mergefile_has_more(MERGEFILE* mf)
 104 {
 105         assert(mf);
 106         assert(mf->fp);
 107         return mf->more;
 108 }
 109