src/common/textfile.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/common/textfile.cpp
   3 // Purpose:     implementation of wxTextFile class
   4 // Author:      Vadim Zeitlin
   5 // Modified by:
   6 // Created:     03.04.98
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) 1998 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
   9 // Licence:     wxWindows licence
  10 ///////////////////////////////////////////////////////////////////////////////
  11
  12 // ============================================================================
  13 // headers
  14 // ============================================================================
  15
  16 #include  "wx/wxprec.h"
  17
  18 #ifdef    __BORLANDC__
  19     #pragma hdrstop
  20 #endif  //__BORLANDC__
  21
  22 #if !wxUSE_FILE || !wxUSE_TEXTBUFFER
  23     #undef wxUSE_TEXTFILE
  24     #define wxUSE_TEXTFILE 0
  25 #endif // wxUSE_FILE
  26
  27 #if wxUSE_TEXTFILE
  28
  29 #ifndef WX_PRECOMP
  30     #include "wx/string.h"
  31     #include "wx/intl.h"
  32     #include "wx/file.h"
  33     #include "wx/log.h"
  34 #endif
  35
  36 #include "wx/textfile.h"
  37 #include "wx/filename.h"
  38 #include "wx/buffer.h"
  39
  40 // ============================================================================
  41 // wxTextFile class implementation
  42 // ============================================================================
  43
  44 wxTextFile::wxTextFile(const wxString& strFileName)
  45           : wxTextBuffer(strFileName)
  46 {
  47 }
  48
  49
  50 // ----------------------------------------------------------------------------
  51 // file operations
  52 // ----------------------------------------------------------------------------
  53
  54 bool wxTextFile::OnExists() const
  55 {
  56     return wxFile::Exists(m_strBufferName);
  57 }
  58
  59
  60 bool wxTextFile::OnOpen(const wxString &strBufferName, wxTextBufferOpenMode OpenMode)
  61 {
  62     wxFile::OpenMode FileOpenMode;
  63
  64     switch ( OpenMode )
  65     {
  66         default:
  67             wxFAIL_MSG( _T("unknown open mode in wxTextFile::Open") );
  68             // fall through
  69
  70         case ReadAccess :
  71             FileOpenMode = wxFile::read;
  72             break;
  73
  74         case WriteAccess :
  75             FileOpenMode = wxFile::write;
  76             break;
  77     }
  78
  79     return m_file.Open(strBufferName.c_str(), FileOpenMode);
  80 }
  81
  82
  83 bool wxTextFile::OnClose()
  84 {
  85     return m_file.Close();
  86 }
  87
  88
  89 bool wxTextFile::OnRead(const wxMBConv& conv)
  90 {
  91     // file should be opened and we must be in it's beginning
  92     wxASSERT( m_file.IsOpened() && m_file.Tell() == 0 );
  93
  94     // read the entire file in memory: this is not the most efficient thing to
  95     // do but there is no good way to avoid it in Unicode build because if we
  96     // read the file block by block we can't convert each block to Unicode
  97     // separately (the last multibyte char in the block might be only partially
  98     // read and so the conversion would fail) and, as the file contents is kept
  99     // in memory by wxTextFile anyhow, it shouldn't be a big problem to read
 100     // the file entirely
 101     const size_t bufSize = (size_t)(m_file.Length() + 4 /* for trailing NULs */ );
 102     size_t bufPos = 0;
 103     wxCharBuffer buf(bufSize - 1 /* it adds 1 internally */);
 104
 105     char block[1024];
 106     for ( bool eof = false; !eof; )
 107     {
 108         // try to read up to the size of the entire block
 109         ssize_t nRead = m_file.Read(block, WXSIZEOF(block));
 110
 111         if ( nRead == wxInvalidOffset )
 112         {
 113             // read error (error message already given in wxFile::Read)
 114             return false;
 115         }
 116
 117         if ( nRead == 0 )
 118             break;
 119
 120         // this shouldn't happen but don't overwrite the buffer if it does
 121         wxCHECK_MSG( bufPos + nRead <= bufSize, false,
 122                      _T("read more than file length?") );
 123
 124         // append to the buffer
 125         memcpy(buf.data() + bufPos, block, nRead);
 126         bufPos += nRead;
 127     }
 128
 129     const wxString str(buf, conv, bufPos);
 130
 131     // this doesn't risk to happen in ANSI build
 132 #if wxUSE_UNICODE
 133     if ( bufSize > 4 && str.empty() )
 134     {
 135         wxLogError(_("Failed to convert file contents to Unicode."));
 136         return false;
 137     }
 138 #endif // wxUSE_UNICODE
 139
 140     free(buf.release()); // we don't need this memory any more
 141
 142
 143     // now break the buffer in lines
 144
 145     // last processed character, we need to know if it was a CR or not
 146     wxChar chLast = '\0';
 147
 148     // the beginning of the current line, changes inside the loop
 149     wxString::const_iterator lineStart = str.begin();
 150     const wxString::const_iterator end = str.end();
 151     for ( wxString::const_iterator p = lineStart; p != end; p++ )
 152     {
 153         const wxChar ch = *p;
 154         switch ( ch )
 155         {
 156             case '\n':
 157                 // could be a DOS or Unix EOL
 158                 if ( chLast == '\r' )
 159                 {
 160                     if ( p - 1 >= lineStart )
 161                     {
 162                         AddLine(wxString(lineStart, p - 1), wxTextFileType_Dos);
 163                     }
 164                     else
 165                     {
 166                         // there were two line endings, so add an empty line:
 167                         AddLine(wxEmptyString, wxTextFileType_Dos);
 168                     }
 169                 }
 170                 else // bare '\n', Unix style
 171                 {
 172                     AddLine(wxString(lineStart, p), wxTextFileType_Unix);
 173                 }
 174
 175                 lineStart = p + 1;
 176                 break;
 177
 178             case '\r':
 179                 if ( chLast == '\r' )
 180                 {
 181                     // Mac empty line
 182                     AddLine(wxEmptyString, wxTextFileType_Mac);
 183                     lineStart = p + 1;
 184                 }
 185                 //else: we don't know what this is yet -- could be a Mac EOL or
 186                 //      start of DOS EOL so wait for next char
 187                 break;
 188
 189             default:
 190                 if ( chLast == '\r' )
 191                 {
 192                     // Mac line termination
 193                     if ( p - 1 >= lineStart )
 194                     {
 195                         AddLine(wxString(lineStart, p - 1), wxTextFileType_Mac);
 196                     }
 197                     else
 198                     {
 199                         // there were two line endings, so add an empty line:
 200                         AddLine(wxEmptyString, wxTextFileType_Mac);
 201                     }
 202                     lineStart = p;
 203                 }
 204         }
 205
 206         chLast = ch;
 207     }
 208
 209     // anything in the last line?
 210     if ( lineStart != end )
 211     {
 212         // add unterminated last line
 213         AddLine(wxString(lineStart, end), wxTextFileType_None);
 214     }
 215
 216     return true;
 217 }
 218
 219
 220 bool wxTextFile::OnWrite(wxTextFileType typeNew, const wxMBConv& conv)
 221 {
 222     wxFileName fn = m_strBufferName;
 223
 224     // We do NOT want wxPATH_NORM_CASE here, or the case will not
 225     // be preserved.
 226     if ( !fn.IsAbsolute() )
 227         fn.Normalize(wxPATH_NORM_ENV_VARS | wxPATH_NORM_DOTS | wxPATH_NORM_TILDE |
 228                      wxPATH_NORM_ABSOLUTE | wxPATH_NORM_LONG);
 229
 230     wxTempFile fileTmp(fn.GetFullPath());
 231
 232     if ( !fileTmp.IsOpened() ) {
 233         wxLogError(_("can't write buffer '%s' to disk."), m_strBufferName.c_str());
 234         return false;
 235     }
 236
 237     size_t nCount = GetLineCount();
 238     for ( size_t n = 0; n < nCount; n++ ) {
 239         fileTmp.Write(GetLine(n) +
 240                       GetEOL(typeNew == wxTextFileType_None ? GetLineType(n)
 241                                                             : typeNew),
 242                       conv);
 243     }
 244
 245     // replace the old file with this one
 246     return fileTmp.Commit();
 247 }
 248
 249 #endif // wxUSE_TEXTFILE