tunned-geobr 0.2.2__py3-none-any.whl → 0.2.4__py3-none-any.whl

tunned_geobr/__init__.py

@@ -67,4 +67,62 @@ from .read_geographic_regions import read_geographic_regions
 from .read_biosphere_reserves import read_biosphere_reserves
 from .read_census_tract_2022 import read_census_tract_2022
 from .read_neighborhoods_2022 import read_neighborhoods_2022
-from .read_baze_sites import read_baze_sites
+from .read_baze_sites import read_baze_sites
+from .read_existent_eolic import read_existent_eolic
+from .read_planned_eolic import read_planned_eolic
+from .read_existent_solar import read_existent_solar
+from .read_planned_solar import read_planned_solar
+from .read_existent_uhe import read_existent_uhe
+from .read_existent_pch import read_existent_pch
+from .read_existent_cgh import read_existent_cgh
+from .read_planned_uhe import read_planned_uhe
+from .read_planned_pch import read_planned_pch
+from .read_planned_cgh import read_planned_cgh
+from .read_existent_biomass_ute import read_existent_biomass_ute
+from .read_existent_fossile_ute import read_existent_fossile_ute
+from .read_existent_nuclear_ute import read_existent_nuclear_ute
+from .read_planned_biomass_ute import read_planned_biomass_ute
+from .read_planned_fossile_ute import read_planned_fossile_ute
+from .read_planned_nuclear_ute import read_planned_nuclear_ute
+from .read_existent_substations import read_existent_substations
+from .read_existent_transmission_lines import read_existent_transmission_lines
+from .read_planned_substations import read_planned_substations
+from .read_planned_transmission_lines import read_planned_transmission_lines
+from .read_subsystem_interconnected import read_subsystem_interconnected
+from .read_isolated_systems import read_isolated_systems
+from .read_etanol_plants import read_etanol_plants
+from .read_biodiesel_plants import read_biodiesel_plants
+from .read_gas_transport_pipelines import read_gas_transport_pipelines
+from .read_gas_distribution_pipelines import read_gas_distribution_pipelines
+from .read_areas_under_contract import read_areas_under_contract
+from .read_federal_union_areas import read_federal_union_areas
+from .read_og_effective_geographic_basin import read_og_effective_geographic_basin
+from .read_og_probabilistic_effective_basin import read_og_probabilistic_effective_basin
+from .read_og_predominant_fluid_type import read_og_predominant_fluid_type
+from .read_og_unconventional_resources import read_og_unconventional_resources
+from .read_og_total_ipa import read_og_total_ipa
+from .read_og_ipa_exploratory_intensity import read_og_ipa_exploratory_intensity
+from .read_og_ipa_exploratory_activity import read_og_ipa_exploratory_activity
+from .read_og_ipa_prospectiveness import read_og_ipa_prospectiveness
+from .read_og_ipa_direct_evidence import read_og_ipa_direct_evidence
+from .read_og_ipa_need_for_knowledge import read_og_ipa_need_for_knowledge
+from .read_og_ipa_supply_infrastructure import read_og_ipa_supply_infrastructure
+from .read_og_legal_pre_salt_polygon import read_og_legal_pre_salt_polygon
+from .read_exploration_production_environment import read_exploration_production_environment
+from .read_sedimentary_basins import read_sedimentary_basins
+from .read_og_basement import read_og_basement
+from .read_hydroelectric_feasibility_studies import read_hydroelectric_feasibility_studies
+from .read_hydroelectric_inventory_aai_studies import read_hydroelectric_inventory_aai_studies
+from .read_ama_anemometric_towers import read_ama_anemometric_towers
+from .read_biomethane_plants import read_biomethane_plants
+from .read_fuel_bases import read_fuel_bases
+from .read_glp_bases import read_glp_bases
+from .read_processing_facilities import read_processing_facilities
+from .read_oil_and_derivatives_terminal import read_oil_and_derivatives_terminal
+from .read_pio_terminals import read_pio_terminals
+from .read_pio_ducts import read_pio_ducts
+from .read_gnl_terminals import read_gnl_terminals
+from .read_natural_gas_processing_hub import read_natural_gas_processing_hub
+from .read_compression_stations import read_compression_stations
+from .read_natural_gas_delivery_points import read_natural_gas_delivery_points
+from .read_quilombola_areas import read_quilombola_areas

