Mercurial > hg > svcore
view data/fileio/CSVFileWriter.cpp @ 1436:d61d6c33f14d streaming-csv-writer
Remove some duplication and make use of utility function for writing in chunks to avoid memory growth.
author | Lucas Thompson <dev@lucas.im> |
---|---|
date | Tue, 17 Apr 2018 10:03:49 +0100 |
parents | 815f82508f96 |
children | 09c2ba31a711 |
line wrap: on
line source
/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */ /* Sonic Visualiser An audio file viewer and annotation editor. Centre for Digital Music, Queen Mary, University of London. This file copyright 2006 Chris Cannam. 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 the Free Software Foundation; either version 2 of the License, or (at your option) any later version. See the file COPYING included with this distribution for more information. */ #include "CSVFileWriter.h" #include "CSVStreamWriter.h" #include "model/Model.h" #include "model/SparseOneDimensionalModel.h" #include "model/SparseTimeValueModel.h" #include "model/NoteModel.h" #include "model/TextModel.h" #include "base/TempWriteFile.h" #include "base/Exceptions.h" #include "base/Selection.h" #include <QFile> #include <QTextStream> #include <exception> CSVFileWriter::CSVFileWriter(QString path, Model *model, QString delimiter, DataExportOptions options) : m_path(path), m_model(model), m_error(""), m_delimiter(delimiter), m_options(options) { } CSVFileWriter::~CSVFileWriter() { } bool CSVFileWriter::isOK() const { return m_error == ""; } QString CSVFileWriter::getError() const { return m_error; } void CSVFileWriter::write() { Selection all { m_model->getStartFrame(), m_model->getEndFrame() }; MultiSelection selections; selections.addSelection(all); writeSelection(&selections); } void CSVFileWriter::writeSelection(MultiSelection *selection) { try { TempWriteFile temp(m_path); QFile file(temp.getTemporaryFilename()); if (!file.open(QIODevice::WriteOnly | QIODevice::Text)) { m_error = tr("Failed to open file %1 for writing") .arg(temp.getTemporaryFilename()); return; } QTextStream out(&file); bool completed = false; for (const auto& bounds : selection->getSelections()) { completed = CSV::writeToStreamInChunks( out, *m_model, bounds, m_reporter, m_delimiter, m_options ); if (!completed) { break; } } file.close(); if (completed) { temp.moveToTarget(); } } catch (FileOperationFailed &f) { m_error = f.what(); } catch (const std::exception &e) { // ProgressReporter could throw m_error = e.what(); } }