Skip to content
Snippets Groups Projects
Select Git revision
  • master default protected
  • utilpy-windows-issue-hotfix
  • pcain-iris
  • Modify-GUT-HOT
  • bgeels-master-patch-68484
  • bgeels-master-patch-39336
  • production protected
  • 1.14.2 protected
  • 1.14.1 protected
  • 1.14.0 protected
  • 1.13.5 protected
  • 1.13.4 protected
  • 1.13.3 protected
  • 1.13.2 protected
  • 1.13.1 protected
  • 1.13.0 protected
  • 1.12.2 protected
  • 1.12.1 protected
  • 1.12.0 protected
  • 1.11.0 protected
  • 1.10.0 protected
  • 1.9.6 protected
  • 1.9.5 protected
  • 1.9.4 protected
  • 1.9.3 protected
  • 1.9.2 protected
  • 1.9.1 protected
27 results
Blame Permalink
Controller.py 2.42 KiB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72 






"""Controller class for geomag algorithms"""

import TimeseriesUtilities


class Controller(object):
    """Controller for geomag algorithms.

    Parameters
    ----------
    inputFactory: TimeseriesFactory
        the factory that will read in timeseries data
    outputFactory: TimeseriesFactory
        the factory that will output the timeseries data
    algorithm: the algorithm(s) that will take procees the timeseries data
    """

    def __init__(self, inputFactory, outputFactory, algorithm, update=False):
        self._inputFactory = inputFactory
        self._algorithm = algorithm
        self._outputFactory = outputFactory
        self._update = update

    def run(self, starttime, endtime):
        """run an algorithm as setup up by the main script.

        Parameters
        ----------
        starttime : UTCDateTime
            time of first sample to be worked on.
        endtime : UTCDateTime
            time of last sample to be worked on.
        """
        if (self._update):
            self.run_as_update(starttime, endtime)
        else:
            self.run_as_timeseries(starttime, endtime)

    def run_as_update(self, starttime, endtime):
        input_channels = self._algorithm.get_input_channels()
        output_channels = self._algorithm.get_output_channels()

        timeseries_in = self._inputFactory.get_timeseries(starttime,
                endtime, channels=input_channels)
        timeseries_out = self._inputFactory.get_timeseries(starttime,
                endtime, channels=output_channels)

        #TODO get input gaps

        output_gaps = TimeseriesUtilities.get_timeseries_gaps(
            timeseries_out, output_channels, starttime, endtime)

        output_merged_gaps = TimeseriesUtilities.get_merged_gaps(output_gaps,
                output_channels)
        # TODO compare gaps.
        #   if there is new data, run algorithm over entire time.
        #   save any new data.

        #TODO iterate is starttime of gaps is starttime and new data found


    def run_as_timeseries(self, starttime, endtime):
        input_channels = self._algorithm.get_input_channels()

        timeseries = self._inputFactory.get_timeseries(starttime, endtime,
            channels=input_channels)

        processed = self._algorithm.process(timeseries)
        output_channels = self._algorithm.get_output_channels()
        self._outputFactory.put_timeseries(timeseries=processed,
                channels=output_channels)