예제 #1
0
 def test_sacpz(self):
     """
     Fetches SAC poles and zeros information.
     """
     client = Client()
     # 1
     t1 = UTCDateTime("2005-01-01")
     t2 = UTCDateTime("2008-01-01")
     result = client.sacpz("IU", "ANMO", "00", "BHZ", t1, t2)
     # drop lines with creation date (current time during request)
     result = result.splitlines()
     sacpz_file = os.path.join(self.path, 'data', 'IU.ANMO.00.BHZ.sacpz')
     expected = open(sacpz_file, 'rt').read().splitlines()
     result.pop(5)
     expected.pop(5)
     self.assertEquals(result, expected)
     # 2 - empty location code
     dt = UTCDateTime("2002-11-01")
     result = client.sacpz('UW', 'LON', '', 'BHZ', dt)
     self.assertTrue("* STATION    (KSTNM): LON" in result)
     self.assertTrue("* LOCATION   (KHOLE):   " in result)
     # 3 - empty location code via '--'
     result = client.sacpz('UW', 'LON', '--', 'BHZ', dt)
     self.assertTrue("* STATION    (KSTNM): LON" in result)
     self.assertTrue("* LOCATION   (KHOLE):   " in result)
예제 #2
0
 def test_issue419(self):
     """
     obspy.iris.Client.availability should work with output='bulkdataselect'
     """
     client = Client()
     # 1 - default output ('bulkdataselect')
     t1 = UTCDateTime("2010-02-27T06:30:00.000")
     t2 = UTCDateTime("2010-02-27T10:30:00.000")
     result = client.availability('IU', channel='B*', starttime=t1,
                                  endtime=t2)
     self.assertTrue(isinstance(result, basestring))
     self.assertTrue('IU YSS 00 BHZ' in result)
     # 2 - explicit set output 'bulkdataselect'
     t1 = UTCDateTime("2010-02-27T06:30:00.000")
     t2 = UTCDateTime("2010-02-27T10:30:00.000")
     result = client.availability('IU', channel='B*', starttime=t1,
                                  endtime=t2, output='bulkdataselect')
     self.assertTrue(isinstance(result, basestring))
     self.assertTrue('IU YSS 00 BHZ' in result)
     # 3 - output 'bulk' (backward compatibility)
     t1 = UTCDateTime("2010-02-27T06:30:00.000")
     t2 = UTCDateTime("2010-02-27T10:30:00.000")
     with warnings.catch_warnings(record=True):
         warnings.simplefilter('ignore', DeprecationWarning)
         result = client.availability('IU', channel='B*', starttime=t1,
                                      endtime=t2, output='bulk')
     self.assertTrue(isinstance(result, basestring))
     self.assertTrue('IU YSS 00 BHZ' in result)
     # 4 - output 'xml'
     t1 = UTCDateTime("2010-02-27T06:30:00.000")
     t2 = UTCDateTime("2010-02-27T10:30:00.000")
     result = client.availability('IU', channel='B*', starttime=t1,
                                  endtime=t2, output='xml')
     self.assertTrue(isinstance(result, basestring))
     self.assertTrue('<?xml' in result)
예제 #3
0
    def test_timeseries(self):
        """
        Tests timeseries Web service interface.

        Examples are inspired by http://www.iris.edu/ws/timeseries/.
        """
        client = Client()
        # 1
        t1 = UTCDateTime("2005-001T00:00:00")
        t2 = UTCDateTime("2005-001T00:01:00")
        # no filter
        st1 = client.timeseries("IU", "ANMO", "00", "BHZ", t1, t2)
        # instrument corrected
        st2 = client.timeseries("IU",
                                "ANMO",
                                "00",
                                "BHZ",
                                t1,
                                t2,
                                filter=["correct"])
        # compare results
        self.assertEquals(st1[0].stats.starttime, st2[0].stats.starttime)
        self.assertEquals(st1[0].stats.endtime, st2[0].stats.endtime)
        self.assertEquals(st1[0].data[0], 24)
        self.assertAlmostEquals(st2[0].data[0], -2.4910707e-06)
예제 #4
0
파일: test_client.py 프로젝트: egdorf/obspy
 def test_sacpz(self):
     """
     Fetches SAC poles and zeros information.
     """
     client = Client()
     # 1
     t1 = UTCDateTime("2005-01-01")
     t2 = UTCDateTime("2008-01-01")
     result = client.sacpz("IU", "ANMO", "00", "BHZ", t1, t2)
     # drop lines with creation date (current time during request)
     result = result.splitlines()
     sacpz_file = os.path.join(self.path, "data", "IU.ANMO.00.BHZ.sacpz")
     expected = open(sacpz_file, "rt").read().splitlines()
     result.pop(5)
     expected.pop(5)
     self.assertEquals(result, expected)
     # 2 - empty location code
     dt = UTCDateTime("2002-11-01")
     result = client.sacpz("UW", "LON", "", "BHZ", dt)
     self.assertTrue("* STATION    (KSTNM): LON" in result)
     self.assertTrue("* LOCATION   (KHOLE):   " in result)
     # 3 - empty location code via '--'
     result = client.sacpz("UW", "LON", "--", "BHZ", dt)
     self.assertTrue("* STATION    (KSTNM): LON" in result)
     self.assertTrue("* LOCATION   (KHOLE):   " in result)
예제 #5
0
 def test_saveResponse(self):
     """
     Fetches and stores response information as SEED RESP file.
     """
     client = Client()
     start = UTCDateTime("2005-001T00:00:00")
     end = UTCDateTime("2008-001T00:00:00")
     # RESP, single channel
     origfile = os.path.join(self.path, 'data', 'RESP.ANMO.IU.00.BHZ')
     with NamedTemporaryFile() as tf:
         tempfile = tf.name
         client.saveResponse(tempfile, "IU", "ANMO", "00", "BHZ", start,
                             end)
         self.assertTrue(filecmp.cmp(origfile, tempfile))
     # RESP, multiple channels
     origfile = os.path.join(self.path, 'data', 'RESP.ANMO.IU._.BH_')
     with NamedTemporaryFile() as tf:
         tempfile = tf.name
         client.saveResponse(tempfile, "IU", "ANMO", "*", "BH?", start, end)
         self.assertTrue(filecmp.cmp(origfile, tempfile))
     # StationXML, single channel
     with NamedTemporaryFile() as tf:
         tempfile = tf.name
         client.saveResponse(tempfile, "IU", "ANMO", "00", "BHZ", start,
                             end, format="StationXML")
         data = open(tempfile).read()
     self.assertTrue('<Station net_code="IU" sta_code="ANMO">' in data)
     # SACPZ, single channel
     with NamedTemporaryFile() as tf:
         tempfile = tf.name
         client.saveResponse(tempfile, "IU", "ANMO", "00", "BHZ", start,
                             end, format="SACPZ")
         data = open(tempfile).read()
     self.assertTrue('NETWORK   (KNETWK): IU' in data)
     self.assertTrue('STATION    (KSTNM): ANMO' in data)
예제 #6
0
파일: test_client.py 프로젝트: egdorf/obspy
 def test_getEvents(self):
     """
     Tests getEvents method.
     """
     client = Client()
     dt = UTCDateTime("2012-03-13T04:49:38")
     # 1
     cat = client.getEvents(
         mindepth=34.9,
         maxdepth=35.1,
         magtype="MB",
         catalog="NEIC PDE",
         lat=-56.1,
         lon=-26.7,
         maxradius=2,
         starttime=dt,
         endtime=dt + 10,
     )
     self.assertEquals(len(cat), 1)
     ev = cat[0]
     self.assertEquals(len(ev.origins), 1)
     self.assertEquals(len(ev.magnitudes), 1)
     self.assertEquals(ev.origins[0].depth, 35.0)
     self.assertEquals(ev.origins[0].latitude, -55.404)
     self.assertEquals(ev.origins[0].longitude, -27.895)
     self.assertEquals(ev.magnitudes[0].magnitude_type, "MB")
예제 #7
0
 def test_traveltime(self):
     """
     Tests calculation of travel-times for seismic phases.
     """
     client = Client()
     result = client.traveltime(evloc=(-36.122, -72.898), evdepth=22.9,
         staloc=[(-33.45, -70.67), (47.61, -122.33), (35.69, 139.69)])
     self.assertTrue(result.startswith('Model: iasp91'))
예제 #8
0
 def test_traveltime(self):
     """
     Tests calculation of travel-times for seismic phases.
     """
     client = Client()
     result = client.traveltime(evloc=(-36.122, -72.898), evdepth=22.9,
         staloc=[(-33.45, -70.67), (47.61, -122.33), (35.69, 139.69)])
     self.assertTrue(result.startswith('Model: iasp91'))
예제 #9
0
 def test_issue623(self):
     """
     obspy.iris bulkdataselect only returns last trace in result
     """
     t1 = UTCDateTime("2011-03-11T06:31:30Z")
     t2 = UTCDateTime("2011-03-11T06:48:00Z")
     client = Client()
     st = client.getWaveform("GE", "EIL", "", "BHZ", t1, t2)
     self.assertEqual(len(st), 5)
예제 #10
0
 def test_issue623(self):
     """
     obspy.iris bulkdataselect only returns last trace in result
     """
     t1 = UTCDateTime("2011-03-11T06:31:30Z")
     t2 = UTCDateTime("2011-03-11T06:48:00Z")
     client = Client()
     st = client.getWaveform("GE", "EIL", "", "BHZ", t1, t2)
     self.assertEqual(len(st), 5)
예제 #11
0
파일: test_client.py 프로젝트: egdorf/obspy
 def test_saveResponse(self):
     """
     Fetches and stores response information as SEED RESP file.
     """
     client = Client()
     start = UTCDateTime("2005-001T00:00:00")
     end = UTCDateTime("2008-001T00:00:00")
     # RESP, single channel
     origfile = os.path.join(self.path, "data", "RESP.ANMO.IU.00.BHZ")
     tempfile = NamedTemporaryFile().name
     client.saveResponse(tempfile, "IU", "ANMO", "00", "BHZ", start, end)
     self.assertTrue(filecmp.cmp(origfile, tempfile))
     os.remove(tempfile)
     # RESP, multiple channels
     origfile = os.path.join(self.path, "data", "RESP.ANMO.IU._.BH_")
     tempfile = NamedTemporaryFile().name
     client.saveResponse(tempfile, "IU", "ANMO", "*", "BH?", start, end)
     self.assertTrue(filecmp.cmp(origfile, tempfile))
     os.remove(tempfile)
     # StationXML, single channel
     tempfile = NamedTemporaryFile().name
     client.saveResponse(tempfile, "IU", "ANMO", "00", "BHZ", start, end, format="StationXML")
     data = open(tempfile).read()
     self.assertTrue('<Station net_code="IU" sta_code="ANMO">' in data)
     os.remove(tempfile)
     # SACPZ, single channel
     tempfile = NamedTemporaryFile().name
     client.saveResponse(tempfile, "IU", "ANMO", "00", "BHZ", start, end, format="SACPZ")
     data = open(tempfile).read()
     self.assertTrue("NETWORK   (KNETWK): IU" in data)
     self.assertTrue("STATION    (KSTNM): ANMO" in data)
     os.remove(tempfile)
예제 #12
0
 def test_flinnengdahl(self):
     """
     Tests calculation of Flinn-Engdahl region code or name.
     """
     client = Client()
     # code
     result = client.flinnengdahl(lat=-20.5, lon=-100.6, rtype="code")
     self.assertEquals(result, 683)
     # w/o kwargs
     result = client.flinnengdahl(-20.5, -100.6, "code")
     self.assertEquals(result, 683)
     # region
     result = client.flinnengdahl(lat=42, lon=-122.24, rtype="region")
     self.assertEquals(result, 'OREGON')
     # w/o kwargs
     result = client.flinnengdahl(42, -122.24, "region")
     self.assertEquals(result, 'OREGON')
     # both
     result = client.flinnengdahl(lat=-20.5, lon=-100.6, rtype="both")
     self.assertEquals(result, (683, 'SOUTHEAST CENTRAL PACIFIC OCEAN'))
     # w/o kwargs
     result = client.flinnengdahl(-20.5, -100.6, "both")
     self.assertEquals(result, (683, 'SOUTHEAST CENTRAL PACIFIC OCEAN'))
     # default rtype
     result = client.flinnengdahl(lat=42, lon=-122.24)
     self.assertEquals(result, (32, 'OREGON'))
     # w/o kwargs
     # outside boundaries
     self.assertRaises(Exception, client.flinnengdahl, lat=-90.1, lon=0)
     self.assertRaises(Exception, client.flinnengdahl, lat=90.1, lon=0)
     self.assertRaises(Exception, client.flinnengdahl, lat=0, lon=-180.1)
     self.assertRaises(Exception, client.flinnengdahl, lat=0, lon=180.1)
예제 #13
0
 def test_saveResponseToStringIO(self):
     """
     Same as test_saveResponse but saves to a StringIO.
     """
     client = Client()
     start = UTCDateTime("2005-001T00:00:00")
     end = UTCDateTime("2008-001T00:00:00")
     # RESP, single channel
     origfile = os.path.join(self.path, 'data', 'RESP.ANMO.IU.00.BHZ')
     with open(origfile, "rb") as fh:
         org_data = fh.read()
     memfile = StringIO.StringIO()
     client.saveResponse(memfile, "IU", "ANMO", "00", "BHZ", start, end)
     memfile.seek(0, 0)
     new_data = memfile.read()
     self.assertEqual(new_data, org_data)
     # RESP, multiple channels
     origfile = os.path.join(self.path, 'data', 'RESP.ANMO.IU._.BH_')
     with open(origfile, "rb") as fh:
         org_data = fh.read()
     memfile = StringIO.StringIO()
     client.saveResponse(memfile, "IU", "ANMO", "*", "BH?", start, end)
     memfile.seek(0, 0)
     new_data = memfile.read()
     self.assertEqual(new_data, org_data)
     # StationXML, single channel
     memfile = StringIO.StringIO()
     client.saveResponse(memfile,
                         "IU",
                         "ANMO",
                         "00",
                         "BHZ",
                         start,
                         end,
                         format="StationXML")
     memfile.seek(0, 0)
     data = memfile.read()
     self.assertTrue('<Station net_code="IU" sta_code="ANMO">' in data)
     # SACPZ, single channel
     memfile = StringIO.StringIO()
     client.saveResponse(memfile,
                         "IU",
                         "ANMO",
                         "00",
                         "BHZ",
                         start,
                         end,
                         format="SACPZ")
     memfile.seek(0, 0)
     data = memfile.read()
     self.assertTrue('NETWORK   (KNETWK): IU' in data)
     self.assertTrue('STATION    (KSTNM): ANMO' in data)
예제 #14
0
파일: rotate.py 프로젝트: dsiervo/Ptgc
def az2(sfile):
	azDic = {}
	client = Client() #Creando objeto cliente
	evtlat,evtlon = evcor(sfile) #Obteniendo las coordenadas del evento del sfile en decimal
	infile = open(sfile,'rU') 
	for line in infile:

		if line[9:11] == 'ES':
			#print 'Dentro del if'
			station_name = line[1:6].split(' ')[0]
			Baz = client.distaz(stalat=stacor('corstations.txt')[station_name][0], stalon=stacor('corstations.txt')[station_name][1], evtlat=evtlat, evtlon=evtlon)
			azDic[station_name] = Baz['backazimuth']
	return azDic
