pretty looks Numo::NArray#inplace alternative

cop-inplace-accessor.rb

module InplaceAccessor
  def self.append_features(klass)
    klass.class_eval do
      alias _getter []
      alias _inplace inplace
      alias _setter []=

      def inplace
        @inplace ||= _inplace
      end

      def [](*arg)
        if arg.empty?
          return inplace
        else
          _getter(*arg)
        end
      end

      def []=(*arg)
        if arg.size == 1
          self
        else
          _setter(*arg)
        end
      end
    end
  end
end

require "numo/narray"

%w(Bit DComplex DFloat Int16 Int32 Int64 Int8 RObject SComplex SFloat UInt16 UInt32 UInt64 UInt8).each do |type|
  Numo.const_get(type).class_eval do
    include InplaceAccessor
  end
end

a = Numo::DFloat.new(3,5).seq
a[] += 100

p a

# Numo::DFloat#shape=[3,5]
# [[100, 101, 102, 103, 104],
#  [105, 106, 107, 108, 109],
#  [110, 111, 112, 113, 114]]