Addition of the PMNDArray class replacing PMArray

Author: YvanGuifo

src/Math-Matrix/PMNDArray.class.st

@@ -0,0 +1,152 @@
+Class {
+	#name : #PMNDArray,
+	#superclass : #Object,
+	#instVars : [
+		'array',
+		'shape',
+		'first',
+		'strides'
+	],
+	#category : #'Math-Matrix'
+}
+
+{ #category : #'instance creation' }
+PMNDArray class >> fromNestedArray: anArray [
+
+	^ self new array: anArray flattened withShape: (self shape:anArray)
+]
+
+{ #category : #'instance creation' }
+PMNDArray class >> newWith: anInteger [
+
+	^ self new array: {anInteger} withShape: #(  )
+]
+
+{ #category : #accessing }
+PMNDArray class >> shape: anArray [
+
+anArray isArray ifFalse:[^#()]
+ifTrue:[
+	^ {anArray size}, (self shape: anArray first)
+	]
+
+]
+
+{ #category : #comparing }
+PMNDArray >> = anArray [
+
+	^ array = anArray array & (first = anArray first)
+	  & (strides = anArray strides) & (shape = anArray shape)
+]
+
+{ #category : #accessing }
+PMNDArray >> array [ 
+^array
+]
+
+{ #category : #'as yet unclassified' }
+PMNDArray >> array: aFlatArray withShape: aShape [
+
+	array := aFlatArray.
+	shape := aShape copy.
+	self updateFirst.
+	shape ifNotEmpty: [ self updateStrides]
+]
+
+{ #category : #accessing }
+PMNDArray >> first [ 
+^first
+]
+
+{ #category : #'primitives - file' }
+PMNDArray >> flattenedIndexOf: coords [
+
+	| position |
+	position := 1.
+	coords withIndexDo: [ :elt :i | 
+		position := (elt - 1) * (strides at: i) + position ].
+	^ position
+]
+
+{ #category : #'as yet unclassified' }
+PMNDArray >> fromNestedArray: aFlatArray withShape: aShape [
+
+	array := aFlatArray.
+	shape := aShape copy.
+	self updateFirst.
+	shape ifNotEmpty: [ self updateStrides]
+]
+
+{ #category : #public }
+PMNDArray >> get: coords [
+
+	| position |
+	position := self flattenedIndexOf: coords.
+	^ array at: position
+]
+
+{ #category : #accessing }
+PMNDArray >> rank [
+	^ shape size
+]
+
+{ #category : #'as yet unclassified' }
+PMNDArray >> reshape: aNewShape [
+
+	^ self viewWithShape: aNewShape.
+
+]
+
+{ #category : #initialization }
+PMNDArray >> set: coords value: aValue [
+
+array at: (self flattenedIndexOf: coords) put: aValue
+]
+
+{ #category : #accessing }
+PMNDArray >> shape [
+
+	^ shape
+]
+
+{ #category : #accessing }
+PMNDArray >> size [
+
+^ shape inject: 1 into:  [  :each :product | each * product].
+	
+]
+
+{ #category : #accessing }
+PMNDArray >> strides [ 
+^strides 
+]
+
+{ #category : #'as yet unclassified' }
+PMNDArray >> updateFirst [
+
+	first := Array new: shape size withAll: 1
+]
+
+{ #category : #'as yet unclassified' }
+PMNDArray >> updateStrides [
+
+	strides := Array new: shape size.
+	strides at: shape size put: 1.
+	((shape size -1) to: 1 by: -1) do: [ :i | 
+		strides at: i put: ((strides at: i + 1) * (shape at: i+1))]
+]
+
+{ #category : #'as yet unclassified' }
+PMNDArray >> view [
+
+	"Share only the array"
+
+	^ self viewWithShape: shape
+]
+
+{ #category : #'as yet unclassified' }
+PMNDArray >> viewWithShape: aNewShape [
+
+	^ PMNDArray new array: self array withShape: aNewShape
+
+]