예제 #15
0
파일: test_client.py 프로젝트: egdorf/obspy
 def test_flinnengdahl(self):
     """
     Tests calculation of Flinn-Engdahl region code or name.
     """
     client = Client()
     # code
     result = client.flinnengdahl(lat=-20.5, lon=-100.6, rtype="code")
     self.assertEquals(result, 683)
     # w/o kwargs
     result = client.flinnengdahl(-20.5, -100.6, "code")
     self.assertEquals(result, 683)
     # region
     result = client.flinnengdahl(lat=42, lon=-122.24, rtype="region")
     self.assertEquals(result, "OREGON")
     # w/o kwargs
     result = client.flinnengdahl(42, -122.24, "region")
     self.assertEquals(result, "OREGON")
     # both
     result = client.flinnengdahl(lat=-20.5, lon=-100.6, rtype="both")
     self.assertEquals(result, (683, "SOUTHEAST CENTRAL PACIFIC OCEAN"))
     # w/o kwargs
     result = client.flinnengdahl(-20.5, -100.6, "both")
     self.assertEquals(result, (683, "SOUTHEAST CENTRAL PACIFIC OCEAN"))
     # default rtype
     result = client.flinnengdahl(lat=42, lon=-122.24)
     self.assertEquals(result, (32, "OREGON"))
     # w/o kwargs
     # outside boundaries
     self.assertRaises(Exception, client.flinnengdahl, lat=-90.1, lon=0)
     self.assertRaises(Exception, client.flinnengdahl, lat=90.1, lon=0)
     self.assertRaises(Exception, client.flinnengdahl, lat=0, lon=-180.1)
     self.assertRaises(Exception, client.flinnengdahl, lat=0, lon=180.1)
예제 #16
0
    def test_event(self):
        """
        Tests event Web service interface.

        Examples are inspired by http://www.iris.edu/ws/event/.
        """
        client = Client()
        # 1
        url = "http://www.iris.edu/ws/event/query?mindepth=34.9&" + \
            "maxdepth=35.1&catalog=NEIC%20PDE&contributor=NEIC%20PDE-Q&" + \
            "magtype=MB&lat=-56.1&lon=-26.7&maxradius=1"
        # direct call
        doc1 = urllib.urlopen(url).read()
        # using client
        doc2 = client.event(mindepth=34.9, maxdepth=35.1, catalog="NEIC PDE",
                            contributor="NEIC PDE-Q", magtype="MB", lat=-56.1,
                            lon=-26.7, maxradius=1)
        self.assertEqual(doc1, doc2)
        client = Client()
        # 2
        url = "http://www.iris.edu/ws/event/query?eventid=3316989"
        # direct call
        doc1 = urllib.urlopen(url).read()
        # using client
        doc2 = client.event(eventid=3316989)
        self.assertEqual(doc1, doc2)
        # 2
        url = "http://www.iris.edu/ws/event/query?eventid=3316989"
        # direct call
        doc1 = urllib.urlopen(url).read()
        # using client
        doc2 = client.event(eventid=3316989)
        self.assertEqual(doc1, doc2)
        # 3
        url = "http://www.iris.edu/ws/event/query?minmag=8.5"
        # direct call
        doc1 = urllib.urlopen(url).read()
        # using client
        doc2 = client.event(minmag=8.5)
        self.assertEqual(doc1, doc2)
        # 4
        url = "http://www.iris.edu/ws/event/query?starttime=2011-01-07T" + \
            "14%3A00%3A00&endtime=2011-02-07&minlat=15&maxlat=40&" + \
            "minlon=-170&maxlon=170&preferredonly=yes&" + \
            "includeallmagnitudes=yes&orderby=magnitude"
        # direct call
        doc1 = urllib.urlopen(url).read()
        # using client
        doc2 = client.event(starttime=UTCDateTime(2011, 1, 7, 14),
                            endtime=UTCDateTime('2011-02-07'), minlat=15.0,
                            maxlat=40.0, minlon=-170, maxlon=170,
                            preferredonly=True, includeallmagnitudes=True,
                            orderby='magnitude')
        self.assertEqual(doc1, doc2)
예제 #17
0
파일: test_client.py 프로젝트: egdorf/obspy
 def test_saveWaveform(self):
     """
     Testing simple waveform file save method.
     """
     # file identical to file retrieved via web interface
     client = Client()
     start = UTCDateTime("2010-02-27T06:30:00")
     end = UTCDateTime("2010-02-27T06:31:00")
     origfile = os.path.join(self.path, "data", "IU.ANMO.00.BHZ.mseed")
     tempfile = NamedTemporaryFile().name
     client.saveWaveform(tempfile, "IU", "ANMO", "00", "BHZ", start, end)
     self.assertTrue(filecmp.cmp(origfile, tempfile))
     os.remove(tempfile)
     # no data raises an exception
     self.assertRaises(Exception, client.saveWaveform, "YY", "XXXX", "00", "BHZ", start, end)
예제 #18
0
 def test_saveWaveformToStringIO(self):
     """
     Same as test_saveWaveform but saves to a StringIO.
     """
     # file identical to file retrieved via web interface
     client = Client()
     start = UTCDateTime("2010-02-27T06:30:00")
     end = UTCDateTime("2010-02-27T06:31:00")
     memfile = StringIO.StringIO()
     client.saveWaveform(memfile, "IU", "ANMO", "00", "BHZ", start, end)
     memfile.seek(0, 0)
     new_data = memfile.read()
     origfile = os.path.join(self.path, 'data', 'IU.ANMO.00.BHZ.mseed')
     with open(origfile, "rb") as fh:
         org_data = fh.read()
     self.assertEqual(new_data, org_data)
예제 #19
0
 def test_saveWaveform(self):
     """
     Testing simple waveform file save method.
     """
     # file identical to file retrieved via web interface
     client = Client()
     start = UTCDateTime("2010-02-27T06:30:00")
     end = UTCDateTime("2010-02-27T06:31:00")
     origfile = os.path.join(self.path, 'data', 'IU.ANMO.00.BHZ.mseed')
     tempfile = NamedTemporaryFile().name
     client.saveWaveform(tempfile, "IU", "ANMO", "00", "BHZ", start, end)
     self.assertTrue(filecmp.cmp(origfile, tempfile))
     os.remove(tempfile)
     # no data raises an exception
     self.assertRaises(Exception, client.saveWaveform, "YY", "XXXX", "00",
                       "BHZ", start, end)
예제 #20
0
 def test_saveWaveformToStringIO(self):
     """
     Same as test_saveWaveform but saves to a StringIO.
     """
     # file identical to file retrieved via web interface
     client = Client()
     start = UTCDateTime("2010-02-27T06:30:00")
     end = UTCDateTime("2010-02-27T06:31:00")
     memfile = StringIO.StringIO()
     client.saveWaveform(memfile, "IU", "ANMO", "00", "BHZ", start, end)
     memfile.seek(0, 0)
     new_data = memfile.read()
     origfile = os.path.join(self.path, 'data', 'IU.ANMO.00.BHZ.mseed')
     with open(origfile, "rb") as fh:
         org_data = fh.read()
     self.assertEqual(new_data, org_data)
예제 #21
0
 def test_getEvents(self):
     """
     Tests getEvents method.
     """
     client = Client()
     dt = UTCDateTime("2012-03-13T04:49:38")
     # 1
     cat = client.getEvents(mindepth=34.9, maxdepth=35.1, magtype="MB",
                            catalog="NEIC PDE", lat=-56.1, lon=-26.7,
                            maxradius=2, starttime=dt, endtime=dt + 10)
     self.assertEqual(len(cat), 1)
     ev = cat[0]
     self.assertEqual(len(ev.origins), 1)
     self.assertEqual(len(ev.magnitudes), 1)
     self.assertEqual(ev.origins[0].depth, 35.0)
     self.assertEqual(ev.origins[0].latitude, -55.404)
     self.assertEqual(ev.origins[0].longitude, -27.895)
     self.assertEqual(ev.magnitudes[0].magnitude_type, 'MB')
예제 #22
0
파일: test_client.py 프로젝트: egdorf/obspy
 def test_getWaveform(self):
     """
     Testing simple waveform request method.
     """
     # simple example
     client = Client()
     start = UTCDateTime("2010-02-27T06:30:00.019538Z")
     end = start + 20
     stream = client.getWaveform("IU", "ANMO", "00", "BHZ", start, end)
     self.assertEquals(len(stream), 1)
     self.assertEquals(stream[0].stats.starttime, start)
     self.assertEquals(stream[0].stats.endtime, end)
     self.assertEquals(stream[0].stats.network, "IU")
     self.assertEquals(stream[0].stats.station, "ANMO")
     self.assertEquals(stream[0].stats.location, "00")
     self.assertEquals(stream[0].stats.channel, "BHZ")
     # no data raises an exception
     self.assertRaises(Exception, client.getWaveform, "YY", "XXXX", "00", "BHZ", start, end)
예제 #23
0
 def test_getWaveform(self):
     """
     Testing simple waveform request method.
     """
     # simple example
     client = Client()
     start = UTCDateTime("2010-02-27T06:30:00.019538Z")
     end = start + 20
     stream = client.getWaveform("IU", "ANMO", "00", "BHZ", start, end)
     self.assertEquals(len(stream), 1)
     self.assertEquals(stream[0].stats.starttime, start)
     self.assertEquals(stream[0].stats.endtime, end)
     self.assertEquals(stream[0].stats.network, 'IU')
     self.assertEquals(stream[0].stats.station, 'ANMO')
     self.assertEquals(stream[0].stats.location, '00')
     self.assertEquals(stream[0].stats.channel, 'BHZ')
     # no data raises an exception
     self.assertRaises(Exception, client.getWaveform, "YY", "XXXX", "00",
                       "BHZ", start, end)
예제 #24
0
파일: test_client.py 프로젝트: egdorf/obspy
    def test_timeseries(self):
        """
        Tests timeseries Web service interface.

        Examples are inspired by http://www.iris.edu/ws/timeseries/.
        """
        client = Client()
        # 1
        t1 = UTCDateTime("2005-001T00:00:00")
        t2 = UTCDateTime("2005-001T00:01:00")
        # no filter
        st1 = client.timeseries("IU", "ANMO", "00", "BHZ", t1, t2)
        # instrument corrected
        st2 = client.timeseries("IU", "ANMO", "00", "BHZ", t1, t2, filter=["correct"])
        # compare results
        self.assertEquals(st1[0].stats.starttime, st2[0].stats.starttime)
        self.assertEquals(st1[0].stats.endtime, st2[0].stats.endtime)
        self.assertEquals(st1[0].data[0], 24)
        self.assertAlmostEquals(st2[0].data[0], -2.4910707e-06)
예제 #25
0
파일: test_client.py 프로젝트: egdorf/obspy
 def test_distaz(self):
     """
     Tests distance and azimuth calculation between two points on a sphere.
     """
     client = Client()
     # normal request
     result = client.distaz(stalat=1.1, stalon=1.2, evtlat=3.2, evtlon=1.4)
     self.assertAlmostEquals(result["distance"], 2.09554)
     self.assertAlmostEquals(result["backazimuth"], 5.46946)
     self.assertAlmostEquals(result["azimuth"], 185.47692)
     # w/o kwargs
     result = client.distaz(1.1, 1.2, 3.2, 1.4)
     self.assertAlmostEquals(result["distance"], 2.09554)
     self.assertAlmostEquals(result["backazimuth"], 5.46946)
     self.assertAlmostEquals(result["azimuth"], 185.47692)
     # missing parameters
     self.assertRaises(Exception, client.distaz, stalat=1.1)
     self.assertRaises(Exception, client.distaz, 1.1)
     self.assertRaises(Exception, client.distaz, stalat=1.1, stalon=1.2)
     self.assertRaises(Exception, client.distaz, 1.1, 1.2)
예제 #26
0
 def test_distaz(self):
     """
     Tests distance and azimuth calculation between two points on a sphere.
     """
     client = Client()
     # normal request
     result = client.distaz(stalat=1.1, stalon=1.2, evtlat=3.2, evtlon=1.4)
     self.assertAlmostEquals(result['distance'], 2.09554)
     self.assertAlmostEquals(result['backazimuth'], 5.46946)
     self.assertAlmostEquals(result['azimuth'], 185.47692)
     # w/o kwargs
     result = client.distaz(1.1, 1.2, 3.2, 1.4)
     self.assertAlmostEquals(result['distance'], 2.09554)
     self.assertAlmostEquals(result['backazimuth'], 5.46946)
     self.assertAlmostEquals(result['azimuth'], 185.47692)
     # missing parameters
     self.assertRaises(Exception, client.distaz, stalat=1.1)
     self.assertRaises(Exception, client.distaz, 1.1)
     self.assertRaises(Exception, client.distaz, stalat=1.1, stalon=1.2)
     self.assertRaises(Exception, client.distaz, 1.1, 1.2)
예제 #27
0
 def test_issue419(self):
     """
     obspy.iris.Client.availability should work with output='bulkdataselect'
     """
     client = Client()
     # 1 - default output ('bulkdataselect')
     t1 = UTCDateTime("2010-02-27T06:30:00.000")
     t2 = UTCDateTime("2010-02-27T10:30:00.000")
     result = client.availability('IU', channel='B*', starttime=t1,
                                  endtime=t2)
     self.assertTrue(isinstance(result, basestring))
     self.assertTrue('IU YSS 00 BHZ' in result)
     # 2 - explicit set output 'bulkdataselect'
     t1 = UTCDateTime("2010-02-27T06:30:00.000")
     t2 = UTCDateTime("2010-02-27T10:30:00.000")
     result = client.availability('IU', channel='B*', starttime=t1,
                                  endtime=t2, output='bulkdataselect')
     self.assertTrue(isinstance(result, basestring))
     self.assertTrue('IU YSS 00 BHZ' in result)
     # 3 - output 'bulk' (backward compatibility)
     t1 = UTCDateTime("2010-02-27T06:30:00.000")
     t2 = UTCDateTime("2010-02-27T10:30:00.000")
     with warnings.catch_warnings(record=True):
         warnings.simplefilter('ignore', DeprecationWarning)
         result = client.availability('IU', channel='B*', starttime=t1,
                                      endtime=t2, output='bulk')
     self.assertTrue(isinstance(result, basestring))
     self.assertTrue('IU YSS 00 BHZ' in result)
     # 4 - output 'xml'
     t1 = UTCDateTime("2010-02-27T06:30:00.000")
     t2 = UTCDateTime("2010-02-27T10:30:00.000")
     result = client.availability('IU', channel='B*', starttime=t1,
                                  endtime=t2, output='xml')
     self.assertTrue(isinstance(result, basestring))
     self.assertTrue('<?xml' in result)
