← Back to branch summary

~ubuntu-branches/ubuntu/trusty/aegisub/trusty

« back to all changes in this revision

Viewing changes to src/text_file_writer.cpp

  • Committer: Package Import Robot
  • Author(s): Sebastian Reichel
  • Date: 2012-03-16 22:58:00 UTC
  • Revision ID: package-import@ubuntu.com-20120316225800-yfb8h9e5n04rk46a
Tags: upstream-2.1.9
Import upstream version 2.1.9

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/text_file_writer.cpp
 
1
// Copyright (c) 2005, Rodrigo Braz Monteiro
 
2
// All rights reserved.
 
3
//
 
4
// Redistribution and use in source and binary forms, with or without
 
5
// modification, are permitted provided that the following conditions are met:
 
6
//
 
7
//   * Redistributions of source code must retain the above copyright notice,
 
8
//     this list of conditions and the following disclaimer.
 
9
//   * Redistributions in binary form must reproduce the above copyright notice,
 
10
//     this list of conditions and the following disclaimer in the documentation
 
11
//     and/or other materials provided with the distribution.
 
12
//   * Neither the name of the Aegisub Group nor the names of its contributors
 
13
//     may be used to endorse or promote products derived from this software
 
14
//     without specific prior written permission.
 
15
//
 
16
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 
17
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 
18
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 
19
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 
20
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 
21
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 
22
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 
23
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 
24
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 
25
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 
26
// POSSIBILITY OF SUCH DAMAGE.
 
27
//
 
28
// -----------------------------------------------------------------------------
 
29
//
 
30
// AEGISUB
 
31
//
 
32
// Website: http://aegisub.cellosoft.com
 
33
// Contact: mailto:zeratul@cellosoft.com
 
34
//
 
35
 
 
36
 
 
37
///////////
 
38
// Headers
 
39
#include "config.h"
 
40
 
 
41
#include <fstream>
 
42
#include "text_file_writer.h"
 
43
#include "options.h"
 
44
 
 
45
 
 
46
///////////////
 
47
// Constructor
 
48
TextFileWriter::TextFileWriter(wxString _filename,wxString enc) {
 
49
        // Setup
 
50
        open = false;
 
51
        customConv = false;
 
52
        IsFirst = true;
 
53
        filename = _filename;
 
54
 
 
55
        // Set encoding
 
56
        encoding = enc;
 
57
        if (encoding == _T("Local") || (encoding.IsEmpty() && Options.AsText(_T("Save Charset")).Lower() == _T("local"))) {
 
58
                conv = &wxConvLocal;
 
59
                wxFontEncoding sysenc = wxLocale::GetSystemEncoding();
 
60
                if (sysenc == wxFONTENCODING_UTF8 || sysenc == wxFONTENCODING_UTF7 ||
 
61
                        sysenc == wxFONTENCODING_UNICODE) // that last one may be a bit questionable
 
62
                        IsUnicode = true;
 
63
                else
 
64
                        IsUnicode = false;
 
65
        }
 
66
        else {
 
67
                if (encoding.IsEmpty()) encoding = Options.AsText(_T("Save Charset"));
 
68
                if (encoding == _T("US-ASCII")) encoding = _T("ISO-8859-1");
 
69
                conv = new wxCSConv(encoding);
 
70
                customConv = true;
 
71
                IsUnicode = encoding.Left(3) == _T("UTF");
 
72
        }
 
73
 
 
74
        // Open file
 
75
        Open();
 
76
}
 
77
 
 
78
 
 
79
//////////////
 
80
// Destructor
 
81
TextFileWriter::~TextFileWriter() {
 
82
        Close();
 
83
}
 
84
 
 
85
 
 
86
/////////////
 
87
// Open file
 
88
void TextFileWriter::Open() {
 
89
        // Open file
 
90
        if (open) return;
 
91
#ifdef WIN32
 
92
        file.open(filename.wc_str(),std::ios::out | std::ios::binary | std::ios::trunc);
 
93
#else
 
94
        file.open(wxFNCONV(filename),std::ios::out | std::ios::binary | std::ios::trunc);
 
95
#endif
 
96
        if (!file.is_open()) {
 
97
                throw _T("Failed opening file for writing.");
 
98
        }
 
99
        open = true;
 
100
 
 
101
        // Set encoding
 
102
        SetEncoding();
 
103
}
 
104
 
 
105
 
 
106
//////////////
 
107
// Close file
 
108
void TextFileWriter::Close() {
 
109
        if (!open) return;
 
110
        file.close();
 
111
        open = false;
 
112
        if (customConv) delete conv;
 
113
}
 
114
 
 
115
 
 
116
/////////////////
 
117
// Write to file
 
118
void TextFileWriter::WriteLineToFile(wxString line,bool addLineBreak) {
 
119
        // Make sure it's loaded
 
120
        if (!open) Open();
 
121
 
 
122
        // Add line break
 
123
        wxString temp = line;
 
124
        if (addLineBreak) temp += _T("\r\n");
 
125
 
 
126
        // Add BOM if it's the first line and the target format is Unicode
 
127
        if (IsFirst && IsUnicode) {
 
128
                wchar_t bom = 0xFEFF;
 
129
                temp = wxString(bom) + temp;
 
130
        }
 
131
        IsFirst = false;
 
132
 
 
133
        // 16-bit
 
134
        if (Is16) {
 
135
                wxWCharBuffer buf = temp.wc_str(*conv);
 
136
                if (!buf.data())
 
137
                        return;
 
138
                size_t len = wcslen(buf.data());
 
139
                file.write((const char*)buf.data(),len*sizeof(wchar_t));
 
140
        }
 
141
 
 
142
        // 8-bit
 
143
        else {
 
144
                wxCharBuffer buf = temp.mb_str(*conv);
 
145
                if (!buf.data())
 
146
                        return;
 
147
                size_t len = strlen(buf.data());
 
148
                file.write(buf.data(),len);
 
149
        }
 
150
}
 
151
 
 
152
 
 
153
////////////////
 
154
// Set encoding
 
155
void TextFileWriter::SetEncoding() {
 
156
        // Prepare
 
157
        Is16 = false;
 
158
 
 
159
        // UTF-16
 
160
        if (encoding.Left(6) == _T("UTF-16")) {
 
161
                Is16 = true;
 
162
        }
 
163
}