merge.c 4.61 KB
Newer Older
Tilman Steinweg's avatar
Tilman Steinweg committed
1
/*-----------------------------------------------------------------------------------------
2
 * Copyright (C) 2016  For the list of authors, see file AUTHORS.
Tilman Steinweg's avatar
Tilman Steinweg committed
3
 *
Florian Wittkamp's avatar
Florian Wittkamp committed
4
 * This file is part of IFOS.
5
 *
Florian Wittkamp's avatar
Florian Wittkamp committed
6
 * IFOS is free software: you can redistribute it and/or modify
Tilman Steinweg's avatar
Tilman Steinweg committed
7 8
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, version 2.0 of the License only.
9
 *
Florian Wittkamp's avatar
Florian Wittkamp committed
10
 * IFOS is distributed in the hope that it will be useful,
Tilman Steinweg's avatar
Tilman Steinweg committed
11 12 13
 * 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.
14
 *
Tilman Steinweg's avatar
Tilman Steinweg committed
15
 * You should have received a copy of the GNU General Public License
Florian Wittkamp's avatar
Florian Wittkamp committed
16
 * along with IFOS. See file COPYING and/or <http://www.gnu.org/licenses/gpl-2.0.html>.
17
 -----------------------------------------------------------------------------------------*/
Tilman Steinweg's avatar
Tilman Steinweg committed
18 19

/*------------------------------------------------------------------------
20 21
 *   merge snapshots files written by the different processes to
 *   a single file
Tilman Steinweg's avatar
Tilman Steinweg committed
22 23 24 25 26 27 28
 *
 *  ----------------------------------------------------------------------*/

#include "fd.h"


void merge(int nsnap, int type){
29 30 31 32 33 34 35 36 37 38 39
    
    
    extern char SNAP_FILE[STRING_SIZE];
    extern int NXG, NYG, SNAP_FORMAT, NPROCX, NPROCY;
    extern int NX, NY, IDX, IDY;
    extern FILE *FP;
    extern float DH;
    extern int SNAPSHOT_START, SNAPSHOT_END, SNAPSHOT_INCR;
    char file[STRING_SIZE], mfile[STRING_SIZE], outfile[STRING_SIZE], ext[10];
    FILE *fp[NPROCY][NPROCX], *fpout;
    int i, j, ip, jp, n;
40
    float a, max = 0.0;
41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
    int shot;
    
    
    
    
    switch(SNAP_FORMAT){
        case 1:
            sprintf(ext,".su");
            break;
        case 2:
            sprintf(ext,".asc");
            break;
        case 3:
            sprintf(ext,".bin");
            break;
    }
    
    switch(type){
        case 1:
            fprintf(FP," x-component of particle velocity");
            strcat(ext,".x");
            break;
        case 2:
            fprintf(FP," y-component of particle velocity");
            strcat(ext,".y");
            break;
        case 4:
            fprintf(FP," P-wave energyfield");
            strcat(ext,".div");
            break;
        case 5:
            fprintf(FP," S-wave energyfield");
            strcat(ext,".rot");
            break;
        case 6:
            fprintf(FP," pressure");
            strcat(ext,".p");
            break;
        case 7:
            fprintf(FP," z-component of particle velocity");
            strcat(ext,".z");
            break;
        default:
84
            declare_error(" merge: cannot find snapfiles! ");
85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103
            break;
    }
    
    sprintf(mfile,"%s%s",SNAP_FILE,ext);
    fprintf(FP," (files: %s.??? ).\n",mfile);
    
    
    fprintf(FP," Opening snapshot files: %s.??? ",mfile);
    

    for(shot=SNAPSHOT_START;shot<=SNAPSHOT_END;shot+=SNAPSHOT_INCR) {
        sprintf(outfile,"%s.shot%i%s",SNAP_FILE,shot,ext);
        fprintf(FP,"\n writing merged snapshot file to  %s \n",outfile);
        fpout=fopen(outfile,"w");
        
        for (ip=0;ip<=NPROCX-1; ip++)
            for (jp=0;jp<=NPROCY-1; jp++){
                sprintf(file,"%s.shot%i.%i.%i",mfile,shot,ip,jp);
                fp[jp][ip]=fopen(file,"r");
104
                if (fp[jp][ip]==NULL) declare_error("merge: can't read snapfile !");
105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148
            }
        
        
        fprintf(FP," ... finished. \n");
        
        
        fprintf(FP," Copying...");
        
        max=0.0;
        for (n=0;n<=nsnap-2; n++){
            for (ip=0;ip<=NPROCX-1; ip++){
                for (i=1;i<=NX;i+=IDX){
                    for (jp=0;jp<=NPROCY-1; jp++){
                        for (j=1;j<=NY;j+=IDY){
                            a=readdsk(fp[jp][ip],SNAP_FORMAT);
                            if (a>max) max=a;
                            writedsk(fpout,a,SNAP_FORMAT);
                        }
                    }
                }
            }
        }
        fprintf(FP," ... finished. \n");
        
        for (ip=0;ip<=NPROCX-1; ip++)
            for (jp=0;jp<=NPROCY-1; jp++)
            {
                fclose(fp[jp][ip]);
            }
        for (ip=0;ip<=NPROCX-1; ip++){
            for (jp=0;jp<=NPROCY-1; jp++){
                sprintf(file,"%s.shot%i.%i.%i",mfile,shot,ip,jp);
                remove(file);
            }
        }
    }
    if (SNAP_FORMAT==3){
        fprintf(FP," Use \n");
        fprintf(FP," xmovie n1=%d n2=%d  < %s loop=1 label1=Y label2=X title=%%g d1=%f d2=%f f1=%f f2=%f clip=%e \n",
                ((NYG-1)/IDY)+1,((NXG-1)/IDY)+1,outfile,DH,DH,DH,DH,max/10.0);
        fprintf(FP," to play the movie. \n");
    }
    
    
Tilman Steinweg's avatar
Tilman Steinweg committed
149 150 151
}