Feature/issue/37 test cre stack in multi layer model

.gitignore

@@ -30,4 +30,4 @@ backup/
 checkpoints
 *.jp2
 .ipynb_checkpoints/
-
+*.asc

Mgetcregather.c

@@ -32,10 +32,12 @@ int main(int argc, char* argv[])
 	float ot; // Time axis origin
 	int nt; // Number of time samples
 	int i,j; // loop counter
-	int cre_n; // Number of CRE Gather traces
+	int cre_n; // Number of traces in CRE vector
 	float cre_d; // CRE Gather sampling
 	float cre_o; // CRE Gather axis origin
 	int trac_m; // CMP sample index
+	int aperture; // Number of traces in CRE Gather
+	float mMax; // maximum CMP coordinate of the model
 
 	/* RSF files I/O */  
 	sf_file in, out, out_m, cremh;
@@ -64,6 +66,13 @@ int main(int argc, char* argv[])
 	if(!sf_histfloat(cremh,"d1",&cre_d)) sf_error("No d1= in cremh file");
 	if(!sf_histfloat(cremh,"o1",&cre_o)) sf_error("No o1= in cremh file");
 
+	if(!sf_getint("aperture",&aperture)) aperture=1;
+	/* Number of traces in a CRE Gather*/
+
+	if(aperture > cre_n){
+		sf_error("The aperture can't be > n1 in cremh file\naperture=%i n2=%i",aperture,cre_n);
+	}
+
 	if(! sf_getbool("verb",&verb)) verb=0;
 	/* 1: active mode; 0: quiet mode */
 
@@ -84,26 +93,28 @@ int main(int argc, char* argv[])
 
 	m = sf_floatalloc(cre_n);
 	sf_floatread(m,cre_n,cremh);
-	creGather = sf_floatalloc2(nt,cre_n);
+	creGather = sf_floatalloc2(nt,aperture);
+
+	mMax = om+dm*nm;
 
