Merge pull request #309 from gridap/add_setindex_to_reindexed_array

Added setindex! method for Reindexed.

Author: fverdugo

NEWS.md

@@ -15,6 +15,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
   - New `CDLagrangianRefFE` struct, that provides a Lagrangian reference FE with different conformity per direction. Since PR [#299](https://github.com/gridap/Gridap.jl/pull/299).
   - New `FESpace` method that takes a model and a `RefFE`. Since PR [#299](https://github.com/gridap/Gridap.jl/pull/299).
   - Possibility to have 0 order in `DISC` directions of a `CDLagrangianRefFE`. Since PR [#308](https://github.com/gridap/Gridap.jl/pull/308).
+  - Added setindex! method for Reindexed. Since PR [#309](https://github.com/gridap/Gridap.jl/pull/309).
 
 ### Changed
 

src/Arrays/Reindex.jl

@@ -43,6 +43,13 @@ end
   a.i_to_v[i]
 end
 
+@propagate_inbounds function Base.setindex!(a::Reindexed,v,j::Integer)
+  i = a.j_to_i[j]
+  a.i_to_v[i]=v
+end
+
+
+
 function testitem(a::Reindexed)
   if length(a.j_to_i) == 0
     testitem(a.i_to_v)

test/ArraysTests/ReindexTests.jl

@@ -5,6 +5,14 @@ using FillArrays
 using Gridap.Arrays
 using Gridap.Arrays: AppliedArray
 
+a = [1,2,3]
+b = reindex(a,[3,2,1])
+for i=1:3
+    b[i]=i
+end
+@test b == [1,2,3]
+@test a == [3,2,1]
+
 a = [[1,2,4,5],[2,4,6,7],[4,3,5,1],[2,3]]
 b = [3,1,2]