X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Fdata%2Fscratch-writer.c;h=952860cdc5887e00104be1239f089f3eaf2c71b8;hb=4e77f0d303d1ef44ee797941a3bebdcbb37119aa;hp=4e2929a24164590e0440ddb42e460c46fa95c53f;hpb=9db3101d0bbbcfb687acd3e442e550557e4e56b1;p=pspp-builds.git diff --git a/src/data/scratch-writer.c b/src/data/scratch-writer.c index 4e2929a2..952860cd 100644 --- a/src/data/scratch-writer.c +++ b/src/data/scratch-writer.c @@ -1,5 +1,5 @@ /* PSPP - a program for statistical analysis. - Copyright (C) 2006 Free Software Foundation, Inc. + Copyright (C) 2006, 2009 Free Software Foundation, Inc. This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by @@ -34,16 +34,19 @@ #include "xalloc.h" +#define N_(msgid) (msgid) + /* A scratch file writer. */ struct scratch_writer { - struct scratch_handle *handle; /* Underlying scratch handle. */ struct file_handle *fh; /* Underlying file handle. */ - struct case_map *compactor; /* Compacts into handle->dictionary. */ + struct fh_lock *lock; /* Exclusive access to file handle. */ + struct dictionary *dict; /* Dictionary for subwriter. */ + struct case_map *compactor; /* Compacts into dictionary. */ struct casewriter *subwriter; /* Data output. */ }; -static struct casewriter_class scratch_writer_casewriter_class; +static const struct casewriter_class scratch_writer_casewriter_class; /* Opens FH, which must have referent type FH_REF_SCRATCH, and returns a scratch_writer for it, or a null pointer on @@ -53,47 +56,36 @@ struct casewriter * scratch_writer_open (struct file_handle *fh, const struct dictionary *dictionary) { - struct scratch_handle *sh; struct scratch_writer *writer; - struct dictionary *scratch_dict; - struct case_map *compactor; struct casewriter *casewriter; + struct fh_lock *lock; size_t dict_value_cnt; - if (!fh_open (fh, FH_REF_SCRATCH, "scratch file", "we")) + /* Get exclusive write access to handle. */ + /* TRANSLATORS: this fragment will be interpolated into + messages in fh_lock() that identify types of files. */ + lock = fh_lock (fh, FH_REF_SCRATCH, N_("scratch file"), FH_ACC_WRITE, true); + if (lock == NULL) return NULL; - /* Destroy previous contents of handle. */ - sh = fh_get_scratch_handle (fh); - if (sh != NULL) - scratch_handle_destroy (sh); + /* Create writer. */ + writer = xmalloc (sizeof *writer); + writer->lock = lock; + writer->fh = fh_ref (fh); - /* Copy the dictionary and compact if needed. */ - scratch_dict = dict_clone (dictionary); - dict_delete_scratch_vars (scratch_dict); - if (dict_count_values (scratch_dict, 0) - < dict_get_next_value_idx (scratch_dict)) + writer->dict = dict_clone (dictionary); + dict_delete_scratch_vars (writer->dict); + if (dict_count_values (writer->dict, 0) + < dict_get_next_value_idx (writer->dict)) { - compactor = case_map_to_compact_dict (scratch_dict, 0); - dict_compact_values (scratch_dict); + writer->compactor = case_map_to_compact_dict (writer->dict, 0); + dict_compact_values (writer->dict); } else - compactor = NULL; - dict_value_cnt = dict_get_next_value_idx (scratch_dict); - - /* Create new contents. */ - sh = xmalloc (sizeof *sh); - sh->dictionary = scratch_dict; - sh->casereader = NULL; - - /* Create writer. */ - writer = xmalloc (sizeof *writer); - writer->handle = sh; - writer->fh = fh; - writer->compactor = compactor; + writer->compactor = NULL; + dict_value_cnt = dict_get_next_value_idx (writer->dict); writer->subwriter = autopaging_writer_create (dict_value_cnt); - fh_set_scratch_handle (fh, sh); casewriter = casewriter_create (dict_value_cnt, &scratch_writer_casewriter_class, writer); taint_propagate (casewriter_get_taint (writer->subwriter), @@ -107,30 +99,44 @@ scratch_writer_casewriter_write (struct casewriter *w UNUSED, void *writer_, struct ccase *c) { struct scratch_writer *writer = writer_; - struct ccase tmp; - if (writer->compactor) - { - case_map_execute (writer->compactor, c, &tmp); - case_destroy (c); - } - else - case_move (&tmp, c); - casewriter_write (writer->subwriter, &tmp); + casewriter_write (writer->subwriter, + case_map_execute (writer->compactor, c)); } /* Closes WRITER. */ static void scratch_writer_casewriter_destroy (struct casewriter *w UNUSED, void *writer_) { + static unsigned int next_unique_id = 0x12345678; + struct scratch_writer *writer = writer_; struct casereader *reader = casewriter_make_reader (writer->subwriter); if (!casereader_error (reader)) - writer->handle->casereader = reader; - fh_close (writer->fh, "scratch file", "we"); + { + /* Destroy previous contents of handle. */ + struct scratch_handle *sh = fh_get_scratch_handle (writer->fh); + if (sh != NULL) + scratch_handle_destroy (sh); + + /* Create new contents. */ + sh = xmalloc (sizeof *sh); + sh->unique_id = ++next_unique_id; + sh->dictionary = writer->dict; + sh->casereader = reader; + fh_set_scratch_handle (writer->fh, sh); + } + else + { + casereader_destroy (reader); + dict_destroy (writer->dict); + } + + fh_unlock (writer->lock); + fh_unref (writer->fh); free (writer); } -static struct casewriter_class scratch_writer_casewriter_class = +static const struct casewriter_class scratch_writer_casewriter_class = { scratch_writer_casewriter_write, scratch_writer_casewriter_destroy,