# -*- coding: utf-8 -*-
"""
Created on Sun Mar 18 21:07:47 2018

@author: Lawrenzz
"""
import matplotlib
matplotlib.use('Agg')
from obspy import read, Trace
from obspy.signal import PPSD
from obspy.io.xseed import Parser
from obspy.signal.invsim import corn_freq_2_paz
import time
from obspy.signal.invsim import corn_freq_2_paz
from obspy.io.sac.sacpz import attach_resp
from obspy.imaging.cm import pqlx
import numpy as np
import math

paz_sts2 = {
    'poles': [-3.691000E-02 + 3.702000E-02j, -3.691000E-02 - 3.702000E-02j, -3.430000E+02 + 0j, -3.700000E+02 + 4.670000E+02j,
              -3.700000E+02 - 4.670000E+02j, -8.360000E+02 + 1.522000E+03j, -8.360000E+02 - 1.522000E+03j, -4.900000E+03 + 4.700000E+03j, -4.900000E+03 - 4.700000E+03j,-6.900000E+03 + 0j, -1.500000E+04 + 0j],
    'zeros': [0j, 0j, -3.920000E+02 + 0j, -1.960000E+03 + 0j, -1.490000E+03 + 1.740000E+03j, -1.490000E+03 - -1.740000E+03j],
    'gain': 1.000000e+00,
    'sensitivity': 7.543000E+02}
paz_1hz=corn_freq_2_paz(1.0, damp=0.707)
paz_1hz = 1.0
#Inputs:
sta = 'GNGN'    #Station name
input_fils_A= "C:/Users/lawre/Desktop/Thesis_Research/Seismometer/GNCS1/2017/09/06/WG.GNCS2.HHZ_centaur-6_1280_20170906_100000.miniseed" #Make sure that "ls" will return files
input_fils_B= "C:/Users/lawre/Desktop/Thesis_Research/Seismometer/GNCS2/09/06/WG.GNCS1.HHZ_centaur-6_1270_20170906_100000.miniseed" #Second Seismometer
win_len = 3600. #Duration of each segment in seconds ppsd_length
ovr_lap = 0.5  #Overlap is the overlap between consecutive segments
sens_resp = "C:/Users/lawre/Desktop/Thesis_Research/S-ConfigurationSensitivityFilesforexistingTrilliumSensor/BHZ.TrilliumCompact.120.754.dataless" #Sensor Response
digi_resp = "C:/Users/lawre/Desktop/Thesis_Research/S-ConfigurationSensitivityFilesforexistingTrilliumSensor/XX.NN443..HHZ.CENTAUR.2.250.OFF.dataless"#Digitizer Response
#=======================================================================================
s = read(input_fils_A)
"""for chn in ['HHZ']: #Loop through channels
	#scur = s.select(channel=chn) #Select the channel"""
s_orig = s.copy() #Copy of original data
s.simulate(paz_remove=paz_sts2, paz_simulate=paz_1hz)
s_orig.plot()	
s.plot()