import_existing_trees.py 346 Bytes
Newer Older
1
import geopandas as gpd
2
from tree import Tree, Forest
3

Eric Duminil's avatar
Eric Duminil committed
4
def get_existing_forest(shp_input):
5
    print(f"Importing {shp_input}")
6
    df = gpd.read_file(shp_input)
7
8
9
10
    trees = [Tree(p.x, p.y) for p in df.geometry]
    return Forest(trees)

if __name__ == "__main__":
Eric Duminil's avatar
Eric Duminil committed
11
    print(repr(get_existing_forest('existing_trees/Trees_ideal_2_20240227.shp')))