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