tunned_geobr/list_geobr.py

@@ -25,13 +25,13 @@ def list_geobr():
         {"Function": "read_country", "Geography": "Country", "Years": "All", "Source": "IBGE"},
         {"Function": "read_region", "Geography": "Region", "Years": "All", "Source": "IBGE"},
         {"Function": "read_state", "Geography": "State", "Years": "All", "Source": "IBGE"},
-        {"Function": "read_state_direct", "Geography": "State (direct)", "Years": "All", "Source": "IBGE"},
+        {"Function": "read_state_direct", "Geography": "State", "Years": "All", "Source": "IBGE"},
         {"Function": "read_meso_region", "Geography": "Meso region", "Years": "1991, 2000, 2010, 2013, 2014, 2015, 2016, 2017, 2018, 2019, 2020", "Source": "IBGE"},
         {"Function": "read_micro_region", "Geography": "Micro region", "Years": "1991, 2000, 2010, 2013, 2014, 2015, 2016, 2017, 2018, 2019, 2020", "Source": "IBGE"},
         {"Function": "read_immediate_region", "Geography": "Immediate region", "Years": "2017, 2019, 2020", "Source": "IBGE"},
         {"Function": "read_intermediate_region", "Geography": "Intermediate region", "Years": "2017, 2019, 2020", "Source": "IBGE"},
         {"Function": "read_municipality", "Geography": "Municipality", "Years": "All", "Source": "IBGE"},
-        {"Function": "read_municipality_direct", "Geography": "Municipality (direct)", "Years": "All", "Source": "IBGE"},
+        {"Function": "read_municipality_direct", "Geography": "Municipality", "Years": "All", "Source": "IBGE"},
         {"Function": "read_weighting_area", "Geography": "Census weighting area", "Years": "2010", "Source": "IBGE"},
         {"Function": "read_census_tract", "Geography": "Census tract", "Years": "2000, 2010", "Source": "IBGE"},
         {"Function": "read_census_tract_2022", "Geography": "Census tract 2022", "Years": "2022", "Source": "IBGE"},
@@ -101,7 +101,78 @@ def list_geobr():
         # Land tenure and property datasets
         {"Function": "read_snci_properties", "Geography": "SNCI Properties", "Years": "All", "Source": "INCRA"},
         {"Function": "read_sigef_properties", "Geography": "SIGEF Properties", "Years": "All", "Source": "INCRA"},
