ostinato/common/abstractfileformat.cpp

/*
Copyright (C) 2011 Srivats P.

This file is part of "Ostinato"

This 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 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>
*/

#include "abstractfileformat.h"

#include "fileformat.h"
#include "pcapfileformat.h"
#include "pdmlfileformat.h"
#include "pythonfileformat.h"

#include <QStringList>

AbstractFileFormat::AbstractFileFormat()
{
    stop_ = false;
}

AbstractFileFormat::~AbstractFileFormat()
{
}

QDialog* AbstractFileFormat::openOptionsDialog()
{
    return NULL;
}

QDialog* AbstractFileFormat::saveOptionsDialog()
{
    return NULL;
}

QStringList AbstractFileFormat::supportedFileTypes()
{
    return QStringList()
        << "Ostinato (*)"
        << "PCAP (*)"
        << "PDML (*.pdml)"
        << "PythonScript (*.py)";
}

void AbstractFileFormat::openStreamsOffline(const QString fileName, 
        OstProto::StreamConfigList &streams, QString &error)
{
    fileName_ = fileName;
    openStreams_ = &streams;
    error_ = &error;
    op_ = kOpen;
    stop_ = false;

    start();
}

void AbstractFileFormat::saveStreamsOffline(
        const OstProto::StreamConfigList streams, 
        const QString fileName, QString &error)
{
    saveStreams_ = streams;
    fileName_ = fileName;
    error_ = &error;
    op_ = kSave;
    stop_ = false;

    start();
}

bool AbstractFileFormat::result()
{
    return result_;
}

AbstractFileFormat* AbstractFileFormat::fileFormatFromFile(
        const QString fileName)
{
    if (fileFormat.isMyFileFormat(fileName))
        return &fileFormat;

    if (pdmlFileFormat.isMyFileFormat(fileName))
        return &pdmlFileFormat;

    if (pcapFileFormat.isMyFileFormat(fileName))
        return &pcapFileFormat;

    return NULL;
}

AbstractFileFormat* AbstractFileFormat::fileFormatFromType(
        const QString fileType)
{

    if (fileFormat.isMyFileType(fileType))
        return &fileFormat;

    if (pdmlFileFormat.isMyFileType(fileType))
        return &pdmlFileFormat;

    if (pcapFileFormat.isMyFileType(fileType))
        return &pcapFileFormat;

    if (pythonFileFormat.isMyFileType(fileType))
        return &pythonFileFormat;

    return NULL;
}

void AbstractFileFormat::cancel()
{
    stop_ = true;
}

void AbstractFileFormat::run()
{
    if (op_ == kOpen)
        result_ = openStreams(fileName_, *openStreams_, *error_);
    else if (op_ == kSave)
        result_ = saveStreams(saveStreams_, fileName_, *error_);
}