import solar
from gradient import *
from shadows import *
import numpy as np

import matplotlib.pyplot as plt # plt 用于顯示圖片
import matplotlib.image as mpimg # mpimg 用于讀取圖片


# dem
import srtm
if __name__ == '__main__':

  
  '''
  # 另一種從網(wǎng)上直接下載DEM數(shù)據(jù)
  geo_elevation_data = srtm.get_data()
  image = geo_elevation_data.get_image((500, 500), (45, 46), (13, 14), 300)
  # the image s a standard PIL object, you can save or show it:
  image.show()

  image = np.asarray(image)

  print(image.shape)

  '''

  # 讀入高程信息
  filename = 'dempyrenees.asc'
  dem = np.loadtxt(filename,skiprows=6,delimiter=' ')

  # 高程信息的維度
  # print(dem)
  print(dem.shape)

  # 定義一個(gè)光線向量
  # 第一個(gè)表示和豎直方向的夾角，第二個(gè)表示由東向西照射
  sv = normal_vector(45, 270)

  # 生成陰影
  shadow = project_shadows(dem=dem,sun_vector=sv,dx=30)
  print(shadow[1,:])
  print(shadow.shape)

  # 顯示dem和shadow圖像
  plt.figure()
  plt.subplot(1,2,1)
  plt.imshow(dem,cmap='gray') # 顯示灰度圖像
  plt.axis('off') # 不顯示坐標(biāo)軸

  plt.subplot(1,2,2)
  plt.imshow(shadow,cmap='gray')
  plt.axis('off')

  plt.show()