Exemple #1
0
    def prepare_dataset_pool(self, recent_years):
        cache_dir = os.path.join(self.urbansim_tmp, 'urbansim_cache')
        SimulationState().set_cache_directory(cache_dir)

        storage = StorageFactory().get_storage('dict_storage')
        dataset_pool = DatasetPool(package_order=['urbansim'], storage=storage)

        storage.write_table(table_name='gridcells',
                            table_data={
                                'grid_id': array([1, 2, 3, 4]),
                                'industrial_sqft': array([4, 0, 1, 0]),
                            })
        gridcell = dataset_pool.get_dataset('gridcell')
        self._write_dataset_to_cache(gridcell, cache_dir, 1998)
        dataset_pool.remove_all_datasets()

        storage.write_table(table_name='gridcells',
                            table_data={
                                'grid_id': array([1, 2, 3, 4]),
                                'industrial_sqft': array([3, 0, 2, 1]),
                            })
        gridcell = dataset_pool.get_dataset('gridcell')
        self._write_dataset_to_cache(gridcell, cache_dir, 1999)
        dataset_pool.remove_all_datasets()

        storage.write_table(table_name='gridcells',
                            table_data={
                                'grid_id': array([1, 2, 3, 4]),
                                'industrial_sqft': array([3, 0, 3, 1]),
                            })
        gridcell = dataset_pool.get_dataset('gridcell')
        self._write_dataset_to_cache(gridcell, cache_dir, 2000)
        dataset_pool.remove_all_datasets()

        storage.write_table(table_name='urbansim_constants',
                            table_data={
                                'recent_years': array([recent_years]),
                            })

        SimulationState().set_current_time(2001)
        attribute_cache = AttributeCache()
        SessionConfiguration(new_instance=True,
                             package_order=['urbansim'],
                             in_storage=attribute_cache)
        dataset_pool = DatasetPool(package_order=['urbansim'],
                                   storage=attribute_cache)

        # Can't write urbansim_constant, so directly add it to the pool.
        temp_dataset_pool = DatasetPool(package_order=['urbansim'],
                                        storage=storage)
        dataset_pool._add_dataset(
            'urbansim_constant',
            temp_dataset_pool.get_dataset('urbansim_constant'))
        return dataset_pool
    def prepare_dataset_pool(self, recent_years):
        cache_dir = os.path.join(self.urbansim_tmp, 'urbansim_cache')
        SimulationState().set_cache_directory(cache_dir)

        storage = StorageFactory().get_storage('dict_storage')
        dataset_pool = DatasetPool(package_order=['urbansim'], storage=storage)

        storage.write_table(table_name='gridcells',
                            table_data={
                                'grid_id': array([1, 2, 3, 4]),
                                'residential_units': array([1, 0, 5, 0]),
                                'commercial_sqft': array([0, 20, 7, 0]),
                                'industrial_sqft': array([0, 0, 0, 10]),
                                'governmental_sqft': array([0, 0, 0, 10]),
                            })
        gridcell = dataset_pool.get_dataset('gridcell')
        self._write_dataset_to_cache(gridcell, cache_dir, 1998)
        dataset_pool.remove_all_datasets()

        storage.write_table(table_name='gridcells',
                            table_data={
                                'grid_id': array([1, 2, 3, 4]),
                                'residential_units': array([1, 0, 6, 0]),
                                'commercial_sqft': array([10, 20, 10, 0]),
                                'industrial_sqft': array([0, 0, 0, 10]),
                                'governmental_sqft': array([0, 0, 0, 10]),
                            })
        gridcell = dataset_pool.get_dataset('gridcell')
        self._write_dataset_to_cache(gridcell, cache_dir, 1999)
        dataset_pool.remove_all_datasets()

        storage.write_table(table_name='gridcells',
                            table_data={
                                'grid_id': array([1, 2, 3, 4]),
                                'residential_units': array([1, 0, 6, 0]),
                                'commercial_sqft': array([10, 20, 10, 0]),
                                'industrial_sqft': array([0, 0, 5, 11]),
                                'governmental_sqft': array([0, 0, 0, 11]),
                            })
        gridcell = dataset_pool.get_dataset('gridcell')
        self._write_dataset_to_cache(gridcell, cache_dir, 2000)
        dataset_pool.remove_all_datasets()

        # For 2001, make sure every gridcell has changed in every way,
        # since this data should *not* be used by this variable.
        storage.write_table(
            table_name='gridcells',
            table_data={
                'grid_id':
                array([1, 2, 3, 4]),
                'residential_units':
                gridcell.get_attribute('residential_units') + 1,
                'commercial_sqft':
                gridcell.get_attribute('commercial_sqft') + 1,
                'industrial_sqft':
                gridcell.get_attribute('industrial_sqft') + 1,
                'governmental_sqft':
                gridcell.get_attribute('governmental_sqft') + 1,
            })
        gridcell = dataset_pool.get_dataset('gridcell')
        self._write_dataset_to_cache(gridcell, cache_dir, 2001)
        dataset_pool.remove_all_datasets()

        storage.write_table(table_name='urbansim_constants',
                            table_data={
                                'recent_years': array([recent_years]),
                            })

        SimulationState().set_current_time(2001)
        attribute_cache = AttributeCache()
        SessionConfiguration(new_instance=True,
                             package_order=['urbansim'],
                             in_storage=attribute_cache)
        dataset_pool = DatasetPool(package_order=['urbansim'],
                                   storage=attribute_cache)

        # Can't write urbansim_constant, so directly add it to the pool.
        temp_dataset_pool = DatasetPool(package_order=['urbansim'],
                                        storage=storage)
        dataset_pool._add_dataset(
            'urbansim_constant',
            temp_dataset_pool.get_dataset('urbansim_constant'))
        return dataset_pool
