]> git.saurik.com Git - wxWidgets.git/blob - src/gtk1/dataobj.cpp
Fix the size of the buffer passed to wxTmemcpy() in wxTextDataObject.
[wxWidgets.git] / src / gtk1 / dataobj.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: src/gtk1/dataobj.cpp
3 // Purpose: wxDataObject class
4 // Author: Robert Roebling
5 // Id: $Id$
6 // Copyright: (c) 1998 Robert Roebling
7 // Licence: wxWindows licence
8 ///////////////////////////////////////////////////////////////////////////////
9
10 // For compilers that support precompilation, includes "wx.h".
11 #include "wx/wxprec.h"
12
13 #if wxUSE_DATAOBJ
14
15 #include "wx/dataobj.h"
16
17 #ifndef WX_PRECOMP
18 #include "wx/log.h"
19 #include "wx/app.h"
20 #include "wx/image.h"
21 #endif
22
23 #include "wx/mstream.h"
24 #include "wx/uri.h"
25
26 #include <gdk/gdk.h>
27
28 //-------------------------------------------------------------------------
29 // global data
30 //-------------------------------------------------------------------------
31
32 GdkAtom g_textAtom = 0;
33 GdkAtom g_altTextAtom = 0;
34 GdkAtom g_pngAtom = 0;
35 GdkAtom g_fileAtom = 0;
36
37 //-------------------------------------------------------------------------
38 // wxDataFormat
39 //-------------------------------------------------------------------------
40
41 wxDataFormat::wxDataFormat()
42 {
43 // do *not* call PrepareFormats() from here for 2 reasons:
44 //
45 // 1. we will have time to do it later because some other Set function
46 // must be called before we really need them
47 //
48 // 2. doing so prevents us from declaring global wxDataFormats because
49 // calling PrepareFormats (and thus gdk_atom_intern) before GDK is
50 // initialised will result in a crash
51 m_type = wxDF_INVALID;
52 m_format = (GdkAtom) 0;
53 }
54
55 wxDataFormat::wxDataFormat( wxDataFormatId type )
56 {
57 PrepareFormats();
58 SetType( type );
59 }
60
61 void wxDataFormat::InitFromString( const wxString &id )
62 {
63 PrepareFormats();
64 SetId( id );
65 }
66
67 wxDataFormat::wxDataFormat( NativeFormat format )
68 {
69 PrepareFormats();
70 SetId( format );
71 }
72
73 void wxDataFormat::SetType( wxDataFormatId type )
74 {
75 PrepareFormats();
76
77 m_type = type;
78
79 #if wxUSE_UNICODE
80 if (m_type == wxDF_UNICODETEXT)
81 m_format = g_textAtom;
82 else if (m_type == wxDF_TEXT)
83 m_format = g_altTextAtom;
84 #else
85 if (m_type == wxDF_TEXT || m_type == wxDF_UNICODETEXT)
86 m_format = g_textAtom;
87 #endif
88 else
89 if (m_type == wxDF_BITMAP)
90 m_format = g_pngAtom;
91 else
92 if (m_type == wxDF_FILENAME)
93 m_format = g_fileAtom;
94 else
95 {
96 wxFAIL_MSG( wxT("invalid dataformat") );
97 }
98 }
99
100 wxDataFormatId wxDataFormat::GetType() const
101 {
102 return m_type;
103 }
104
105 wxString wxDataFormat::GetId() const
106 {
107 gchar* atom_name = gdk_atom_name( m_format );
108 wxString ret = wxString::FromAscii( atom_name );
109 g_free(atom_name);
110 return ret;
111 }
112
113 void wxDataFormat::SetId( NativeFormat format )
114 {
115 PrepareFormats();
116 m_format = format;
117
118 if (m_format == g_textAtom)
119 #if wxUSE_UNICODE
120 m_type = wxDF_UNICODETEXT;
121 #else
122 m_type = wxDF_TEXT;
123 #endif
124 else
125 if (m_format == g_altTextAtom)
126 m_type = wxDF_TEXT;
127 else
128 if (m_format == g_pngAtom)
129 m_type = wxDF_BITMAP;
130 else
131 if (m_format == g_fileAtom)
132 m_type = wxDF_FILENAME;
133 else
134 m_type = wxDF_PRIVATE;
135 }
136
137 void wxDataFormat::SetId( const wxString& id )
138 {
139 PrepareFormats();
140 m_type = wxDF_PRIVATE;
141 m_format = gdk_atom_intern( id.ToAscii(), FALSE );
142 }
143
144 void wxDataFormat::PrepareFormats()
145 {
146 // VZ: GNOME included in RedHat 6.1 uses the MIME types below and not the
147 // atoms STRING and file:ALL as the old code was, but normal X apps
148 // use STRING for text selection when transfering the data via
149 // clipboard, for example, so do use STRING for now (GNOME apps will
150 // probably support STRING as well for compatibility anyhow), but use
151 // text/uri-list for file dnd because compatibility is not important
152 // here (with whom?)
153 if (!g_textAtom)
154 #if wxUSE_UNICODE
155 g_textAtom = gdk_atom_intern( "UTF8_STRING", FALSE );
156 g_altTextAtom = gdk_atom_intern( "STRING", FALSE );
157 #else
158 g_textAtom = gdk_atom_intern( "STRING" /* "text/plain" */, FALSE );
159 #endif
160 if (!g_pngAtom)
161 g_pngAtom = gdk_atom_intern( "image/png", FALSE );
162 if (!g_fileAtom)
163 g_fileAtom = gdk_atom_intern( "text/uri-list", FALSE );
164 }
165
166 //-------------------------------------------------------------------------
167 // wxDataObject
168 //-------------------------------------------------------------------------
169
170 wxDataObject::wxDataObject()
171 {
172 }
173
174 wxDataObject::~wxDataObject()
175 {
176 // dtor is empty but needed for Darwin and AIX -- otherwise it doesn't link
177 }
178
179 bool wxDataObject::IsSupportedFormat(const wxDataFormat& format, Direction dir) const
180 {
181 size_t nFormatCount = GetFormatCount(dir);
182 if ( nFormatCount == 1 )
183 {
184 return format == GetPreferredFormat();
185 }
186 else
187 {
188 wxDataFormat *formats = new wxDataFormat[nFormatCount];
189 GetAllFormats(formats,dir);
190
191 size_t n;
192 for ( n = 0; n < nFormatCount; n++ )
193 {
194 if ( formats[n] == format )
195 break;
196 }
197
198 delete [] formats;
199
200 // found?
201 return n < nFormatCount;
202 }
203 }
204
205 // ----------------------------------------------------------------------------
206 // wxFileDataObject
207 // ----------------------------------------------------------------------------
208
209 bool wxFileDataObject::GetDataHere(void *buf) const
210 {
211 wxString filenames;
212
213 for (size_t i = 0; i < m_filenames.GetCount(); i++)
214 {
215 filenames += wxT("file:");
216 filenames += m_filenames[i];
217 filenames += wxT("\r\n");
218 }
219
220 memcpy( buf, filenames.mbc_str(), filenames.length() + 1 );
221
222 return true;
223 }
224
225 size_t wxFileDataObject::GetDataSize() const
226 {
227 size_t res = 0;
228
229 for (size_t i = 0; i < m_filenames.GetCount(); i++)
230 {
231 // This is junk in UTF-8
232 res += m_filenames[i].length();
233 res += 5 + 2; // "file:" (5) + "\r\n" (2)
234 }
235
236 return res + 1;
237 }
238
239 bool wxFileDataObject::SetData(size_t WXUNUSED(size), const void *buf)
240 {
241 m_filenames.Empty();
242
243 // we get data in the text/uri-list format, i.e. as a sequence of URIs
244 // (filenames prefixed by "file:") delimited by "\r\n"
245 wxString filename;
246 for ( const char *p = (const char *)buf; ; p++ )
247 {
248 // some broken programs (testdnd GTK+ sample!) omit the trailing
249 // "\r\n", so check for '\0' explicitly here instead of doing it in
250 // the loop statement to account for it
251 if ( (*p == '\r' && *(p+1) == '\n') || !*p )
252 {
253 size_t lenPrefix = 5; // strlen("file:")
254 if ( filename.Left(lenPrefix).MakeLower() == wxT("file:") )
255 {
256 // sometimes the syntax is "file:filename", sometimes it's
257 // URL-like: "file://filename" - deal with both
258 if ( filename[lenPrefix] == wxT('/') &&
259 filename[lenPrefix + 1] == wxT('/') )
260 {
261 // skip the slashes
262 lenPrefix += 2;
263 }
264
265 AddFile(wxURI::Unescape(filename.c_str() + lenPrefix));
266 filename.Empty();
267 }
268 else
269 {
270 wxLogDebug(wxT("Unsupported URI '%s' in wxFileDataObject"),
271 filename.c_str());
272 }
273
274 if ( !*p )
275 break;
276
277 // skip '\r'
278 p++;
279 }
280 else
281 {
282 filename += *p;
283 }
284 }
285
286 return true;
287 }
288
289 void wxFileDataObject::AddFile( const wxString &filename )
290 {
291 m_filenames.Add( filename );
292 }
293
294 // ----------------------------------------------------------------------------
295 // wxBitmapDataObject
296 // ----------------------------------------------------------------------------
297
298 wxBitmapDataObject::wxBitmapDataObject()
299 {
300 Init();
301 }
302
303 wxBitmapDataObject::wxBitmapDataObject( const wxBitmap& bitmap )
304 : wxBitmapDataObjectBase(bitmap)
305 {
306 Init();
307
308 DoConvertToPng();
309 }
310
311 wxBitmapDataObject::~wxBitmapDataObject()
312 {
313 Clear();
314 }
315
316 void wxBitmapDataObject::SetBitmap( const wxBitmap &bitmap )
317 {
318 ClearAll();
319
320 wxBitmapDataObjectBase::SetBitmap(bitmap);
321
322 DoConvertToPng();
323 }
324
325 bool wxBitmapDataObject::GetDataHere(void *buf) const
326 {
327 if ( !m_pngSize )
328 {
329 wxFAIL_MSG( wxT("attempt to copy empty bitmap failed") );
330
331 return false;
332 }
333
334 memcpy(buf, m_pngData, m_pngSize);
335
336 return true;
337 }
338
339 bool wxBitmapDataObject::SetData(size_t size, const void *buf)
340 {
341 Clear();
342
343 wxCHECK_MSG( wxImage::FindHandler(wxBITMAP_TYPE_PNG) != NULL,
344 false, wxT("You must call wxImage::AddHandler(new wxPNGHandler); to be able to use clipboard with bitmaps!") );
345
346 m_pngSize = size;
347 m_pngData = malloc(m_pngSize);
348
349 memcpy(m_pngData, buf, m_pngSize);
350
351 wxMemoryInputStream mstream((char*) m_pngData, m_pngSize);
352 wxImage image;
353 if ( !image.LoadFile( mstream, wxBITMAP_TYPE_PNG ) )
354 {
355 return false;
356 }
357
358 m_bitmap = wxBitmap(image);
359
360 return m_bitmap.IsOk();
361 }
362
363 void wxBitmapDataObject::DoConvertToPng()
364 {
365 if ( !m_bitmap.IsOk() )
366 return;
367
368 wxCHECK_RET( wxImage::FindHandler(wxBITMAP_TYPE_PNG) != NULL,
369 wxT("You must call wxImage::AddHandler(new wxPNGHandler); to be able to use clipboard with bitmaps!") );
370
371 wxImage image = m_bitmap.ConvertToImage();
372
373 wxCountingOutputStream count;
374 image.SaveFile(count, wxBITMAP_TYPE_PNG);
375
376 m_pngSize = count.GetSize() + 100; // sometimes the size seems to vary ???
377 m_pngData = malloc(m_pngSize);
378
379 wxMemoryOutputStream mstream((char*) m_pngData, m_pngSize);
380 image.SaveFile(mstream, wxBITMAP_TYPE_PNG);
381 }
382
383 #endif // wxUSE_DATAOBJ