[Audacity-cvs] audacity-src/src FileIO.h, NONE, 1.1 FileIO.cpp, NONE, 1.1

[Leland Lucius]

Update of /cvsroot/audacity/audacity-src/src
In directory sc8-pr-cvs11.sourceforge.net:/tmp/cvs-serv6715

Added Files:
        FileIO.h FileIO.cpp 
Log Message:
New file I/O wrapper class to handle local and remote (ftp) files.

--- NEW FILE: FileIO.h ---
/**********************************************************************

  Audacity: A Digital Audio Editor
  
  FileIO.h
 
  Leland Lucius

**********************************************************************/

#ifndef __AUDACITY_FILEIO__
#define __AUDACITY_FILEIO__

#include <wx/object.h>
#include <wx/wfstream.h>
#include <wx/protocol/ftp.h>

class FileIO
{
 public:
   typedef enum FileIOMode
   {
      Input,
      Output
   } FileIOMode;

 public:
   FileIO(const wxString name, FileIOMode mode);
   ~FileIO();

   bool IsOpened();
   
   void Close();

   wxInputStream & Read(void *buffer, size_t size);
   wxOutputStream & Write(const void *buffer, size_t size);

   // On the Mac, this sets the file type and creator.  It does nothing on
   // other platforms.
   void SetCatalogInfo();
   void SetCatalogInfo(wxUint32 type);

 private:
   wxString mName;
   wxURI mUri;
   wxFTP *mFtp;
   wxInputStream *mInputStream;
   wxOutputStream *mOutputStream;
   FileIOMode mMode;
   bool mOpen;
};

#endif

// Indentation settings for Vim and Emacs.
// Please do not modify past this point.
//
// Local Variables:
// c-basic-offset: 3
// indent-tabs-mode: nil
// End:
//
// vim: et sts=3 sw=3

--- NEW FILE: FileIO.cpp ---
/**********************************************************************

  Audacity: A Digital Audio Editor
  
  FileIO.cpp
 
  Leland Lucius

**********************************************************************/

#include "Audacity.h"

#include <wx/defs.h>
#include <wx/filename.h>
#include <wx/wfstream.h>

#include "FileIO.h"

FileIO::FileIO(const wxString name, FileIOMode mode)
: mName(name),
  mMode(mode),
  mFtp(NULL),
  mInputStream(NULL),
  mOutputStream(NULL),
  mOpen(false)
{
   wxString scheme;

   mUri.Create(mName);
   scheme = mUri.GetScheme();

   if (scheme == wxT("ftp")) {
      wxString user = mUri.GetUser();
      wxString pass = mUri.GetPassword();;

      mFtp = new wxFTP();
      mFtp->SetDefaultTimeout(60);

      if (!user.IsEmpty()) {
         mFtp->SetUser(user);
      }

      if (!pass.IsEmpty()) {
         mFtp->SetPassword(pass);
      }

      if (!mFtp->Connect(mUri.GetServer())) {
         wxPrintf(wxT("Couldn't connect to server\n"));

         delete mFtp;
         mFtp = NULL;

         return;
      }

      mFtp->SetBinary();
      mFtp->SetPassive(true);

      if (mMode == FileIO::Input) {
         mInputStream = mFtp->GetInputStream(mUri.GetPath());
         if (mInputStream == NULL) {
            wxPrintf(wxT("Couldn't get input stream: %s\n"), 
mUri.GetPath().c_str());
   
            delete mFtp;
            mFtp = NULL;
   
            return;
         }
      }
      else {
         mOutputStream = mFtp->GetOutputStream(mUri.GetPath());
         if (mOutputStream == NULL) {
            wxPrintf(wxT("Couldn't get output stream: %s\n"), 
mUri.GetPath().c_str());
   
            delete mFtp;
            mFtp = NULL;
   
            return;
         }
      }

      mOpen = true;
   }
   else {
      if (mMode == FileIO::Input) {
         mInputStream = new wxFFileInputStream(mName);
         if (mInputStream == NULL) {
            wxPrintf(wxT("Couldn't get input stream: %s\n"), name.c_str());
            return;
         }
      }
      else {
         mOutputStream = new wxFFileOutputStream(mName);
         if (mOutputStream == NULL) {
            wxPrintf(wxT("Couldn't get output stream: %s\n"), name.c_str());
            return;
         }
      }

      mOpen = true;
   }
}

FileIO::~FileIO()
{
   Close();
}

bool FileIO::IsOpened()
{
   return mOpen;
}

void FileIO::Close()
{
   if (mOutputStream) {
      delete mOutputStream;
      mOutputStream = NULL;
   }

   if (mInputStream) {
      delete mInputStream;
      mInputStream = NULL;
   }

   if (mFtp) {
      delete mFtp;
      mFtp = NULL;
   }

   mOpen = false;
}

// MacOS: set the file type/creator so that the OS knows it's an MP3
// file which was created by Audacity
      
void FileIO::SetCatalogInfo()
{
#ifdef __WXMAC__
   if (!mOpen || mFtp != NULL) {
      return;
   }

   wxUint32 type;
   wxFileName fn(mName);
   wxString ext = fn.GetExt().ToUpper() + wxT("    ");

   type = (ext[0] & 0xff) << 24 |
          (ext[1] & 0xff) << 16 |
          (ext[2] & 0xff) << 8  |
          (ext[3] & 0xff);

   SetCatalogInfo(type);
#endif

   return;
}

void FileIO::SetCatalogInfo(wxUint32 type)
{
#ifdef __WXMAC__
   if (!mOpen || mFtp != NULL) {
      return;
   }

   wxFileName fn(mName);

   fn.MacSetTypeAndCreator(type, AUDACITY_CREATOR);
#endif

   return;
}

wxInputStream & FileIO::Read(void *buf, size_t size)
{
   if (mInputStream == NULL) {
      return *mInputStream;
   }

   return mInputStream->Read(buf, size);
}

wxOutputStream & FileIO::Write(const void *buf, size_t size)
{
   if (mOutputStream == NULL) {
      return *mOutputStream;
   }
   
   return mOutputStream->Write(buf, size);
}

// Indentation settings for Vim and Emacs.
// Please do not modify past this point.
//
// Local Variables:
// c-basic-offset: 3
// indent-tabs-mode: nil
// End:
//
// vim: et sts=3 sw=3


-------------------------------------------------------------------------
Take Surveys. Earn Cash. Influence the Future of IT
Join SourceForge.net's Techsay panel and you'll get the chance to share your
opinions on IT & business topics through brief surveys-and earn cash
http://www.techsay.com/default.php?page=join.php&p=sourceforge&CID=DEVDEV
_______________________________________________
Audacity-cvs mailing list
[EMAIL PROTECTED]
https://lists.sourceforge.net/lists/listinfo/audacity-cvs