예제 #28
0
    def test_event(self):
        """
        Tests event Web service interface.

        Examples are inspired by http://www.iris.edu/ws/event/.
        """
        client = Client()
        # 1
        url = "http://www.iris.edu/ws/event/query?mindepth=34.9&" + \
            "maxdepth=35.1&catalog=NEIC%20PDE&contributor=NEIC%20PDE-Q&" + \
            "magtype=MB&lat=-56.1&lon=-26.7&maxradius=1"
        # direct call
        doc1 = urllib.urlopen(url).read()
        # using client
        doc2 = client.event(mindepth=34.9, maxdepth=35.1, catalog="NEIC PDE",
                            contributor="NEIC PDE-Q", magtype="MB", lat=-56.1,
                            lon=-26.7, maxradius=1)
        self.assertEqual(doc1, doc2)
        client = Client()
        # 2
        url = "http://www.iris.edu/ws/event/query?eventid=3316989"
        # direct call
        doc1 = urllib.urlopen(url).read()
        # using client
        doc2 = client.event(eventid=3316989)
        self.assertEqual(doc1, doc2)
        # 2
        url = "http://www.iris.edu/ws/event/query?eventid=3316989"
        # direct call
        doc1 = urllib.urlopen(url).read()
        # using client
        doc2 = client.event(eventid=3316989)
        self.assertEqual(doc1, doc2)
        # 3
        url = "http://www.iris.edu/ws/event/query?minmag=8.5"
        # direct call
        doc1 = urllib.urlopen(url).read()
        # using client
        doc2 = client.event(minmag=8.5)
        self.assertEqual(doc1, doc2)
        # 4
        url = "http://www.iris.edu/ws/event/query?starttime=2011-01-07T" + \
            "14%3A00%3A00&endtime=2011-02-07&minlat=15&maxlat=40&" + \
            "minlon=-170&maxlon=170&preferredonly=yes&" + \
            "includeallmagnitudes=yes&orderby=magnitude"
        # direct call
        doc1 = urllib.urlopen(url).read()
        # using client
        doc2 = client.event(starttime=UTCDateTime(2011, 1, 7, 14),
                            endtime=UTCDateTime('2011-02-07'), minlat=15.0,
                            maxlat=40.0, minlon=-170, maxlon=170,
                            preferredonly=True, includeallmagnitudes=True,
                            orderby='magnitude')
        self.assertEqual(doc1, doc2)
예제 #29
0
    def test_availability(self):
        """
        Tests availability of waveform data at the DMC.

        Examples are inspired by http://www.iris.edu/ws/availability/.
        """
        client = Client()
        # 1
        t1 = UTCDateTime("2010-02-27T06:30:00.000")
        t2 = UTCDateTime("2010-02-27T10:30:00.000")
        result = client.availability('IU',
                                     channel='B*',
                                     starttime=t1,
                                     endtime=t2)
        self.assertTrue(isinstance(result, basestring))
        self.assertTrue('IU YSS 00 BHZ' in result)
        # 2
        dt = UTCDateTime("2011-11-13T07:00:00")
        result = client.availability(network='GE',
                                     starttime=dt,
                                     endtime=dt + 10)
        self.assertTrue(isinstance(result, basestring))
        self.assertTrue('GE DAG -- BHE' in result)
        # 3 - unknown network results in empty string
        dt = UTCDateTime(2011, 11, 16)
        result = client.availability(network='XX',
                                     starttime=dt,
                                     endtime=dt + 10)
        # 4 - location=None
        t1 = UTCDateTime("2010-02-27T06:30:00")
        t2 = UTCDateTime("2010-02-27T06:40:00")
        result = client.availability("IU", "K*", starttime=t1, endtime=t2)
        self.assertTrue(isinstance(result, basestring))
        self.assertTrue('IU KBL -- BHZ' in result)
        self.assertTrue('IU KBS 00 BHE' in result)
        # 5 - empty location
        result = client.availability("IU", "K*", "", starttime=t1, endtime=t2)
        self.assertTrue(isinstance(result, basestring))
        self.assertTrue('IU KBL -- BHZ' in result)
        self.assertFalse('IU KBS 00 BHE' in result)
        # 6 - empty location code via '--'
        result = client.availability("IU",
                                     "K*",
                                     "--",
                                     starttime=t1,
                                     endtime=t2)
        self.assertTrue(isinstance(result, basestring))
        self.assertTrue('IU KBL -- BHZ' in result)
        self.assertFalse('IU KBS 00 BHE' in result)
예제 #30
0
파일: test_client.py 프로젝트: egdorf/obspy
    def test_resp(self):
        """
        Tests resp Web service interface.

        Examples are inspired by http://www.iris.edu/ws/resp/.
        """
        client = Client()
        # 1
        t1 = UTCDateTime("2005-001T00:00:00")
        t2 = UTCDateTime("2008-001T00:00:00")
        result = client.resp("IU", "ANMO", "00", "BHZ", t1, t2)
        self.assertTrue("B050F03     Station:     ANMO" in result)
        # 2 - empty location code
        result = client.resp("UW", "LON", "", "EHZ")
        self.assertTrue("B050F03     Station:     LON" in result)
        self.assertTrue("B052F03     Location:    ??" in result)
        # 3 - empty location code via '--'
        result = client.resp("UW", "LON", "--", "EHZ")
        self.assertTrue("B050F03     Station:     LON" in result)
        self.assertTrue("B052F03     Location:    ??" in result)
        # 4
        dt = UTCDateTime("2010-02-27T06:30:00.000")
        result = client.resp("IU", "ANMO", "*", "*", dt)
        self.assertTrue("B050F03     Station:     ANMO" in result)
예제 #31
0
 def test_saveResponseToStringIO(self):
     """
     Same as test_saveResponse but saves to a StringIO.
     """
     client = Client()
     start = UTCDateTime("2005-001T00:00:00")
     end = UTCDateTime("2008-001T00:00:00")
     # RESP, single channel
     origfile = os.path.join(self.path, 'data', 'RESP.ANMO.IU.00.BHZ')
     with open(origfile, "rb") as fh:
         org_data = fh.read()
     memfile = StringIO.StringIO()
     client.saveResponse(memfile, "IU", "ANMO", "00", "BHZ", start, end)
     memfile.seek(0, 0)
     new_data = memfile.read()
     self.assertEqual(new_data, org_data)
     # RESP, multiple channels
     origfile = os.path.join(self.path, 'data', 'RESP.ANMO.IU._.BH_')
     with open(origfile, "rb") as fh:
         org_data = fh.read()
     memfile = StringIO.StringIO()
     client.saveResponse(memfile, "IU", "ANMO", "*", "BH?", start, end)
     memfile.seek(0, 0)
     new_data = memfile.read()
     self.assertEqual(new_data, org_data)
     # StationXML, single channel
     memfile = StringIO.StringIO()
     client.saveResponse(memfile, "IU", "ANMO", "00", "BHZ", start, end,
             format="StationXML")
     memfile.seek(0, 0)
     data = memfile.read()
     self.assertTrue('<Station net_code="IU" sta_code="ANMO">' in data)
     # SACPZ, single channel
     memfile = StringIO.StringIO()
     client.saveResponse(memfile, "IU", "ANMO", "00", "BHZ", start, end,
             format="SACPZ")
     memfile.seek(0, 0)
     data = memfile.read()
     self.assertTrue('NETWORK   (KNETWK): IU' in data)
     self.assertTrue('STATION    (KSTNM): ANMO' in data)
예제 #32
0
    def test_availability(self):
        """
        Tests availability of waveform data at the DMC.

        Examples are inspired by http://www.iris.edu/ws/availability/.
        """
        client = Client()
        # 1
        t1 = UTCDateTime("2010-02-27T06:30:00.000")
        t2 = UTCDateTime("2010-02-27T10:30:00.000")
        result = client.availability('IU', channel='B*', starttime=t1,
                                     endtime=t2)
        self.assertTrue(isinstance(result, basestring))
        self.assertTrue('IU YSS 00 BHZ' in result)
        # 2
        dt = UTCDateTime("2011-11-13T07:00:00")
        result = client.availability(network='GE', starttime=dt,
                                     endtime=dt + 10)
        self.assertTrue(isinstance(result, basestring))
        self.assertTrue('GE DAG -- BHE' in result)
        # 3 - unknown network results in empty string
        dt = UTCDateTime(2011, 11, 16)
        result = client.availability(network='XX', starttime=dt,
                                     endtime=dt + 10)
        # 4 - location=None
        t1 = UTCDateTime("2010-02-27T06:30:00")
        t2 = UTCDateTime("2010-02-27T06:40:00")
        result = client.availability("IU", "K*", starttime=t1, endtime=t2)
        self.assertTrue(isinstance(result, basestring))
        self.assertTrue('IU KBL -- BHZ' in result)
        self.assertTrue('IU KBS 00 BHE' in result)
        # 5 - empty location
        result = client.availability("IU", "K*", "", starttime=t1, endtime=t2)
        self.assertTrue(isinstance(result, basestring))
        self.assertTrue('IU KBL -- BHZ' in result)
        self.assertFalse('IU KBS 00 BHE' in result)
        # 6 - empty location code via '--'
        result = client.availability("IU", "K*", "--", starttime=t1,
                                     endtime=t2)
        self.assertTrue(isinstance(result, basestring))
        self.assertTrue('IU KBL -- BHZ' in result)
        self.assertFalse('IU KBS 00 BHE' in result)
예제 #33
0
    def test_resp(self):
        """
        Tests resp Web service interface.

        Examples are inspired by http://www.iris.edu/ws/resp/.
        """
        client = Client()
        # 1
        t1 = UTCDateTime("2005-001T00:00:00")
        t2 = UTCDateTime("2008-001T00:00:00")
        result = client.resp("IU", "ANMO", "00", "BHZ", t1, t2)
        self.assertTrue('B050F03     Station:     ANMO' in result)
        # 2 - empty location code
        result = client.resp("UW", "LON", "", "EHZ")
        self.assertTrue('B050F03     Station:     LON' in result)
        self.assertTrue('B052F03     Location:    ??' in result)
        # 3 - empty location code via '--'
        result = client.resp("UW", "LON", "--", "EHZ")
        self.assertTrue('B050F03     Station:     LON' in result)
        self.assertTrue('B052F03     Location:    ??' in result)
        # 4
        dt = UTCDateTime("2010-02-27T06:30:00.000")
        result = client.resp("IU", "ANMO", "*", "*", dt)
        self.assertTrue('B050F03     Station:     ANMO' in result)
예제 #34
0
 def test_evalresp(self):
     """
     Tests evaluating instrument response information.
     """
     client = Client()
     dt = UTCDateTime("2005-01-01")
     # plot as PNG file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU",
                     station="ANMO",
                     location="00",
                     channel="BHZ",
                     time=dt,
                     output='plot',
                     filename=tempfile)
     self.assertEqual(open(tempfile, 'rb').read(4)[1:4], 'PNG')
     os.remove(tempfile)
     # plot-amp as PNG file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU",
                     station="ANMO",
                     location="00",
                     channel="BHZ",
                     time=dt,
                     output='plot-amp',
                     filename=tempfile)
     self.assertEqual(open(tempfile, 'rb').read(4)[1:4], 'PNG')
     os.remove(tempfile)
     # plot-phase as PNG file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU",
                     station="ANMO",
                     location="00",
                     channel="BHZ",
                     time=dt,
                     output='plot-phase',
                     filename=tempfile)
     self.assertEqual(open(tempfile, 'rb').read(4)[1:4], 'PNG')
     os.remove(tempfile)
     # fap as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU",
                     station="ANMO",
                     location="00",
                     channel="BHZ",
                     time=dt,
                     output='fap',
                     filename=tempfile)
     self.assertEqual(
         open(tempfile, 'rt').readline(),
         '1.000000E-05  1.202802E+04  1.792007E+02\n')
     os.remove(tempfile)
     # cs as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU",
                     station="ANMO",
                     location="00",
                     channel="BHZ",
                     time=dt,
                     output='cs',
                     filename=tempfile)
     self.assertEqual(
         open(tempfile, 'rt').readline(),
         '1.000000E-05 -1.202685E+04 1.677835E+02\n')
     os.remove(tempfile)
     # fap & def as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU",
                     station="ANMO",
                     location="00",
                     channel="BHZ",
                     time=dt,
                     output='fap',
                     units='def',
                     filename=tempfile)
     self.assertEqual(
         open(tempfile, 'rt').readline(),
         '1.000000E-05  1.202802E+04  1.792007E+02\n')
     os.remove(tempfile)
     # fap & dis as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU",
                     station="ANMO",
                     location="00",
                     channel="BHZ",
                     time=dt,
                     output='fap',
                     units='dis',
                     filename=tempfile)
     self.assertEqual(
         open(tempfile, 'rt').readline(),
         '1.000000E-05  7.557425E-01  2.692007E+02\n')
     os.remove(tempfile)
     # fap & vel as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU",
                     station="ANMO",
                     location="00",
                     channel="BHZ",
                     time=dt,
                     output='fap',
                     units='vel',
                     filename=tempfile)
     self.assertEqual(
         open(tempfile, 'rt').readline(),
         '1.000000E-05  1.202802E+04  1.792007E+02\n')
     os.remove(tempfile)
     # fap & acc as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU",
                     station="ANMO",
                     location="00",
                     channel="BHZ",
                     time=dt,
                     output='fap',
                     units='acc',
                     filename=tempfile)
     self.assertEqual(
         open(tempfile, 'rt').readline(),
         '1.000000E-05  1.914318E+08  8.920073E+01\n')
     os.remove(tempfile)
     # fap as NumPy ndarray
     data = client.evalresp(network="IU",
                            station="ANMO",
                            location="00",
                            channel="BHZ",
                            time=dt,
                            output='fap')
     np.testing.assert_array_equal(
         data[0], [1.00000000e-05, 1.20280200e+04, 1.79200700e+02])
     # cs as NumPy ndarray
     data = client.evalresp(network="IU",
                            station="ANMO",
                            location="00",
                            channel="BHZ",
                            time=dt,
                            output='cs')
     np.testing.assert_array_equal(
         data[0], [1.00000000e-05, -1.20268500e+04, 1.67783500e+02])
예제 #35
0
 def test_evalresp(self):
     """
     Tests evaluating instrument response information.
     """
     client = Client()
     dt = UTCDateTime("2005-01-01")
     # plot as PNG file
     with NamedTemporaryFile() as tf:
         tempfile = tf.name
         client.evalresp(network="IU", station="ANMO", location="00",
                         channel="BHZ", time=dt, output='plot',
                         filename=tempfile)
         with open(tempfile, 'rb') as fp:
             self.assertEqual(fp.read(4)[1:4], b'PNG')
     # plot-amp as PNG file
     with NamedTemporaryFile() as tf:
         tempfile = tf.name
         client.evalresp(network="IU", station="ANMO", location="00",
                         channel="BHZ", time=dt, output='plot-amp',
                         filename=tempfile)
         with open(tempfile, 'rb') as fp:
             self.assertEqual(fp.read(4)[1:4], b'PNG')
     # plot-phase as PNG file
     with NamedTemporaryFile() as tf:
         tempfile = tf.name
         client.evalresp(network="IU", station="ANMO", location="00",
                         channel="BHZ", time=dt, output='plot-phase',
                         filename=tempfile)
         with open(tempfile, 'rb') as fp:
             self.assertEqual(fp.read(4)[1:4], b'PNG')
     # fap as ASCII file
     with NamedTemporaryFile() as tf:
         tempfile = tf.name
         client.evalresp(network="IU", station="ANMO", location="00",
                         channel="BHZ", time=dt, output='fap',
                         filename=tempfile)
         with open(tempfile, 'rt') as fp:
             self.assertEqual(fp.readline(),
                              '1.000000E-05  1.055999E+04  1.792007E+02\n')
     # cs as ASCII file
     with NamedTemporaryFile() as tf:
         tempfile = tf.name
         client.evalresp(network="IU", station="ANMO", location="00",
                         channel="BHZ", time=dt, output='cs',
                         filename=tempfile)
         with open(tempfile, 'rt') as fp:
             self.assertEqual(fp.readline(),
                              '1.000000E-05 -1.055896E+04 1.473054E+02\n')
     # fap & def as ASCII file
     with NamedTemporaryFile() as tf:
         tempfile = tf.name
         client.evalresp(network="IU", station="ANMO", location="00",
                         channel="BHZ", time=dt, output='fap', units='def',
                         filename=tempfile)
         with open(tempfile, 'rt') as fp:
             self.assertEqual(fp.readline(),
                              '1.000000E-05  1.055999E+04  1.792007E+02\n')
     # fap & dis as ASCII file
     with NamedTemporaryFile() as tf:
         tempfile = tf.name
         client.evalresp(network="IU", station="ANMO", location="00",
                         channel="BHZ", time=dt, output='fap', units='dis',
                         filename=tempfile)
         with open(tempfile, 'rt') as fp:
             self.assertEqual(fp.readline(),
                              '1.000000E-05  6.635035E-01  2.692007E+02\n')
     # fap & vel as ASCII file
     with NamedTemporaryFile() as tf:
         tempfile = tf.name
         client.evalresp(network="IU", station="ANMO", location="00",
                         channel="BHZ", time=dt, output='fap', units='vel',
                         filename=tempfile)
         with open(tempfile, 'rt') as fp:
             self.assertEqual(fp.readline(),
                              '1.000000E-05  1.055999E+04  1.792007E+02\n')
     # fap & acc as ASCII file
     with NamedTemporaryFile() as tf:
         tempfile = tf.name
         client.evalresp(network="IU", station="ANMO", location="00",
                         channel="BHZ", time=dt, output='fap', units='acc',
                         filename=tempfile)
         with open(tempfile, 'rt') as fp:
             self.assertEqual(fp.readline(),
                              '1.000000E-05  1.680674E+08  8.920073E+01\n')
     # fap as NumPy ndarray
     data = client.evalresp(network="IU", station="ANMO", location="00",
                            channel="BHZ", time=dt, output='fap')
     np.testing.assert_array_equal(
         data[0], [1.00000000e-05, 1.05599900e+04, 1.79200700e+02])
     # cs as NumPy ndarray
     data = client.evalresp(network="IU", station="ANMO", location="00",
                            channel="BHZ", time=dt, output='cs')
     np.testing.assert_array_equal(
         data[0], [1.00000000e-05, -1.05589600e+04, 1.47305400e+02])
