ebm 0.99.3__py3-none-any.whl

ebm/model/energy_requirement.py

@@ -0,0 +1,353 @@
+import typing
+
+from loguru import logger
+import numpy as np
+import pandas as pd
+
+from ebm.model.database_manager import DatabaseManager
+from ebm.model.building_category import BuildingCategory
+from ebm.model.building_condition import BuildingCondition
+from ebm.model.data_classes import YearRange
+from ebm.model.energy_purpose import EnergyPurpose
+from ebm.model.file_handler import FileHandler
+from ebm.services.files import make_unique_path
+
+
+def yearly_reduction(x):
+    if x.year < x.period_start_year:
+        return 1.0
+    if x.year > x.period_end_year:
+        return round(1.0 - x.improvement_at_period_end, 15)
+    ls = np.linspace(1.0, 1.0 - x.improvement_at_period_end, int(x.period_end_year - x.period_start_year + 1.0))[
+        x.year_no]
+    return round(ls, 15)  # x.year_no.astype(int)
+
+
+
+class EnergyRequirement:
+    def __init__(self,
+                 building_code_list: typing.List[str],
+                 period: YearRange = YearRange(2010, 2050),
+                 calibration_year: int = 1999,
+                 database_manager = None):
+        self.building_code_list = building_code_list
+        self.period = period
+        self.calibration_year = calibration_year
+        if calibration_year == period.start:
+            logger.trace(f'Calibration year {calibration_year} is same as start year {period.start}')
+        elif calibration_year not in period.subset(1):
+            logger.debug(f'Calibration year {calibration_year} is outside period {period.start}-{period.end}')
+
+        self.database_manager = database_manager
+
+
+    def calculate_for_building_category(self, database_manager: DatabaseManager = None) -> pd.DataFrame:
+        """
+        Calculates energy requirements for a single building category
+
+        Parameters
+        ----------
+        database_manager: DatabaseManager
+            optional database_manager used to load input parameters
+
+        Returns
+        -------
+        Iterable of pd.Series
+            indexed by year, building_category, TEK, purpose, building_condition
+            column kwh_m2 representing energy requirement
+
+        """
+        database_manager = database_manager if database_manager else self.database_manager
+
+        all_building_codes = database_manager.get_building_code_list().tolist()
+        all_building_categories = list(BuildingCategory)
+        all_purpose = list(EnergyPurpose)
+        most_conditions = list(BuildingCondition.existing_conditions())
+        model_years = YearRange(2020, 2050)
+        erq_oc = database_manager.get_energy_req_original_condition()
+
+        merged = self.calculate_energy_requirement(all_building_categories, all_purpose, all_building_codes, erq_oc, model_years,
+                                                   most_conditions, database_manager)
+        
+        merged = merged.drop_duplicates('building_category,building_code,building_condition,year,purpose'.split(','), keep='first')
+        return merged[['building_category', 'building_code', 'building_condition','year', 'purpose',
+                       'original_kwh_m2', 'reduction_yearly', 'reduction_policy', 'reduction_condition',
+                       'reduced_kwh_m2', 'behaviour_factor', 'kwh_m2']]
+
+    def calculate_energy_requirement(self, all_building_categories, all_purpose, all_building_codes, energy_requirement_original_condition, model_years,
+                                     most_conditions, database_manager) -> pd.DataFrame:
+        df_bc = pd.DataFrame(all_building_categories, columns=['building_category'])
+        df_building_code = pd.merge(df_bc, pd.DataFrame({'building_code': all_building_codes}), how='cross')
+        df_purpose = pd.merge(df_building_code, pd.DataFrame(all_purpose, columns=['purpose']), how='cross')
+        df_condition = pd.merge(df_purpose, pd.DataFrame({'building_condition': most_conditions}), how='cross')
+        df_years = pd.merge(df_condition, pd.DataFrame({'year': model_years}), how='cross')
+
+        energy_requirement_original_condition = energy_requirement_original_condition.copy()
+
+        energy_requirement_original_condition = energy_requirement_original_condition.join(
+            pd.DataFrame({'building_condition_r': most_conditions}),
+            how='cross',
+        )
+        energy_requirement_original_condition['building_condition'] = energy_requirement_original_condition.building_condition_r
+        energy_requirement_original_condition = energy_requirement_original_condition.drop(columns=['building_condition_r'])
+
+        erq_all_years = pd.merge(left=df_years, right=energy_requirement_original_condition, how='left')
+        energy_requirements = erq_all_years.drop(columns=['index', 'level_0'], errors='ignore')
+
+        reduction_per_condition = database_manager.get_energy_req_reduction_per_condition()
+        policy_improvement = database_manager.get_energy_need_policy_improvement()
+        yearly_improvement = database_manager.get_energy_need_yearly_improvements()
+
+        return self.calculate_energy_reduction(energy_requirements, model_years, policy_improvement,
+                                               reduction_per_condition, yearly_improvement)
+
+    def calculate_energy_reduction(self, energy_requirements: pd.DataFrame,
+                                   model_years: YearRange,
+                                   policy_improvement: pd.DataFrame,
+                                   reduction_per_condition: pd.DataFrame,
+                                   yearly_improvement: pd.DataFrame) -> pd.DataFrame:
+        """
+        Calculate and combine all reduction factors for energy needs into a single Dataframe.
+
+        Parameters
+        ----------
+        energy_requirements : pd.DataFrame
+        model_years : YearRange
+        policy_improvement : pd.DataFrame
+        reduction_per_condition : pd.DataFrame
+        yearly_improvement : pd.DataFrame
+
+        Returns
+        -------
+        pd.DataFrame
+        """
+        reduction_condition = self.calculate_reduction_condition(reduction_per_condition)
+        condition_factor = pd.merge(left=energy_requirements, right=reduction_condition,
+                                    on=['building_category', 'building_code', 'building_condition', 'purpose'],
+                                    how='left')
+
+        reduction_policy = self.calculate_reduction_policy(policy_improvement, energy_requirements)
+        reduction_yearly = self.calculate_reduction_yearly(energy_requirements, yearly_improvement)
+
+        merged = self.merge_energy_requirement_reductions(condition_factor, reduction_yearly, reduction_policy)
+
+        return merged
+
+    def merge_energy_requirement_reductions(self, condition_factor, yearly_improvements, reduction_policy):
+        m_nrg_yi = pd.merge(left=condition_factor,
+                            right=yearly_improvements.copy(),
+                            on=['building_category', 'building_code', 'purpose', 'year'],
+                            how='left')
+        m_nrg_yi = pd.merge(left=m_nrg_yi,
+                            right=reduction_policy.copy(),
+                            on=['building_category', 'building_code', 'purpose', 'year'],
+                            how='left')
+        merged = m_nrg_yi.copy()
+        merged.loc[:, 'reduction_yearly'] = merged.loc[:, 'reduction_yearly'].fillna(1.0)
+
+        merged.loc[:, 'reduction_policy'] = merged.loc[:, 'reduction_policy'].fillna(1.0)
+        merged['reduction_condition'] = merged['reduction_condition'].fillna(1.0)
+        merged['reduced_kwh_m2'] = (merged['kwh_m2'] * merged['reduction_condition'].fillna(1.0) *
+                                    merged['reduction_yearly'].fillna(1.0) * merged['reduction_policy'].fillna(1.0))
+        merged['behavior_kwh_m2'] = merged['reduced_kwh_m2'] * merged['behaviour_factor'].fillna(1.0)
+        merged = merged.rename(columns={'kwh_m2': 'original_kwh_m2'})
+        merged['kwh_m2'] = merged['behavior_kwh_m2']
+        return merged
+
+    def calculate_reduction_yearly(self, df_years: pd.DataFrame, yearly_improvement: pd.DataFrame) -> pd.DataFrame:
+        """
+        Calculate factor for yearly reduction for each entry in the DataFrame yearly_improvement.
+
+        This method merges the yearly improvement data with the policy improvement data, adjusts the
+        efficiency start year if the period end year is greater, and calculates the yearly reduction
+        based on the yearly efficiency improvement.
+
+        Parameters
+        ----------
+        df_years : pd.DataFrame
+            DataFrame containing all years for which to calculate factors. Must include column 'year'.
+        yearly_improvement : pd.DataFrame
+            DataFrame containing yearly improvement information. Must include columns 'yearly_efficiency_improvement', and 'efficiency_start_year'.
+
+        Returns
+        -------
+        pd.DataFrame
+            DataFrame with the calculated 'reduction_yearly' column and updated entries.
+        """
+        required_in_yearly_improvement = {'yearly_efficiency_improvement', 'start_year', 'end_year'}
+        if not required_in_yearly_improvement.issubset(yearly_improvement.columns):
+            logger.debug(f'Got columns {", ".join(yearly_improvement.columns)}')
+            missing = required_in_yearly_improvement.difference(yearly_improvement.columns)
+            raise ValueError('Required column{} not found in yearly_improvement: {}'.format(
+                's' if len(missing) != 1 else '', missing
+            ))
+        if 'year' not in df_years:
+            logger.debug(f'Got columns {", ".join(df_years.columns)}')
+            raise ValueError('df_years does not contain column year')
+
+        years = pd.DataFrame(data=[y for y in df_years.year.unique()], columns=['year'])
+
+        df = pd.merge(left=yearly_improvement, right=years, how='cross')
+        rows_in_range = df[(df.year >= df.start_year) & (df.year <= df.end_year)].index
+
+        df.loc[rows_in_range, 'yearly_change'] = (1.0 - df.loc[rows_in_range, 'yearly_efficiency_improvement'])
+        df.loc[rows_in_range, 'pow'] = (df.loc[rows_in_range, 'year'] - df.loc[rows_in_range, 'start_year']) + 1
+        df.loc[rows_in_range, 'reduction_yearly'] =  df.loc[rows_in_range, 'yearly_change'] ** df.loc[rows_in_range, 'pow']
+
+        df.loc[df[df.start_year > df.year].index, 'reduction_yearly'] = df.loc[
+            df[df.start_year > df.year].index, 'reduction_yearly'].fillna(1.0)
+        df.loc[:, 'reduction_yearly'] = df.loc[:, 'reduction_yearly'].ffill()
+
+        return df[['building_category', 'building_code', 'purpose', 'year', 'reduction_yearly']]
+
+
+    def calculate_reduction_policy(self, policy_improvement: pd.DataFrame, all_things) -> pd.DataFrame:
+        """
+        Calculate the reduction policy for each entry in the DataFrame.
+
+        This method computes the reduction policy by first calculating the number of years since the
+        start of the period. It then applies the `yearly_reduction` function to each relevant entry
+        to determine the reduction policy.
+
+        Parameters
+        ----------
+        policy_improvement : pd.DataFrame
+            DataFrame containing policy improvement information. Must include columns 'year' and 'period_start_year'.
+        all_things: pd.DataFrame
+            DataFrame containing every combination of building_category, TEK, purpose, year
+
+        Returns
+        -------
+        pd.DataFrame
+            DataFrame with the calculated 'reduction_policy' column and updated entries.
+        """
+        policy_improvement = policy_improvement.sort_values(
+            by=['building_category', 'building_code', 'purpose', 'start_year', 'end_year'])
+
+        policy_improvement[['building_category_s', 'TEK_s', 'purpose_s', 'start_year_s', 'end_year_s']] = \
+        policy_improvement[
+            ['building_category', 'building_code', 'purpose', 'start_year', 'end_year']]
+
+        policy_improvement = policy_improvement.set_index(
+            ['building_category', 'building_code', 'purpose', 'start_year', 'end_year'], drop=True)
+
+        shifted = policy_improvement.shift(1).reset_index()
+
+        shifted = shifted.query('building_category==building_category_s & building_code==TEK_s & purpose==purpose_s')
+        shifted['improvement_at_start_year'] = shifted['improvement_at_end_year']
+        shifted = shifted[['building_category', 'building_code', 'purpose', 'start_year', 'end_year', 'improvement_at_start_year']]
+
+        start_year_from_previous = shifted
+
+        policy_improvement = pd.merge(left=policy_improvement,
+                                      right=start_year_from_previous,
+                                      left_on=['building_category', 'building_code', 'purpose', 'start_year', 'end_year'],
+                                      right_on=['building_category', 'building_code', 'purpose', 'start_year', 'end_year'],
+                                      how='left'
+                                      )
+
+        policy_improvement[['start_year', 'end_year']] = policy_improvement[['start_year', 'end_year']].astype(int)
+        policy_improvement = policy_improvement.set_index(
+            ['building_category', 'building_code', 'purpose', 'start_year', 'end_year'], drop=True)
+        policy_improvement['improvement_at_start_year'] = 1.0-policy_improvement['improvement_at_start_year'].fillna(0.0)
+
+        policy_improvement = policy_improvement[['improvement_at_start_year', 'improvement_at_end_year']].reset_index()
+
+        df = pd.merge(left=all_things[['building_category', 'building_code', 'purpose', 'year']],
+                      right=policy_improvement,
+                      on=['building_category', 'building_code', 'purpose'], how='left')
+
+        df['num_values'] = df['end_year'] - df['start_year'] + 1.0
+        df['n'] = (df.year - df.start_year).clip(upper=df.num_values-1, lower=0)
+
+        df['step'] = ((1.0-df['improvement_at_end_year']) - df['improvement_at_start_year']) / (df['num_values']-1.0)
+
+        df['reduction_policy'] = df['improvement_at_start_year'] + (df['n']) * df['step']
+        df['reduction_policy'] = df['reduction_policy'].fillna(1.0)
+        df['_col_to_filter'] = (df['year'] < df['start_year']) | (df['year'] > df['end_year'])
+        df = df.sort_values(by=['building_category', 'building_code', 'purpose', 'year', '_col_to_filter'])
+        df = df.drop_duplicates(['building_category', 'building_code', 'purpose', 'year'])
+
+        return df[['building_category', 'building_code', 'purpose', 'year', 'reduction_policy']]
+
+
+    def calculate_reduction_condition(self, reduction_per_condition: pd.DataFrame) -> pd.DataFrame:
+        """
+        Calculate the reduction condition for each entry in the DataFrame.
+
+        This method computes the reduction condition by subtracting the reduction share from 1.0.
+        It also fills any NaN values in the 'reduction_condition' column with 1.0.
+
+        Parameters
+        ----------
+        reduction_per_condition : pd.DataFrame
+            DataFrame containing the reduction share information. Must include columns 'reduction_share' and 'building_code'.
+
+        Returns
+        -------
+        pd.DataFrame
+            DataFrame with the calculated 'reduction_condition' column and filtered entries.
+        """
+        reduction_per_condition['reduction_condition'] = 1.0 - reduction_per_condition['reduction_share']
+        reduction_per_condition.loc[:, 'reduction_condition'] = reduction_per_condition.loc[:,
+                                                                'reduction_condition'].fillna(1.0)
+        return reduction_per_condition
+
+    def calculate_energy_requirements(
+            self,
+            building_categories: typing.Iterable[BuildingCategory] = None) -> pd.DataFrame:
+        """
+        Calculates energy requirements for building categories
+
+        Parameters
+        ----------
+        building_categories : Iterable[BuildingCategory]
+            Iterable containing building categories on which to calculate energy requirements.
+
+        Returns
+        -------
+        Iterable of pd.Series
+            indexed by year, building_category, TEK, purpose, building_condition
+            column kwh_m2 representing energy requirement
+
+        """
+        building_categories = building_categories if building_categories else iter(BuildingCategory)
+
+        return self.calculate_for_building_category(self.database_manager)
+
+    @staticmethod
+    def new_instance(period, calibration_year, database_manager=None):
+        if period.start != 2010 and calibration_year != calibration_year:
+            logger.warning(f'EnergyRequirements {period.start=} {calibration_year=}')
+        dm = database_manager if isinstance(database_manager, DatabaseManager) else DatabaseManager()
+        instance = EnergyRequirement(building_code_list=dm.get_building_code_list(), period=period, calibration_year=calibration_year,
+                                     database_manager=dm)
+        return instance
+
+
+def main():
+    import os
+    import sys
+    import pathlib
+    log_format = """
+    <blue>{elapsed}</blue> | <level>{level: <8}</level> | <cyan>{function: <20}</cyan>:<cyan>{line: <3}</cyan> - <level>{message}</level>
+    """.strip()
+    logger.remove()
+    logger.add(sys.stderr, format=log_format, level='WARNING')
+
+    dm = DatabaseManager(FileHandler(directory='kalibrering'))
+    er = EnergyRequirement.new_instance(YearRange(2020, 2050), calibration_year=2020, database_manager=dm)
+
+    logger.error('Calculating')
+    df =  er.calculate_energy_requirements()
+    logger.error('Writing to file')
+
+    xlsx_filename = make_unique_path(pathlib.Path('output/er.xlsx'))
+    df.to_excel(xlsx_filename)
+
+    logger.error('DONE')
+    os.startfile(xlsx_filename, 'open')
+
+
+if __name__ == '__main__':
+    main()