-        {"Function": "read_quilombola_areas", "Geography": "Quilombola Areas", "Years": "All", "Source": "INCRA"}
+        {"Function": "read_quilombola_areas", "Geography": "Quilombola Areas", "Years": "All", "Source": "INCRA"},
+        
+        # Energy infrastructure datasets - Solar
+        {"Function": "read_existent_solar", "Geography": "Existing Solar Power Plants", "Years": "All", "Source": "EPE"},
+        {"Function": "read_planned_solar", "Geography": "Planned Solar Power Plants", "Years": "All", "Source": "EPE"},
+        
+        # Energy infrastructure datasets - Wind
+        {"Function": "read_existent_eolic", "Geography": "Existing Wind Power Plants", "Years": "All", "Source": "EPE"},
+        {"Function": "read_planned_eolic", "Geography": "Planned Wind Power Plants", "Years": "All", "Source": "EPE"},
+        
+        # Energy infrastructure datasets - Hydroelectric
+        {"Function": "read_existent_uhe", "Geography": "Existing Large Hydroelectric Plants", "Years": "All", "Source": "EPE"},
+        {"Function": "read_existent_pch", "Geography": "Existing Small Hydroelectric Plants", "Years": "All", "Source": "EPE"},
+        {"Function": "read_existent_cgh", "Geography": "Existing Mini Hydroelectric Plants", "Years": "All", "Source": "EPE"},
+        {"Function": "read_planned_uhe", "Geography": "Planned Large Hydroelectric Plants", "Years": "All", "Source": "EPE"},
+        {"Function": "read_planned_pch", "Geography": "Planned Small Hydroelectric Plants", "Years": "All", "Source": "EPE"},
+        {"Function": "read_planned_cgh", "Geography": "Planned Mini Hydroelectric Plants", "Years": "All", "Source": "EPE"},
+        
+        # Energy infrastructure datasets - Thermoelectric
+        {"Function": "read_existent_biomass_ute", "Geography": "Existing Biomass Thermoelectric Plants", "Years": "All", "Source": "EPE"},
+        {"Function": "read_existent_fossile_ute", "Geography": "Existing Fossil Thermoelectric Plants", "Years": "All", "Source": "EPE"},
+        {"Function": "read_existent_nuclear_ute", "Geography": "Existing Nuclear Thermoelectric Plants", "Years": "All", "Source": "EPE"},
+        {"Function": "read_planned_biomass_ute", "Geography": "Planned Biomass Thermoelectric Plants", "Years": "All", "Source": "EPE"},
+        {"Function": "read_planned_fossile_ute", "Geography": "Planned Fossil Thermoelectric Plants", "Years": "All", "Source": "EPE"},
+        {"Function": "read_planned_nuclear_ute", "Geography": "Planned Nuclear Thermoelectric Plants", "Years": "All", "Source": "EPE"},
+        
+        # Energy infrastructure datasets - Transmission
+        {"Function": "read_existent_substations", "Geography": "Existing Electrical Substations", "Years": "All", "Source": "EPE"},
+        {"Function": "read_existent_transmission_lines", "Geography": "Existing Transmission Lines", "Years": "All", "Source": "EPE"},
+        {"Function": "read_planned_substations", "Geography": "Planned Electrical Substations", "Years": "All", "Source": "EPE"},
+        {"Function": "read_planned_transmission_lines", "Geography": "Planned Transmission Lines", "Years": "All", "Source": "EPE"},
+        {"Function": "read_subsystem_interconnected", "Geography": "National Interconnected System Subsystems", "Years": "All", "Source": "EPE"},
+        {"Function": "read_isolated_systems", "Geography": "Isolated Electrical Systems", "Years": "All", "Source": "EPE"},
+        
+        # Energy infrastructure datasets - Biofuel Plants
+        {"Function": "read_etanol_plants", "Geography": "Ethanol Production Plants", "Years": "All", "Source": "EPE"},
+        {"Function": "read_biodiesel_plants", "Geography": "Biodiesel Production Plants", "Years": "All", "Source": "EPE"},
+        {"Function": "read_biomethane_plants", "Geography": "Biomethane Production Plants", "Years": "All", "Source": "EPE"},
+        
+        # Energy infrastructure datasets - Oil and Gas Infrastructure
+        {"Function": "read_fuel_bases", "Geography": "Fuel Bases", "Years": "All", "Source": "EPE"},
+        {"Function": "read_glp_bases", "Geography": "GLP (LPG) Bases", "Years": "All", "Source": "EPE"},
+        {"Function": "read_processing_facilities", "Geography": "Oil and Gas Processing Facilities", "Years": "All", "Source": "EPE"},
+        {"Function": "read_oil_and_derivatives_terminal", "Geography": "Oil and Derivatives Terminals", "Years": "All", "Source": "EPE"},
+        {"Function": "read_pio_terminals", "Geography": "PIO Terminals", "Years": "All", "Source": "EPE"},
+        {"Function": "read_pio_ducts", "Geography": "PIO Ducts", "Years": "All", "Source": "EPE"},
+        {"Function": "read_gnl_terminals", "Geography": "GNL (LNG) Terminals", "Years": "All", "Source": "EPE"},
+        {"Function": "read_natural_gas_processing_hub", "Geography": "Natural Gas Processing Hubs", "Years": "All", "Source": "EPE"},
+        {"Function": "read_compression_stations", "Geography": "Natural Gas Compression Stations", "Years": "All", "Source": "EPE"},
+        {"Function": "read_natural_gas_delivery_points", "Geography": "Natural Gas Delivery Points", "Years": "All", "Source": "EPE"},
+        {"Function": "read_gas_transport_pipelines", "Geography": "Gas Transport Pipelines", "Years": "All", "Source": "EPE"},
+        {"Function": "read_gas_distribution_pipelines", "Geography": "Gas Distribution Pipelines", "Years": "All", "Source": "EPE"},
+        {"Function": "read_areas_under_contract", "Geography": "Oil and Gas Areas Under Contract", "Years": "All", "Source": "EPE"},
+        {"Function": "read_federal_union_areas", "Geography": "Federal Union Areas for Oil and Gas", "Years": "All", "Source": "EPE"},
+        {"Function": "read_og_effective_geographic_basin", "Geography": "Oil and Gas Effective Geographic Basins", "Years": "All", "Source": "EPE"},
+        {"Function": "read_og_probabilistic_effective_basin", "Geography": "Oil and Gas Probabilistic Effective Basins", "Years": "All", "Source": "EPE"},
+        {"Function": "read_og_predominant_fluid_type", "Geography": "Oil and Gas Predominant Fluid Type Areas", "Years": "All", "Source": "EPE"},
+        {"Function": "read_og_unconventional_resources", "Geography": "Oil and Gas Unconventional Resources", "Years": "All", "Source": "EPE"},
+        {"Function": "read_og_total_ipa", "Geography": "Oil and Gas Total IPA", "Years": "All", "Source": "EPE"},
+        {"Function": "read_og_ipa_exploratory_intensity", "Geography": "Oil and Gas IPA Exploratory Intensity", "Years": "All", "Source": "EPE"},
+        {"Function": "read_og_ipa_exploratory_activity", "Geography": "Oil and Gas IPA Exploratory Activity", "Years": "All", "Source": "EPE"},
+        {"Function": "read_og_ipa_prospectiveness", "Geography": "Oil and Gas IPA Prospectiveness", "Years": "All", "Source": "EPE"},
+        {"Function": "read_og_ipa_direct_evidence", "Geography": "Oil and Gas IPA Direct Evidence of Hydrocarbons", "Years": "All", "Source": "EPE"},
+        {"Function": "read_og_ipa_need_for_knowledge", "Geography": "Oil and Gas IPA Need for Knowledge", "Years": "All", "Source": "EPE"},
+        {"Function": "read_og_ipa_supply_infrastructure", "Geography": "Oil and Gas IPA Supply Infrastructure", "Years": "All", "Source": "EPE"},
+        {"Function": "read_og_legal_pre_salt_polygon", "Geography": "Oil and Gas Legal Pre-Salt Polygon", "Years": "All", "Source": "EPE"},
+        {"Function": "read_exploration_production_environment", "Geography": "Exploration and Production Environment", "Years": "All", "Source": "EPE"},
+        {"Function": "read_sedimentary_basins", "Geography": "Sedimentary Basins", "Years": "All", "Source": "EPE"},
+        {"Function": "read_og_basement", "Geography": "Oil and Gas Basement", "Years": "All", "Source": "EPE"},
+        {"Function": "read_hydroelectric_feasibility_studies", "Geography": "Hydroelectric Feasibility Studies", "Years": "All", "Source": "EPE"},
+        {"Function": "read_hydroelectric_inventory_aai_studies", "Geography": "Hydroelectric Inventory and AAI Studies", "Years": "All", "Source": "EPE"},
+        {"Function": "read_ama_anemometric_towers", "Geography": "AMA Anemometric Towers", "Years": "All", "Source": "EPE"}
     ]
     
     # Create DataFrame