예제 #36
0
파일: test_client.py 프로젝트: egdorf/obspy
 def test_evalresp(self):
     """
     Tests evaluating instrument response information.
     """
     client = Client()
     dt = UTCDateTime("2005-01-01")
     # plot as PNG file
     tempfile = NamedTemporaryFile().name
     client.evalresp(
         network="IU", station="ANMO", location="00", channel="BHZ", time=dt, output="plot", filename=tempfile
     )
     self.assertEqual(open(tempfile, "rb").read(4)[1:4], "PNG")
     os.remove(tempfile)
     # plot-amp as PNG file
     tempfile = NamedTemporaryFile().name
     client.evalresp(
         network="IU", station="ANMO", location="00", channel="BHZ", time=dt, output="plot-amp", filename=tempfile
     )
     self.assertEqual(open(tempfile, "rb").read(4)[1:4], "PNG")
     os.remove(tempfile)
     # plot-phase as PNG file
     tempfile = NamedTemporaryFile().name
     client.evalresp(
         network="IU", station="ANMO", location="00", channel="BHZ", time=dt, output="plot-phase", filename=tempfile
     )
     self.assertEqual(open(tempfile, "rb").read(4)[1:4], "PNG")
     os.remove(tempfile)
     # fap as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(
         network="IU", station="ANMO", location="00", channel="BHZ", time=dt, output="fap", filename=tempfile
     )
     self.assertEqual(open(tempfile, "rt").readline(), "1.000000E-05  1.202802E+04  1.792007E+02\n")
     os.remove(tempfile)
     # cs as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(
         network="IU", station="ANMO", location="00", channel="BHZ", time=dt, output="cs", filename=tempfile
     )
     self.assertEqual(open(tempfile, "rt").readline(), "1.000000E-05 -1.202685E+04 1.677835E+02\n")
     os.remove(tempfile)
     # fap & def as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(
         network="IU",
         station="ANMO",
         location="00",
         channel="BHZ",
         time=dt,
         output="fap",
         units="def",
         filename=tempfile,
     )
     self.assertEqual(open(tempfile, "rt").readline(), "1.000000E-05  1.202802E+04  1.792007E+02\n")
     os.remove(tempfile)
     # fap & dis as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(
         network="IU",
         station="ANMO",
         location="00",
         channel="BHZ",
         time=dt,
         output="fap",
         units="dis",
         filename=tempfile,
     )
     self.assertEqual(open(tempfile, "rt").readline(), "1.000000E-05  7.557425E-01  2.692007E+02\n")
     os.remove(tempfile)
     # fap & vel as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(
         network="IU",
         station="ANMO",
         location="00",
         channel="BHZ",
         time=dt,
         output="fap",
         units="vel",
         filename=tempfile,
     )
     self.assertEqual(open(tempfile, "rt").readline(), "1.000000E-05  1.202802E+04  1.792007E+02\n")
     os.remove(tempfile)
     # fap & acc as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(
         network="IU",
         station="ANMO",
         location="00",
         channel="BHZ",
         time=dt,
         output="fap",
         units="acc",
         filename=tempfile,
     )
     self.assertEqual(open(tempfile, "rt").readline(), "1.000000E-05  1.914318E+08  8.920073E+01\n")
     os.remove(tempfile)
     # fap as NumPy ndarray
     data = client.evalresp(network="IU", station="ANMO", location="00", channel="BHZ", time=dt, output="fap")
     np.testing.assert_array_equal(data[0], [1.00000000e-05, 1.20280200e04, 1.79200700e02])
     # cs as NumPy ndarray
     data = client.evalresp(network="IU", station="ANMO", location="00", channel="BHZ", time=dt, output="cs")
     np.testing.assert_array_equal(data[0], [1.00000000e-05, -1.20268500e04, 1.67783500e02])
from obspy.iris import Client
import numpy as np
import matplotlib.pyplot as plt
from obspy.core import UTCDateTime
from obspy.signal import cornFreq2Paz, seisSim

client = Client()
# dt = UTCDateTime(2005, 4, 26, 21, 0, 0, 31225480)
dt = UTCDateTime("2006-5-26")
data = client.evalresp("TA", "M03C", "--", "BHE", dt) 
# data[0]
예제 #38
0
        lat = str(station.Lat)
        lon = str(station.Lon)
        ele = str(station.Elevation)

        for j in range(0, len(station.Channel)):
            cha = station.Channel[j].get('chan_code')
            loc = station.Channel[j].get('loc_code')

            sta_req.append([net, sta, loc, cha, lat, lon, ele])

    return sta_req

#################

client = Client()

#t1 = UTCDateTime(raw_input("Please enter the start time: (format: 1998-01-01)\n"))
#t2 = UTCDateTime(raw_input("Please enter the end time: (format: 1998-01-02)\n"))
t1 = UTCDateTime(2012,5,19)
t2=UTCDateTime(2012,5,21)
file_name = raw_input('Enter the name of the stream file!\n')


print t1
print t2

######## 
#st = client.saveWaveform("./G.SSB.00.BHE", "G", "SSB", "*", "BHE", t1, t2)
#st = client.getWaveform("G", "SSB", "10", "*", t1, t2)
#### Permete d avoir toutes les donnes meeme s il y a des trous
print (st[0].stats)
st.write('REF.EHZ.2009:082.mseed', format='MSEED')
#st.write('REF.EHZ.2009:082.wave', format='WAV', framerate=6000)
st.write('REF.EHZ.2009:082.sac', format='SAC')

# This needs basemap and demonstrates the event and catalog classes
from obspy.core.event import *
cat = readEvents(\
"http://www.seismicportal.eu/services/event/search?magMin=8.0")
cat.plot()



# IRIS DMC example
from obspy.iris import Client
from obspy.core import UTCDateTime
client = Client()
t = UTCDateTime("2012-08-05T06:00:00.000")
st = client.getWaveform('IU', 'ANMO', '00', 'BHZ', t, t + 300)
st.plot()

# Earthworm wave server example - connection is refused though
from obspy.earthworm import Client
client = Client("pele.ess.washington.edu", 16017)
response = client.availability("UW", "TUCA", channel="BHZ")
print response
t = response[0][4]
st = client.getWaveform('UW', 'TUCA', '', 'BH*', t + 100, t + 130)
st.plot()

from obspy.iris import Client
from obspy.core import UTCDateTime
from obspy.core.util import NamedTemporaryFile
import matplotlib.pyplot as plt
import numpy as np

# MW 7.1 Darfield earthquake, New Zealand
t1 = UTCDateTime("2010-09-3T16:30:00.000")
t2 = UTCDateTime("2010-09-3T17:00:00.000")

# Fetch waveform from IRIS web service into a ObsPy stream object
client = Client()
st = client.getWaveform('NZ', 'BFZ', '10', 'HHZ', t1, t2)