ebm/model/energy_use.py

@@ -0,0 +1,202 @@
+import numpy as np
+import pandas as pd
+
+from ebm import extractors
+from ebm.energy_consumption import HEATING_SYSTEM_SHARE, GRUNNLAST_ANDEL, BASE_LOAD_EFFICIENCY, BASE_LOAD_ENERGY_PRODUCT, \
+    PEAK_LOAD_ENERGY_PRODUCT, PEAK_LOAD_COVERAGE, PEAK_LOAD_EFFICIENCY, TERTIARY_LOAD_COVERAGE, TERTIARY_LOAD_EFFICIENCY, \
+    TERTIARY_LOAD_ENERGY_PRODUCT, COOLING_EFFICIENCY, DHW_EFFICIENCY, DOMESTIC_HOT_WATER_ENERGY_PRODUCT
+from ebm.model import energy_need as e_n, heating_systems_parameter as h_s_param
+from ebm.model.data_classes import YearRange
+from ebm.model.database_manager import DatabaseManager
+from ebm.s_curve import calculate_s_curves
+
+
+def base_load(heating_systems_projection: pd.DataFrame) -> pd.DataFrame:
+    heating_systems_projection['heating_system'] = '-'
+    df = heating_systems_projection[
+        ['building_category', 'building_code', 'year', 'heating_systems', HEATING_SYSTEM_SHARE, GRUNNLAST_ANDEL, BASE_LOAD_EFFICIENCY,
+         BASE_LOAD_ENERGY_PRODUCT, 'heating_system']].copy()
+    df = df.rename(columns={GRUNNLAST_ANDEL: 'load_share', BASE_LOAD_EFFICIENCY: 'load_efficiency',
+                            BASE_LOAD_ENERGY_PRODUCT: 'energy_product'})
+    df.loc[:, 'load'] = 'base'
+    df.loc[:, 'purpose'] = 'heating_rv'
+    df['heating_system'] = df.heating_systems.apply(lambda s: s.split('-')[0])
+    df['heating_system'] = df['heating_system'].str.strip()
+    return df
+
+
+def peak_load(heating_systems_projection:pd.DataFrame) -> pd.DataFrame:
+    df = heating_systems_projection[
+        ['building_category', 'building_code', 'year', 'heating_systems', HEATING_SYSTEM_SHARE, PEAK_LOAD_COVERAGE, PEAK_LOAD_EFFICIENCY,
+         PEAK_LOAD_ENERGY_PRODUCT, 'heating_system']].copy()
+    df = df.rename(columns={PEAK_LOAD_COVERAGE: 'load_share', PEAK_LOAD_EFFICIENCY: 'load_efficiency',
+                            PEAK_LOAD_ENERGY_PRODUCT: 'energy_product'})
+    df.loc[:, 'load'] = 'peak'
+    df.loc[:, 'purpose'] = 'heating_rv'
+    df['heating_system'] = df.heating_systems.apply(lambda s: s.split('-')[1:2]).explode('heating_system')
+    df['heating_system'] = df['heating_system'].str.strip()
+    return df
+
+
+def tertiary_load(heating_systems_projection: pd.DataFrame) ->pd.DataFrame:
+    df = heating_systems_projection[
+        ['building_category', 'building_code', 'year', 'heating_systems', HEATING_SYSTEM_SHARE, TERTIARY_LOAD_COVERAGE, TERTIARY_LOAD_EFFICIENCY,
+         TERTIARY_LOAD_ENERGY_PRODUCT, 'heating_system']].copy()
+    df = df.rename(columns={TERTIARY_LOAD_COVERAGE: 'load_share', TERTIARY_LOAD_EFFICIENCY: 'load_efficiency',
+                            TERTIARY_LOAD_ENERGY_PRODUCT: 'energy_product'})
+    df.loc[:, 'load'] = 'tertiary'
+    df.loc[:, 'purpose'] = 'heating_rv'
+    df['heating_system'] = df.heating_systems.apply(lambda s: s.split('-')[2:3]).explode('heating_system')
+    df['heating_system'] = df['heating_system'].str.strip()
+    return df
+
+
+def heating_rv(heating_systems_projection: pd.DataFrame) -> pd.DataFrame:
+    df = heating_systems_projection.copy()
+
+    base = base_load(df)
+    peak = peak_load(df)
+    tertiary = tertiary_load(df)
+
+    return pd.concat([base, peak, tertiary])
+
+
+def heating_dhw(heating_systems_projection: pd.DataFrame) ->pd.DataFrame:
+    df = heating_systems_projection[
+        ['building_category', 'building_code', 'year', 'heating_systems', HEATING_SYSTEM_SHARE, GRUNNLAST_ANDEL, DHW_EFFICIENCY,
+         DOMESTIC_HOT_WATER_ENERGY_PRODUCT]].copy()
+    df.loc[:, GRUNNLAST_ANDEL] = 1.0
+    df = df.rename(columns={GRUNNLAST_ANDEL: 'load_share', DHW_EFFICIENCY: 'load_efficiency',
+                            DOMESTIC_HOT_WATER_ENERGY_PRODUCT: 'energy_product'})
+
+    df.loc[:, 'load'] = 'dhw'
+    df['purpose'] = 'heating_dhw'
+    return df
+
+
+def cooling(heating_systems_projection: pd.DataFrame) -> pd.DataFrame:
+    df = heating_systems_projection[
+        ['building_category', 'building_code', 'year', 'heating_systems', HEATING_SYSTEM_SHARE, GRUNNLAST_ANDEL, COOLING_EFFICIENCY,
+         BASE_LOAD_ENERGY_PRODUCT]].copy()
+    df.loc[:, GRUNNLAST_ANDEL] = 1.0
+    df.loc[:, BASE_LOAD_ENERGY_PRODUCT] = 'Electricity'
+    df = df.rename(columns={GRUNNLAST_ANDEL: 'load_share', COOLING_EFFICIENCY: 'load_efficiency',
+                            BASE_LOAD_ENERGY_PRODUCT: 'energy_product'})
+
+    df.loc[:, 'load'] = 'base'
+    df.loc[:, 'purpose'] = 'cooling'
+    return df
+
+
+def other(heating_systems_projection: pd.DataFrame) -> pd.DataFrame:
+    df = heating_systems_projection[
+        ['building_category', 'building_code', 'year', 'heating_systems', HEATING_SYSTEM_SHARE, GRUNNLAST_ANDEL, BASE_LOAD_EFFICIENCY,
+         BASE_LOAD_ENERGY_PRODUCT]].copy()
+    df.loc[:, GRUNNLAST_ANDEL] = 1.0
+    df.loc[:, BASE_LOAD_EFFICIENCY] = 1.0
+    df.loc[:, BASE_LOAD_ENERGY_PRODUCT] = 'Electricity'
+    df = df.rename(columns={GRUNNLAST_ANDEL: 'load_share', BASE_LOAD_EFFICIENCY: 'load_efficiency',
+                            BASE_LOAD_ENERGY_PRODUCT: 'energy_product'})
+    df.loc[:, 'load'] = 'base'
+    df.loc[:, '_purposes'] = 'electrical_equipment,fans_and_pumps,lighting'
+    df = df.assign(**{'purpose': df['_purposes'].str.split(',')}).explode('purpose')
+    df = df.drop(columns=['_purposes'], errors='ignore')
+
+    return df.reset_index().drop(columns=['index'], errors='ignore')
+
+
+def all_purposes(heating_systems_projection: pd.DataFrame) -> pd.DataFrame:
+    return pd.concat([heating_rv(heating_systems_projection), heating_dhw(heating_systems_projection),
+               cooling(heating_systems_projection), other(heating_systems_projection)])
+
+
+def efficiency_factor(heating_systems: pd.DataFrame) -> pd.DataFrame:
+    df = heating_systems
+    df.loc[:, 'efficiency_factor'] = df.loc[:, 'heating_system_share'] * df.loc[:, 'load_share'] / df.loc[:, 'load_efficiency']
+
+    return df
+
+
+def energy_use_kwh(energy_need: pd.DataFrame, efficiency_factor: pd.DataFrame) -> pd.DataFrame:
+    nrj = energy_need.copy()
+
+    df = nrj.reset_index().merge(efficiency_factor,
+                                 left_on=['building_category', 'building_code', 'purpose', 'year'],
+                                 right_on=['building_category', 'building_code', 'purpose', 'year'])
+
+    df['kwh'] = df['energy_requirement'] * df['heating_system_share'] * df['load_share'] / df['load_efficiency']
+    if 'kwh_m2' in df.columns:
+        df['kwh_m2'] = df['kwh_m2'] * df['efficiency_factor']
+    else:
+        df['kwh_m2'] = np.nan
+    return df
+
+
+def building_group_energy_use_kwh(heating_systems_parameter: pd.DataFrame, energy_need: pd.DataFrame) -> pd.DataFrame:
+    df = all_purposes(heating_systems_parameter)
+    df.loc[:, 'building_group'] = 'yrkesbygg'
+    df.loc[df.building_category.isin(['house', 'apartment_block']), 'building_group'] = 'bolig'
+
+    efficiency_factor_df = efficiency_factor(df)
+    df = energy_use_kwh(energy_need=energy_need, efficiency_factor=efficiency_factor_df)
+
+    return df
+
+
+def energy_use_gwh_by_building_group(energy_use_kwh: pd.DataFrame) -> pd.DataFrame:
+    energy_use_by_building_group = energy_use_kwh[['building_group', 'year', 'energy_product', 'kwh']].groupby(
+        by=['building_group', 'energy_product', 'year']).sum() / 1_000_000
+    energy_use_wide = energy_use_by_building_group.reset_index().pivot(columns=['year'],
+                                                                       index=['building_group', 'energy_product'],
+                                                                       values=['kwh'])
+    energy_use_wide = energy_use_wide.reset_index()
+    energy_use_wide.columns = ['building_group', 'energy_source'] + [c for c in energy_use_wide.columns.get_level_values(1)[2:]]
+    return energy_use_wide
+
+
+def calculate_energy_use(database_manager: 'DatabaseManager',
+                         years: YearRange|None=YearRange(2020, 2050),
+                         area_parameters:pd.DataFrame|None=None,
+                         scurve_parameters: pd.DataFrame|None=None,
+                         building_code_parameters:pd.DataFrame|None=None) -> pd.DataFrame:
+    """
+    calculates energy use in KWh by building_category, TEK, building_condition, year, purpose.
+
+    The dataframe is index by row index. (subject to change)
+
+    extra columns
+    m2, original_kwh_m2, reduction_yearly, reduction_policy, reduction_condition, reduced_kwh_m2,
+       behaviour_factor, kwh_m2, energy_requirement, heating_systems,
+       heating_system_share, load_share, load_efficiency, energy_product,
+       heating_system, load, building_group, efficiency_factor
+
+    Parameters
+    ----------
+    database_manager : pd.DataFrame
+    years : YearRange, optional
+    area_parameters : pd.DataFrame, optional
+    scurve_parameters : pd.DataFrame, optional
+    building_code_parameters : pd.DataFrame, optional
+
+    Returns
+    -------
+    pd.DataFrame
+        energy use in KWh by building_category, TEK, building_condition, year, purpose
+        ,
+    """
+    scurve_parameters = database_manager.get_scurve_params() if scurve_parameters is None else scurve_parameters
+    area_parameters = database_manager.get_area_parameters() if area_parameters is None else area_parameters
+    area_parameters['year'] = years.start
+    building_code_parameters = database_manager.file_handler.get_building_code() if building_code_parameters is None else building_code_parameters
+
+    s_curves_by_condition = calculate_s_curves(scurve_parameters, building_code_parameters, years)  # 📌
+    area_forecast = extractors.extract_area_forecast(years, s_curves_by_condition, building_code_parameters, area_parameters, database_manager)  # 📍
+
+    energy_need_kwh_m2 = extractors.extract_energy_need(years, database_manager)  # 📍
+    total_energy_need = e_n.transform_total_energy_need(energy_need_kwh_m2, area_forecast)  # 📌
+
+    heating_systems_projection = extractors.extract_heating_systems_forecast(years, database_manager)  # 📍
+    heating_systems_parameter = h_s_param.heating_systems_parameter_from_projection(heating_systems_projection)  # 📌
+    energy_use_kwh_with_building_group = building_group_energy_use_kwh(heating_systems_parameter, total_energy_need)
+
+    return energy_use_kwh_with_building_group

ebm/model/enums.py

@@ -0,0 +1,8 @@
+from enum import Enum
+
+class ReturnCode(Enum):
+    """Enum used by __main__ for return codes"""
+    OK = 0
+    FILE_EXISTS = 1
+    FILE_NOT_ACCESSIBLE = 2
+    MISSING_INPUT_FILES = 3

ebm/model/exceptions.py

@@ -0,0 +1,4 @@
+
+class AmbiguousDataError(Exception):
+    """Custom exception raised when conflicting data is found."""
+    pass