Python TTYProgressBar.poke方法代码示例

# 需要导入模块: from sunpy.util.progressbar import TTYProgressBar [as 别名]
# 或者: from sunpy.util.progressbar.TTYProgressBar import poke [as 别名]
    def translate_and_query(self, hek_results, limit=None,
                            full_query=False, use_progress_bar=True):
        """
        Translates HEK results, makes a VSO query, then returns the results.

        Takes the results from a HEK query, translates them, then makes a VSO
        query, returning the results in a list organized by their
        corresponding HEK query.

        Parameters
        ----------
        hek_results: sunpy.net.hek.hek.Response or list of sunpy.-.-.-.Response
            The results from a HEK query in the form of a list.
        limit: int
            An approximate limit to the desired number of VSO results.
        full_query: Boolean
            A simple flag that determines if the method is being
            called from the full_query() method.
        use_progress_bar: Boolean
            A flag to turn off the progress bar, defaults to "on"

        Examples
        --------
        >>> from sunpy.net import hek, hek2vso
        >>> h = hek.HEKClient()
        >>> tstart = '2011/08/09 07:23:56'
        >>> t_end = '2011/08/09 12:40:29'
        >>> t_event = 'FL'
        >>> q = h.query(hek.attrs.Time(tstart, tend), hek.attts.EventType(event_type))
        >>> h2v = hek2vso.H2VClient()
        >>> res = h2v.translate_and_query(q)
        """
        if full_query is False:
            self.quick_clean()
            self.hek_results = hek_results
        vso_query = translate_results_to_query(hek_results)
        result_size = len(vso_query)
        for query in vso_query:
            if use_progress_bar:
                pbar = TTYProgressBar('Querying VSO webservice', result_size)
            temp = self.vso_client.query(*query)
            self.vso_results.append(temp)
            self.num_of_records += len(temp)
            if limit is not None:
                if self.num_of_records >= limit:
                    break
            pbar.poke()
        if use_progress_bar:
            pbar.finish()
        
        return self.vso_results

# 需要导入模块: from sunpy.util.progressbar import TTYProgressBar [as 别名]
# 或者: from sunpy.util.progressbar.TTYProgressBar import poke [as 别名]
    def translate_and_query(self, hek_results, limit=None, progress=False):
        """
        Translates HEK results, makes a VSO query, then returns the results.

        Takes the results from a HEK query, translates them, then makes a VSO
        query, returning the results in a list organized by their
        corresponding HEK query.

        Parameters
        ----------
        hek_results: sunpy.net.hek.hek.Response or list of Responses
            The results from a HEK query in the form of a list.
        limit: int
            An approximate limit to the desired number of VSO results.
        progress: Boolean
            A flag to turn off the progress bar, defaults to "off"

        Examples
        --------
        >>> from sunpy.net import hek, hek2vso
        >>> h = hek.HEKClient()
        >>> tstart = '2011/08/09 07:23:56'
        >>> tend = '2011/08/09 12:40:29'
        >>> event_type = 'FL'
        >>> q = h.query(hek.attrs.Time(tstart, tend), hek.attts.EventType(event_type))
        >>> h2v = hek2vso.H2VClient()
        >>> res = h2v.translate_and_query(q)
        """
        vso_query = translate_results_to_query(hek_results)
        result_size = len(vso_query)
        if progress:
            sys.stdout.write('\rQuerying VSO webservice')
            sys.stdout.flush()
            pbar = TTYProgressBar(result_size)

        for query in vso_query:
            temp = self.vso_client.query(*query)
            self.vso_results.append(temp)
            self.num_of_records += len(temp)
            if limit is not None:
                if self.num_of_records >= limit:
                    break
            if progress:
                pbar.poke()

        if progress:
            pbar.finish()

        return self.vso_results

# 需要导入模块: from sunpy.util.progressbar import TTYProgressBar [as 别名]
# 或者: from sunpy.util.progressbar.TTYProgressBar import poke [as 别名]
    def _gaussian_fits(self, line_guess=None, *extra_lines, **kwargs):
        """
        Returns an array of fit objects from which parameters can be extracted
        corresponding to the line guesses provided.

        Parameters
        ----------
        line_guess and extra_lines: 3-tuples of floats
            There must be at least one guess, in the format (intensity,
            position, stddev). The closer these guesses are to the true values
            the better the fit will be. If left to None, each of the individual
            spectra will come up with their own guesses. This only works for
            cubes with clean, single-line spectra.
        recalc=False: boolean
            If True, the gaussian fits will be recalculated, even if there's an
            existing fit for the given wavelengths already in the memo. This
            keyword should be set to True if changing the amplitude or width of
            the fit.
        **kwargs: dict
            Extra keyword arguments are ultimately passed on to the astropy
            fitter.
        """
        recalc = kwargs.pop('recalc', False)
        if line_guess is None:
            key = 'auto'
        else:
            key = tuple([guess[1] for guess in (line_guess,) + extra_lines])
        if recalc or key not in self._memo:
            gaussian_array = np.empty(self.spectra.shape, dtype=object)
            bar = PB(self.spectra.shape[0] * self.spectra.shape[1])
            drawbar = kwargs.pop('progress_bar', False)
            for i in range(self.spectra.shape[0]):
                for j in range(self.spectra.shape[1]):
                    fit = self.spectra[i, j].gaussian_fit(line_guess,
                                                          *extra_lines,
                                                          **kwargs)
                    gaussian_array[i, j] = fit
                    if drawbar:
                        bar.poke()
            self._memo[key] = gaussian_array
            bar.finish()
            return gaussian_array
        else:
            return self._memo[key]