# Download and save instrument response file into a temporary file
with NamedTemporaryFile() as tf:
    respf = tf.name
    client.saveResponse(respf, 'NZ', 'BFZ', '10', 'HHZ', t1, t2, format="RESP")

    # make a copy to keep our original data
    st_orig = st.copy()

    # define a filter band to prevent amplifying noise during the deconvolution
    pre_filt = (0.005, 0.006, 30.0, 35.0)

    # this can be the date of your raw data or any date for which the
    # SEED RESP-file is valid
    date = t1

    seedresp = {'filename': respf,  # RESP filename
                # when using Trace/Stream.simulate() the "date" parameter can
from obspy.core.util import NamedTemporaryFile
import matplotlib.pyplot as plt
import numpy as np

# MW 7.1 Darfield earthquake, New Zealand
t1 = UTCDateTime("2010-09-3T16:30:00.000")
t2 = UTCDateTime("2010-09-3T17:00:00.000")

# Fetch waveform from IRIS FDSN web service into a ObsPy stream object
fdsn_client = FDSN_Client("IRIS")
st = fdsn_client.get_waveforms('NZ', 'BFZ', '10', 'HHZ', t1, t2)

# Download and save instrument response file into a temporary file
with NamedTemporaryFile() as tf:
    respf = tf.name
    old_iris_client = OldIris_Client()
    # fetch RESP information from "old" IRIS web service, see obspy.fdsn
    # for accessing the new IRIS FDSN web services
    old_iris_client.resp('NZ', 'BFZ', '10', 'HHZ', t1, t2, filename=respf)

    # make a copy to keep our original data
    st_orig = st.copy()

    # define a filter band to prevent amplifying noise during the deconvolution
    pre_filt = (0.005, 0.006, 30.0, 35.0)

    # this can be the date of your raw data or any date for which the
    # SEED RESP-file is valid
    date = t1

    seedresp = {'filename': respf,  # RESP filename
예제 #42
0
from obspy.core import UTCDateTime
from obspy.core.util import NamedTemporaryFile
import matplotlib.pyplot as plt

# MW 7.1 Darfield earthquake, New Zealand
t1 = UTCDateTime("2010-09-3T16:30:00.000")
t2 = UTCDateTime("2010-09-3T17:00:00.000")

# Fetch waveform from IRIS FDSN web service into a ObsPy stream object
fdsn_client = FDSN_Client("IRIS")
st = fdsn_client.get_waveforms('NZ', 'BFZ', '10', 'HHZ', t1, t2)

# Download and save instrument response file into a temporary file
with NamedTemporaryFile() as tf:
    respf = tf.name
    old_iris_client = OldIris_Client()
    # fetch RESP information from "old" IRIS web service, see obspy.fdsn
    # for accessing the new IRIS FDSN web services
    old_iris_client.resp('NZ', 'BFZ', '10', 'HHZ', t1, t2, filename=respf)

    # make a copy to keep our original data
    st_orig = st.copy()

    # define a filter band to prevent amplifying noise during the deconvolution
    pre_filt = (0.005, 0.006, 30.0, 35.0)

    # this can be the date of your raw data or any date for which the
    # SEED RESP-file is valid
    date = t1

    seedresp = {'filename': respf,  # RESP filename
예제 #43
0
def wavesdownloader(args):



    #################################################################
    # ---- Check if internet connection is active 
    #
    # check servers
    args.server=list2servers(args.server,'N')
    # Do not apply if only LOCAL is active
    if (args.server[0] == 1 or args.server[1] == 1 or args.server[3] == 1 or args.server[4] == 1):
       try:
          urllib2.urlopen("http://google.com", timeout=2)
       except urllib2.URLError:
          print "No internet connection"
          sys.exit()
    
     
    #################################################################
    # ----  C. Initialize variables ----                            #
    dataStreamEida = Stream()
    dataStreamIris = Stream()
    dataStreamLoca = Stream()
    dataStreamWbDc = Stream()
    dataStream     = Stream()
    center=args.center.split(' ')
    grradius=args.radius.split(' ')
    kmradius=args.radius.split(' ')
    ev_lat=eval(center[0])
    ev_lon=eval(center[1])
    kmradius[0]=eval(kmradius[0])*100
    kmradius[1]=eval(kmradius[1])*100
    bbox=getBBox(args.supCor,args.infCor) #minlat minlon maxlat maxlon
    aFormat="fseed"
    aFORMAT="FSEED"
    irisClient = IClient()
    ID = "0"

    # check consistency of arguments
    checkConsistency(args)

    # define t1 and t2
    if args.end=="None":
       t1 = UTCDateTime(args.beg)
       t1.getTimeStamp()
       t2 = t1+int(args.len)
       t2= str(t2)[:19] #this must be a string for
       args.end=t2
    else:
       t1 = UTCDateTime(args.beg)
       t2 = UTCDateTime(args.end)
       if t1 >= t2:
          print "Wrong begin and end time entries"
          sys.exit()


    # make outdir if not exist (from myUsefullFuncs)
    if args.outdir != ".":
       mkdir(args.outdir)
    # if filtering require data writing
    if args.wfiltr == "Y":
       mkdir(args.outdir + os.sep + "_f")

    # write  summary
    makeSummary(1,dataStream,args)
    

    # check for local fseed archive
    if args.server[2] == 1 and args.fsfile == "None":
       print "local fseed file non spcified. Check --fsfile option"
       sys.exit(0)
    
    #################################################################
    # ----  D. Download waveforms (fseed and stream)
    # if eida is on
    if args.server[0] == 1:

      # Convert some args for eida syntax 
      args.net=reformatNetStaList(args.net,"eida")
      args.sta=reformatNetStaList(args.sta,"eida")
      args.cha=reformatChaStaList(args.cha,"eida")
    
      # initialize eida requestor 
      user=INGV_requestor(args.usr,args.pas)
    
      # download
      print "\n \nDownloading from eida"
      if args.mode == "circular":
        (downloadPath,ID) = user.run_circular_query( \
                            args.net,args.sta,args.cha, \
                            center[0],center[1],kmradius[0],kmradius[1], \
                            args.beg,args.end,aFORMAT)
      else: # i.e.: rectangular
        (downloadPath,ID) = user.run_rectangular_query( \
                            args.net,args.sta,args.cha, \
                            bbox[0],bbox[1],bbox[2],bbox[3], \
                            args.beg,args.end,aFORMAT)
    
      #set name and location of the fseed 
      archiveFile = downloadPath + os.sep + ID + "_data.tgz"
      (root,seed)=grabIngvEidaArchive(archiveFile,aFormat)
    
      # get stream
      try: 
        dataStreamEida = read(seed)
      except:
        print "no data found or no response from Eida server"

    
    # if iris is on
    if args.server[1] == 1:
    
       # Convert some args for eida syntax 
       args.net=reformatNetStaList(args.net,"iris")
       args.sta=reformatNetStaList(args.sta,"iris")
       args.cha=reformatNetStaList(args.cha,"iris")
    
       # initialize downloader
       print "\n \nDownloading from iris"
       if args.mode == "circular":
          response = irisClient.availability( \
                     network=args.net, station=args.sta, channel=args.cha, \
                     location=args.loc,starttime=t1, endtime=t2, \
                     lat=center[0], lon=center[1], minradius=grradius[0], maxradius=grradius[1])

       else: # i.e.: rectangular
          IrisBox = setIrisRectBox(bbox)
          response = irisClient.availability( \
                     network=args.net, station=args.sta, channel=args.cha, \
                     location=args.loc,starttime=t1, endtime=t2,\
                     minlat=IrisBox[0], minlon=IrisBox[1], maxlat=IrisBox[2], maxlon=IrisBox[3])
    
       # Download
       try:
           dataStreamIris = irisClient.bulkdataselect(response)
           # here save mseed file format
           dataStreamIris.write(args.outdir + os.sep + 'iris.' + ID + '_data.mseed',format='MSEED', encoding='STEIM2')
       except:
           print "IRIS bulkdataselect returns False. No data from IRIS\n\n"
    
    
    #
    # if local is on
    if args.server[2] == 1:
       print "\n \nExtracting from local fseed"
       dataStreamLoca=read(args.fsfile)


    # if ARClink - WEBDC is on
    if args.server[3] == 1:
    
       print "\n \nDownloading from WEBDC via ArcLink"
       # Convert some args for eida syntax 
       args.net=reformatNetStaList(args.net,"iris")
       args.sta=reformatNetStaList(args.sta,"iris")
       args.cha=reformatNetStaList(args.cha,"iris")

       # get station list available
       Inventory=getInventoryViaArcLink(t1,t2,center,grradius,bbox,args)
       # download data
       dataStreamWbDc=getDataViaArcLink(t1,t2,Inventory,args)
       if len(dataStreamWbDc) > 0:
          dataStreamWbDc.write(args.outdir + os.sep + 'webdc.' + ID + '_data.mseed',format='MSEED', encoding='STEIM2')
       
    ############################################################
    # Check if only empty data
    if(len(dataStreamIris) + len(dataStreamEida) + len(dataStreamLoca) + len(dataStreamWbDc)<=0):
      print "\nNo data found! Exit!"
      sys.exit()

    ############################################################
    # ----  E. data to sac (or other formats)
    
    # trim data
    if args.beg!="None" or  args.end!="None":
       Tb = UTCDateTime(args.beg)
       Te = UTCDateTime(args.end)
       dataStreamIris.trim(starttime=Tb, endtime=Te)
       dataStreamEida.trim(starttime=Tb, endtime=Te)
       dataStreamLoca.trim(starttime=Tb, endtime=Te)
       dataStreamWbDc.trim(starttime=Tb, endtime=Te)
    
    # find gaps and remove those traces
    if args.rmgaps=="Y":
       dataStreamIris = removeGaps(dataStreamIris, args.mingap, args.maxgap, verbose="true")
       dataStreamEida = removeGaps(dataStreamEida, args.mingap, args.maxgap, verbose="true")
       dataStreamLoca = removeGaps(dataStreamLoca, args.mingap, args.maxgap, verbose="true")
       dataStreamWbDc = removeGaps(dataStreamWbDc, args.mingap, args.maxgap, verbose="true")
    
    # Remove traces shorter than expected tolerance
    args.reject=eval(args.reject)
    if args.reject <= 100:
       dataStreamIris=removeShortTraces(dataStreamIris,args.reject,Tb,Te)
       dataStreamEida=removeShortTraces(dataStreamEida,args.reject,Tb,Te)
       dataStreamLoca=removeShortTraces(dataStreamLoca,args.reject,Tb,Te)
       dataStreamWbDc=removeShortTraces(dataStreamWbDc,args.reject,Tb,Te)
    
    # Remove mean and trend if required
    if args.demean == "Y":
       dataStreamIris=removeMeanTrend(dataStreamIris)
       dataStreamEida=removeMeanTrend(dataStreamEida)
       dataStreamLoca=removeMeanTrend(dataStreamLoca)
       dataStreamWbDc=removeMeanTrend(dataStreamWbDc)

    # Decimation if required
    if args.deci != "None":
       dataStreamIris=decimateStream(dataStreamIris,args.deci)
       dataStreamEida=decimateStream(dataStreamEida,args.deci)
       dataStreamLoca=decimateStream(dataStreamLoca,args.deci)
       dataStreamWbDc=decimateStream(dataStreamWbDc,args.deci)
    
    
    ############################################################
    # ----  F. metadata, response, and paz files and update header
    #          for arcklink data Paz already exists. Resp file can't 
    #          be downloaded now (12.10.2012)
    
    # eida - very easy
    if args.server[0] == 1:
       #if args.res != "0": # and args.reqFileFormat != "MSEED":

       # first test if fseed file exists
       aa = os.path.isfile(seed)
       if aa == "False":
          print "No data found in fseed file from EIDA"
       else:
          (respFiles, pazFiles) = extractResponse(seed,args.res,args.outdir)
          if args.res == "2" or  args.res == "3":
            dataStreamEida = addPazStats(dataStreamEida,args.outdir,pazFiles)
          dataStreamEida = updateStats(dataStreamEida,args.outdir,"rdseed.stations",ev_lat,ev_lon)
          try:
            shutil.move(seed,args.outdir + os.sep + 'eida.' + ID + '_data.fseed')
          except:
            pass
          try:
            shutil.move(args.outdir + os.sep + 'rdseed.stations',args.outdir + os.sep + 'eida_rdseed.stations')
          except:
            pass


    # local same than eida  - very easy
    if args.server[2] == 1:
       #if args.res != "0":
       (respFiles, pazFiles) = extractResponse(args.fsfile,args.res,args.outdir)
       if args.res == "2" or args.res == "3":
          dataStreamLoca = addPazStats(dataStreamLoca,args.outdir,pazFiles)
       dataStreamLoca = updateStats(dataStreamLoca,args.outdir,"rdseed.stations",ev_lat,ev_lon)
       dataStreamLoca = purgeListStation(dataStreamLoca,args,'d')
       shutil.move(args.outdir + os.sep + 'rdseed.stations',args.outdir + os.sep + 'local_rdseed.stations')   
    
    # iris - longer
    if args.server[1] == 1:
       pazFiles = []
       temp_iris = open(args.outdir + os.sep + "tmp_iris.station","w")   
       # here store lat lon staz locId for later update of iris stream
    
       for i in range(len(dataStreamIris)):
         n = dataStreamIris[i].stats['network']
         s = dataStreamIris[i].stats['station']
         l = dataStreamIris[i].stats['location']
         c = dataStreamIris[i].stats['channel']
         nameresp = "RESP." + n + "." + s + "." + l + "." + c 
         namepaz  = "SAC.PZs." + n + "." + s + "." + c
    
         # get metadata station latitude and longitude
         meta = irisClient.station(network=n, station=s, location=l, channel=c, starttime=t1, endtime=t2)
         (lat,lon) = getXmlTagData(meta)
         # update temp_iris
         temp_iris.write(s + " " + n + " " + lat + " " + lon + "0.0\n")
    
         # write resp files
         if args.res == "1" or args.res == "3":
             irisClient.saveResponse(filename=args.outdir + os.sep + nameresp ,network=n, station=s, location=args.loc, channel=c,\
                 starttime=t1, endtime=t2, format='RESP')
    
         if args.res == "2" or args.res == "3":
             sacpz = irisClient.sacpz(network=n, station=s, location=args.loc, channel=c, starttime=t1, endtime=t2)
             sacpz = sac4sac(sacpz)
             f = open(args.outdir + os.sep + namepaz, 'w')
             f.write(sacpz)
             f.close()     # close close close!!!! porcoIddio che mi dimentico sempre!!!
             pazFiles.append(namepaz)
    
       # lose temp_iris
       temp_iris.close()
       
       # update stream
       dataStreamIris = updateStats(dataStreamIris,args.outdir,"tmp_iris.station",ev_lat,ev_lon)
       if args.res == '2' or  args.res == '3':
          dataStreamIris = addPazStats(dataStreamIris,args.outdir,pazFiles)

    if args.server[3] == 1:

       # Write webdc.stations file and extract PZ file
       #   and update lat lon and event loc into statz for consistency
       dataStreamWbDc=writeWbDcStation(dataStreamWbDc,args)

       if args.res == "2" or args.res == "3":
           PzFileFromStat(dataStreamWbDc,args)
           print "\nWarning, RESP files can not be downloaded from WebDC vie ArcLink\n"
       if args.res == "1":
           print "\nWarning, RESP files can not be downloaded from WebDC vie ArcLink\n"

    #############################################################
    ## ---- G. Join all traces and write mseed files with original data
    for i in range(len(dataStreamEida)):
           dataStream.append(dataStreamEida[i])
    for i in range(len(dataStreamIris)):
           dataStream.append(dataStreamIris[i])
    for i in range(len(dataStreamLoca)):
           dataStream.append(dataStreamLoca[i])
    for i in range(len(dataStreamWbDc)):
           dataStream.append(dataStreamWbDc[i])
    # Stop and exit id no data downloaded
    if len(dataStream) == 0:
       print "\n\nNo data available!\n"
       sys.exit()

    #############################################################
    ## ---- join stations files
    files = []
    if (args.server[0] == 1):
       if(args.outdir + os.sep + "tmp_iris.station"  == True):
         files.append(args.outdir + os.sep + "tmp_iris.station")
    if (args.server[1] == 1):
       if(args.outdir + os.sep + "eida_rdseed.stations" == True):
          files.append(args.outdir + os.sep + "eida_rdseed.stations")
    content = ''
    for f in files:
       content = content + '\n' + open(f).read()
    open(args.outdir + os.sep + 'list.stations','w').write(content)
    
 
    #############################################################
    ## ---- H. Make deconvolution of instrument
    if args.deco == "Y":
       dataStream = removeInstrument(dataStream,args)

    #############################################################
    ## ----  I. Rotate orizontal to GCP
    if args.rot == "Y":
       rotatDataStream = rotateToGCP(dataStream)
       # join horizontal with rotated
       dataStream=join_NERT(dataStream,rotatDataStream)

    #############################################################
    # Write fseed to sac
    if(args.format!="None"):
      RsacFilesEida = fromFseed2sac(dataStream, args.format, args.outdir, args.format)
      for i in range(len(RsacFilesEida)):
         lat=dataStream[i].stats['stla']
         lon=dataStream[i].stats['stlo']
         ev_lat=dataStream[i].stats['evla']
         ev_lon=dataStream[i].stats['evlo']
         ok = updateSacHeader(RsacFilesEida[i],args.outdir,lat,lon,ev_lat,ev_lon)

    #############################################################
    ## ---- L. Write mseed to store semi-processed data.
    #  ---- and other output files
    # if needs to repeat analysis with different parameters on these data just add option 
    # --redo "Y" to your data request line. --server and --res will be automatically disabled
    # and the following mseed file will be loaded. Do not change the name of this file if you want to use 
    # --redo "Y"
    # analysis includes only --filter --cFreq --Sta/Lta --deco 
    #dataStream.write(args.outdir + os.sep + 'downloadedData.mseed', format='MSEED',encoding='FLOAT64')
    # and make summary level 2 (write stats.elements not writable on mseed
    makeSummary(2,dataStream,args)


    ############################################################
    # ----  Write output files
    # kml file station for google
    createKML(dataStream,args.outdir)
     


    ############################################################
    # ---- START DATA ANALYSIS:

    ############################################################
    # ----  M: filter data if required (deafult=none) 
    if args.bandpass == "0" and args.lowpass == "0" and args.highpass == "0":
       pass
    else:
       dataStream=FilterData(dataStream, args.bandpass, args.highpass, args.lowpass)

    
    ############################################################# 
    ## ---- N. Write processed data with filter . This include also rotated filtered data 
    if args.wfiltr != "N":
       outP = args.outdir + os.sep + args.wfiltr
       mkdir(outP)
       FsacFiles = fromFseed2sac(dataStream, args.format, outP, args.format)
    

    #############################################################
    ## ----  O. compute central frequency, Sta/Lta
    # Initialize stats for this section
    dataStream=initStats(dataStream)

    # central frequency
    """
    The period is determined as the period of the maximum value of the
    Fourier amplitude spectrum.
    """
    if args.cfreq=="Y":
       dataStream=cFreqStream(dataStream) 

    # pick using sta/lta
    staLtaStream=Stream()
    if args.slta!="None":
       staLtaStream=StaLta(dataStream,args.slta)
       staLtaStream=trig(staLtaStream,args.slta)
       staLtaStream.write(args.outdir + os.sep + 'picks.mseed', format='MSEED',encoding='FLOAT64')
       dataStream=syncStat(staLtaStream,dataStream)
       # if required to write 
       if args.wcf == "Y":
          SL = fromFseed2sac(staLtaStream,"SAC",args.outdir,'slt')

    # get PGMs:
    if args.pgm=="Y":
       dataStream=get_PGMs(dataStream,args)
       if(args.shake == 'Y'):
         (xmlHeader,shakeLines)=export4ShakeMap(dataStream) 
         writeShake(xmlHeader,shakeLines,args)


    # write summary level 3: data analysis
    makeSummary(3,dataStream,args)


    ############################################################
    # ----  P. Plot 
    args.pltmode = int(args.pltmode)
    if args.pltmode!=0:   
       plotWaves(dataStream,args.pltmode, \
                 kmradius, args.mode,args.pltchan, \
                 args.pltNERT,args.pltazi, \
                 staLtaStream,args.slta, args.outdir, args.rot)
예제 #44
0
#!/usr/bin/python

from obspy.iris import Client
#from obspy.fdsn import Client
client = Client()
result = client.distaz(stalat=1.1, stalon=1.2, evtlat=3.2, evtlon=1.4)
print(result['distance'])
print(result['backazimuth'])
print(result['azimuth'])

예제 #45
0
 def test_evalresp(self):
     """
     Tests evaluating instrument response information.
     """
     client = Client()
     dt = UTCDateTime("2005-01-01")
     # plot as PNG file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU", station="ANMO", location="00",
                     channel="BHZ", time=dt, output='plot',
                     filename=tempfile)
     self.assertEqual(open(tempfile, 'rb').read(4)[1:4], 'PNG')
     os.remove(tempfile)
     # plot-amp as PNG file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU", station="ANMO", location="00",
                     channel="BHZ", time=dt, output='plot-amp',
                     filename=tempfile)
     self.assertEqual(open(tempfile, 'rb').read(4)[1:4], 'PNG')
     os.remove(tempfile)
     # plot-phase as PNG file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU", station="ANMO", location="00",
                     channel="BHZ", time=dt, output='plot-phase',
                     filename=tempfile)
     self.assertEqual(open(tempfile, 'rb').read(4)[1:4], 'PNG')
     os.remove(tempfile)
     # fap as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU", station="ANMO", location="00",
                     channel="BHZ", time=dt, output='fap',
                     filename=tempfile)
     self.assertEqual(open(tempfile, 'rt').readline(),
                      '1.000000E-05  1.202802E+04  1.792007E+02\n')
     os.remove(tempfile)
     # cs as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU", station="ANMO", location="00",
                     channel="BHZ", time=dt, output='cs',
                     filename=tempfile)
     self.assertEqual(open(tempfile, 'rt').readline(),
                      '1.000000E-05 -1.202685E+04 1.677835E+02\n')
     os.remove(tempfile)
     # fap & def as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU", station="ANMO", location="00",
                     channel="BHZ", time=dt, output='fap', units='def',
                     filename=tempfile)
     self.assertEqual(open(tempfile, 'rt').readline(),
                      '1.000000E-05  1.202802E+04  1.792007E+02\n')
     os.remove(tempfile)
     # fap & dis as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU", station="ANMO", location="00",
                     channel="BHZ", time=dt, output='fap', units='dis',
                     filename=tempfile)
     self.assertEqual(open(tempfile, 'rt').readline(),
                      '1.000000E-05  7.557425E-01  2.692007E+02\n')
     os.remove(tempfile)
     # fap & vel as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU", station="ANMO", location="00",
                     channel="BHZ", time=dt, output='fap', units='vel',
                     filename=tempfile)
     self.assertEqual(open(tempfile, 'rt').readline(),
                      '1.000000E-05  1.202802E+04  1.792007E+02\n')
     os.remove(tempfile)
     # fap & acc as ASCII file
     tempfile = NamedTemporaryFile().name
     client.evalresp(network="IU", station="ANMO", location="00",
                     channel="BHZ", time=dt, output='fap', units='acc',
                     filename=tempfile)
     self.assertEqual(open(tempfile, 'rt').readline(),
                      '1.000000E-05  1.914318E+08  8.920073E+01\n')
     os.remove(tempfile)
     # fap as NumPy ndarray
     data = client.evalresp(network="IU", station="ANMO", location="00",
                            channel="BHZ", time=dt, output='fap')
     np.testing.assert_array_equal(data[0],
         [1.00000000e-05, 1.20280200e+04, 1.79200700e+02])
     # cs as NumPy ndarray
     data = client.evalresp(network="IU", station="ANMO", location="00",
                            channel="BHZ", time=dt, output='cs')
     np.testing.assert_array_equal(data[0],
         [1.00000000e-05, -1.20268500e+04, 1.67783500e+02])
