Add Indexer

[fzimmermann89]

Implements broadcasted indexing on tensors.

Main idea is to parse the indexing expression once, then apply it to all tensor fields.

The indexing rules are such that indexing not violate the shape inveraints of our tensors:
no axes will be removed. new axes are only introduced at the beginning of the tensor (in the other dimensions)

[fzimmermann89]

Stack from ghstack (oldest at bottom):

• getnested #594
• IHeader #591
• -> Add Indexer #580
• Add CheckDataMixin #568
• Refactor AcqInfo and Trajectory Calculation #560
• Ensure compressed number of coils is not greater than existing number #567
• Move KData to data #559

[github-actions]

Coverage Report

File	Stmts	Miss	Cover	Missing
src/mrpro/algorithms/csm
inati.py	24	1	96%	44
walsh.py	16	1	94%	34
src/mrpro/algorithms/dcf
dcf_voronoi.py	53	4	92%	15, 48–49, 76
src/mrpro/algorithms/optimizers
adam.py	20	1	95%	69
src/mrpro/algorithms/reconstruction
DirectReconstruction.py	28	16	43%	51–71, 85
IterativeSENSEReconstruction.py	13	1	92%	76
Reconstruction.py	50	22	56%	42, 54–56, 80–87, 104–113
RegularizedIterativeSENSEReconstruction.py	41	17	59%	96–100, 114–139
src/mrpro/data
AcqInfo.py	146	5	97%	51, 55, 59, 229, 267
CheckDataMixin.py	329	36	89%	94, 100, 102, 119, 127, 129, 131, 164–165, 188, 197, 214, 242, 258, 306, 311, 317–322, 352, 359, 445, 447, 479, 481, 483, 487, 503, 544, 549, 631, 647, 669–673, 705
CsmData.py	28	3	89%	13, 80–82
DcfData.py	44	8	82%	16, 64, 76–81
EncodingLimits.py	73	3	96%	33, 123, 126
IData.py	67	9	87%	119, 125, 129, 159–167
IHeader.py	75	7	91%	75, 109, 127–131
KData.py	222	33	85%	112–113, 128, 135, 147–158, 167, 175, 184–188, 240–241, 281, 298, 303–304, 323–334, 487, 489, 547, 562, 626, 635
KHeader.py	139	19	86%	23, 108–112, 139, 187, 194–195, 198, 205, 222–229, 237–248
KNoise.py	31	15	52%	39–52, 56–61
KTrajectory.py	81	12	85%	108–113, 116–118, 203–207
MoveDataMixin.py	140	18	87%	15, 113, 129, 143–145, 207, 323–325, 338, 417, 437–438, 440, 455–456, 458
QData.py	39	7	82%	42, 65–73
Rotation.py	674	35	95%	100, 198, 335, 433, 477, 495, 581, 583, 592, 626, 628, 691, 768, 773, 776, 791, 808, 813, 889, 1077, 1082, 1085, 1109, 1113, 1240, 1242, 1250–1251, 1315, 1397, 1690, 1846, 1881, 1885, 1996
SpatialDimension.py	232	21	91%	34, 104, 141, 148, 154, 274–276, 289–291, 325, 343, 356, 369, 382, 395, 404–405, 420, 429
acq_filters.py	12	1	92%	47
src/mrpro/data/traj_calculators
KTrajectoryCalculator.py	27	1	96%	85
KTrajectoryIsmrmrd.py	13	2	85%	41, 50
src/mrpro/operators
CartesianSamplingOp.py	89	3	97%	118, 157, 280
ConstraintsOp.py	60	2	97%	46, 48
EndomorphOperator.py	65	2	97%	228, 234
FiniteDifferenceOp.py	27	2	93%	40, 105
FourierOp.py	158	3	98%	263, 381, 386
Functional.py	71	5	93%	20–22, 117, 119
GridSamplingOp.py	136	9	93%	72–73, 82–83, 90–91, 94, 96, 98
LinearOperator.py	168	10	94%	55, 91, 190, 220, 261, 270, 278, 287, 295, 320
LinearOperatorMatrix.py	158	16	90%	82, 119, 152, 161, 166, 175–178, 191–194, 203, 215, 304, 331, 359
MultiIdentityOp.py	13	2	85%	43, 48
Operator.py	78	2	97%	25, 74
ProximableFunctionalSeparableSum.py	39	3	92%	50, 103, 110
SliceProjectionOp.py	173	8	95%	44, 61, 63, 69, 206, 227, 260, 300
WaveletOp.py	120	5	96%	152, 170, 205, 210, 233
ZeroPadOp.py	16	1	94%	30
src/mrpro/utils
filters.py	62	2	97%	44, 49
indexing.py	170	1	99%	325
reshape.py	60	1	98%	191
slice_profiles.py	46	6	87%	20, 36, 113–116, 149
sliding_window.py	34	1	97%	34
split_idx.py	10	2	80%	43, 47
summarize_tensorvalues.py	11	9	18%	20–29
typing.py	18	11	39%	8–23
zero_pad_or_crop.py	31	6	81%	26, 30, 54, 57, 60, 63
TOTAL	5405	410	92%

Tests	Skipped	Failures	Errors	Time
2323	0 💤	0 ❌	0 🔥	1m 50s ⏱️

[github-actions]

📚 Documentation

📁 Download as zip
🔍 View online