response_spectra_map.py

# /*
#  ============================================================================
# Loads the plane displacement data to produce acceleration signals at each grid
# point, then generates a response spectra map of signals.
# Version: Sep 25, 2015.
#  ============================================================================
#  */
import numpy as np
import sys
sys.path.insert(0, '/Users/kelicheng/seismtools') # insert the path to seismtools to import stools program.
from stools import max_osc_response
from htools import plot, saveDat, dis_to_acc, show_progress, loadFile
from userInput import ResponseInput
np.seterr(divide='ignore', invalid='ignore')


def response_spectra_map(userInput):
	compoDic = {'x':0, 'y':1, 'z':2}
	typeDic = {'a':2, 'v':1, 'd':0}
	fp = userInput.fp
	dimensionX = userInput.dimensionX
	dimensionY = userInput.dimensionY
	spaceX = userInput.spaceX
	spaceY = userInput.spaceY
	outSpaceX = userInput.outspaceX
	outSpaceY = userInput.outspaceY

	simulationTime = userInput.simulationTime
	deltaT = userInput.deltaT
	period = userInput.period
	component = userInput.component
	responseType = userInput.plotType
	colorMap = userInput.colorMap


	numLayer = int(simulationTime/deltaT)
	downDip = dimensionX/spaceX+1
	alongStrike = dimensionY/spaceY+1
	numGridX = int(dimensionX/outSpaceX)+1
	numGridY = int(dimensionY/outSpaceY)+1


	response = np.empty((numGridY, numGridX))
	# iterate through each grid points
	for i in range(0, numGridX):
		for j in range(0, numGridY):
			dis = np.array([],float)

			# read the same grid point at different layer
			for k in range(0, numLayer):
				x_coor = i * (outSpaceX/spaceX)
				y_coor = j * (outSpaceY/spaceY)

				data = loadFile(fp, alongStrike, downDip, k, y_coor, x_coor, 1)
				dis = np.append(dis, data[compoDic[component]])
			acc = dis_to_acc(dis, deltaT)

			# calculate response and put into the matrix
			response[j][i] = max_osc_response(acc, deltaT, 0.05, period, 0, 0)[typeDic[responseType]]

		show_progress(i, numGridX)

	sys.stdout.write('\n')
	if userInput.printDat:
		saveDat(userInput.out_path, dimensionX, outSpaceX, outSpaceY, response)
	# plot(response, colorMap)
	plot(response, userInput)
# end of response_spectra_map



if __name__ == "__main__":
	if len(sys.argv) > 1:
		argument = tuple(sys.argv[1:])
		userInput = ResponseInput(*argument)
	else:
		userInput = ResponseInput()
	response_spectra_map(userInput)