tunned_geobr/read_ama_anemometric_towers.py

@@ -0,0 +1,119 @@
+import geopandas as gpd
+import requests
+import zipfile
+import tempfile
+import os
+import warnings
+import shutil
+
+
+def read_ama_anemometric_towers(simplified=True, verbose=False):
+    """Download data for AMA Anemometric Towers in Brazil.
+
+    This function downloads, processes, and returns data for AMA Anemometric Towers
+    in Brazil as a geopandas GeoDataFrame.
+
+    Parameters
+    ----------
+    simplified : bool, optional
+        If True, returns a simplified version of the dataset with only essential columns.
+        If False, returns the complete dataset with all columns.
+        Default is True.
+    verbose : bool, optional
+        If True, prints detailed information about the data download and processing.
+        Default is False.
+
+    Returns
+    -------
+    geopandas.GeoDataFrame
+        A GeoDataFrame containing AMA Anemometric Towers data.
+
+    Examples
+    --------
+    >>> # Download AMA Anemometric Towers data
+    >>> df = read_ama_anemometric_towers()
+    >>> df.head()
+    """
+
+    url = "https://gisepeprd2.epe.gov.br/arcgis/rest/services/Download_Dados_Webmap_EPE/GPServer/Extract%20Data%20Task/execute?f=json&env%3AoutSR=102100&Layers_to_Clip=%5B%22Torres%20Anemom%C3%A9tricas%20do%20AMA%22%5D&Area_of_Interest=%7B%22geometryType%22%3A%22esriGeometryPolygon%22%2C%22features%22%3A%5B%7B%22geometry%22%3A%7B%22rings%22%3A%5B%5B%5B-9237395.881983705%2C-4650539.310904562%5D%2C%5B-9237395.881983705%2C1219824.4613954136%5D%2C%5B-2349502.3891517334%2C1219824.4613954136%5D%2C%5B-2349502.3891517334%2C-4650539.310904562%5D%2C%5B-9237395.881983705%2C-4650539.310904562%5D%5D%5D%2C%22spatialReference%22%3A%7B%22wkid%22%3A102100%7D%7D%7D%5D%2C%22sr%22%3A%7B%22wkid%22%3A102100%7D%7D&Feature_Format=Shapefile%20-%20SHP%20-%20.shp&Raster_Format=Tagged%20Image%20File%20Format%20-%20TIFF%20-%20.tif"
+
+    if verbose:
+        print("Downloading data...")
+
+    try:
+        response = requests.get(url)
+        response.raise_for_status()
+        response_json = response.json()
+
+        if "value" not in response_json or not response_json["value"]:
+            raise ValueError("No data found in the response")
+
+        download_url = response_json["value"]["itemUrl"]
+        
+        if verbose:
+            print(f"Download URL: {download_url}")
+            print("Downloading zip file...")
+
+        zip_response = requests.get(download_url)
+        zip_response.raise_for_status()
+
+        # Create a temporary directory to extract the files
+        with tempfile.TemporaryDirectory() as temp_dir:
+            zip_path = os.path.join(temp_dir, "data.zip")
+            
+            # Save the zip file
+            with open(zip_path, "wb") as f:
+                f.write(zip_response.content)
+            
+            if verbose:
+                print(f"Zip file saved to {zip_path}")
+                print("Extracting files...")
+            
+            # Extract the zip file
+            with zipfile.ZipFile(zip_path, "r") as zip_ref:
+                zip_ref.extractall(temp_dir)
+            
+            # Find the shapefile
+            shp_files = [f for f in os.listdir(temp_dir) if f.endswith(".shp")]
+            
+            if not shp_files:
+                raise FileNotFoundError("No shapefile found in the downloaded zip file")
+            
+            shp_path = os.path.join(temp_dir, shp_files[0])
+            
+            if verbose:
+                print(f"Reading shapefile from {shp_path}")
+            
+            # Read the shapefile
+            gdf = gpd.read_file(shp_path)
+            
+            # Convert to SIRGAS 2000 (EPSG:4674)
+            gdf = gdf.to_crs(epsg=4674)
+            
+            if simplified:
+                # Select only essential columns
+                if verbose:
+                    print("Simplifying the dataset...")
+                
+                # Identify the essential columns
+                essential_cols = ["geometry"]
+                
+                # Add any other essential columns that exist in the dataset
+                for col in ["NOME", "MUNICIPIO", "UF", "ALTURA", "SITUACAO"]:
+                    if col in gdf.columns:
+                        essential_cols.append(col)
+                
+                # Select only the essential columns
+                gdf = gdf[essential_cols]
+            
+            return gdf
+    
+    except requests.exceptions.RequestException as e:
+        warnings.warn(f"Error downloading data: {e}")
+        return None
+    except (ValueError, FileNotFoundError, zipfile.BadZipFile) as e:
+        warnings.warn(f"Error processing data: {e}")
+        return None
+    except Exception as e:
+        warnings.warn(f"Unexpected error: {e}")
+        return None

