#
# UrbanSim software. Copyright (C) 1998-2004 University of Washington
# 
# You can redistribute this program and/or modify it under the terms of the
# GNU General Public License as published by the Free Software Foundation
# (http://www.gnu.org/copyleft/gpl.html).
# 
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE. See the file LICENSE.html for copyright
# and licensing information, and the file ACKNOWLEDGMENTS.html for funding and
# other acknowledgments.
# 

import os

from opus_core.configurations.database_configuration import DatabaseConfiguration
from urbansim.configs.base_configuration import AbstractUrbansimConfiguration
from urbansim.configs.general_configuration_with_changed_elcm import GeneralConfiguration

class Baseline(GeneralConfiguration):
    """stc's baseline configuration.
    """
    def __init__(self):
        config = AbstractUrbansimConfiguration()
        
        config_changes = {
            'description':'StClair baseline',
            'in_storage_type':'mysql',
            'models': [
                'prescheduled_events',
                'events_coordinator',
                'residential_land_share_model',
                'land_price_model',
                'development_project_transition_model',
                'residential_development_project_location_choice_model',
                'commercial_development_project_location_choice_model',
                'industrial_development_project_location_choice_model',
                'development_event_transition_model',
                'events_coordinator',
                'residential_land_share_model',
                'household_transition_model',
                'employment_transition_model',
                'household_relocation_model',
                'household_location_choice_model',
                'employment_relocation_model',
                {'employment_location_choice_model': {'group_members': '_all_'}},
                'distribute_unplaced_jobs_model'
                ],
            'creating_baseyear_cache_configuration': {
                'cache_directory_root': 'D:/urbansim_cache/stc_11_21_06',
                'cache_from_mysql': True,
                'cache_mysql_data': 'urbansim.model_coordinators.cache_mysql_data',
                'tables_to_cache': [
                    'annual_employment_control_totals',
                    'annual_household_control_totals',
                    #'buildings',
                    #'building_types',
                    'development_event_history',
                    'gridcells',
                    'households',
                    'job_building_types',
                    'jobs',
                    'travel_data',
                    'zones',
                    'counties',
                    'commercial_development_location_choice_model_coefficients',
                    'commercial_development_location_choice_model_specification',
                    'commercial_employment_location_choice_model_coefficients',
                    'commercial_employment_location_choice_model_specification',
                    'home_based_employment_location_choice_model_specification',
                    'home_based_employment_location_choice_model_coefficients',
                    'industrial_employment_location_choice_model_coefficients',
                    'industrial_employment_location_choice_model_specification',
                    'industrial_development_location_choice_model_coefficients',
                    'industrial_development_location_choice_model_specification',
                    'residential_development_location_choice_model_coefficients',
                    'residential_development_location_choice_model_specification',
                    'fazes',
                    'urbansim_constants',
                    'household_location_choice_model_coefficients',
                    'household_location_choice_model_specification',
                    'land_price_model_coefficients',
                    'land_price_model_specification',
                    'residential_land_share_model_coefficients',
                    'residential_land_share_model_specification',
                    'plan_type_group_definitions',
                    'plan_type_groups',
                    'large_areas',
                    'household_characteristics_for_ht',
                    'development_types',
                    'development_type_group_definitions',
                    'development_constraints',
                    'annual_relocation_rates_for_households',
                    'annual_relocation_rates_for_jobs',
                    'base_year',
                    'cities',
                    'development_constraint_events',
                    'development_events',
                    'development_type_groups',
                    'employment_adhoc_sector_group_definitions',
                    'employment_adhoc_sector_groups',
                    'employment_events',
                    'employment_sectors',
                    'land_use_events',
                    'plan_types',
                    'race_names',
                    'target_vacancies',
                    'jobs_for_estimation',
                    'households_for_estimation',
                    #'development_events_exogenous',
                    'job_building_types'
                    ],
                'tables_to_cache_nchunks': {'gridcells': 1},
                'baseyear_cache': {
                    'directory_to_cache': 'D:/urbansim_cache/stc_11_21_06'
                    },
                },
            'datasets_to_preload': {
                'development_constraint': {},
                'development_event_history': {},
                'development_type': {},
                'gridcell': {'nchunks': 2},
                'household': {},
                'job': {},
                'job_building_type': {},
                'target_vacancy': {},
                'zone': {}
                },
            'input_configuration': DatabaseConfiguration(
                host_name     = os.environ.get('MYSQLHOSTNAME','localhost'),
                user_name     = os.environ.get('MYSQLUSERNAME',''),
                password      = os.environ.get('MYSQLPASSWORD',''),
                database_name = 'stc_baseyear',
                ),
            'base_year':2005,
            'years':(2006, 2035),
            }
        config.merge(config_changes)
        self.merge(config)