예제 #46
0
print st
print(st[0].stats)
st.write('REF.EHZ.2009:082.mseed', format='MSEED')
#st.write('REF.EHZ.2009:082.wave', format='WAV', framerate=6000)
st.write('REF.EHZ.2009:082.sac', format='SAC')

# This needs basemap and demonstrates the event and catalog classes
from obspy.core.event import *
cat = readEvents(\
"http://www.seismicportal.eu/services/event/search?magMin=8.0")
cat.plot()

# IRIS DMC example
from obspy.iris import Client
from obspy.core import UTCDateTime

client = Client()
t = UTCDateTime("2012-08-05T06:00:00.000")
st = client.getWaveform('IU', 'ANMO', '00', 'BHZ', t, t + 300)
st.plot()

# Earthworm wave server example - connection is refused though
from obspy.earthworm import Client

client = Client("pele.ess.washington.edu", 16017)
response = client.availability("UW", "TUCA", channel="BHZ")
print response
t = response[0][4]
st = client.getWaveform('UW', 'TUCA', '', 'BH*', t + 100, t + 130)
st.plot()
예제 #47
0
def main(argv=sys.argv): 

    global eplat, eplon, epdepth, orig
    
    
    GFdir = "/home/roberto/data/GFS/"
    beta = 4.e3 #m/s 
    rho = 3.e3 #kg/m^3 
    mu = rho*beta*beta
    mu =40e9
    
    Lbdm0min = 1e-26*np.array([125.])
    Lbdsmooth = 1e-26*np.array([100.])
    
    #~ Lbdm0min = 1e-26*np.linspace(60.,500,40)
    #~ Lbdsmooth = 1e-26*np.linspace(60.,500,40)#*0.5

    corners = 4.
    fmin = 0.001
    fmax = 0.005
    
    ### Data from Chilean 2010 EQ (Same as W phase inv.) 
    strike = 18.
    dip    = 18.
    rake   = 104. # 109.
    #rake = 45.
    
    rakeA = rake + 45.
    rakeB = rake - 45.
    ####################
    nsx = 21
    nsy = 11
    Min_h = 10.
    flen  = 600. #Fault's longitude [km] along strike
    fwid  = 300. #Fault's longitude [km] along dip
    sflen = flen/float(nsx)
    sfwid = fwid/float(nsy)    
    swp = [1, 0, 2]
    nsf = nsx*nsy
    ###################
    
    t_h     = 10.
    MISFIT = np.array([])
    #RUPVEL = np.arange(1.0, 5.0, 0.05)
    RupVel = 2.1 # Best fit
    #RupVel = 2.25 #From Lay et al.
    
    
    #for RupVel in RUPVEL:
    print "****************************"
    print RupVel
    print "****************************"
    NP = [strike, dip, rake]
    NPA = [strike, dip, rakeA]
    NPB = [strike, dip, rakeB]
    
    M  = np.array(NodalPlanetoMT(NP))  
    MA = np.array(NodalPlanetoMT(NPA)) 
    MB = np.array(NodalPlanetoMT(NPB)) 
    
    Mp = np.sum(M**2)/np.sqrt(2)
    
    
    #############
        #Loading req file and EQparameters
    parameters={}    
    with open(argv[1],'r') as file:
        for line in file:
            line = line.split()
            key = line[0]
            val = line[1:]
            parameters[key] = val 
    #~ cmteplat = float(parameters['eplat'][0])
    #~ cmteplon = float(parameters['eplon'][0])
    #~ cmtepdepth=float(parameters['epdepth'][0])    
    orig = UTCDateTime(parameters['origin_time'][0])
    
    ####Hypocentre from
    ### http://earthquake.usgs.gov/earthquakes/eqinthenews/2010/us2010tfan/    
    cmteplat = -35.91#-35.85#-36.03#-35.83
    cmteplon = -72.73#-72.72#-72.83# -72.67
    cmtepdepth= 35.
    eq_hyp = (cmteplat,cmteplon,cmtepdepth)
      ############

    
    
    grid, sblt = fault_grid('CL-2010',cmteplat,cmteplon,cmtepdepth,0, Min_h,\
                             strike, dip, rake, flen, fwid, nsx, nsy,                            Verbose=False, ffi_io=True, gmt_io=True)
                             
    print ('CL-2010',cmteplat,cmteplon,cmtepdepth,0, Min_h,\
                             strike, dip, rake, flen, fwid, nsx, nsy,\
                            )
    print grid[0][1]
    #sys.exit()
    #############
    #Loading files and setting dirs:
    
    inputfile =  os.path.abspath(argv[1]) 
    if not os.path.exists(inputfile): print inputfile, "does not exist."; exit() 
    workdir = "/".join(inputfile.split("/")[:-1]) 
    basename = inputfile.split("/")[-1][:-4]
    if workdir[-1] != "/": workdir += "/"
    
    try :
        os.mkdir(workdir+"WPinv")
    except OSError:
        pass#print "Directory WPtraces already exists. Skipping"
    
    trfile = open(workdir+"goodtraces.dat")
    trlist = []
    #Loading Good traces files:
    while 1:
        line = trfile.readline().rstrip('\r\n')
        if not line: break        
        trlist.append(line.split()[0])        
    trfile.close()
    #############
    
    # Reading traces:    
    st = read(workdir+"WPtraces/" + basename + ".decov.trim.mseed")  
    #############################################################################
    ######Determining the sf closest to the hypocentre:    
    min_Dist_hyp_subf = flen *fwid
    for subf in range(nsf):
        sblat   = grid[subf][1]
        sblon   = grid[subf][0]
        sbdepth = grid[subf][2]              
        sf_hyp =  (sblat,sblon, sbdepth)        
        Dist_hyp_subf = hypo2dist(eq_hyp,sf_hyp)
        if Dist_hyp_subf < min_Dist_hyp_subf:
            min_Dist_hyp_subf = Dist_hyp_subf
            min_sb_hyp = sf_hyp
            hyp_subf = subf
    print hyp_subf,  min_sb_hyp,  min_Dist_hyp_subf    
    
    
    ####Determining trimming times:
    
    test_tr = read(GFdir + "H003.5/PP/GF.0001.SY.LHZ.SAC")[0]
    t0 = test_tr.stats.starttime
    TrimmingTimes = {}   # Min. Distace from the fault to each station. 
    A =0
    for trid in trlist:     
        tr = st.select(id=trid)[0]
        metafile = workdir + "DATA/" + "META." + tr.id + ".xml"
        META = DU.getMetadataFromXML(metafile)[tr.id]
        stlat = META['latitude']
        stlon = META['longitude'] 
        dist =   locations2degrees(min_sb_hyp[0],min_sb_hyp[1],\
                                   stlat,stlon) 
        parrivaltime = getTravelTimes(dist,min_sb_hyp[2])[0]['time']        
        ta = t0 + parrivaltime
        tb = ta + round(15.*dist) 
        TrimmingTimes[trid] = (ta, tb)
        
    
    ##############################################################################
     
    #####

    DIST = []
    # Ordering the stations in terms of distance
    for trid in trlist: 
        metafile = workdir + "DATA/" + "META." + trid + ".xml"
        META = DU.getMetadataFromXML(metafile)[trid]
        lat = META['latitude']
        lon = META['longitude']
        trdist = locations2degrees(cmteplat,cmteplon,lat,lon) 
        DIST.append(trdist)   

    
    DistIndex = lstargsort(DIST)
    
    if len(argv) == 3:
        trlist = [argv[2]]
        OneStation = True
    else:
        trlist = [trlist[i] for i in DistIndex]
        OneStation = False
        
   ##### 

    client = Client()
    ObservedDisp = np.array([])   
    gridlat = []
    gridlon = []
    griddepth = []
    sbarea = []
    mindist = flen*fwid # min distance hyp-subfault 
    

    ##########Loop for each subfault
    for subf in range(nsf):
        print "**********"
        print subf
        eplat   = grid[subf][1]
        eplon   = grid[subf][0]           
        epdepth = grid[subf][2]
        
        ## Storing the subfault's location centered in the hypcenter 
        gridlat.append(eplat-cmteplat)
        gridlon.append(eplon-cmteplon)
        griddepth.append(epdepth)
        
        
        strike = grid[subf][3] #+ 360.
        dip    = grid[subf][4]
        rake   = grid[subf][5] #     
        NP = [strike, dip, rake]
        
        M = np.array(NodalPlanetoMT(NP))   

        
                #Calculating the time dalay:
            
        sf_hyp = (eplat,eplon, epdepth)        
        Dist_ep_subf = hypo2dist(eq_hyp,sf_hyp)
        t_d = round(Dist_ep_subf/RupVel) #-59.
        print eplat,eplon, epdepth
    
        #t_d  = 0.
        
    
        # Determining depth dir:
        depth = []
        depthdir = []
        for file in os.listdir(GFdir):
            if file[-2:] == ".5":
                depthdir.append(file)
                depth.append(float(file[1:-2]))            
        BestDirIndex = np.argsort(abs(epdepth-np.array(depth)))[0]      
        hdir = GFdir + depthdir[BestDirIndex] + "/"   
        # hdir is the absolute path to the closest deepth. 
        
        
        
        SYN = np.array([])
        SYNA = np.array([])
        SYNB = np.array([])

        
        #Main loop :
        for trid in trlist:  
                       
            tr = st.select(id=trid)[0]    
            metafile = workdir + "DATA/" + "META." + tr.id + ".xml"
            META = DU.getMetadataFromXML(metafile)[tr.id]
            lat = META['latitude']
            lon = META['longitude']    
            trPPsy,  trRRsy, trRTsy,  trTTsy = \
                                   GFSelectZ(lat,lon,hdir) 
            
            tr.stats.delta = trPPsy.stats.delta
            azi =   -np.pi/180.*gps2DistAzimuth(lat,lon,\
                                               eplat,eplon)[2]
            trROT = MTrotationZ(azi, trPPsy,  trRRsy, trRTsy,  trTTsy)        
                        
            
                    #Triangle 
            dt = trROT[0].stats.delta          
            trianglen = 2.*t_h/dt-1.
            window = triang(trianglen)
            window /= np.sum(window)
            #window = np.array([1.])
            
            FirstValid = int(trianglen/2.) + 1
            dist =   locations2degrees(eplat,eplon,lat,lon) 
            parrivaltime = getTravelTimes(dist,epdepth)[0]['time']
            
            t1 = TrimmingTimes[trid][0] - t_d
            t2 = TrimmingTimes[trid][1] - t_d
            
   
            #~ t1 = trROT[0].stats.starttime + parrivaltime- t_d
            #~ t2 = t1+ round(MinDist[tr.id]*15. )
                           
           
            N = len(trROT[0])
            for trR in trROT:
                trR.data *= 10.**-21 ## To get M in Nm                   
                trR.data -= trR.data[0]
                AUX1 = len(trR)
                trR.data = convolve(trR.data,window,mode='valid') 
                AUX2 = len(trR)
                mean = np.mean(np.hstack((trR.data[0]*np.ones(FirstValid),\
                               trR.data[:60./trR.stats.delta*1.-FirstValid+1])))
                #mean = np.mean(trR.data[:60])
                trR.data -= mean      
                trR.data = bp.bandpassfilter(trR.data,len(trR), trR.stats.                                              delta,
                corners , 1 , fmin, fmax)  
                t_l = dt*0.5*(AUX1 - AUX2)                             
                trR.trim(t1-t_l,t2-t_l, pad=True, fill_value=trR.data[0])  #We lost t_h due to the convolution                  
            
            
         
            #~ for trR in trROT:
                #~ trR.data *= 10.**-23 ## To get M in Nm
                #~ trR.data -= trR.data[0]                
                #~ trR.data = convolve(trR.data,window,mode='same')
                #~ # mean = np.mean(np.hstack((trR.data[0]*np.ones(FirstValid),\
                             #~ # trR.data[:60./trR.stats.delta*1.-FirstValid+1])))
                #~ mean = np.mean(trR.data[:60])               
                #~ trR.data -= mean
                #~ trR.data = bp.bandpassfilter(trR.data,len(trR), trR.stats.delta,\
                                             #~ corners ,1 , fmin, fmax)
                #~ trR.trim(t1,t2,pad=True, fill_value=trR.data[0])  

            nmin = min(len(tr.data),len(trROT[0].data))             
            tr.data = tr.data[:nmin]
            for trR in trROT:
                trR.data = trR.data[:nmin]
              
                
             #############            
            trROT = np.array(trROT)  
            syn  =  np.dot(trROT.T,M) 
            synA =  np.dot(trROT.T,MA)
            synB =  np.dot(trROT.T,MB)
            
            SYN = np.append(SYN,syn)  
            SYNA = np.append(SYNA,synA)
            SYNB = np.append(SYNB,synB)
            
            if subf == 0 : ObservedDisp =  np.append(ObservedDisp,tr.data,0) 
            
  
        sbarea.append(grid[subf][6])
   
        print np.shape(A), np.shape(np.array([SYN]))
        if subf == 0: 
            A  = np.array([SYN])
            AA = np.array([SYNA])
            AB = np.array([SYNB])
        else:
            A = np.append(A,np.array([SYN]),0) 
            AA = np.append(AA,np.array([SYNA]),0)
            AB = np.append(AB,np.array([SYNB]),0)
        
    
    
    #Full matrix with the two rake's component
    AC = np.vstack((AA,AB))

#MISFIT = np.array([])
########## Stabilizing the solution:         

