QvisDownloader.C 6.34 KB
Newer Older
whitlocb's avatar
whitlocb committed
1 2
/*****************************************************************************
*
3
* Copyright (c) 2000 - 2013, Lawrence Livermore National Security, LLC
whitlocb's avatar
whitlocb committed
4
* Produced at the Lawrence Livermore National Laboratory
5
* LLNL-CODE-442911
whitlocb's avatar
whitlocb committed
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
* All rights reserved.
*
* This file is  part of VisIt. For  details, see https://visit.llnl.gov/.  The
* full copyright notice is contained in the file COPYRIGHT located at the root
* of the VisIt distribution or at http://www.llnl.gov/visit/copyright.html.
*
* Redistribution  and  use  in  source  and  binary  forms,  with  or  without
* modification, are permitted provided that the following conditions are met:
*
*  - Redistributions of  source code must  retain the above  copyright notice,
*    this list of conditions and the disclaimer below.
*  - Redistributions in binary form must reproduce the above copyright notice,
*    this  list of  conditions  and  the  disclaimer (as noted below)  in  the
*    documentation and/or other materials provided with the distribution.
*  - Neither the name of  the LLNS/LLNL nor the names of  its contributors may
*    be used to endorse or promote products derived from this software without
*    specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT  HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR  IMPLIED WARRANTIES, INCLUDING,  BUT NOT  LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND  FITNESS FOR A PARTICULAR  PURPOSE
* ARE  DISCLAIMED. IN  NO EVENT  SHALL LAWRENCE  LIVERMORE NATIONAL  SECURITY,
* LLC, THE  U.S.  DEPARTMENT OF  ENERGY  OR  CONTRIBUTORS BE  LIABLE  FOR  ANY
* DIRECT,  INDIRECT,   INCIDENTAL,   SPECIAL,   EXEMPLARY,  OR   CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT  LIMITED TO, PROCUREMENT OF  SUBSTITUTE GOODS OR
* SERVICES; LOSS OF  USE, DATA, OR PROFITS; OR  BUSINESS INTERRUPTION) HOWEVER
* CAUSED  AND  ON  ANY  THEORY  OF  LIABILITY,  WHETHER  IN  CONTRACT,  STRICT
* LIABILITY, OR TORT  (INCLUDING NEGLIGENCE OR OTHERWISE)  ARISING IN ANY  WAY
* OUT OF THE  USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
* DAMAGE.
*
*****************************************************************************/
#include <QvisDownloader.h>

#include <QByteArray>
#include <QFile>
42
#include <QNetworkReply>
whitlocb's avatar
whitlocb committed
43 44 45 46 47 48 49 50 51 52 53 54 55 56

// ****************************************************************************
// Method: QvisDownloader::QvisDownloader
//
// Purpose: 
//   Constructor.
//
// Arguments:
//   parent : The object's parent.
//
// Programmer: Brad Whitlock
// Creation:   Thu Oct  2 10:28:55 PDT 2008
//
// Modifications:
57 58
//   Brad Whitlock, Tue Nov  1 13:02:42 PDT 2011
//   Rewrite.
whitlocb's avatar
whitlocb committed
59 60 61 62 63
//   
// ****************************************************************************

QvisDownloader::QvisDownloader(QObject *parent) : QObject(parent)
{
64 65
    file = NULL;
    bytes = NULL;
whitlocb's avatar
whitlocb committed
66

67 68 69
    manager = new QNetworkAccessManager(this);
    connect(manager, SIGNAL(finished(QNetworkReply *)),
            this, SLOT(finished(QNetworkReply *)));
whitlocb's avatar
whitlocb committed
70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107
}

// ****************************************************************************
// Method: QvisDownloader::~QvisDownloader
//
// Purpose: 
//   Destructor.
//
// Programmer: Brad Whitlock
// Creation:   Thu Oct  2 10:29:19 PDT 2008
//
// Modifications:
//   
// ****************************************************************************

QvisDownloader::~QvisDownloader()
{
}

