Chris@148: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@148: 
Chris@148: /*
Chris@148:     Sonic Visualiser
Chris@148:     An audio file viewer and annotation editor.
Chris@148:     Centre for Digital Music, Queen Mary, University of London.
Chris@148:     This file copyright 2006 Chris Cannam.
Chris@148:     
Chris@148:     This program is free software; you can redistribute it and/or
Chris@148:     modify it under the terms of the GNU General Public License as
Chris@148:     published by the Free Software Foundation; either version 2 of the
Chris@148:     License, or (at your option) any later version.  See the file
Chris@148:     COPYING included with this distribution for more information.
Chris@148: */
Chris@148: 
Chris@148: #include "CSVFileWriter.h"
dev@1436: #include "CSVStreamWriter.h"
Chris@148: 
Chris@150: #include "model/Model.h"
Chris@148: #include "model/SparseOneDimensionalModel.h"
Chris@148: #include "model/SparseTimeValueModel.h"
Chris@148: #include "model/NoteModel.h"
Chris@148: #include "model/TextModel.h"
Chris@148: 
Chris@675: #include "base/TempWriteFile.h"
Chris@675: #include "base/Exceptions.h"
Chris@839: #include "base/Selection.h"
Chris@675: 
Chris@148: #include <QFile>
Chris@148: #include <QTextStream>
dev@1436: #include <exception>
Chris@148: 
Chris@1063: CSVFileWriter::CSVFileWriter(QString path,
Chris@1063:                              Model *model,
Chris@1063:                              QString delimiter,
Chris@1063:                              DataExportOptions options) :
Chris@148:     m_path(path),
Chris@148:     m_model(model),
Chris@148:     m_error(""),
Chris@1063:     m_delimiter(delimiter),
Chris@1063:     m_options(options)
Chris@148: {
Chris@148: }
Chris@148: 
Chris@148: CSVFileWriter::~CSVFileWriter()
Chris@148: {
Chris@148: }
Chris@148: 
Chris@148: bool
Chris@148: CSVFileWriter::isOK() const
Chris@148: {
Chris@148:     return m_error == "";
Chris@148: }
Chris@148: 
Chris@148: QString
Chris@148: CSVFileWriter::getError() const
Chris@148: {
Chris@148:     return m_error;
Chris@148: }
Chris@148: 
Chris@148: void
Chris@148: CSVFileWriter::write()
Chris@148: {
dev@1436:     Selection all {
dev@1436:         m_model->getStartFrame(),
dev@1436:         m_model->getEndFrame()
dev@1436:     };
dev@1436:     MultiSelection selections;
dev@1436:     selections.addSelection(all);
cannam@1452:     writeSelection(selections); 
Chris@148: }
Chris@148: 
Chris@839: void
cannam@1452: CSVFileWriter::writeSelection(MultiSelection selection)
Chris@839: {
Chris@839:     try {
Chris@839:         TempWriteFile temp(m_path);
Chris@148: 
Chris@839:         QFile file(temp.getTemporaryFilename());
Chris@839:         if (!file.open(QIODevice::WriteOnly | QIODevice::Text)) {
Chris@839:             m_error = tr("Failed to open file %1 for writing")
Chris@839:                 .arg(temp.getTemporaryFilename());
Chris@839:             return;
Chris@839:         }
Chris@839:     
Chris@839:         QTextStream out(&file);
Chris@839: 
Chris@1816:         if (m_options & DataExportIncludeHeader) {
Chris@1833:             out << StringBits::joinDelimited
Chris@1833:                 (m_model->getStringExportHeaders(m_options),
Chris@1833:                  m_delimiter)
Chris@1816:                 << endl;
Chris@1816:         }
Chris@1815:         
cannam@1452:         sv_frame_t blockSize = 65536;
cannam@1452: 
cannam@1452:         if (m_model->isSparse()) {
cannam@1452:             // Write the whole in one go, as re-seeking for each block
cannam@1452:             // may be very costly otherwise
cannam@1452:             sv_frame_t startFrame, endFrame;
cannam@1452:             selection.getExtents(startFrame, endFrame);
cannam@1452:             blockSize = endFrame - startFrame;
cannam@1452:         }
cannam@1452:         
dev@1445:         bool completed = CSVStreamWriter::writeInChunks(
dev@1445:             out,
dev@1445:             *m_model,
cannam@1452:             selection,
dev@1445:             m_reporter,
dev@1445:             m_delimiter,
cannam@1452:             m_options,
cannam@1452:             blockSize
dev@1442:         );
Chris@839: 
Chris@839:         file.close();
dev@1436:         if (completed) {
dev@1436:             temp.moveToTarget();
dev@1436:         }
Chris@839: 
Chris@839:     } catch (FileOperationFailed &f) {
Chris@839:         m_error = f.what();
dev@1436:     } catch (const std::exception &e) { // ProgressReporter could throw
dev@1436:         m_error = e.what();
Chris@839:     }
Chris@839: }