lib/wx/shapes/shape_data_object.rb in wxruby3-shapes-0.9.0.pre.beta.3 vs lib/wx/shapes/shape_data_object.rb in wxruby3-shapes-0.9.5

@@ -2,42 +2,123 @@
 # Copyright (c) M.J.N. Corino, The Netherlands
 
 module Wx::SF
 
   # Class encapsulating data object used during clipboard operations with shapes.
-  class ShapeDataObject < Wx::DataObjectSimpleBase
+  class ShapeDataObject < Wx::DataObject
 
-    DataFormatID = 'ShapeFrameWorkDataFormat1_0'
+    DataFormatID = Wx::DataFormat.new('ShapeFrameWorkDataFormat1_0')
 
     # @overload initialize()
     #   Default constructor
     # @overload initialize(selection)
     #   User constructor
     #   @param [Array<Wx::SF::Shape>] selection List of shapes which should be stored in the data object
     def initialize(selection = nil)
-      super(Wx::DataFormat.new(DataFormatID))
-      @data = selection ? selection.serialize : ''
+      unless selection.nil? || (selection.is_a?(Array) && selection.all? { |e| e.is_a?(Shape) })
+        raise SFException, 'Expected nil or Array<Wx::SF::Shape>'
+      end
+      super()
+      @shapes = selection
+      @data = nil
+      @format = DataFormatID.get_type
     end
 
-    # Returns size of the data object
-    # @return [Integer]
-    def _get_data_size
-      @data.bytesize
+    def get_as_text
+      if  @format == DataFormatID.get_type
+        @data ||= (@shapes ? @shapes.serialize(format: :yaml) : nil)
+      end
+      @data || ''
     end
 
-    # Exports data from data object.
-    # @return [Boolean] true on success, otherwise false
-    def _get_data
-      @data
+    def get_as_shapes
+      if @format == Wx::DataFormatId::DF_TEXT || @format == Wx::DataFormatId::DF_UNICODETEXT
+        @shapes ||= (@data && @data.size>0 ? FIRM.deserialize(@data, format: :yaml) : nil)
+      end
+      @shapes || []
     end
 
-    # Function should inport data from data object from given buffer.
-    # @param [String] buf External input data buffer
-    # @return [Boolean] true on success, otherwise false
-    def _set_data(buf)
-      @data = buf ? buf : ''
-      true
+    # List all the formats that we support. By default, the first is
+    # treated as the 'preferred' format; this can be overridden by
+    # providing a get_preferred format.
+    def get_all_formats(direction)
+      [ DataFormatID, Wx::DF_TEXT, Wx::DF_UNICODETEXT ]
     end
+
+    # Do setting the data
+    def set_data(format, the_data)
+      case format.get_type
+      when DataFormatID.get_type
+        @shapes = if the_data.size > 0
+                    begin
+                      FIRM.deserialize(the_data, format: :yaml)
+                    rescue Exception
+                      $stderr.puts "#{$!}\n#{$!.backtrace.join("\n")}"
+                      return false
+                    end
+                  else
+                    nil
+                  end
+        @data = nil
+        @format = format.get_type
+        true
+      when Wx::DataFormatId::DF_TEXT, Wx::DataFormatId::DF_UNICODETEXT
+        @data = the_data
+        @shapes = nil
+        @format = format.get_type
+        true
+      else
+        false
+      end
+    end
+
+    def get_data_size(format)
+      case format.get_type
+      when Wx::DataFormatId::DF_TEXT, Wx::DataFormatId::DF_UNICODETEXT
+        @data ? @data.size : 0
+      when DataFormatID.get_type
+        get_as_text.bytesize
+      else
+        0
+      end
+    end
+
+    # Do getting the data
+    def get_data_here(format)
+      case format.get_type
+      when Wx::DataFormatId::DF_TEXT, Wx::DataFormatId::DF_UNICODETEXT
+        @data
+      when DataFormatID.get_type
+        begin
+          get_as_text
+        rescue Exception
+          $stderr.puts "#{$!}\n#{$!.backtrace.join("\n")}"
+          nil
+        end
+      else
+        nil
+      end
+    end
+
+    # # Returns size of the data object
+    # # @return [Integer]
+    # def _get_data_size
+    #   @data.bytesize
+    # end
+    #
+    # # Exports data from data object.
+    # # @return [Boolean] true on success, otherwise false
+    # def _get_data
+    #   @data
+    # end
+    #
+    # # Function should inport data from data object from given buffer.
+    # # @param [String] buf External input data buffer
+    # # @return [Boolean] true on success, otherwise false
+    # def _set_data(buf)
+    #   @data = buf ? buf : ''
+    #   true
+    # end
 
   end
 
 end