예제 #1
0
    def chopper_selected_traces(self, fallback=False, marker_selector=None, *args, **kwargs ):
        '''Iterate over selected traces.
        
        This is a shortcut to get all trace data contained in the selected 
        markers in the running snuffler. For each selected marker, 
        :py:meth:`pyrocko.pile.Pile.chopper` is called with the arguments *tmin*, *tmax*, and 
        *trace_selector* set to values according to the marker. Additional
        arguments to the chopper are handed over from *\*args* and *\*\*kwargs*.
        
        :param fallback: if ``True``, if no selection has been marked, use the content
               currently visible in the viewer.
        :param marker_selector: if not ``None`` a callback to filter markers.
               
        '''
       
        try:
            viewer = self.get_viewer()
            markers = viewer.selected_markers()
            if marker_selector is not None:
                markers = [  marker for marker in markers if marker_selector(marker) ] 
            pile = self.get_pile()

            if markers:
                
                for marker in markers:
                    if not marker.nslc_ids:
                        trace_selector = None
                    else:
                        trace_selector = lambda tr: marker.match_nslc(tr.nslc_id)

                    for traces in pile.chopper(
                            tmin = marker.tmin,
                            tmax = marker.tmax,
                            trace_selector = trace_selector,
                            *args,
                            **kwargs):
                        
                        yield traces
                        
            elif fallback:
                if 'trace_selector' not in kwargs:
                    kwargs['trace_selector'] = viewer.trace_selector
                
                tmin, tmax = viewer.get_time_range()
                for traces in pile.chopper(
                        tmin = tmin,
                        tmax = tmax,
                        *args,
                        **kwargs):
                    
                    yield traces
            else:
                raise NoTracesSelected()
        
        except NoViewerSet:

            pile = self.get_pile()
            for traces in  pile.chopper(*args, **kwargs):
                yield traces
예제 #2
0
파일: iris_pull.py 프로젝트: simondaout/bat
    def iter_prepare(self, pile, event, stations):
        stations_by_nsl = dict((get_nsl(s), s) for s in stations)
        nsl_avail = pile.gather_keys(gather=get_nsl)
        for nsl in nsl_avail:
            station = stations_by_nsl[nsl]
            tmin = self.get_tmin(event, station)
            tmax = self.get_tmax(event, station)
            tpad = self.get_tpad(event, station)

            for traces in pile.chopper(
                    tmin=tmin,
                    tmax=tmax,
                    tpad=tpad,
                    trace_selector=lambda tr: get_nsl(tr) == nsl,
                    want_incomplete=False):

                if not traces:
                    continue

                station = stations_by_nsl[nsl]
                prepared_traces = self.prepare_station_traces(
                    traces, event, station)

                yield prepared_traces
예제 #3
0
파일: stalta.py 프로젝트: megies/pyrocko
    def call(self):
        '''Main work routine of the snuffling.'''
        
        self.cleanup()
        
        swin, ratio = self.swin, self.ratio
        lwin = swin * ratio
        tpad = lwin/2.
        
        pile = self.get_pile()
        tmin, tmax = pile.get_tmin() + tpad, pile.get_tmax() - tpad
        
        if not self.apply_to_all:
            vtmin, vtmax = self.get_viewer().get_time_range()
            tmin = max(vtmin, tmin)
            tmax = min(vtmax, tmax)

        tinc = min(lwin * self.block_factor, tmax-tmin)
        
        show_level_traces = self.show_level_traces
        
       # if show_level_traces and tmax-tmin > lwin * 150:
       #     self.error('Processing time window is longer than 50 x LTA window. Turning off display of level traces.')
       #     show_level_traces = False
        
        markers = []
        for traces in pile.chopper(tmin=tmin, tmax=tmax, tinc=tinc, tpad=tpad, want_incomplete=False):
            sumtrace = None
            isum = 0
            for trace in traces:
                if self.lowpass is not None:
                    trace.lowpass(4, self.lowpass, nyquist_exception=True)

                if self.highpass is not None:
                    trace.highpass(4, self.highpass, nyquist_exception=True)
                
                trace.sta_lta_centered(swin, lwin, scalingmethod=scalingmethod_map[self.scalingmethod])
                trace.chop(trace.wmin, min(trace.wmax,tmax))
                trace.set_codes(location='cg')
                trace.meta = { 'tabu': True }
                
                #print trace.ydata.max()
                if sumtrace is None:
                    sumtrace = trace.copy()
                    sumtrace.set_codes(network='', station='SUM', location='cg', channel='')
                else:
                    sumtrace.add(trace)
                isum += 1
    
            if show_level_traces:
                self.add_traces(traces)
    
            if sumtrace is not None:
                tpeaks, apeaks = sumtrace.peaks(self.level*isum, swin)
    
                for t, a in zip(tpeaks, apeaks):
                    mark = Marker([  ], t, t)
                    print mark, a #'%s.%s.%s.%s' % ('', trace.station, '', trace.channel)
                    markers.append(mark)
                
                if show_level_traces:
                    self.add_trace(sumtrace)
                    
        self.add_markers(markers)