@@ -462,10 +462,9 @@ macro_rules! arith_assign_func {
462
462
fn $fn_name( & mut self , rhs: Array ) {
463
463
let tmp_seq = Seq :: <f32 >:: default ( ) ;
464
464
let mut idxrs = Indexer :: new( ) ;
465
- idxrs. set_index( & tmp_seq, 0 , Some ( false ) ) ;
466
- idxrs. set_index( & tmp_seq, 1 , Some ( false ) ) ;
467
- idxrs. set_index( & tmp_seq, 2 , Some ( false ) ) ;
468
- idxrs. set_index( & tmp_seq, 3 , Some ( false ) ) ;
465
+ for n in 0 ..self . numdims( ) {
466
+ idxrs. set_index( & tmp_seq, n, Some ( false ) ) ;
467
+ }
469
468
let tmp = assign_gen( self as & Array , & idxrs,
470
469
& $func( self as & Array , & rhs, false ) ) ;
471
470
let old = mem:: replace( self , tmp) ;
@@ -490,10 +489,9 @@ macro_rules! bit_assign_func {
490
489
fn $fn_name( & mut self , rhs: Array ) {
491
490
let tmp_seq = Seq :: <f32 >:: default ( ) ;
492
491
let mut idxrs = Indexer :: new( ) ;
493
- idxrs. set_index( & tmp_seq, 0 , Some ( false ) ) ;
494
- idxrs. set_index( & tmp_seq, 1 , Some ( false ) ) ;
495
- idxrs. set_index( & tmp_seq, 2 , Some ( false ) ) ;
496
- idxrs. set_index( & tmp_seq, 3 , Some ( false ) ) ;
492
+ for n in 0 ..self . numdims( ) {
493
+ idxrs. set_index( & tmp_seq, n, Some ( false ) ) ;
494
+ }
497
495
let tmp = assign_gen( self as & Array , & idxrs, & $func( self as & Array , & rhs, false ) ) ;
498
496
let old = mem:: replace( self , tmp) ;
499
497
}
0 commit comments