-	for(i=0;i<cre_n;i++){
+	for(i=0;i<aperture;i++){
 		trac_m = (int)((double)m[i]/dm);
 
 		for(j=0;j<nt;j++){
-			creGather[i][j] = t[trac_m][i][j];
+			creGather[i][j] = (m[i] <= mMax)? t[trac_m][i][j] : 0.;
 		}
 	}
 
 	/* eixo = sf_maxa(n,o,d)*/
 	ax = sf_maxa(nt, ot, dt);
-	ay = sf_maxa(nh, oh, dh);
+	ay = sf_maxa(aperture, oh, dh);
 	az = sf_maxa(1, 0, 1);
 
 	/* sf_oaxa(arquivo, eixo, índice do eixo) */
 	sf_oaxa(out,ax,1);
 	sf_oaxa(out,ay,2);
 	sf_oaxa(out,az,3);
-	sf_floatwrite(creGather[0],cre_n*nt,out);
+	sf_floatwrite(creGather[0],aperture*nt,out);
 
 	/* eixo do vetor m */
 	am = sf_maxa(cre_n,oh,dh);

experiments/multiLayerModel/cds/SConscript

@@ -0,0 +1,84 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# SConscript  (Madagascar Script)
+#
+# Purpose: Generate model and data cubes to CRE stacking process.
+# This script generates model, original and PEF interpolated data cubes.
+# It depends of the 'creStack.py' script that defines functions.
+#
+# Site: http://www.dirackslounge.online
+# 
+# Version 1.0
+#
+# Programer: Rodolfo A. C. Neves (Dirack) 04/03/2020
+#
+# Email: rodolfo_profissional@hotmail.com
+#
+# License: GPL-3.0 <https://www.gnu.org/licenses/gpl-3.0.txt>.
+
+# Madagascar package
+from rsf.proj import *
+
+# CRE recipe
+from madagascarRecipes.pefInterpolation import pefInterpolation as pefin
+from madagascarRecipes.kimodel import multiLayerModel as mlmod
+from madagascarRecipes.kimodel import kirchoffNewtonModeling as kinewmod
+
+xmax = 6.0
+zmax = 2.0
+
+layers = ((0.30,0.50,0.20,0.30),
+	  (1.65,1.85,1.55,1.65))
+
+velocities = (1.508,
+	      1.690,
+	      2.0)
+
+# Generate multi layer model and data cube
+mlmod(interfaces='interfaces',
+	dipsfile='interfacesDip',
+	modelfile='mod1',
+	xmax=xmax,
+	zmax=zmax,
+	layers=layers,
+	velocities=velocities)
+
+kinewmod(reflectors='interfaces',
+	reflectorsDip='interfacesDip',
+	filename='multiLayerDataCube',
+	velocities=velocities)
+
+# PEF interpolation of the data cube
+a1=10
+a2=2
+rect1=50
+rect2=2
+
+pefin(dataCube='multiLayerDataCube',
+    interpolated='interpolatedDataCube',
+    nm=201,
+    dm=0.025,
+    nt=1001,
+    dt=0.004,
+    nhi=161,
+    a1=a1,
+    a2=a2,
+    rect1=rect1,
+    rect2=rect2)
+
+# Do the PEF interpolation one more time
+# to increase the CMP sampling
+pefin('interpolatedDataCube',
+    'interpolatedDataCube2',
+    nm=402,
+    dm=0.0125,
+    nt=1001,
+    dt=0.004,
+    nhi=161,
+    a1=a1,
+    a2=a2,
+    rect1=rect1,
+    rect2=rect2)
+
+End()

experiments/multiLayerModel/cds/SConscript2

@@ -0,0 +1,61 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# SConscript  (Madagascar Script)
+#
+# Purpose: Contatenate CRE stacked traces.
+#
+# Site: http://www.dirackslounge.online
+# 
+# Version 1.0
+#
+# Programer: Rodolfo A. C. Neves (Dirack) 04/03/2020
+#
+# Email: rodolfo_profissional@hotmail.com
+#
+# License: GPL-3.0 <https://www.gnu.org/licenses/gpl-3.0.txt>.
+
+# Madagascar package
+from rsf.proj import *
+
+# Error report function
+import atexit
+from madagascarRecipes.errorReport import print_build_failures
+
+import glob
+
+# Build the cre stacked section
+# throughout cre stacked traces sorting
+files = glob.glob('creStackedTrace-*.rsf')
+length = len(files)
+
+sortedFiles = []
+for i in range(length):
+	string = 'creStackedTrace-m0-%i.rsf' % i
+	sortedFiles.append(string)  
+
+Flow('stackedSection',sortedFiles,
+	'''
+	rcat axis=2 ${SOURCES[1:%d]} --out=stdout
+	''' % len(files))
+
+Result('stackedSection','grey title="Stacked Section"')
+
+Flow('filtStackedSection','stackedSection',
+	'''
+	bandpass fhi=20 --out=stdout
+	''')
+
+Result('filtStackedSection','grey title="Filtered Stacked Section"')
+
+# Use aliases to split building
+Alias('stack',['stackedSection.rsf','filtStackedSection.rsf'])
+
+# Show error message if fail
+
+message = '''
+SCosncript Failed build
+'''
+
+atexit.register(print_build_failures,message)
+End()

experiments/multiLayerModel/cds/SConstruct

@@ -0,0 +1,167 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# SConstruct  (Madagascar Script)
+#
+# Purpose: Build the interpolation process twice to improve CMP sampling
+# and CRE stacking results (Use CDS condition).
+#
+# Site: http://www.dirackslounge.online
+# 
+# Version 1.0
+#
+# Programer: Rodolfo A. C. Neves (Dirack) 04/03/2020
+#
+# Email: rodolfo_profissional@hotmail.com
+#
+# License: GPL-3.0 <https://www.gnu.org/licenses/gpl-3.0.txt>.
+
+# Madagascar package
+from rsf.proj import *
+
+# Import glob python library
+import glob
+
+# Error report function
+import atexit
+from madagascarRecipes.errorReport import print_build_failures
+
+# Call SConscript to generate input: original and interpolated data cubes
+# SConscript('SConscript')
+
+# CRE stacking
+# It uses Very Fast Simulated Aneeling and non hyperbolic CRS
+# to get zero offset CRS parameters (RN, RNIP and BETA) from data cube
+v0 = 1.5
+ot0 = 0.3
+dt0 = 0.008
+nt0 = 331
+om0 = 2
+dm0 = 0.025
+nm0 = 108
+dataCube='multiLayerDataCube'
+
+for i in range(nm0):
+
+	parametersCube = []
+	creGatherCube = []
+	creTimeCurveCube = []
+	creGatherCubeForM0List = []
+	creTimeCurveCubeForM0List = []
+	creGatherIndex = 0
+
+	m0 = om0 + (i * dm0)
+
+	for j in range(nt0):
+
+		t0 = ot0 + (dt0 * j)
+
+		crsParameters = 'crsParameters-m0-%g-t0-%g' % (i,j)
+		creMhCoordinates = 'creMhCoordinates-m0-%g-t0-%g' % (i,j)
+		creGather = 'creGather-m0-%g-t0-%g' % (i,j)
+		creMcoordinate = 'creMcoordinate-m0-%g-t0-%g' % (i,j)
+		creTimeCurve = 'creTimeCurve-m0-%g-t0-%g' % (i,j)
+		creGatherPlot = 'crePlot-m0-%g-t0-%g' % (i,j)
+
+		# Very Fast Simulated Aneelling Global Optimization (VFSA)
+		Flow(crsParameters,'interpolatedDataCube2',
+			'''
+			vfsacrsnh m0=%g v0=%g t0=%g verb=y repeat=3
+			''' % (m0,v0,t0))
+
+		# Calculate CRE trajectory
+		Flow(creMhCoordinates,['interpolatedDataCube2',crsParameters],
+			'''
+			cretrajec verb=y m0=%g param=${SOURCES[1]} |
+			put unit1="Offset" label1="Km"
+			''' % (m0))
+
+		#Get CRE Gather from interpolated Data Cube
+		Flow([creGather,creMcoordinate],['interpolatedDataCube2',creMhCoordinates],
+			'''
+			getcregather aperture=50 verb=y cremh=${SOURCES[1]} m=${TARGETS[1]} |
+                        put label1="Time" unit1="s" label2="Offset" unit2="km" 
+			''')
+
+		# Calculate CRE traveltime curve t(m,h)
+		Flow(creTimeCurve,[creMcoordinate, crsParameters],
+			'''
+			getcretimecurve cds=y param=${SOURCES[1]} t0=%g m0=%g v0=%g verb=y |
+                        put label1="Offset" unit1="Km" 
+			''' % (t0,m0,v0))
+
+		parametersCube.append(crsParameters)
+		creGatherCube.append(creGather)
+		creTimeCurveCube.append(creTimeCurve)
+		creGatherIndex = creGatherIndex + 1
+
+	# Concatenate cre gathers for each m0
+	creGatherCubeForM0 = "creGatherCube-m0-%i" % i
+	creGatherCubeForM0List.append(creGatherCubeForM0)
+	Flow(creGatherCubeForM0,creGatherCube,
+		'''
+		rcat axis=3 ${SOURCES[1:%d]}
+		''' % nt0)
+
+	# Concatenate cre Time curves for each m0
+	creTimeCurveCubeForM0 = "creTimeCurveCube-m0-%i" % i
+	creTimeCurveCubeForM0List.append(creTimeCurveCubeForM0)
+	Flow(creTimeCurveCubeForM0,creTimeCurveCube,
+		'''
+		rcat axis=2 ${SOURCES[1:%d]}
+		'''% nt0)
+
+	# Stack one m0 per turn
+	creTimeCurveCube = []
+	creGatherCube = []
+	creGatherTrace = "creGatherTrace-m0-%i" % i
+	creTimeCurveTrace = "creTimeCurveTrace-m0-%i" % i
+	creStackedTrace = "creStackedTrace-m0-%i" % i
+
+	# Get cre Gathers organized by t0 and m0
+	Flow(creGatherTrace,creGatherCubeForM0List,
+		'''
+		put n4=1 d4=%g o4=%g d3=%g o3=%g label3=t0 unit3=s label4=m0 unit4=Km
+		''' % (dm0,m0,dt0,ot0))
+
+	# Get all the traveltime curves organized by t0 and m0
+	Flow(creTimeCurveTrace,creTimeCurveCubeForM0List,
+		'''
+		put label3="m0" unit3="Km" label2="t0" unit2="s" d2=%g o2=%g d3=%g o3=%g
+		''' % (dt0,ot0,dm0,m0))
+
+	# CRE stacking
+	Flow(creStackedTrace,[creGatherTrace,creTimeCurveTrace],
+		'''
+		crestack timeCurves=${SOURCES[1]} verb=y |
+		put label1=t0 unit1=s label2=m0 unit2=Km --out=stdout
+		''')
+
+# Build the cre stacked section
+# throughout cre stacked traces sorting
+files = glob.glob('creStackedTrace-*.rsf')
+length = len(files)
+
+sortedFiles = []
+for i in range(length):
+	string = 'creStackedTrace-m0-%i.rsf' % i
+	sortedFiles.append(string)  
+
+Flow('stackedSection',sortedFiles,
+	'''
+	rcat axis=2 ${SOURCES[1:%d]} --out=stdout
+	''' % len(files))
+
+Flow('filtStackedSection','stackedSection',
+	'''
+	bandpass fhi=20 --out=stdout
+	''')
+
+# Show error message if fail
+message = '''
+SConstruct Failed build when m0={m0} and t0={t0}
+'''.format(m0=(i*dm0+om0),t0=(j*dt0+ot0))
+
+atexit.register(print_build_failures,message)
+
+End()