/**
 * 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."
 *
 * @file p3dFileDownload.h
 * @author drose
 * @date 2009-06-11
 */

#ifndef P3DFILEDOWNLOAD_H
#define P3DFILEDOWNLOAD_H

#include "p3d_plugin_common.h"
#include "p3dDownload.h"

#include <fstream>

/**
 * This is a specialization on P3DDownload that specifically writes a disk
 * file.
 */
class P3DFileDownload : public P3DDownload {
public:
  P3DFileDownload();
  P3DFileDownload(const P3DFileDownload &copy);

  bool set_filename(const std::string &filename);
  inline const std::string &get_filename() const;

protected:
  virtual bool open_file();
  void close_file();
  virtual bool receive_data(const unsigned char *this_data,
                            size_t this_data_size);
  virtual void download_finished(bool success);

protected:
  ofstream _file;

private:
  std::string _filename;
};

#include "p3dFileDownload.I"

#endif