#### Moment minimization:
#~ constraintD  = np.zeros(nsf)
#~ ObservedDispcons = np.append(ObservedDisp,constraintD)
#~ for lbd in Lbd:
    #~ constraintF  = lbd*np.eye(nsf,nsf)         
    #~ Acons = np.append(A,constraintF,1)   
    #~ print np.shape(Acons.T), np.shape(ObservedDispcons)
    #~ R = nnls(Acons.T,ObservedDispcons)
    #~ M = R[0]
    #~ #M = np.zeros(nsf)
    #~ #M[::2] = 1
    #~ fit = np.dot(A.T,M)
    #~ misfit = 100.*np.sum(np.abs(fit-ObservedDisp))\
             #~ /np.sum(np.abs(ObservedDisp))
    
    #~ MISFIT = np.append(MISFIT,misfit)
#~ plt.figure()
#~ plt.plot(Lbd,MISFIT)
#~ ###########################################
#~ ### Smoothing:
#~ constraintF_base = SmoothMatrix(nsx,nsy)
#~ constraintD  = np.zeros(np.shape(constraintF_base)[0])
#~ ObservedDispcons = np.append(ObservedDisp,constraintD)
#~ for lbd in Lbd:
    #~ constraintF  = lbd*constraintF_base   
    #~ Acons = np.append(A,constraintF.T,1)   
    #~ #print np.shape(Acons.T), np.shape(ObservedDispcons)
    #~ R = nnls(Acons.T,ObservedDispcons)
    #~ M = R[0]
    #~ fit = np.dot(A.T,M)
    #~ misfit = 100.*np.sum(np.abs(fit-ObservedDisp))\
             #~ /np.sum(np.abs(ObservedDisp))
    #~ print lbd, misfit
    #~ MISFIT = np.append(MISFIT,misfit)
#~ ###########################################    
###########################################
#~ ##### Moment Minimization (including rake projections):
#~ constraintD  = np.zeros(2*nsf)
#~ ObservedDispcons = np.append(ObservedDisp,constraintD)
#~ for lbd in Lbd:
    #~ constraintF  = lbd*np.eye(2*nsf,2*nsf)         
    #~ ACcons = np.append(AC,constraintF,1)   
    #~ print np.shape(ACcons.T), np.shape(ObservedDispcons)
    #~ R = nnls(ACcons.T,ObservedDispcons)
    #~ M = R[0]
    #~ fit = np.dot(AC.T,M)
    #~ misfit = 100.*np.sum(np.abs(fit-ObservedDisp))\
             #~ /np.sum(np.abs(ObservedDisp))        
    #~ MISFIT = np.append(MISFIT,misfit)  
    #~ M = np.sqrt(M[:nsf]**2+M[nsf:]**2)

##############################################
### Smoothing (including rake projections):
#~ constraintF_base = SmoothMatrix(nsx,nsy)
#~ Nbase = np.shape(constraintF_base)[0]
#~ constraintD  = np.zeros(2*Nbase)
#~ constraintF_base_big = np.zeros((2*Nbase, 2*nsf))
#~ constraintF_base_big[:Nbase,:nsf]= constraintF_base
#~ constraintF_base_big[Nbase:,nsf:]= constraintF_base 
#~ ObservedDispcons = np.append(ObservedDisp,constraintD)
#~ for lbd in Lbd:
    #~ constraintF  = lbd*constraintF_base_big   
    #~ ACcons = np.append(AC,constraintF.T,1)   
    #~ #print np.shape(Acons.T), np.shape(ObservedDispcons)
    #~ R = nnls(ACcons.T,ObservedDispcons)
    #~ M = R[0]
    #~ fit = np.dot(AC.T,M)
    #~ misfit = 100.*np.sum(np.abs(fit-ObservedDisp))\
             #~ /np.sum(np.abs(ObservedDisp))
    #~ print lbd, misfit
    #~ MISFIT = np.append(MISFIT,misfit)
#~ M = np.sqrt(M[:nsf]**2+M[nsf:]**2)    
###########################################    
#~ ##### Moment Minimization and  Smoothing
  #~ #### (including rake projections):
    #~ mom0 = []
    #~ constraintF_base = SmoothMatrix(nsx,nsy)
    #~ Nbase = np.shape(constraintF_base)[0]
    #~ constraintDsmoo = np.zeros(2*Nbase)
    #~ constraintDmin  = np.zeros(2*nsf)
    #~ constraintF_base_big = np.zeros((2*Nbase, 2*nsf))
    #~ constraintF_base_big[:Nbase,:nsf]= constraintF_base
    #~ constraintF_base_big[Nbase:,nsf:]= constraintF_base 
    #~ ObservedDispcons = np.concatenate((ObservedDisp,
                                  #~ constraintDmin,
                             #~ constraintDsmoo  ))    
   
    #~ for lbdm0 in Lbdm0min:
        #~ constraintFmin  = lbdm0*np.eye(2*nsf,2*nsf)
        #~ for lbdsm in Lbdsmooth:              
            #~ constraintFsmoo  = lbdsm*constraintF_base_big 
            #~ ACcons = np.hstack((AC, constraintFmin, constraintFsmoo.T))   
            #~ print lbdm0, lbdsm
            #~ R = nnls(ACcons.T,ObservedDispcons)
            #~ M = R[0]
            #~ fit = np.dot(AC.T,M)
            #~ misfit = 100.*np.sum(np.abs(fit-ObservedDisp))\
                     #~ /np.sum(np.abs(ObservedDisp))        
            #~ MISFIT = np.append(MISFIT,misfit) 
            #~ MA = M[:nsf]
            #~ MB = M[nsf:]
            #~ M = np.sqrt(MA**2+MB**2)
            #~ mom0.append(np.sum(M))
    ##############################################

    # Rotation to the rake's conventional angle:  
    #MB, MA = Rot2D(MB,MA,-rakeB)
    print np.shape(M), np.shape(A.T)
    R = nnls(A.T,ObservedDisp)
    M = R[0]
    
    #~ M = np.zeros(nsf)
    #~ M[::2] = 1 
    fit = np.dot(A.T,M)
    MA = M
    MB = M
    
    np.save("RealSol", M)
      
    nm0 = np.size(Lbdm0min) 
    nsmth = np.size(Lbdsmooth)
    #~ plt.figure()
    #~ plt.pcolor(1./Lbdsmooth, 1./Lbdm0min,MISFIT.reshape(nm0,nsmth))
    #~ plt.xlabel(r'$1/ \lambda_{2}$', fontsize = 24)
    #~ plt.ylabel(r'$1/ \lambda_{1}$',fontsize = 24 )
    #~ plt.ylim((1./Lbdm0min).min(),(1./Lbdm0min).max() )
    #~ plt.ylim((1./Lbdsmooth).min(),(1./Lbdsmooth).max() )
    #~ cbar = plt.colorbar()
    #~ cbar.set_label("Misfit %")
    #~ print np.shape(Lbdm0min), np.shape(mom0)
    
    #~ plt.figure()
    #~ CS = plt.contour(1./Lbdsmooth, 1./Lbdm0min,MISFIT.reshape(nm0,nsmth) )
    #~ plt.xlabel(r'$1/ \lambda_{2}$', fontsize = 24)
    #~ plt.ylabel(r'$1/ \lambda_{1}$',fontsize = 24 )
    #~ plt.clabel(CS, inline=1, fontsize=10)
    #~ plt.title('Misfit')
    
    
    
    #~ plt.figure()
    #~ plt.plot(1./Lbdm0min,MISFIT)
    #~ plt.xlabel(r'$1/ \lambda_{2}$', fontsize = 24)
    #~ plt.ylabel("Misfit %")
    #~ plt.figure()
    #~ plt.plot(Lbdm0min,mom0)
    #~ plt.ylabel(r'$M_0\, [Nm]$', fontsize = 24)
    #~ plt.xlabel(r'$\lambda_{M0}$', fontsize = 24)

   
    misfit = 100.*np.sum(np.abs(fit-ObservedDisp))/np.sum(np.abs(ObservedDisp))
    print "Residual: ", 1000.*R[1]
    print misfit
    
    
    #SLIP = M*Mp/mu/(1.e6*np.array(sbarea))
    
    sbarea = sflen*sfwid
    SLIP = M/(mu*1.e6*sbarea)
    SLIP = SLIP.reshape(nsx,nsy).T[::-1]
    moment = M.reshape(nsx,nsy).T[::-1]
    
    plt.figure(figsize = (13,5))
    plt.plot(fit,'b' ,label="Fit")
    plt.plot(ObservedDisp,'r',label="Observed")
    plt.xlabel("Time [s]")
    plt.ylabel("Displacement [m]")
    plt.legend()
    
    
    np.set_printoptions(linewidth=1000,precision=3)
    print "***********"
    print sbarea
    print SLIP
    print np.mean(SLIP)
    print "Moment:"
    print np.sum(M)
 

    ### SLIPS Distribution (as the synthetics) :
    SLIPS = M.reshape(nsx,nsy).T
    SLIPS /=  mu*1.e6*sbarea
    
    
    #~ #########Ploting slip distribution:
    #~ #we are going to reflect the y axis later, so:
    hypsbloc = [hyp_subf / nsy , -(hyp_subf % nsy) - 2]

    #Creating the strike and dip axis:
    StrikeAx= np.linspace(0,flen,nsx+1)
    DipAx= np.linspace(0,fwid,nsy+1)
    DepthAx = DipAx*np.sin(np.pi/180.*dip) + Min_h
    print DepthAx
    hlstrike = StrikeAx[hypsbloc[0]] + sflen*0.5
    #we are going to reflect the axis later, so:
    hldip = DipAx[hypsbloc[1]] + sfwid*0.5 
    hldepth = DepthAx[hypsbloc[1]] + sfwid*0.5*np.sin(np.pi/180.*dip)    
    
    StrikeAx = StrikeAx - hlstrike
    DipAx =     DipAx   - hldip
    
    XX, YY = np.meshgrid(StrikeAx, DepthAx)
    XX, ZZ = np.meshgrid(StrikeAx, DipAx )  

   ######Plot: (Old colormap: "gist_rainbow_r")
    plt.figure(figsize = (13,6))
    ax = host_subplot(111)
    im = ax.pcolor(XX, YY, SLIPS, cmap="jet")    
    ax.set_ylabel('Depth [km]')       
    ax.set_ylim(DepthAx[-1],DepthAx[0])  
    
    # Creating a twin plot 
    ax2 = ax.twinx()
    im2 = ax2.pcolor(XX, ZZ, SLIPS[::-1,:], cmap="jet")    
    ax2.set_ylabel('Distance along the dip [km]')
    ax2.set_xlabel('Distance along the strike [km]')    
    ax2.set_ylim(DipAx[0],DipAx[-1])
    ax2.set_xlim(StrikeAx[0],StrikeAx[-1])       
                         
                         
    ax.axis["bottom"].major_ticklabels.set_visible(False) 
    ax2.axis["bottom"].major_ticklabels.set_visible(False)
    ax2.axis["top"].set_visible(True)
    ax2.axis["top"].label.set_visible(True)
    
    
    divider = make_axes_locatable(ax)
    cax = divider.append_axes("bottom", size="5%", pad=0.1)
    cb = plt.colorbar(im, cax=cax, orientation="horizontal")
    cb.set_label("Slip [m]")             
    ax2.plot([0], [0], '*', ms=225./(nsy+4))
    ax2.set_xticks(ax2.get_xticks()[1:-1])

    
    #~ ### Rake plot:
    plt.figure(figsize = (13,6))
    fig = host_subplot(111)
    XXq, ZZq = np.meshgrid(StrikeAx[:-1]+sflen, DipAx[:-1]+sfwid )
    Q = plt.quiver(XXq,ZZq, MB.reshape(nsx,nsy).T[::-1,:]/(mu*1.e6*sbarea), 
                    MA.reshape(nsx,nsy).T[::-1,:]/(mu*1.e6*sbarea),
                    SLIPS[::-1,:],
                units='xy',scale = 0.5  ,  linewidths=(2,), 
                edgecolors=('k'), headaxislength=5  )
    fig.set_ylim([ZZq.min()-80,ZZq.max()+80])
    fig.set_xlim([XXq.min()-20, XXq.max()+20 ])
    fig.set_ylabel('Distance along dip [km]') 
    fig.set_xlabel('Distance along the strike [km]') 
    
    fig2 = fig.twinx()
    fig2.set_xlabel('Distance along the strike [km]') 
    
    fig.axis["bottom"].major_ticklabels.set_visible(False) 
    fig.axis["bottom"].label.set_visible(False)
    fig2.axis["top"].set_visible(True)
    fig2.axis["top"].label.set_visible(True)
    fig2.axis["right"].major_ticklabels.set_visible(False)

    divider = make_axes_locatable(fig)
    cax = divider.append_axes("bottom", size="5%", pad=0.1)
    cb = plt.colorbar(im, cax=cax, orientation="horizontal")
    cb.set_label("Slip [m]") 
    

    
    
    plt.show()

    
        #############
    #~ print np.shape(MISFIT),  np.shape(RUPVEL)
    #~ plt.figure()
    #~ plt.plot(RUPVEL,MISFIT)
    #~ plt.xlabel("Rupture Velocity [km/s]")
    #~ plt.ylabel("Misfit %")
    #~ plt.show()
     

    print np.shape(MB.reshape(nsx,nsy).T)
    print np.shape(ZZ)
