# 必要なライブラリをインポート
import streamlit as st
from snowflake.snowpark.context import get_active_session
import json
#geo版のPandas
import geopandas as gpd
#geoデータの補正関数
from shapely.geometry import shape
from shapely.validation import make_valid
#geoデータの表示
import pydeck as pdk

# Snowflakeからデータを取得する関数をキャッシュ化
@st.cache_resource
def create_source():
    session = get_active_session()
    # Snowflakeからデータをクエリして取得
    df = session.sql('''
        SELECT
            PREF_NAME
            ,PREF_CODE
            ,POPULATION
            ,ST_ASGEOJSON(POLYGON_JP_PREF_LIGHT) AS GEO_JSON //GeographyからGeojsonに変換
        FROM
            PREPPER_OPEN_DATA_BANK__JAPANESE_PREFECTURE_DATA.E_PODB.E_PR_FD20
    ''').to_pandas()
    return df

# データの取得
df = create_source()

#GeojsonからGeometryに変換
def trans_shape(x):
    geometry = shape(json.loads(x))
    if geometry.is_valid:
        return geometry
    else:
        return make_valid(geometry)

# GeoDataFrameに都道府県名・人口・ポリゴン（GeojsonからGeometryに変換したもの）を格納
# COLORカラム：人口をもとにしたポリゴンの色
geometry = [trans_shape(row) for row in df['GEO_JSON'] ]
gdf = gpd.GeoDataFrame(df[['PREF_NAME', 'PREF_CODE', 'POPULATION']].reset_index(), geometry=geometry)
gdf["color"] = [[i*255, (1-i)*255, 0, 140] for i in gdf['POPULATION']/gdf['POPULATION'].max()]

# ポリゴン地図の可視化
layer = pdk.Layer(
    "GeoJsonLayer",
    data=gdf, # GeoDataFrameを指定
    get_polygon="geometry.coordinates", # Geometry辞書のgeometry.coodinatesの中に地理データが入っているので指定
    get_fill_color='color',  # ポリゴン色はCOLORカラムを指定
)

fig = pdk.Deck(
        layers=[layer],
        map_style="mapbox://styles/mapbox/light-v9", # ベースマップを指定
        initial_view_state=pdk.ViewState( # マップの初期表示の拡大度、中心を指定
            latitude=38,
            longitude=138,
            zoom=3.8,
        )
)

# 地図を表示
st.pydeck_chart(fig)

//GeographyからGeojsonに変換
ST_ASGEOJSON(POLYGON_JP_PREF_LIGHT) AS GEO_JSON

#GeojsonからGeometryに変換
def trans_shape(x):
    geometry = shape(json.loads(x))
    if geometry.is_valid: #ポリゴンが使用できるか？
        return geometry
    else:
        return make_valid(geometry) #ポリゴン補正

geometry = [trans_shape(row) for row in df['GEO_JSON'] ] #47都道府県のポリゴンリストを作成

geometry = [trans_shape(row) for row in df['GEO_JSON'] ] #47都道府県のポリゴンリスト

#GeoDataFrameの作成
gdf = gpd.GeoDataFrame(df[['PREF_NAME', 'PREF_CODE', 'POPULATION']].reset_index(), geometry=geometry)

# ポリゴン地図の可視化
layer = pdk.Layer(
    "GeoJsonLayer",
    data=gdf, # GeoDataFrameを指定
    get_polygon="geometry.coordinates", # Geometry辞書のgeometry.coodinatesの中に地理データが入っているので指定
    get_fill_color='color',  # ポリゴン色はCOLORカラムを指定
)

fig = pdk.Deck(
        layers=[layer], #表示する地図レイヤーを指定
        map_style="mapbox://styles/mapbox/light-v9", # ベースマップを指定
        initial_view_state=pdk.ViewState( # マップの初期表示の拡大度、中心を指定
            latitude=38,
            longitude=138,
            zoom=3.8,
        )
)

# タイトルを表示
st.title('ポリゴン地図を表示できたよ～')
# 地図を表示
st.pydeck_chart(fig)