openego / eGon-data

    )

def get_foreign_gas_bus_id(scn_name="eGon2035", carrier="CH4"):

    """Calculate the etrago bus id based on the geometry for eGon2035

    Map node_ids from TYNDP and etragos bus_id

    Parameters

    ----------

    scn_name : str

        Name of the scenario

    carrier : str

        Name of the carrier

    Returns

    -------

    pandas.Series

        List of mapped node_ids from TYNDP and etragos bus_id

    """

    sources = config.datasets()["gas_neighbours"]["sources"]

    bus_id = db.select_geodataframe(

        f"""

        SELECT bus_id, ST_Buffer(geom, 1) as geom, country

        FROM grid.egon_etrago_bus

        WHERE scn_name = '{scn_name}'

        AND carrier = '{carrier}'

        AND country != 'DE'

        """,

        epsg=3035,

    )

    # insert installed capacities

    file = zipfile.ZipFile(f"tyndp/{sources['tyndp_capacities']}")

    # Select buses in neighbouring countries as geodataframe

    buses = pd.read_excel(

        file.open("TYNDP-2020-Scenario-Datafile.xlsx").read(),

        sheet_name="Nodes - Dict",

    ).query("longitude==longitude")

    buses = gpd.GeoDataFrame(

        buses,

        crs=4326,

        geometry=gpd.points_from_xy(buses.longitude, buses.latitude),

    ).to_crs(3035)

    buses["bus_id"] = 0

    # Select bus_id from etrago with shortest distance to TYNDP node

    for i, row in buses.iterrows():

        distance = bus_id.set_index("bus_id").geom.distance(row.geometry)

        buses.loc[i, "bus_id"] = distance[

            distance == distance.min()

        ].index.values[0]

    return buses.set_index("node_id").bus_id

    }

def get_foreign_bus_id():

    """Calculte the etrago bus id from Nodes of TYNDP based on the geometry

    Returns

    -------

    pandas.Series

        List of mapped node_ids from TYNDP and etragos bus_id

    """

    sources = config.datasets()["electrical_neighbours"]["sources"]

    bus_id = db.select_geodataframe(

        """SELECT bus_id, ST_Buffer(geom, 1) as geom, country

        FROM grid.egon_etrago_bus

        WHERE scn_name = 'eGon2035'

        AND carrier = 'AC'

        AND v_nom = 380.

        AND country != 'DE'

        AND bus_id NOT IN (

            SELECT bus_i

            FROM osmtgmod_results.bus_data)

        """,

        epsg=3035,

    )

    # insert installed capacities

    file = zipfile.ZipFile(f"tyndp/{sources['tyndp_capacities']}")

    # Select buses in neighbouring countries as geodataframe

    buses = pd.read_excel(

        file.open("TYNDP-2020-Scenario-Datafile.xlsx").read(),

        sheet_name="Nodes - Dict",

    ).query("longitude==longitude")

    buses = gpd.GeoDataFrame(

        buses,

        crs=4326,

        geometry=gpd.points_from_xy(buses.longitude, buses.latitude),

    ).to_crs(3035)

    buses["bus_id"] = 0

    # Select bus_id from etrago with shortest distance to TYNDP node

    for i, row in buses.iterrows():

        distance = bus_id.set_index("bus_id").geom.distance(row.geometry)

        buses.loc[i, "bus_id"] = distance[

            distance == distance.min()

        ].index.values[0]

    return buses.set_index("node_id").bus_id

def calc_capacities():