refactor: reduce duplication

Signed-off-by: Athan <kgryte@gmail.com>

Author: kgryte

lib/node_modules/@stdlib/blas/ext/join-between/lib/assign.js

@@ -89,7 +89,8 @@ function assign( x, separators, out, options ) {
 	var opts;
 	var sep;
 	var shx;
-	var sh;
+	var shc;
+	var flg;
 	var M;
 	var p;
 	var s;
@@ -124,50 +125,46 @@ function assign( x, separators, out, options ) {
 	}
 	shx = getShape( x );
 
+	// Resolve the complement shape...
+	if ( hasOwnProp( opts, 'dims' ) ) {
+		shc = nonCoreShape( shx, opts.dims );
+		flg = true;
+	} else {
+		shc = [];
+		flg = false;
+	}
 	// Broadcast prefix to the complement shape...
 	if ( isndarrayLike( p ) ) {
-		if ( hasOwnProp( opts, 'dims' ) ) {
-			prefix = maybeBroadcastArray( p, nonCoreShape( shx, opts.dims ) );
+		if ( flg ) {
+			prefix = maybeBroadcastArray( p, shc );
 		} else if ( ndims( p ) === 0 ) {
 			prefix = p;
 		} else {
 			throw new TypeError( format( 'invalid option. `%s` option must be a zero-dimensional ndarray when not specifying dimensions. Value: `%s`.', 'prefix', p ) );
 		}
 	} else {
-		if ( hasOwnProp( opts, 'dims' ) ) {
-			sh = nonCoreShape( shx, opts.dims );
-		} else {
-			sh = [];
-		}
-		prefix = broadcastScalar( p, 'generic', sh, getOrder( x ) );
+		prefix = broadcastScalar( p, 'generic', shc, getOrder( x ) );
 	}
 	// Broadcast suffix to the complement shape...
 	if ( isndarrayLike( s ) ) {
-		if ( hasOwnProp( opts, 'dims' ) ) {
-			suffix = maybeBroadcastArray( s, nonCoreShape( shx, opts.dims ) );
+		if ( flg ) {
+			suffix = maybeBroadcastArray( s, shc );
 		} else if ( ndims( s ) === 0 ) {
 			suffix = s;
 		} else {
 			throw new TypeError( format( 'invalid option. `%s` option must be a zero-dimensional ndarray when not specifying dimensions. Value: `%s`.', 'suffix', s ) );
 		}
 	} else {
-		if ( hasOwnProp( opts, 'dims' ) ) {
-			sh = nonCoreShape( shx, opts.dims );
-		} else {
-			sh = [];
-		}
-		suffix = broadcastScalar( s, 'generic', sh, getOrder( x ) );
+		suffix = broadcastScalar( s, 'generic', shc, getOrder( x ) );
 	}
 	// Broadcast separators to [ ...complementShape, M-1 ] where M is the number of elements along the reduced dimensions...
-	if ( hasOwnProp( opts, 'dims' ) ) {
-		sh = nonCoreShape( shx, opts.dims );
-		M = numel( shx ) / numel( sh );
+	if ( flg ) {
+		M = numel( shx ) / numel( shc );
 	} else {
-		sh = [];
 		M = numel( shx );
 	}
-	sh.push( M - 1 );
-	sep = maybeBroadcastArray( separators, sh );
+	shc.push( M - 1 );
+	sep = maybeBroadcastArray( separators, shc );
 
 	return base( x, prefix, suffix, sep, out, opts );
 }