add destructive twins for most methods [see ievgrafov#32]

Author: ievgrafov

lib/gnuplotrb/animation.rb

@@ -27,6 +27,11 @@ class Animation < Multiplot
     alias_method :add_frame, :add_plot
     alias_method :add_frames, :add_plots
     alias_method :remove_frame, :remove_plot
+    alias_method :update_frame!, :update_plot!
+    alias_method :replace_frame!, :replace_plot!
+    alias_method :add_frame!, :add_plot!
+    alias_method :add_frames!, :add_plots!
+    alias_method :remove_frame!, :remove_plot!
 
     ##
     # ====== Overview

lib/gnuplotrb/mixins/option_handling.rb

@@ -147,6 +147,13 @@ def options(**options)
       end
     end
 
+    ##
+    # Destructive twin of #options
+    def options!(**options)
+      @options = @options ? @options.merge(options) : Hamster::Hash.new(options)
+      self
+    end
+
     private
 
     ##
@@ -162,5 +169,14 @@ def option(key, *value)
         options(key => value)
       end
     end
+
+    def option!(key, *value)
+      options!(key => value)
+    end
+
+    def option=(key, *value)
+      options!(key => value)
+      @options[key]
+    end
   end
 end

lib/gnuplotrb/mixins/plottable.rb

@@ -46,10 +46,16 @@ def plot(*_)
     #   content = multiplot.to_svg(size: [100, 100], fname: 'Arial', fsize: 12)
     def method_missing(meth_id, *args)
       meth = meth_id.id2name
-      if meth[0..2] == 'to_'
+      case
+      when meth[0..2] == 'to_'
         term = meth[3..-1]
         super unless OptionHandling.valid_terminal?(term)
         to_specific_term(term, *args)
+      when meth[-1] == '!'
+        option!(meth[0..-2].to_sym, *args)
+      when meth[-1] == '='
+        option!(meth[0..-2].to_sym, *args)
+        option(meth[0..-2].to_sym)
       else
         option(meth_id, *args)
       end

lib/gnuplotrb/multiplot.rb

@@ -83,6 +83,14 @@ def update_plot(position = 0, **options)
 
     alias_method :update, :update_plot
 
+    def update_plot!(position = 0, **options)
+      return self unless block_given? if options.empty?
+      replacement = @plots[position].options!(options)
+      yield(replacement) if block_given?
+    end
+
+    alias_method :update!, :update_plot!
+
     ##
     # ====== Overview
     # Create new Multiplot object where plot (Plot or Splot object)
@@ -100,6 +108,13 @@ def replace_plot(position = 0, plot)
 
     alias_method :replace, :replace_plot
 
+    def replace_plot!(position = 0, plot)
+      @plots = @plots.set(position, plot)
+    end
+
+    alias_method :replace!, :replace_plot!
+    alias_method :[]=, :replace_plot!
+
     ##
     # ====== Overview
     # Create new Multiplot with given *plots* added before plot at given *position*.
@@ -119,6 +134,14 @@ def add_plots(*plots)
     alias_method :<<, :add_plots
     alias_method :add, :add_plots
 
+    def add_plots!(*plots)
+      plots.unshift(0) unless plots[0].is_a?(Numeric)
+      @plots = @plots.insert(*plots)
+    end
+
+    alias_method :add_plot!, :add_plots!
+    alias_method :add!, :add_plots!
+
     ##
     # ====== Overview
     # Create new Multiplot without plot at given position
@@ -134,6 +157,12 @@ def remove_plot(position = -1)
 
     alias_method :remove, :remove_plot
 
+    def remove_plot!(position = -1)
+      @plots = @plots.delete_at(position)
+    end
+
+    alias_method :remove!, :remove_plot!
+
     ##
     # ====== Overview
     # Equal to #plots[*args]

lib/gnuplotrb/plot.rb

@@ -107,6 +107,10 @@ def update_dataset(position = 0, data: nil, **options)
       new_ds.equal?(old_ds) ? self : replace_dataset(position, new_ds)
     end
 
+    def update_dataset!(position = 0, data: nil, **options)
+      @datasets[position].update!(data, options)
+    end
+
     ##
     # ====== Overview
     # Create new Plot object where dataset at *position* will
@@ -124,6 +128,12 @@ def replace_dataset(position = 0, dataset)
       self.class.new(@datasets.set(position, dataset_from_any(dataset)), @options)
     end
 
+    def replace_dataset!(position = 0, dataset)
+      @datasets = @datasets.set(position, dataset_from_any(dataset))
+    end
+
+    alias_method :[]=, :replace_dataset!
+
     ##
     # ====== Overview
     # Create new Plot object where given datasets will
@@ -139,13 +149,21 @@ def replace_dataset(position = 0, dataset)
     #   cosx_and_sinx = sinx << ['cos(x)']
     def add_datasets(*datasets)
       datasets.map! { |ds| ds.is_a?(Numeric) ? ds : dataset_from_any(ds) }
+      # first element is position where to add datasets
       datasets.unshift(0) unless datasets[0].is_a?(Numeric)
       self.class.new(@datasets.insert(*datasets), @options)
     end
 
     alias_method :add_dataset, :add_datasets
     alias_method :<<, :add_datasets
 
+    def add_datasets(*datasets)
+      datasets.map! { |ds| ds.is_a?(Numeric) ? ds : dataset_from_any(ds) }
+      # first element is position where to add datasets
+      datasets.unshift(0) unless datasets[0].is_a?(Numeric)
+      @datasets = @datasets.insert(*datasets)
+    end
+
     ##
     # ====== Overview
     # Create new Plot object where dataset at given position
@@ -161,6 +179,10 @@ def remove_dataset(position = -1)
       self.class.new(@datasets.delete_at(position), @options)
     end
 
+    def remove_dataset!(position = -1)
+      @datasets = @datasets.delete_at(position)
+    end
+
     ##
     # ====== Overview
     # The same as Plot#datasets[args]

lib/gnuplotrb/staff/datablock.rb

@@ -41,6 +41,16 @@ def update(data)
       end
     end
 
+    def update!(data)
+      data_str = data.to_gnuplot_points
+      if @stored_in_file
+        File.open(@file_name, 'a') { |f| f.puts "\n#{data_str}" }
+      else
+        @data = "#{@data}\n#{data_str}"
+      end
+      self
+    end
+
     ##
     # ====== Overview
     # Returns quoted filename if datablock stored in file or outputs

lib/gnuplotrb/staff/dataset.rb

@@ -123,6 +123,12 @@ def update(data = nil, **options)
       end
     end
 
+    def update!(data = nil, **options)
+      @data.update!(data) if data
+      options!(options)
+      self
+    end
+
     ##
     # ====== Overview
     # Own implementation of #clone. Creates new Dataset if