// ****************************************************************************
// Method: QvisDownloader::get
//
// Purpose: 
//   Gets the named remote file and puts its contents in the passed byte array.
//
// Arguments:
//   remoteFile : The name of the remote file.
//   b          : A pointer to the byte array that will be used to store the
//                file's bytes.
//
// Returns:    True.
//
// Note:       
//
// Programmer: Brad Whitlock
// Creation:   Thu Oct  2 10:35:28 PDT 2008
//
// Modifications:
108 109
//   Brad Whitlock, Tue Nov  1 13:02:42 PDT 2011
//   Rewrite.
whitlocb's avatar
whitlocb committed
110 111 112 113
//   
// ****************************************************************************

bool
114
QvisDownloader::get(const QUrl &url, QByteArray *b)
whitlocb's avatar
whitlocb committed
115
{
116 117 118 119 120
    QNetworkRequest request(url);
    QNetworkReply *reply = manager->get(request);
    connect(reply, SIGNAL(downloadProgress(qint64,qint64)),
            this, SIGNAL(downloadProgress(qint64,qint64)));

whitlocb's avatar
whitlocb committed
121
    bytes = b;
122
    file = NULL;
whitlocb's avatar
whitlocb committed
123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144

    return true;
}

// ****************************************************************************
// Method: QvisDownloader::get
//
// Purpose: 
//   Gets the named remote file and puts its contents in the passed byte array.
//
// Arguments:
//   remoteFile : The name of the remote file.
//   localFile  : The name of the local file to create.
//
// Returns:    True if the local file can be created; false otherwise.
//
// Note:       
//
// Programmer: Brad Whitlock
// Creation:   Thu Oct  2 10:35:28 PDT 2008
//
// Modifications:
145 146 147
//   Brad Whitlock, Tue Nov  1 13:02:42 PDT 2011
//   Rewrite.
//
whitlocb's avatar
whitlocb committed
148 149 150 151
// ****************************************************************************


bool
152
QvisDownloader::get(const QUrl &url, const QString &localFile)
whitlocb's avatar
whitlocb committed
153
{
154
    bytes = NULL;
whitlocb's avatar
whitlocb committed
155 156 157 158
    file = new QFile(localFile);
    if (!file->open(QIODevice::WriteOnly))
    {
        delete file;
159
        file = NULL;
whitlocb's avatar
whitlocb committed
160 161 162
        return false;
    }

163 164 165 166 167
    QNetworkRequest request(url);
    QNetworkReply *reply = manager->get(request);
    connect(reply, SIGNAL(downloadProgress(qint64,qint64)),
            this, SIGNAL(downloadProgress(qint64,qint64)));

whitlocb's avatar
whitlocb committed
168 169 170 171 172 173 174 175
    return true;
}

//
// Slots
//

// ****************************************************************************
176
// Method: QvisDownloader::finished
whitlocb's avatar
whitlocb committed
177 178
//
// Purpose: 
179
//   This method gets called when our network request is finished.
whitlocb's avatar
whitlocb committed
180 181
//
// Arguments:
182
//   reply : The network reply.
whitlocb's avatar
whitlocb committed
183 184 185 186 187 188
//
// Returns:    
//
// Note:       
//
// Programmer: Brad Whitlock
189
// Creation:   Tue Nov  1 13:02:57 PDT 2011
whitlocb's avatar
whitlocb committed
190 191 192 193 194 195
//
// Modifications:
//   
// ****************************************************************************

void
196
QvisDownloader::finished(QNetworkReply *reply)
whitlocb's avatar
whitlocb committed
197
{
198 199
    disconnect(reply, SIGNAL(downloadProgress(qint64,qint64)),
               this, SIGNAL(downloadProgress(qint64,qint64)));
whitlocb's avatar
whitlocb committed
200

201
    if(!reply->error())
whitlocb's avatar
whitlocb committed
202
    {
203 204 205 206 207 208 209 210 211 212 213 214 215
        if(bytes != NULL)
        {
             *bytes = reply->readAll();
             bytes = NULL;
        }
        if(file != NULL)
        {
            file->write(reply->readAll());
            file->close();
        }

        bytes = NULL;
        file = NULL;
whitlocb's avatar
whitlocb committed
216 217
    }

218
    emit done(reply->error());
whitlocb's avatar
whitlocb committed
219
}