Skip to content

Commit 97a4ebc

Browse files
committed
rename domain_axes to domain
1 parent fec6193 commit 97a4ebc

File tree

3 files changed

+37
-36
lines changed

3 files changed

+37
-36
lines changed

src/fusiontensor/fusiontensor.jl

Lines changed: 13 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -176,8 +176,8 @@ end
176176
# empty matrix
177177
function FusionTensor(elt::Type, legs::FusionTensorAxes)
178178
S = sector_type(legs)
179-
row_axis, codomain_trees_to_ranges = fuse_axes(S, codomain_axes(legs))
180-
col_axis, domain_trees_to_ranges = flip_domain(fuse_axes(S, dual.(domain_axes(legs)))...)
179+
row_axis, codomain_trees_to_ranges = fuse_axes(S, codomain(legs))
180+
col_axis, domain_trees_to_ranges = flip_domain(fuse_axes(S, dual.(domain(legs)))...)
181181

182182
mat = initialize_data_matrix(elt, row_axis, col_axis)
183183
tree_to_block_mapping = intersect_codomain_domain(
@@ -228,16 +228,17 @@ end
228228
# ============================== FusionTensor interface ==================================
229229

230230
# misc access
231-
for f in [
232-
:(codomain_axes),
233-
:(codomain_axis),
234-
:(domain_axes),
235-
:(domain_axis),
236-
:(ndims_codomain),
237-
:(ndims_domain),
238-
]
239-
@eval $f(ft::FusionTensor) = $f(axes(ft))
240-
end
231+
codomain_axes(ft::FusionTensor) = codomain(axes(ft))
232+
233+
domain_axes(ft::FusionTensor) = domain(axes(ft))
234+
235+
codomain_axis(ft::FusionTensor) = fused_codomain(axes(ft))
236+
237+
domain_axis(ft::FusionTensor) = fused_domain(axes(ft))
238+
239+
ndims_codomain(ft::FusionTensor) = length_codomain(axes(ft))
240+
241+
ndims_domain(ft::FusionTensor) = length_domain(axes(ft))
241242

242243
function charge_block_size(ft::FusionTensor, f1::SectorFusionTree, f2::SectorFusionTree)
243244
b = Tuple(findblock(ft, f1, f2))

src/fusiontensor/fusiontensoraxes.jl

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -112,24 +112,24 @@ end
112112

113113
# ============================== FusionTensor interface ==================================
114114

115-
codomain_axes(fta::FusionTensorAxes) = fta[Block(1)]
115+
codomain(fta::FusionTensorAxes) = fta[Block(1)]
116116

117-
domain_axes(fta::FusionTensorAxes) = fta[Block(2)]
117+
domain(fta::FusionTensorAxes) = fta[Block(2)]
118118

119-
function codomain_axis(fta::FusionTensorAxes)
120-
if ndims_codomain(fta) == 0
119+
function fused_codomain(fta::FusionTensorAxes)
120+
if length_codomain(fta) == 0
121121
return trivial_axis(fta)
122122
end
123-
return (codomain_axes(fta)...)
123+
return (codomain(fta)...)
124124
end
125125

126-
function domain_axis(fta::FusionTensorAxes)
127-
if ndims_domain(fta) == 0
126+
function fused_domain(fta::FusionTensorAxes)
127+
if length_domain(fta) == 0
128128
return dual(trivial_axis(fta))
129129
end
130-
return dual((dual.(domain_axes(fta))...))
130+
return dual((dual.(domain(fta))...))
131131
end
132132

133-
ndims_codomain(fta::FusionTensorAxes) = length(codomain_axes(fta))
133+
length_codomain(fta::FusionTensorAxes) = length(codomain(fta))
134134

135-
ndims_domain(fta::FusionTensorAxes) = length(domain_axes(fta))
135+
length_domain(fta::FusionTensorAxes) = length(domain(fta))

test/test_fusiontensoraxes.jl

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,10 @@ using FusionTensors:
88
FusionTensorAxes,
99
ndims_domain,
1010
ndims_codomain,
11-
codomain_axes,
12-
codomain_axis,
13-
domain_axes,
14-
domain_axis,
11+
codomain,
12+
domain,
13+
fused_codomain,
14+
fused_domain,
1515
promote_sector_type,
1616
promote_sectors
1717
using GradedArrays:
@@ -68,12 +68,12 @@ end
6868
@test blocks(fta) == blocks(bt)
6969

7070
@test sector_type(fta) === sector_type(g2)
71-
@test length(codomain_axes(fta)) == 2
72-
@test all(map(r -> space_isequal(r, g2), codomain_axes(fta)))
73-
@test length(domain_axes(fta)) == 2
74-
@test all(map(r -> space_isequal(r, g2b), domain_axes(fta)))
75-
@test space_isequal(codomain_axis(fta), g2 g2)
76-
@test space_isequal(domain_axis(fta), dual(g2 g2))
71+
@test length(codomain(fta)) == 2
72+
@test all(map(r -> space_isequal(r, g2), codomain(fta)))
73+
@test length(domain(fta)) == 2
74+
@test all(map(r -> space_isequal(r, g2b), domain(fta)))
75+
@test space_isequal(fused_codomain(fta), g2 g2)
76+
@test space_isequal(fused_domain(fta), dual(g2 g2))
7777
@test space_isequal(trivial_axis(fta), trivial_axis(typeof(s2)))
7878

7979
@test fta == fta
@@ -95,8 +95,8 @@ end
9595
@test blocklengths(fta) == (0, 0)
9696
@test sector_type(fta) == TrivialSector
9797

98-
@test codomain_axes(fta) == ()
99-
@test space_isequal(codomain_axis(fta), trivial_axis(TrivialSector))
100-
@test domain_axes(fta) == ()
101-
@test space_isequal(domain_axis(fta), dual(trivial_axis(TrivialSector)))
98+
@test codomain(fta) == ()
99+
@test space_isequal(fused_codomain(fta), trivial_axis(TrivialSector))
100+
@test domain(fta) == ()
101+
@test space_isequal(fused_domain(fta), dual(trivial_axis(TrivialSector)))
102102
end

0 commit comments

Comments
 (0)