Пример #1
0
class DriftChamber:
    """
    Objects of this class represent an instance of the program.
    """
    def __init__(self, p_args=None):
        """
        Constructor.
        
        :param p_args     The command line arguments. Only use this for testing purposes and let it default to sys.argv otherwise.
        """

        # The configuration class also parses the command line arguments.
        # Hence, in order for the argument parser to be the first to stream to the standard output. No output should be streamed before the next line is executed.
        self._args = p_args
        self._init_commandline_argument_parser()
        self._init_confiuration()
        self._init_data_store()
        self._init_run_engine()

    def _init_commandline_argument_parser(self):
        self._commandLineKeyOfConfigurationFile = "config"
        self._commandLineFlagOfConfigurationFile = "--" + self._commandLineKeyOfConfigurationFile
        self._argumentParser = ArgumentParser(
            'Drift Chamber Project of Team A')
        self._argumentParser.add_argument(
            self._commandLineFlagOfConfigurationFile,
            type=str,
            help=
            'Path to a file that holds the configuration for this program run.'
        )
        self._parsedArgs = vars(self._argumentParser.parse_args(self._args))

    def _init_confiuration(self):
        self._configuration = Configuration(
            self._parsedArgs[self._commandLineKeyOfConfigurationFile],
            driftChamberConfig_generalSpecification)

    def _init_data_store(self):
        self._dataStore = DataStore()
        self._dataStore.put('nEvent', self._configuration['General_nEvent'],
                            ObjectLifetime.Application)
        self._dataStore.put('configuration', self._configuration,
                            ObjectLifetime.Application)
        self._dataStore.put('driftChamber', self, ObjectLifetime.Application)

    def _init_run_engine(self):
        self._run_engine = RunEngineFactory(
            self._configuration["Modules_moduleSequence"],
            self._dataStore).get_run_engine()

    def start_simulation(self):
        logging.info("'Drift Chamber Simulation' started.")
        self._run_engine.log_configuration()
        self._run_engine.run()
        logging.info("'Drift Chamber Simulation' done.")
class DriftChamber:
    """
    Objects of this class represent an instance of the program.
    """
    def __init__(self, p_args = None):
        """
        Constructor.
        
        :param p_args     The command line arguments. Only use this for testing purposes and let it default to sys.argv otherwise.
        """
        
        # The configuration class also parses the command line arguments.
        # Hence, in order for the argument parser to be the first to stream to the standard output. No output should be streamed before the next line is executed.
        self._args = p_args
        self._init_commandline_argument_parser()
        self._init_confiuration()
        self._init_data_store()
        self._init_run_engine()      
        
        
    def _init_commandline_argument_parser(self):
        self._commandLineKeyOfConfigurationFile = "config"
        self._commandLineFlagOfConfigurationFile = "--" + self._commandLineKeyOfConfigurationFile
        self._argumentParser = ArgumentParser('Drift Chamber Project of Team A')
        self._argumentParser.add_argument(
            self._commandLineFlagOfConfigurationFile, 
            type=str,
            help='Path to a file that holds the configuration for this program run.')
        self._parsedArgs = vars(self._argumentParser.parse_args(self._args))
        
        
    def _init_confiuration(self):
        self._configuration = Configuration(
            self._parsedArgs[self._commandLineKeyOfConfigurationFile],
            driftChamberConfig_generalSpecification)
        
        
    def _init_data_store(self):
        self._dataStore = DataStore()
        self._dataStore.put('nEvent', self._configuration['General_nEvent'], ObjectLifetime.Application)
        self._dataStore.put('configuration', self._configuration, ObjectLifetime.Application)
        self._dataStore.put('driftChamber', self, ObjectLifetime.Application)
        
            
    def _init_run_engine(self):
        self._run_engine = RunEngineFactory(self._configuration["Modules_moduleSequence"], self._dataStore).get_run_engine()
            
        
    def start_simulation(self):
        logging.info("'Drift Chamber Simulation' started.")
        self._run_engine.log_configuration()
        self._run_engine.run()
        logging.info("'Drift Chamber Simulation' done.")
 def test_run(self):
     moduleFactory = ModuleFactory('TestModule', 
                                   self.path_to_default_test_config_file,
                                   self.path_to_test_modules,
                                   self.path_to_test_modules_py)
     module = moduleFactory.get_module_instance()
     data_store = DataStore()
     data_store.put('nEvent', 2)
     run_engine = RunEngineFactory([moduleFactory], data_store).get_run_engine()
     self.assertIsInstance(run_engine, RunEngine)
     run_engine.run()
     self.assertEqual(module.beginCalled, 1)
     self.assertEqual(module.eventCalled, 2)
     self.assertEqual(module.endCalled, 1)
 def test_run(self):
     moduleFactory = ModuleFactory('TestModule',
                                   self.path_to_default_test_config_file,
                                   self.path_to_test_modules,
                                   self.path_to_test_modules_py)
     module = moduleFactory.get_module_instance()
     data_store = DataStore()
     data_store.put('nEvent', 2)
     run_engine = RunEngineFactory([moduleFactory],
                                   data_store).get_run_engine()
     self.assertIsInstance(run_engine, RunEngine)
     run_engine.run()
     self.assertEqual(module.beginCalled, 1)
     self.assertEqual(module.eventCalled, 2)
     self.assertEqual(module.endCalled, 1)
class MultiModule(Module):
    def begin(self, datastore):
        configuration = datastore.get(self)
        self._run_engine = RunEngineFactory(
            configuration['MultiModule_moduleSequence'],
            datastore).get_run_engine()
        self._run_engine.call_all_begin_methods()

    def event(self, datastore):
        self._run_engine.call_next_event_methods()

    def end(self, datastore):
        self._run_engine.call_all_end_methods()
class MultiModule(Module):

    def begin(self, datastore):
        configuration = datastore.get(self)
        self._run_engine = RunEngineFactory(configuration['MultiModule_moduleSequence'], datastore).get_run_engine()
        self._run_engine.call_all_begin_methods()
        

    def event(self, datastore):
        self._run_engine.call_next_event_methods()
        

    def end(self, datastore):
        self._run_engine.call_all_end_methods()
 def begin(self, datastore):
     configuration = datastore.get(self)
     self._run_engine = RunEngineFactory(
         configuration['MultiModule_moduleSequence'],
         datastore).get_run_engine()
     self._run_engine.call_all_begin_methods()
Пример #8
0
 def _init_run_engine(self):
     self._run_engine = RunEngineFactory(
         self._configuration["Modules_moduleSequence"],
         self._dataStore).get_run_engine()
 def begin(self, datastore):
     configuration = datastore.get(self)
     self._run_engine = RunEngineFactory(configuration['MultiModule_moduleSequence'], datastore).get_run_engine()
     self._run_engine.call_all_begin_methods()
 def _init_run_engine(self):
     self._run_engine = RunEngineFactory(self._configuration["Modules_moduleSequence"], self._dataStore).get_run_engine()