tunned_geobr/read_areas_under_contract.py

@@ -0,0 +1,129 @@
+import geopandas as gpd
+import requests
+import shutil
+import zipfile
+import tempfile
+import warnings
+import os
+from shapely.geometry.point import Point
+
+
+def read_areas_under_contract(simplified=False, verbose=False):
+    """Download data of oil and gas areas under contract in Brazil.
+
+    This function downloads and returns data of oil and gas areas under contract
+    (Áreas sob Contrato) in Brazil as a GeoPandas GeoDataFrame. The data comes 
+    from EPE (Energy Research Company).
+
+    Parameters
+    ----------
+    simplified : bool, optional
+        If True, returns a simplified version of the dataset with only the most
+        important columns. If False, returns the complete dataset. Default is False.
+    verbose : bool, optional
+        If True, displays detailed messages about the download and processing
+        steps. Default is False.
+
+    Returns
+    -------
+    gpd.GeoDataFrame
+        A GeoDataFrame containing data on oil and gas areas under contract in Brazil.
+
+    Raises
+    ------
+    Exception
+        If the download or processing of the data fails.
+
+    Example
+    -------
+    >>> from tunned_geobr import read_areas_under_contract
+    >>> 
+    >>> # Read the data
+    >>> contract_areas = read_areas_under_contract()
+    >>> 
+    >>> # Plot the data
+    >>> contract_areas.plot()
+    """
+    
+    if verbose:
+        print("Downloading data of oil and gas areas under contract in Brazil")
+    
+    # Define the URL for the API request
+    url = "https://gisepeprd2.epe.gov.br/arcgis/rest/services/Download_Dados_Webmap_EPE/GPServer/Extract%20Data%20Task/execute?f=json&env%3AoutSR=102100&Layers_to_Clip=%5B%22%C3%81reas%20sob%20Contrato%22%5D&Area_of_Interest=%7B%22geometryType%22%3A%22esriGeometryPolygon%22%2C%22features%22%3A%5B%7B%22geometry%22%3A%7B%22rings%22%3A%5B%5B%5B-9237395.881983705%2C-4650539.310904562%5D%2C%5B-9237395.881983705%2C1219824.4613954136%5D%2C%5B-2349502.3891517334%2C1219824.4613954136%5D%2C%5B-2349502.3891517334%2C-4650539.310904562%5D%2C%5B-9237395.881983705%2C-4650539.310904562%5D%5D%5D%2C%22spatialReference%22%3A%7B%22wkid%22%3A102100%7D%7D%7D%5D%2C%22sr%22%3A%7B%22wkid%22%3A102100%7D%7D&Feature_Format=Shapefile%20-%20SHP%20-%20.shp&Raster_Format=Tagged%20Image%20File%20Format%20-%20TIFF%20-%20.tif"
+    
+    try:
+        # Make the API request
+        response = requests.get(url)
+        response.raise_for_status()
+        
+        # Parse the JSON response
+        data = response.json()
+        
+        # Extract the URL for the zip file
+        if 'results' in data and len(data['results']) > 0 and 'value' in data['results'][0]:
+            download_url = data['results'][0]['value']['url']
+        else:
+            raise Exception("Failed to extract download URL from API response")
+        
+        # Create a temporary directory to store the downloaded files
+        with tempfile.TemporaryDirectory() as temp_dir:
+            # Download the zip file
+            zip_path = os.path.join(temp_dir, "areas_under_contract.zip")
+            if verbose:
+                print("Downloading zip file")
+            
+            response = requests.get(download_url, stream=True)
+            response.raise_for_status()
+            
+            with open(zip_path, 'wb') as f:
+                response.raw.decode_content = True
+                shutil.copyfileobj(response.raw, f)
+            
+            # Extract the zip file
+            if verbose:
+                print("Extracting files")
+            
+            with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+                zip_ref.extractall(temp_dir)
+            
+            # Find the shapefile in the extracted files
+            shp_files = [f for f in os.listdir(temp_dir) if f.endswith('.shp')]
+            
+            if not shp_files:
+                raise Exception("No shapefile found in the downloaded zip file")
+            
+            # Read the shapefile
+            if verbose:
+                print("Reading shapefile")
+            
+            shp_path = os.path.join(temp_dir, shp_files[0])
+            gdf = gpd.read_file(shp_path)
+            
+            # Convert to SIRGAS 2000 (EPSG:4674)
+            if verbose:
+                print("Converting to SIRGAS 2000 (EPSG:4674)")
+            
+            gdf = gdf.to_crs(epsg=4674)
+            
+            # Simplify the dataset if requested
+            if simplified:
+                if verbose:
+                    print("Simplifying the dataset")
+                
+                # Select only the most important columns
+                # Adjust these columns based on the actual data structure
+                cols_to_keep = ['NOME', 'OPERADOR', 'BACIA', 'CONTRATO', 'FASE', 'UF', 'geometry']
+                cols_available = [col for col in cols_to_keep if col in gdf.columns]
+                
+                if not cols_available:
+                    warnings.warn("None of the specified columns for simplification are available. Returning the full dataset.")
+                else:
+                    gdf = gdf[cols_available]
+            
+            if verbose:
+                print("Finished processing oil and gas areas under contract data")
+            
+            return gdf
+    
+    except Exception as e:
+        raise Exception(f"Failed to download or process oil and gas areas under contract data: {str(e)}")

