758 |
|
int |
759 |
|
rmx_transpose(RMATRIX *rm) |
760 |
|
{ |
761 |
< |
RMATRIX dnew; |
762 |
< |
int i, j; |
761 |
> |
#define bmbyte(r,c) bmap[((r)*rm->ncols+(c))>>3] |
762 |
> |
#define bmbit(r,c) (1 << ((r)*rm->ncols+(c) & 7)) |
763 |
> |
#define bmop(r,c, op) (bmbyte(r,c) op bmbit(r,c)) |
764 |
> |
#define bmtest(r,c) bmop(r,c,&) |
765 |
> |
#define bmset(r,c) bmop(r,c,|=) |
766 |
> |
#define bmtestandset(r,c) (!bmtest(r,c) && bmset(r,c)) |
767 |
> |
uby8 *bmap; |
768 |
> |
rmx_dtype val[MAXCOMP]; |
769 |
> |
RMATRIX dold; |
770 |
> |
int i, j; |
771 |
|
|
772 |
|
if (!rm || !rm->mtx | (rm->ncomp > MAXCOMP)) |
773 |
|
return(0); |
780 |
|
return(1); |
781 |
|
} |
782 |
|
if (rm->nrows == rm->ncols) { /* square matrix case */ |
783 |
< |
rmx_dtype val[MAXCOMP]; |
784 |
< |
for (j = rm->ncols; j--; ) |
785 |
< |
for (i = rm->nrows; i--; ) { |
786 |
< |
if (i == j) continue; |
779 |
< |
memcpy(val, rmx_val(rm,i,j), |
783 |
> |
for (i = rm->nrows; i--; ) |
784 |
> |
for (j = rm->ncols; j--; ) { |
785 |
> |
if (i == j) continue; |
786 |
> |
memcpy(val, rmx_val(rm,i,j), |
787 |
|
sizeof(rmx_dtype)*rm->ncomp); |
788 |
< |
memcpy(rmx_lval(rm,i,j), rmx_val(rm,j,i), |
788 |
> |
memcpy(rmx_lval(rm,i,j), rmx_val(rm,j,i), |
789 |
|
sizeof(rmx_dtype)*rm->ncomp); |
790 |
< |
memcpy(rmx_val(rm,j,i), val, |
790 |
> |
memcpy(rmx_val(rm,j,i), val, |
791 |
|
sizeof(rmx_dtype)*rm->ncomp); |
792 |
< |
} |
792 |
> |
} |
793 |
|
return(1); |
794 |
|
} |
795 |
< |
memset(&dnew, 0, sizeof(dnew)); |
796 |
< |
dnew.ncols = rm->nrows; dnew.nrows = rm->ncols; |
797 |
< |
dnew.ncomp = rm->ncomp; |
791 |
< |
if (!rmx_prepare(&dnew)) |
795 |
> |
/* clear completion bitmap */ |
796 |
> |
bmap = (uby8 *)calloc(((size_t)rm->nrows*rm->ncols+7)>>3, 1); |
797 |
> |
if (!bmap) |
798 |
|
return(0); |
799 |
< |
rmx_addinfo(&dnew, rm->info); |
800 |
< |
dnew.dtype = rm->dtype; |
801 |
< |
copycolor(dnew.cexp, rm->cexp); |
802 |
< |
memcpy(dnew.wlpart, rm->wlpart, sizeof(dnew.wlpart)); |
803 |
< |
for (j = dnew.ncols; j--; ) |
804 |
< |
for (i = dnew.nrows; i--; ) |
805 |
< |
memcpy(rmx_lval(&dnew,i,j), rmx_val(rm,j,i), |
806 |
< |
sizeof(rmx_dtype)*dnew.ncomp); |
807 |
< |
/* and reassign result */ |
808 |
< |
return(rmx_transfer_data(rm, &dnew, 1)); |
799 |
> |
dold = *rm; |
800 |
> |
rm->ncols = dold.nrows; rm->nrows = dold.ncols; |
801 |
> |
for (i = rm->nrows; i--; ) |
802 |
> |
for (j = rm->ncols; j--; ) { |
803 |
> |
int i0, j0; |
804 |
> |
int i1 = i; |
805 |
> |
size_t j1 = j; |
806 |
> |
if (!bmtestandset(i,j)) continue; |
807 |
> |
memcpy(val, rmx_val(rm,i,j), |
808 |
> |
sizeof(rmx_dtype)*rm->ncomp); |
809 |
> |
for ( ; ; ) { /* value transpose loop */ |
810 |
> |
const rmx_dtype *src; |
811 |
> |
i0 = i1; j0 = (int)j1; |
812 |
> |
src = rmx_val(&dold, j0, i0); |
813 |
> |
j1 = (src - dold.mtx)/dold.ncomp; |
814 |
> |
i1 = j1 / rm->ncols; |
815 |
> |
j1 -= i1*rm->ncols; |
816 |
> |
if (!bmtestandset(i1,j1)) break; |
817 |
> |
memcpy(rmx_lval(rm,i0,j0), src, |
818 |
> |
sizeof(rmx_dtype)*rm->ncomp); |
819 |
> |
} |
820 |
> |
/* close the loop */ |
821 |
> |
memcpy(rmx_lval(rm,i0,j0), val, |
822 |
> |
sizeof(rmx_dtype)*rm->ncomp); |
823 |
> |
} |
824 |
> |
free(bmap); /* all done! */ |
825 |
> |
return(1); |
826 |
> |
#undef bmbyte |
827 |
> |
#undef bmbit |
828 |
> |
#undef bmop |
829 |
> |
#undef bmtest |
830 |
> |
#undef bmset |
831 |
> |
#undef bmtestandset |
832 |
|
} |
833 |
|
|
834 |
|
/* Multiply (concatenate) two matrices and allocate the result */ |
1065 |
|
case 1: |
1066 |
|
setcolor(cv, dp[0], dp[0], dp[0]); |
1067 |
|
break; |
1068 |
< |
default: { |
1068 |
> |
default: |
1069 |
> |
if (sizeof(COLORV) == sizeof(rmx_dtype)) { |
1070 |
> |
scolor2color(cv, (const COLORV *)dp, |
1071 |
> |
rm->ncomp, rm->wlpart); |
1072 |
> |
} else { |
1073 |
|
COLORV scol[MAXCOMP]; |
1074 |
< |
int k; |
1075 |
< |
for (k = rm->ncomp; k--; ) |
1043 |
< |
scol[k] = dp[k]; |
1074 |
> |
int k = rm->ncomp; |
1075 |
> |
while (k--) scol[k] = dp[k]; |
1076 |
|
scolor2color(cv, scol, rm->ncomp, rm->wlpart); |
1077 |
< |
} break; |
1077 |
> |
} |
1078 |
> |
break; |
1079 |
|
} |
1080 |
|
} |
1081 |
|
} |