Exemple #3
0
    def prepare_dataset_pool(self, recent_years):
        cache_dir = os.path.join(self.urbansim_tmp, 'urbansim_cache')
        SimulationState().set_cache_directory(cache_dir)

        storage = StorageFactory().get_storage('dict_storage')
        dataset_pool = DatasetPool(package_order=['urbansim'],
                                   storage=storage)

        storage.write_table(
            table_name='gridcells',
            table_data={
                'grid_id': array([1,2,3,4]),
                'industrial_sqft': array([4,0,1,0]),
            }
        )
        gridcell = dataset_pool.get_dataset('gridcell')
        self._write_dataset_to_cache(gridcell, cache_dir, 1998)
        dataset_pool.remove_all_datasets()

        storage.write_table(
            table_name='gridcells',
            table_data={
                'grid_id': array([1,2,3,4]),
                'industrial_sqft': array([3,0,2,1]),
            }
        )
        gridcell = dataset_pool.get_dataset('gridcell')
        self._write_dataset_to_cache(gridcell, cache_dir, 1999)
        dataset_pool.remove_all_datasets()

        storage.write_table(
            table_name='gridcells',
            table_data={
                'grid_id': array([1,2,3,4]),
                'industrial_sqft': array([3,0,3,1]),
            }
        )
        gridcell = dataset_pool.get_dataset('gridcell')
        self._write_dataset_to_cache(gridcell, cache_dir, 2000)
        dataset_pool.remove_all_datasets()

        storage.write_table(
            table_name='urbansim_constants',
            table_data={
                'recent_years': array([recent_years]),
            }
        )

        SimulationState().set_current_time(2001)
        attribute_cache = AttributeCache()
        SessionConfiguration(new_instance=True,
                             package_order=['urbansim'],
                             in_storage=attribute_cache)
        dataset_pool = DatasetPool(package_order=['urbansim'],
                                   storage=attribute_cache)

        # Can't write urbansim_constant, so directly add it to the pool.
        temp_dataset_pool = DatasetPool(package_order=['urbansim'],
                                        storage=storage)
        dataset_pool._add_dataset('urbansim_constant',
                                  temp_dataset_pool.get_dataset('urbansim_constant'))
        return dataset_pool
Exemple #4
0
    def prepare_dataset_pool(self, recent_years):
        cache_dir = os.path.join(self.urbansim_tmp, 'urbansim_cache')
        SimulationState().set_cache_directory(cache_dir)

        storage = StorageFactory().get_storage('dict_storage')
        dataset_pool = DatasetPool(package_order=['urbansim'],
                                   storage=storage)

        storage.write_table(
            table_name='gridcells',
            table_data={
                'grid_id':array([1,2,3,4]),
                'residential_units':array([1,0,5,0]),
                'commercial_sqft':array([0,20,7,0]),
                'industrial_sqft':array([0,0,0,10]),
                'governmental_sqft':array([0,0,0,10]),
            }
        )
        gridcell = dataset_pool.get_dataset('gridcell')
        self._write_dataset_to_cache(gridcell, cache_dir, 1998)
        dataset_pool.remove_all_datasets()

        storage.write_table(
            table_name='gridcells',
            table_data={
                'grid_id':array([1,2,3,4]),
                'residential_units':array([1,0,6,0]),
                'commercial_sqft':array([10,20,10,0]),
                'industrial_sqft':array([0,0,0,10]),
                'governmental_sqft':array([0,0,0,10]),
            }
        )
        gridcell = dataset_pool.get_dataset('gridcell')
        self._write_dataset_to_cache(gridcell, cache_dir, 1999)
        dataset_pool.remove_all_datasets()

        storage.write_table(
            table_name='gridcells',
            table_data={
                'grid_id':array([1,2,3,4]),
                'residential_units':array([1,0,6,0]),
                'commercial_sqft':array([10,20,10,0]),
                'industrial_sqft':array([0,0,5,11]),
                'governmental_sqft':array([0,0,0,11]),
            }
        )
        gridcell = dataset_pool.get_dataset('gridcell')
        self._write_dataset_to_cache(gridcell, cache_dir, 2000)
        dataset_pool.remove_all_datasets()

        # For 2001, make sure every gridcell has changed in every way,
        # since this data should *not* be used by this variable.
        storage.write_table(
            table_name='gridcells',
            table_data={
            'grid_id':array([1,2,3,4]),
            'residential_units':gridcell.get_attribute('residential_units') + 1,
            'commercial_sqft':gridcell.get_attribute('commercial_sqft') + 1,
            'industrial_sqft':gridcell.get_attribute('industrial_sqft') + 1,
            'governmental_sqft':gridcell.get_attribute('governmental_sqft') + 1,
            }
        )
        gridcell = dataset_pool.get_dataset('gridcell')
        self._write_dataset_to_cache(gridcell, cache_dir, 2001)
        dataset_pool.remove_all_datasets()

        storage.write_table(
            table_name='urbansim_constants',
            table_data={
                'recent_years': array([recent_years]),
            }
        )

        SimulationState().set_current_time(2001)
        attribute_cache = AttributeCache()
        SessionConfiguration(new_instance=True,
                             package_order=['urbansim'],
                             in_storage=attribute_cache)
        dataset_pool = DatasetPool(package_order=['urbansim'],
                                   storage=attribute_cache)

        # Can't write urbansim_constant, so directly add it to the pool.
        temp_dataset_pool = DatasetPool(package_order=['urbansim'],
                                        storage=storage)
        dataset_pool._add_dataset('urbansim_constant',
                                  temp_dataset_pool.get_dataset('urbansim_constant'))
        return dataset_pool