import_existing_trees.py

import geopandas as gpd
from tree import Tree, Forest

def get_existing_forest(shp_input):
    print(f"Importing {shp_input}")
    df = gpd.read_file(shp_input)
    trees = []
    for tree_row in df.itertuples():
        point = tree_row.geometry
        trees.append(Tree(point.x, point.y,
                          description=tree_row.Bezeichnun,
                          radius=tree_row.Kroneradi,
                          type=tree_row.Baumart
                          ))
    return Forest(trees)

if __name__ == "__main__":
    print(repr(get_existing_forest('existing_trees/Trees_ideal_2_20240227.shp')))