tunned_geobr/read_biodiesel_plants.py

@@ -0,0 +1,128 @@
+import geopandas as gpd
+import requests
+import shutil
+import zipfile
+import tempfile
+import warnings
+import os
+from shapely.geometry.point import Point
+
+
+def read_biodiesel_plants(simplified=False, verbose=False):
+    """Download data of biodiesel production plants in Brazil.
+
+    This function downloads and returns data of biodiesel production plants
+    in Brazil as a GeoPandas GeoDataFrame. The data comes from EPE (Energy Research Company).
+
+    Parameters
+    ----------
+    simplified : bool, optional
+        If True, returns a simplified version of the dataset with only the most
+        important columns. If False, returns the complete dataset. Default is False.
+    verbose : bool, optional
+        If True, displays detailed messages about the download and processing
+        steps. Default is False.
+
+    Returns
+    -------
+    gpd.GeoDataFrame
+        A GeoDataFrame containing data on biodiesel production plants in Brazil.
+
+    Raises
+    ------
+    Exception
+        If the download or processing of the data fails.
+
+    Example
+    -------
+    >>> from tunned_geobr import read_biodiesel_plants
+    >>> 
+    >>> # Read the data
+    >>> biodiesel_plants = read_biodiesel_plants()
+    >>> 
+    >>> # Plot the data
+    >>> biodiesel_plants.plot()
+    """
+    
+    if verbose:
+        print("Downloading data of biodiesel production plants in Brazil")
+    
+    # Define the URL for the API request
+    url = "https://gisepeprd2.epe.gov.br/arcgis/rest/services/Download_Dados_Webmap_EPE/GPServer/Extract%20Data%20Task/execute?f=json&env%3AoutSR=102100&Layers_to_Clip=%5B%22Plantas%20de%20biodiesel%22%5D&Area_of_Interest=%7B%22geometryType%22%3A%22esriGeometryPolygon%22%2C%22features%22%3A%5B%7B%22geometry%22%3A%7B%22rings%22%3A%5B%5B%5B-8655251.47456396%2C-4787514.465591563%5D%2C%5B-8655251.47456396%2C1229608.401015912%5D%2C%5B-3508899.2341809804%2C1229608.401015912%5D%2C%5B-3508899.2341809804%2C-4787514.465591563%5D%2C%5B-8655251.47456396%2C-4787514.465591563%5D%5D%5D%2C%22spatialReference%22%3A%7B%22wkid%22%3A102100%7D%7D%7D%5D%2C%22sr%22%3A%7B%22wkid%22%3A102100%7D%7D&Feature_Format=Shapefile%20-%20SHP%20-%20.shp&Raster_Format=Tagged%20Image%20File%20Format%20-%20TIFF%20-%20.tif"
+    
+    try:
+        # Make the API request
+        response = requests.get(url)
+        response.raise_for_status()
+        
+        # Parse the JSON response
+        data = response.json()
+        
+        # Extract the URL for the zip file
+        if 'results' in data and len(data['results']) > 0 and 'value' in data['results'][0]:
+            download_url = data['results'][0]['value']['url']
+        else:
+            raise Exception("Failed to extract download URL from API response")
+        
+        # Create a temporary directory to store the downloaded files
+        with tempfile.TemporaryDirectory() as temp_dir:
+            # Download the zip file
+            zip_path = os.path.join(temp_dir, "biodiesel_plants.zip")
+            if verbose:
+                print("Downloading zip file")
+            
+            response = requests.get(download_url, stream=True)
+            response.raise_for_status()
+            
+            with open(zip_path, 'wb') as f:
+                response.raw.decode_content = True
+                shutil.copyfileobj(response.raw, f)
+            
+            # Extract the zip file
+            if verbose:
+                print("Extracting files")
+            
+            with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+                zip_ref.extractall(temp_dir)
+            
+            # Find the shapefile in the extracted files
+            shp_files = [f for f in os.listdir(temp_dir) if f.endswith('.shp')]
+            
+            if not shp_files:
+                raise Exception("No shapefile found in the downloaded zip file")
+            
+            # Read the shapefile
+            if verbose:
+                print("Reading shapefile")
+            
+            shp_path = os.path.join(temp_dir, shp_files[0])
+            gdf = gpd.read_file(shp_path)
+            
+            # Convert to SIRGAS 2000 (EPSG:4674)
+            if verbose:
+                print("Converting to SIRGAS 2000 (EPSG:4674)")
+            
+            gdf = gdf.to_crs(epsg=4674)
+            
+            # Simplify the dataset if requested
+            if simplified:
+                if verbose:
+                    print("Simplifying the dataset")
+                
+                # Select only the most important columns
+                # Adjust these columns based on the actual data structure
+                cols_to_keep = ['NOME', 'CAPACIDADE', 'PROPRIETAR', 'UF', 'MUNICIPIO', 'geometry']
+                cols_available = [col for col in cols_to_keep if col in gdf.columns]
+                
+                if not cols_available:
+                    warnings.warn("None of the specified columns for simplification are available. Returning the full dataset.")
+                else:
+                    gdf = gdf[cols_available]
+            
+            if verbose:
+                print("Finished processing biodiesel production plants data")
+            
+            return gdf
+    
+    except Exception as e:
+        raise Exception(f"Failed to download or process biodiesel production plants data: {str(e)}")