use Accessors

Author: ogauthe

Project.toml

@@ -4,6 +4,7 @@ authors = ["ITensor developers <[email protected]> and contributors"]
 version = "0.2.0"
 
 [deps]
+Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"
 BlockArrays = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"
 BlockSparseArrays = "2c9a651f-6452-4ace-a6ac-809f4280fbb4"
 BroadcastMapConversion = "4a4adec5-520f-4750-bb37-d5e66b4ddeb2"
@@ -20,6 +21,7 @@ TypeParameterAccessors = "7e5a90cf-f82e-492e-a09b-e3e26432c138"
 WignerSymbols = "9f57e263-0b3d-5e2e-b1be-24f2bb48858b"
 
 [compat]
+Accessors = "0.1.39"
 BlockArrays = "1.2.0"
 BlockSparseArrays = "0.2.0"
 BroadcastMapConversion = "0.1.0"

src/fusiontensor/base_interface.jl

@@ -1,18 +1,13 @@
 # This files defines Base functions for FusionTensor
 
+using Accessors: @set
+
 using BlockSparseArrays: @view!
 
-function Base.:*(x::Number, ft::FusionTensor)
-  return FusionTensor(
-    x * data_matrix(ft), codomain_axes(ft), domain_axes(ft), trees_block_mapping(ft)
-  )
-end
+set_data_matrix(ft::FusionTensor, data_matrix) = @set ft.data_matrix = data_matrix
 
-function Base.:*(ft::FusionTensor, x::Number)
-  return FusionTensor(
-    x * data_matrix(ft), codomain_axes(ft), domain_axes(ft), trees_block_mapping(ft)
-  )
-end
+Base.:*(x::Number, ft::FusionTensor) = set_data_matrix(ft, x * data_matrix(ft))
+Base.:*(ft::FusionTensor, x::Number) = set_data_matrix(ft, x * data_matrix(ft))
 
 # tensor contraction is a block data_matrix product.
 function Base.:*(left::FusionTensor, right::FusionTensor)
@@ -26,32 +21,17 @@ Base.:+(ft::FusionTensor) = ft
 # tensor addition is a block data_matrix add.
 function Base.:+(left::FusionTensor, right::FusionTensor)
   @assert matching_axes(axes(left), axes(right))
-  new_data_matrix = data_matrix(left) + data_matrix(right)
-  return FusionTensor(
-    new_data_matrix, codomain_axes(left), domain_axes(left), trees_block_mapping(left)
-  )
+  return set_data_matrix(left, data_matrix(left) + data_matrix(right))
 end
 
-function Base.:-(ft::FusionTensor)
-  new_data_matrix = -data_matrix(ft)
-  return FusionTensor(
-    new_data_matrix, codomain_axes(ft), domain_axes(ft), trees_block_mapping(ft)
-  )
-end
+Base.:-(ft::FusionTensor) = set_data_matrix(ft, -data_matrix(ft))
 
 function Base.:-(left::FusionTensor, right::FusionTensor)
   @assert matching_axes(axes(left), axes(right))
-  new_data_matrix = data_matrix(left) - data_matrix(right)
-  return FusionTensor(
-    new_data_matrix, codomain_axes(left), domain_axes(left), trees_block_mapping(left)
-  )
+  return set_data_matrix(left, data_matrix(left) - data_matrix(right))
 end
 
-function Base.:/(ft::FusionTensor, x::Number)
-  return FusionTensor(
-    data_matrix(ft) / x, codomain_axes(ft), domain_axes(ft), trees_block_mapping(ft)
-  )
-end
+Base.:/(ft::FusionTensor, x::Number) = set_data_matrix(ft, data_matrix(ft) / x)
 
 Base.Array(ft::FusionTensor) = Array(to_array(ft))
 
@@ -75,11 +55,7 @@ Base.axes(ft::FusionTensor) = (codomain_axes(ft)..., domain_axes(ft)...)
 
 # conj is defined as coefficient wise complex conjugation, without axis dual
 Base.conj(ft::FusionTensor{<:Real}) = ft   # same object for real element type
-function Base.conj(ft::FusionTensor)
-  return FusionTensor(
-    conj(data_matrix(ft)), codomain_axes(ft), domain_axes(ft), trees_block_mapping(ft)
-  )
-end
+Base.conj(ft::FusionTensor) = set_data_matrix(ft, conj(data_matrix(ft)))
 
 function Base.copy(ft::FusionTensor)
   return FusionTensor(