开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 并行计算 > 查看源码
dwt.c
资源名称：mpi.zip　[点击查看]
上传用户：hhyinxing
上传日期：2013-09-10
资源大小：833k
文件大小：4k
源码类别：

并行计算

开发平台：

Unix_Linux

dwt.c：源码内容
  1. #include <stdio.h>
  2. #include <stdlib.h>
  3. #include "mpi.h"
  5. #define  MAX_PROCESSOR_NUM       12
  6. #define  MAX_N                 1024
  7. #define  MAX_L                  100
  9. void print(int b[], int start, int len)
  10. {
  11.     int end = start+len;
  12.     int i;
  14.     for(i = start; i < end; i ++)
  15.     {
  16.         printf("%d  ",b[i]);
  17.     }
  18.     printf("n");
  20. }
  22. int main(int argc, char *argv[])
  23. {
  24.     int c[MAX_N], d[MAX_N];
  25.     int h[MAX_L], g[MAX_L];
  26.     int pred[MAX_N], curr[MAX_N];
  27.     int N,L;
  28.     FILE *fin;
  29.     int n;
  30.     int offsetAddress;
  31.     int t;
  32.     int i, j, k;
  33.     int size, rank;
  34.     double  transTime = 0 ,tempCurrentTime, beginTime;
  36.     MPI_Status status;
  38.     MPI_Init(&argc, &argv);
  39.     MPI_Comm_size(MPI_COMM_WORLD,&size);
  40.     MPI_Comm_rank(MPI_COMM_WORLD,&rank);
  42.     if(rank == 0)
  43.     {
  44.         fin = fopen("dataIn.txt","r");
  45.         if (fin == NULL)
  46.         {
  47.             puts("Not find input data file");
  48.             puts("Please create a file "dataIn.txt"");
  49.             puts("<example for dataIn.txt> ");
  50.             puts("10");
  51.             puts("1 9 40 5 67 2 6 8 49 5");
  52.             puts("1 3  4 5  6 7 8 8  0 4");
  53.             puts("16");
  54.             puts("1 0 4 5  5 6 7 8  5 4 0 3  4 6 1 2");
  55.             MPI_Finalize();
  56.             exit(-1);
  57.         }
  58.         else
  59.         {
  60.             /* read array h,g from dataIn.txt */
  61.             fscanf(fin,"%d",&L);
  62.             if ((L < 1)||(L > MAX_L))
  63.             {
  64.                 puts("Input the length of array h is out of range!");
  65.                 exit(-1);
  66.             }
  67.             for(i = 0; i < L; i ++) fscanf(fin,"%d", h+i);
  68.             for(i = 0; i < L; i ++) fscanf(fin,"%d", g+i);
  70.             /* read array c from dataIn.txt */
  71.             fscanf(fin,"%d",&N);
  72.             if ((N < 1)||(N > MAX_N))
  73.             {
  74.                 puts("Input the length of array c is out of range!");
  75.                 exit(-1);
  76.             }
  77.             for(i = 0; i < N; i ++) fscanf(fin,"%d", pred+i);
  78.             /* if  2^(t-1) < N < 2^t , then attach zero to expend N to 2^t */
  80.             puts("One dimensional wavelet transform's input data from file dataIn.txt");
  81.             printf("h[] : ");
  82.             print(h, 0, L);
  83.             printf("g[] : ");
  84.             print(g, 0, L);
  85.             printf("c[] : ");
  86.             print(pred, 0, N);
  87.             printf("n");
  88.         }
  89.     }
  91.     MPI_Bcast(&N, 1, MPI_INT, 0, MPI_COMM_WORLD);
  92.     MPI_Bcast(&L, 1, MPI_INT, 0, MPI_COMM_WORLD);
  93.     MPI_Bcast(h, L, MPI_INT, 0, MPI_COMM_WORLD);
  94.     MPI_Bcast(g, L, MPI_INT, 0, MPI_COMM_WORLD);
  96.     n = N;
  97.     offsetAddress = 0;
  98.     t = 0;
  99.     while(n > 1)
  100.     {
  101.         MPI_Bcast(pred, n, MPI_INT, 0, MPI_COMM_WORLD);
  102.         n /=2;
  104.         int jLength =  n/size;
  105.         int jStart  =  n%size+rank*jLength;
  107.         if (rank == 0)
  108.         {
  109.             for(j = 0; j < jStart+jLength; j ++)
  110.             {
  111.                 curr[j] = 0;
  112.                 d[offsetAddress+j] = 0;
  113.                 for(k = 0; k < L; k ++)
  114.                 {
  115.                     curr[j] += h[k]*pred[(2*j+k)%(2*n)];
  116.                     d[offsetAddress+j] += g[k]*pred[(2*j+k)%(2*n)];
  117.                 }
  118.             }
  119.             /* curr[0 .. jStart-1] ==> pred[0 .. jStart-1] */
  120.             for(j = 0; j < jStart; j ++) pred[j] = curr[j];
  121.         }
  122.         else
  123.         {
  124.             for(j = jStart; j < jStart+jLength; j ++)
  125.             {
  126.                 curr[j] = 0;
  127.                 d[offsetAddress+j] = 0;
  128.                 for(k = 0; k < L; k ++)
  129.                 {
  130.                     curr[j] += h[k]*pred[(2*j+k)%(2*n)];
  131.                     d[offsetAddress+j] += g[k]*pred[(2*j+k)%(2*n)];
  132.                 }
  133.             }
  134.         }
  136.         /* gather into &(pred+jStart) */
  137.         MPI_Gather( (curr+jStart), jLength, MPI_INT, (pred+n%size), jLength, MPI_INT, 0, MPI_COMM_WORLD);
  138.         MPI_Gather( (d+offsetAddress+jStart),jLength, MPI_INT, (d+offsetAddress+jStart), jLength, MPI_INT, 0, MPI_COMM_WORLD);
  140.         /* store in c[] */
  141.         if (rank == 0)
  142.         {
  143.             for(j = 0, k = offsetAddress; j < n;  j++, k ++) c[k] = pred[j];
  144.         }
  146.         if (rank == 0)
  147.         {
  148.             t ++;
  149.             printf("Running after loop  %dn", t);
  150.             printf("c[] : ");
  151.             print(c, offsetAddress, n);
  152.             printf("d[] : ");
  153.             print(d, offsetAddress, n);
  154.             printf("n");
  155.         }
  156.         offsetAddress += n;
  158.     }
  160.     MPI_Finalize();
  161. }