transpose_generic.c [plain text]
#include "config.h"
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include "libgfortran.h"
extern void transpose (gfc_array_char *, gfc_array_char *);
export_proto(transpose);
void
transpose (gfc_array_char *ret, gfc_array_char *source)
{
index_type rxstride, rystride;
char *rptr;
index_type sxstride, systride;
const char *sptr;
index_type xcount, ycount;
index_type x, y;
index_type size;
assert (GFC_DESCRIPTOR_RANK (source) == 2
&& GFC_DESCRIPTOR_RANK (ret) == 2);
size = GFC_DESCRIPTOR_SIZE (source);
if (ret->data == NULL)
{
assert (ret->dtype == source->dtype);
ret->dim[0].lbound = 0;
ret->dim[0].ubound = source->dim[1].ubound - source->dim[1].lbound;
ret->dim[0].stride = 1;
ret->dim[1].lbound = 0;
ret->dim[1].ubound = source->dim[0].ubound - source->dim[0].lbound;
ret->dim[1].stride = ret->dim[0].ubound+1;
ret->data = internal_malloc_size (size * size0 ((array_t*)ret));
ret->base = 0;
}
sxstride = source->dim[0].stride * size;
if (sxstride == 0)
sxstride = size;
systride = source->dim[1].stride * size;
xcount = source->dim[0].ubound + 1 - source->dim[0].lbound;
ycount = source->dim[1].ubound + 1 - source->dim[1].lbound;
rxstride = ret->dim[0].stride * size;
if (rxstride == 0)
rxstride = size;
rystride = ret->dim[1].stride * size;
rptr = ret->data;
sptr = source->data;
for (y = 0; y < ycount; y++)
{
for (x = 0; x < xcount; x++)
{
memcpy (rptr, sptr, size);
sptr += sxstride;
rptr += rystride;
}
sptr += systride - (sxstride * xcount);
rptr += rxstride - (rystride * xcount);
}
}