예제 #48
0
from obspy.iris import Client
from obspy import UTCDateTime
client = Client()
starttime = UTCDateTime("1978-07-17")
endtime = UTCDateTime("2013-03-01")
lat="40.76873"
lon="-73.96491"
radius="5.00"
cat = client.getEvents(starttime=starttime, endtime=endtime, minmag=3.0, lat=lat, lon=lon, maxradius=radius)
#cat = client.getEvents(starttime=starttime, endtime=endtime, minmag=6.7);
print(cat);
cat.plot(projection="local");
예제 #49
0
def wavesdownloader(args):

    #################################################################
    # ---- Check if internet connection is active
    #
    # check servers
    args.server = list2servers(args.server, 'N')
    # Do not apply if only LOCAL is active
    if (args.server[0] == 1 or args.server[1] == 1 or args.server[3] == 1
            or args.server[4] == 1):
        try:
            urllib2.urlopen("http://google.com", timeout=2)
        except urllib2.URLError:
            print "No internet connection"
            sys.exit()

    #################################################################
    # ----  C. Initialize variables ----                            #
    dataStreamEida = Stream()
    dataStreamIris = Stream()
    dataStreamLoca = Stream()
    dataStreamWbDc = Stream()
    dataStream = Stream()
    center = args.center.split(' ')
    grradius = args.radius.split(' ')
    kmradius = args.radius.split(' ')
    ev_lat = eval(center[0])
    ev_lon = eval(center[1])
    kmradius[0] = eval(kmradius[0]) * 100
    kmradius[1] = eval(kmradius[1]) * 100
    bbox = getBBox(args.supCor, args.infCor)  #minlat minlon maxlat maxlon
    aFormat = "fseed"
    aFORMAT = "FSEED"
    irisClient = IClient()
    ID = "0"

    # check consistency of arguments
    checkConsistency(args)

    # define t1 and t2
    if args.end == "None":
        t1 = UTCDateTime(args.beg)
        t1.getTimeStamp()
        t2 = t1 + int(args.len)
        t2 = str(t2)[:19]  #this must be a string for
        args.end = t2
    else:
        t1 = UTCDateTime(args.beg)
        t2 = UTCDateTime(args.end)
        if t1 >= t2:
            print "Wrong begin and end time entries"
            sys.exit()

    # make outdir if not exist (from myUsefullFuncs)
    if args.outdir != ".":
        mkdir(args.outdir)
    # if filtering require data writing
    if args.wfiltr == "Y":
        mkdir(args.outdir + os.sep + "_f")

    # write  summary
    makeSummary(1, dataStream, args)

    # check for local fseed archive
    if args.server[2] == 1 and args.fsfile == "None":
        print "local fseed file non spcified. Check --fsfile option"
        sys.exit(0)

    #################################################################
    # ----  D. Download waveforms (fseed and stream)
    # if eida is on
    if args.server[0] == 1:

        # Convert some args for eida syntax
        args.net = reformatNetStaList(args.net, "eida")
        args.sta = reformatNetStaList(args.sta, "eida")
        args.cha = reformatChaStaList(args.cha, "eida")

        # initialize eida requestor
        user = INGV_requestor(args.usr, args.pas)

        # download
        print "\n \nDownloading from eida"
        if args.mode == "circular":
            (downloadPath,ID) = user.run_circular_query( \
                                args.net,args.sta,args.cha, \
                                center[0],center[1],kmradius[0],kmradius[1], \
                                args.beg,args.end,aFORMAT)
        else:  # i.e.: rectangular
            (downloadPath,ID) = user.run_rectangular_query( \
                                args.net,args.sta,args.cha, \
                                bbox[0],bbox[1],bbox[2],bbox[3], \
                                args.beg,args.end,aFORMAT)

        #set name and location of the fseed
        archiveFile = downloadPath + os.sep + ID + "_data.tgz"
        (root, seed) = grabIngvEidaArchive(archiveFile, aFormat)

        # get stream
        try:
            dataStreamEida = read(seed)
        except:
            print "no data found or no response from Eida server"

    # if iris is on
    if args.server[1] == 1:

        # Convert some args for eida syntax
        args.net = reformatNetStaList(args.net, "iris")
        args.sta = reformatNetStaList(args.sta, "iris")
        args.cha = reformatNetStaList(args.cha, "iris")

        # initialize downloader
        print "\n \nDownloading from iris"
        if args.mode == "circular":
            response = irisClient.availability( \
                       network=args.net, station=args.sta, channel=args.cha, \
                       location=args.loc,starttime=t1, endtime=t2, \
                       lat=center[0], lon=center[1], minradius=grradius[0], maxradius=grradius[1])

        else:  # i.e.: rectangular
            IrisBox = setIrisRectBox(bbox)
            response = irisClient.availability( \
                       network=args.net, station=args.sta, channel=args.cha, \
                       location=args.loc,starttime=t1, endtime=t2,\
                       minlat=IrisBox[0], minlon=IrisBox[1], maxlat=IrisBox[2], maxlon=IrisBox[3])

        # Download
        try:
            dataStreamIris = irisClient.bulkdataselect(response)
            # here save mseed file format
            dataStreamIris.write(args.outdir + os.sep + 'iris.' + ID +
                                 '_data.mseed',
                                 format='MSEED',
                                 encoding='STEIM2')
        except:
            print "IRIS bulkdataselect returns False. No data from IRIS\n\n"

    #
    # if local is on
    if args.server[2] == 1:
        print "\n \nExtracting from local fseed"
        dataStreamLoca = read(args.fsfile)

    # if ARClink - WEBDC is on
    if args.server[3] == 1:

        print "\n \nDownloading from WEBDC via ArcLink"
        # Convert some args for eida syntax
        args.net = reformatNetStaList(args.net, "iris")
        args.sta = reformatNetStaList(args.sta, "iris")
        args.cha = reformatNetStaList(args.cha, "iris")

        # get station list available
        Inventory = getInventoryViaArcLink(t1, t2, center, grradius, bbox,
                                           args)
        # download data
        dataStreamWbDc = getDataViaArcLink(t1, t2, Inventory, args)
        if len(dataStreamWbDc) > 0:
            dataStreamWbDc.write(args.outdir + os.sep + 'webdc.' + ID +
                                 '_data.mseed',
                                 format='MSEED',
                                 encoding='STEIM2')

    ############################################################
    # Check if only empty data
    if (len(dataStreamIris) + len(dataStreamEida) + len(dataStreamLoca) +
            len(dataStreamWbDc) <= 0):
        print "\nNo data found! Exit!"
        sys.exit()

    ############################################################
    # ----  E. data to sac (or other formats)

    # trim data
    if args.beg != "None" or args.end != "None":
        Tb = UTCDateTime(args.beg)
        Te = UTCDateTime(args.end)
        dataStreamIris.trim(starttime=Tb, endtime=Te)
        dataStreamEida.trim(starttime=Tb, endtime=Te)
        dataStreamLoca.trim(starttime=Tb, endtime=Te)
        dataStreamWbDc.trim(starttime=Tb, endtime=Te)

    # find gaps and remove those traces
    if args.rmgaps == "Y":
        dataStreamIris = removeGaps(dataStreamIris,
                                    args.mingap,
                                    args.maxgap,
                                    verbose="true")
        dataStreamEida = removeGaps(dataStreamEida,
                                    args.mingap,
                                    args.maxgap,
                                    verbose="true")
        dataStreamLoca = removeGaps(dataStreamLoca,
                                    args.mingap,
                                    args.maxgap,
                                    verbose="true")
        dataStreamWbDc = removeGaps(dataStreamWbDc,
                                    args.mingap,
                                    args.maxgap,
                                    verbose="true")

    # Remove traces shorter than expected tolerance
    args.reject = eval(args.reject)
    if args.reject <= 100:
        dataStreamIris = removeShortTraces(dataStreamIris, args.reject, Tb, Te)
        dataStreamEida = removeShortTraces(dataStreamEida, args.reject, Tb, Te)
        dataStreamLoca = removeShortTraces(dataStreamLoca, args.reject, Tb, Te)
        dataStreamWbDc = removeShortTraces(dataStreamWbDc, args.reject, Tb, Te)

    # Remove mean and trend if required
    if args.demean == "Y":
        dataStreamIris = removeMeanTrend(dataStreamIris)
        dataStreamEida = removeMeanTrend(dataStreamEida)
        dataStreamLoca = removeMeanTrend(dataStreamLoca)
        dataStreamWbDc = removeMeanTrend(dataStreamWbDc)

    # Decimation if required
    if args.deci != "None":
        dataStreamIris = decimateStream(dataStreamIris, args.deci)
        dataStreamEida = decimateStream(dataStreamEida, args.deci)
        dataStreamLoca = decimateStream(dataStreamLoca, args.deci)
        dataStreamWbDc = decimateStream(dataStreamWbDc, args.deci)

    ############################################################
    # ----  F. metadata, response, and paz files and update header
    #          for arcklink data Paz already exists. Resp file can't
    #          be downloaded now (12.10.2012)

    # eida - very easy
    if args.server[0] == 1:
        #if args.res != "0": # and args.reqFileFormat != "MSEED":

        # first test if fseed file exists
        aa = os.path.isfile(seed)
        if aa == "False":
            print "No data found in fseed file from EIDA"
        else:
            (respFiles, pazFiles) = extractResponse(seed, args.res,
                                                    args.outdir)
            if args.res == "2" or args.res == "3":
                dataStreamEida = addPazStats(dataStreamEida, args.outdir,
                                             pazFiles)
            dataStreamEida = updateStats(dataStreamEida, args.outdir,
                                         "rdseed.stations", ev_lat, ev_lon)
            try:
                shutil.move(
                    seed, args.outdir + os.sep + 'eida.' + ID + '_data.fseed')
            except:
                pass
            try:
                shutil.move(args.outdir + os.sep + 'rdseed.stations',
                            args.outdir + os.sep + 'eida_rdseed.stations')
            except:
                pass

    # local same than eida  - very easy
    if args.server[2] == 1:
        #if args.res != "0":
        (respFiles, pazFiles) = extractResponse(args.fsfile, args.res,
                                                args.outdir)
        if args.res == "2" or args.res == "3":
            dataStreamLoca = addPazStats(dataStreamLoca, args.outdir, pazFiles)
        dataStreamLoca = updateStats(dataStreamLoca, args.outdir,
                                     "rdseed.stations", ev_lat, ev_lon)
        dataStreamLoca = purgeListStation(dataStreamLoca, args, 'd')
        shutil.move(args.outdir + os.sep + 'rdseed.stations',
                    args.outdir + os.sep + 'local_rdseed.stations')

    # iris - longer
    if args.server[1] == 1:
        pazFiles = []
        temp_iris = open(args.outdir + os.sep + "tmp_iris.station", "w")
        # here store lat lon staz locId for later update of iris stream

        for i in range(len(dataStreamIris)):
            n = dataStreamIris[i].stats['network']
            s = dataStreamIris[i].stats['station']
            l = dataStreamIris[i].stats['location']
            c = dataStreamIris[i].stats['channel']
            nameresp = "RESP." + n + "." + s + "." + l + "." + c
            namepaz = "SAC.PZs." + n + "." + s + "." + c

            # get metadata station latitude and longitude
            meta = irisClient.station(network=n,
                                      station=s,
                                      location=l,
                                      channel=c,
                                      starttime=t1,
                                      endtime=t2)
            (lat, lon) = getXmlTagData(meta)
            # update temp_iris
            temp_iris.write(s + " " + n + " " + lat + " " + lon + "0.0\n")

            # write resp files
            if args.res == "1" or args.res == "3":
                irisClient.saveResponse(filename=args.outdir + os.sep + nameresp ,network=n, station=s, location=args.loc, channel=c,\
                    starttime=t1, endtime=t2, format='RESP')

            if args.res == "2" or args.res == "3":
                sacpz = irisClient.sacpz(network=n,
                                         station=s,
                                         location=args.loc,
                                         channel=c,
                                         starttime=t1,
                                         endtime=t2)
                sacpz = sac4sac(sacpz)
                f = open(args.outdir + os.sep + namepaz, 'w')
                f.write(sacpz)
                f.close(
                )  # close close close!!!! porcoIddio che mi dimentico sempre!!!
                pazFiles.append(namepaz)

        # lose temp_iris
        temp_iris.close()

        # update stream
        dataStreamIris = updateStats(dataStreamIris, args.outdir,
                                     "tmp_iris.station", ev_lat, ev_lon)
        if args.res == '2' or args.res == '3':
            dataStreamIris = addPazStats(dataStreamIris, args.outdir, pazFiles)

    if args.server[3] == 1:

        # Write webdc.stations file and extract PZ file
        #   and update lat lon and event loc into statz for consistency
        dataStreamWbDc = writeWbDcStation(dataStreamWbDc, args)

        if args.res == "2" or args.res == "3":
            PzFileFromStat(dataStreamWbDc, args)
            print "\nWarning, RESP files can not be downloaded from WebDC vie ArcLink\n"
        if args.res == "1":
            print "\nWarning, RESP files can not be downloaded from WebDC vie ArcLink\n"

    #############################################################
    ## ---- G. Join all traces and write mseed files with original data
    for i in range(len(dataStreamEida)):
        dataStream.append(dataStreamEida[i])
    for i in range(len(dataStreamIris)):
        dataStream.append(dataStreamIris[i])
    for i in range(len(dataStreamLoca)):
        dataStream.append(dataStreamLoca[i])
    for i in range(len(dataStreamWbDc)):
        dataStream.append(dataStreamWbDc[i])
    # Stop and exit id no data downloaded
    if len(dataStream) == 0:
        print "\n\nNo data available!\n"
        sys.exit()

    #############################################################
    ## ---- join stations files
    files = []
    if (args.server[0] == 1):
        if (args.outdir + os.sep + "tmp_iris.station" == True):
            files.append(args.outdir + os.sep + "tmp_iris.station")
    if (args.server[1] == 1):
        if (args.outdir + os.sep + "eida_rdseed.stations" == True):
            files.append(args.outdir + os.sep + "eida_rdseed.stations")
    content = ''
    for f in files:
        content = content + '\n' + open(f).read()
    open(args.outdir + os.sep + 'list.stations', 'w').write(content)

    #############################################################
    ## ---- H. Make deconvolution of instrument
    if args.deco == "Y":
        dataStream = removeInstrument(dataStream, args)

    #############################################################
    ## ----  I. Rotate orizontal to GCP
    if args.rot == "Y":
        rotatDataStream = rotateToGCP(dataStream)
        # join horizontal with rotated
        dataStream = join_NERT(dataStream, rotatDataStream)

    #############################################################
    # Write fseed to sac
    if (args.format != "None"):
        RsacFilesEida = fromFseed2sac(dataStream, args.format, args.outdir,
                                      args.format)
        for i in range(len(RsacFilesEida)):
            lat = dataStream[i].stats['stla']
            lon = dataStream[i].stats['stlo']
            ev_lat = dataStream[i].stats['evla']
            ev_lon = dataStream[i].stats['evlo']
            ok = updateSacHeader(RsacFilesEida[i], args.outdir, lat, lon,
                                 ev_lat, ev_lon)

    #############################################################
    ## ---- L. Write mseed to store semi-processed data.
    #  ---- and other output files
    # if needs to repeat analysis with different parameters on these data just add option
    # --redo "Y" to your data request line. --server and --res will be automatically disabled
    # and the following mseed file will be loaded. Do not change the name of this file if you want to use
    # --redo "Y"
    # analysis includes only --filter --cFreq --Sta/Lta --deco
    #dataStream.write(args.outdir + os.sep + 'downloadedData.mseed', format='MSEED',encoding='FLOAT64')
    # and make summary level 2 (write stats.elements not writable on mseed
    makeSummary(2, dataStream, args)

    ############################################################
    # ----  Write output files
    # kml file station for google
    createKML(dataStream, args.outdir)

    ############################################################
    # ---- START DATA ANALYSIS:

    ############################################################
    # ----  M: filter data if required (deafult=none)
    if args.bandpass == "0" and args.lowpass == "0" and args.highpass == "0":
        pass
    else:
        dataStream = FilterData(dataStream, args.bandpass, args.highpass,
                                args.lowpass)

    #############################################################
    ## ---- N. Write processed data with filter . This include also rotated filtered data
    if args.wfiltr != "N":
        outP = args.outdir + os.sep + args.wfiltr
        mkdir(outP)
        FsacFiles = fromFseed2sac(dataStream, args.format, outP, args.format)

    #############################################################
    ## ----  O. compute central frequency, Sta/Lta
    # Initialize stats for this section
    dataStream = initStats(dataStream)

    # central frequency
    """
    The period is determined as the period of the maximum value of the
    Fourier amplitude spectrum.
    """
    if args.cfreq == "Y":
        dataStream = cFreqStream(dataStream)

    # pick using sta/lta
    staLtaStream = Stream()
    if args.slta != "None":
        staLtaStream = StaLta(dataStream, args.slta)
        staLtaStream = trig(staLtaStream, args.slta)
        staLtaStream.write(args.outdir + os.sep + 'picks.mseed',
                           format='MSEED',
                           encoding='FLOAT64')
        dataStream = syncStat(staLtaStream, dataStream)
        # if required to write
        if args.wcf == "Y":
            SL = fromFseed2sac(staLtaStream, "SAC", args.outdir, 'slt')

    # get PGMs:
    if args.pgm == "Y":
        dataStream = get_PGMs(dataStream, args)
        if (args.shake == 'Y'):
            (xmlHeader, shakeLines) = export4ShakeMap(dataStream)
            writeShake(xmlHeader, shakeLines, args)

    # write summary level 3: data analysis
    makeSummary(3, dataStream, args)

    ############################################################
    # ----  P. Plot
    args.pltmode = int(args.pltmode)
    if args.pltmode != 0:
        plotWaves(dataStream,args.pltmode, \
                  kmradius, args.mode,args.pltchan, \
                  args.pltNERT,args.pltazi, \
                  staLtaStream,args.slta, args.outdir, args.rot)