fastsum_benchomp_createdataset.c
/*
* Copyright (c) 2002, 2017 Jens Keiner, Stefan Kunis, Daniel Potts
*
* This program is free software; you can redistribute it and/or modify it under
* the terms of the GNU General Public License as published by the Free Software
* Foundation; either version 2 of the License, or (at your option) any later
* version.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
* details.
*
* You should have received a copy of the GNU General Public License along with
* this program; if not, write to the Free Software Foundation, Inc., 51
* Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <stdlib.h>
#include <complex.h>
#include "config.h"
#include "nfft3.h"
#include "infft.h"
void fastsum_benchomp_createdataset(unsigned int d, int L, int M)
{
int t, j, k;
R *x;
R *y;
C *alpha;
x = (R*) NFFT(malloc)((size_t)(d * L) * sizeof(R));
y = (R*) NFFT(malloc)((size_t)(d * L) * sizeof(R));
alpha = (C*) NFFT(malloc)((size_t)(L) * sizeof(C));
k = 0;
while (k < L)
{
R r_max = K(1.0);
R r2 = K(0.0);
for (j = 0; j < d; j++)
x[k * d + j] = K(2.0) * r_max * NFFT(drand48)() - r_max;
for (j = 0; j < d; j++)
r2 += x[k * d + j] * x[k * d + j];
if (r2 >= r_max * r_max)
continue;
k++;
}
NFFT(vrand_unit_complex)(alpha, L);
k = 0;
while (k < M)
{
R r_max = K(1.0);
R r2 = K(0.0);
for (j = 0; j < d; j++)
y[k * d + j] = K(2.0) * r_max * NFFT(drand48)() - r_max;
for (j = 0; j < d; j++)
r2 += y[k * d + j] * y[k * d + j];
if (r2 >= r_max * r_max)
continue;
k++;
}
printf("%d %d %d\n", d, L, M);
for (j = 0; j < L; j++)
{
for (t = 0; t < d; t++)
printf("%.16" __FES__ " ", x[d * j + t]);
printf("\n");
}
for (j = 0; j < L; j++)
printf("%.16" __FES__ " %.16" __FES__ "\n", CREAL(alpha[j]), CIMAG(alpha[j]));
for (j = 0; j < M; j++)
{
for (t = 0; t < d; t++)
printf("%.16" __FES__ " ", y[d * j + t]);
printf("\n");
}
NFFT(free)(x);
NFFT(free)(y);
NFFT(free)(alpha);
}
int main(int argc, char **argv)
{
int d;
int L;
int M;
if (argc < 4)
{
fprintf(stderr, "usage: d L M\n");
return -1;
}
d = atoi(argv[1]);
L = atoi(argv[2]);
M = atoi(argv[3]);
fprintf(stderr, "d=%d, L=%d, M=%d\n", d, L, M);
fastsum_benchomp_createdataset(d, L, M);
return 0;
}
nfft3.h
Header file for the nfft3 library.