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
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