124 lines
4.0 KiB
C++
Executable File
124 lines
4.0 KiB
C++
Executable File
// Filename: p3dFileDownload.cxx
|
|
// Created by: drose (11Jun09)
|
|
//
|
|
////////////////////////////////////////////////////////////////////
|
|
//
|
|
// PANDA 3D SOFTWARE
|
|
// Copyright (c) Carnegie Mellon University. All rights reserved.
|
|
//
|
|
// All use of this software is subject to the terms of the revised BSD
|
|
// license. You should have received a copy of this license along
|
|
// with this source code in a file named "LICENSE."
|
|
//
|
|
////////////////////////////////////////////////////////////////////
|
|
|
|
#include "p3dFileDownload.h"
|
|
#include "p3dInstanceManager.h"
|
|
#include "mkdir_complete.h"
|
|
#include "wstring_encode.h"
|
|
|
|
////////////////////////////////////////////////////////////////////
|
|
// Function: P3DFileDownload::Constructor
|
|
// Access: Public
|
|
// Description:
|
|
////////////////////////////////////////////////////////////////////
|
|
P3DFileDownload::
|
|
P3DFileDownload() {
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////
|
|
// Function: P3DFileDownload::Copy Constructor
|
|
// Access: Public
|
|
// Description:
|
|
////////////////////////////////////////////////////////////////////
|
|
P3DFileDownload::
|
|
P3DFileDownload(const P3DFileDownload ©) :
|
|
P3DDownload(copy)
|
|
{
|
|
// We don't copy the filename. You have to copy it yourself.
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////
|
|
// Function: P3DFileDownload::set_filename
|
|
// Access: Public
|
|
// Description: Supplies the target local filename for the download.
|
|
// Returns true on success, false on failure.
|
|
////////////////////////////////////////////////////////////////////
|
|
bool P3DFileDownload::
|
|
set_filename(const string &filename) {
|
|
_filename = filename;
|
|
|
|
return open_file();
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////
|
|
// Function: P3DFileDownload::open_file
|
|
// Access: Protected, Virtual
|
|
// Description: Opens the local file for receiving the download.
|
|
// Returns true on success, false on failure.
|
|
////////////////////////////////////////////////////////////////////
|
|
bool P3DFileDownload::
|
|
open_file() {
|
|
if (!mkfile_complete(_filename, nout)) {
|
|
nout << "Failed to create " << _filename << "\n";
|
|
return false;
|
|
}
|
|
|
|
_file.clear();
|
|
#ifdef _WIN32
|
|
wstring filename_w;
|
|
if (string_to_wstring(filename_w, _filename)) {
|
|
_file.open(filename_w.c_str(), ios::out | ios::trunc | ios::binary);
|
|
}
|
|
#else // _WIN32
|
|
_file.open(_filename.c_str(), ios::out | ios::trunc | ios::binary);
|
|
#endif // _WIN32
|
|
if (!_file) {
|
|
nout << "Failed to open " << _filename << " in write mode\n";
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////
|
|
// Function: P3DFileDownload::close_file
|
|
// Access: Protected
|
|
// Description: Closes the local file.
|
|
////////////////////////////////////////////////////////////////////
|
|
void P3DFileDownload::
|
|
close_file() {
|
|
_file.close();
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////
|
|
// Function: P3DFileDownload::receive_data
|
|
// Access: Protected, Virtual
|
|
// Description: Called as new data is downloaded. Returns true on
|
|
// success, false on failure.
|
|
////////////////////////////////////////////////////////////////////
|
|
bool P3DFileDownload::
|
|
receive_data(const unsigned char *this_data, size_t this_data_size) {
|
|
_file.write((const char *)this_data, this_data_size);
|
|
|
|
if (!_file) {
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////
|
|
// Function: P3DFileDownload::download_finished
|
|
// Access: Protected, Virtual
|
|
// Description: Intended to be overloaded to generate a callback
|
|
// when the download finishes, either successfully or
|
|
// otherwise. The bool parameter is true if the
|
|
// download was successful.
|
|
////////////////////////////////////////////////////////////////////
|
|
void P3DFileDownload::
|
|
download_finished(bool success) {
|
|
P3DDownload::download_finished(success);
|
|
_file.close();
|
|
}
|