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