Commit d3ed2281 authored by Cristian Lalescu's avatar Cristian Lalescu
Browse files

add back full.cpp

parent 6d238707
/***********************************************************************
*
* Copyright 2015 Johns Hopkins University
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
* Contact: turbulence@pha.jhu.edu
* Website: http://turbulence.pha.jhu.edu/
*
************************************************************************/
#include "p3DFFT_to_iR.hpp"
#include "Morton_shuffler.hpp"
#include <iostream>
int myrank, nprocs;
int main(int argc, char *argv[])
{
MPI_Init(&argc, &argv);
MPI_Comm_rank(MPI_COMM_WORLD, &myrank);
MPI_Comm_size(MPI_COMM_WORLD, &nprocs);
int n, N, nfiles, nfields;
if (argc == 5)
{
n = atoi(argv[1]);
N = atoi(argv[2]);
nfiles = atoi(argv[3]);
nfields = atoi(argv[4]);
}
else
{
std::cerr <<
"not enough (or too many) parameters.\naborting." <<
std::endl;
MPI_Finalize();
return EXIT_SUCCESS;
}
p3DFFT_to_iR *r = new p3DFFT_to_iR(
(n/2+1), n, n,
N, N, N,
nfields);
// initialize file names
char* ifile[nfields];
for (int i=0; i<nfields; i++)
{
ifile[i] = (char*)malloc(100*sizeof(char));
sprintf(ifile[i], "Kdata%d", i);
}
//read
r->read(ifile);
for (int i = 0; i<nfields; i++)
free(ifile[i]);
Morton_shuffler *s = new Morton_shuffler(
N, N, N, nfields, nfiles);
s->shuffle(r->r3, "Rdata");
delete s;
delete r;
// clean up
fftwf_mpi_cleanup();
MPI_Finalize();
return EXIT_SUCCESS;
}
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment