Skip to content

Commit

Permalink
Permiere relecture equipe processus
Browse files Browse the repository at this point in the history
  • Loading branch information
@leavauchier
leavauchier committed Jul 2, 2024
1 parent ce329b6 commit 0b49b82
Showing 1 changed file with 89 additions and 22 deletions.
111 changes: 89 additions & 22 deletions scripts/mark_points_to_use_for_digital_models_with_new_dimension.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,14 @@ def main(input_las, output_las, dsm_dimension, dtm_dimension, output_dsm, output
pipeline = pdal.Pipeline() | pdal.Reader.las(input_las)

# 0 - ajout de dimensions temporaires et de sortie
added_dimensions = [dtm_dimension, dsm_dimension, "PT_VEG_DSM", "PT_ON_BRIDGE"]
added_dimensions = [
dtm_dimension,
dsm_dimension,
"PT_VEG_DSM",
"PT_ON_BRIDGE",
"PT_ON_BUILDING",
"PT_ON_VEGET",
]
pipeline |= pdal.Filter.ferry(dimensions="=>" + ", =>".join(added_dimensions))

# 1 - recherche des points max de végétation (4,5) sur une grille régulière, avec prise en
Expand All @@ -67,14 +74,44 @@ def main(input_las, output_las, dsm_dimension, dtm_dimension, output_dsm, output
condition_ref=macro.build_condition("Classification", [4, 5]),
condition_out="PT_VEG_DSM=1",
)
pipeline = macro.add_radius_assign(
pipeline,
1,
False,
condition_src=macro.build_condition("Classification", [6, 17]),
condition_ref=macro.build_condition("Classification", [4, 5]),
condition_out="PT_ON_VEGET=1",
max2d_above=0, # ne pas prendre les points qui sont au dessus des points pont (condition_ref)
max2d_below=900, # prendre tous les points qui sont en dessous des points pont (condition_ref)
)
pipeline = macro.add_radius_assign(
pipeline,
1,
False,
condition_src="PT_VEG_DSM==1 && Classification==2",
condition_ref="Classification==2",
condition_ref="Classification==2 && PT_VEG_DSM==0",
condition_out="PT_VEG_DSM=0",
)
pipeline = macro.add_radius_assign(
pipeline,
1,
False,
condition_src="PT_ON_VEGET==1 && Classification==6",
condition_ref="Classification==6 && PT_ON_VEGET==0",
condition_out="PT_ON_VEGET=0",
max2d_above=0.5, # ne pas prendre les points qui sont au dessus des points pont (condition_ref)
max2d_below=0.5, # prendre tous les points qui sont en dessous des points pont (condition_ref)
)
pipeline = macro.add_radius_assign(
pipeline,
1,
False,
condition_src="PT_ON_VEGET==1 && Classification==17",
condition_ref="Classification==17 && PT_ON_VEGET==0",
condition_out="PT_ON_VEGET=0",
max2d_above=0.5, # ne pas prendre les points qui sont au dessus des points pont (condition_ref)
max2d_below=0.5, # prendre tous les points qui sont en dessous des points pont (condition_ref)
)

# selection des points de veget basse proche de la veget haute
pipeline = macro.add_radius_assign(
Expand All @@ -95,7 +132,7 @@ def main(input_las, output_las, dsm_dimension, dtm_dimension, output_dsm, output
resolution=0.75, output_dimension=dsm_dimension, output_type="max", where="PT_VEG_DSM==1"
)

# 2 - sélection des points pour DTM et DSM
# 2 - sélection des points pour DTM et DSM#

# selection de points DTM (max) sur une grille régulière
# TODO: remplacer par GridDecimation une fois le correctif mergé dans PDAL
Expand All @@ -107,7 +144,7 @@ def main(input_las, output_las, dsm_dimension, dtm_dimension, output_dsm, output
output_dimension=dtm_dimension,
output_type="max",
where="Classification==2",
)
) #

# selection de points DSM (max) sur une grille régulière
# TODO: remplacer par GridDecimation une fois le correctif mergé dans PDAL
Expand All @@ -123,10 +160,10 @@ def main(input_las, output_las, dsm_dimension, dtm_dimension, output_dsm, output
resolution=0.5,
output_dimension=dsm_dimension,
output_type="max",
where="("
where="(PT_ON_VEGET==0 && ("
+ macro.build_condition("Classification", [6, 9, 17, 64])
+ f") || {dsm_dimension}==1",
)
+ f") || {dsm_dimension}==1)",
) #

# assigne des points sol sélectionnés : les points proches de la végétation, des ponts, de l'eau, 64
pipeline = macro.add_radius_assign(
Expand All @@ -135,45 +172,75 @@ def main(input_las, output_las, dsm_dimension, dtm_dimension, output_dsm, output
False,
condition_src=f"{dtm_dimension}==1",
condition_ref=macro.build_condition("Classification", [4, 5, 6, 9, 17, 64]),
condition_out=f"{dsm_dimension}=0",
)
# Test proximité batiment
pipeline = macro.add_radius_assign(
pipeline,
1.25,
False,
condition_src="Classification==2 && PT_VEG_DSM==0",
condition_ref="Classification==6",
condition_out="PT_ON_BUILDING=1",
)
# BUFFER INVERSE Se mettre
pipeline = macro.add_radius_assign(
pipeline,
1,
False,
condition_src=f"Classification==2 && {dsm_dimension}==0 && PT_ON_BUILDING==1 && {dtm_dimension}==1",
condition_ref="Classification==2 && PT_ON_BUILDING==0 && PT_VEG_DSM==0",
condition_out=f"{dsm_dimension}=1",
)

# 3 - gestion des ponts
# bouche trou : on filtre les points (2,3,4,5,9) au milieu du pont en les mettant à PT_ON_BRIDGE=1

# bouche trou : on filtre les points (2,3,4,5,9) au milieu du pont en les mettant à PT_ON_BRIDGE=1##
pipeline = macro.add_radius_assign(
pipeline,
1.5,
False,
condition_src=macro.build_condition("Classification", [2, 3, 4, 5, 9]),
condition_src=macro.build_condition("Classification", [2, 3, 4, 5, 6, 9]),
condition_ref="Classification==17",
condition_out="PT_ON_BRIDGE=1",
max2d_above=0, # ne pas prendre les points qui sont au dessus des points pont (condition_ref)
max2d_below=-1, # prendre tous les points qui sont en dessous des points pont (condition_ref)
max2d_above=0, # ne pas prendre les points qui sont au dessus des points pont (condition_ref)
max2d_below=900, # prendre tous les points qui sont en dessous des points pont (condition_ref)
)
pipeline = macro.add_radius_assign(
pipeline,
1.5,
1.25,
False,
# condition_ref=macro.build_condition("Classification", [2, 3, 4, 5]),
condition_src="PT_ON_BRIDGE==1",
condition_ref=macro.build_condition("Classification", [2, 3, 4, 5]),
condition_ref="PT_ON_BRIDGE==0 && ( "
+ macro.build_condition("Classification", [2, 3, 4, 5, 6, 9])
+ " )",
condition_out="PT_ON_BRIDGE=0",
max2d_above=0.5, # ne pas prendre les points qui sont au dessus des points pont (condition_ref)
max2d_below=0.5, # prendre tous les points qui sont en dessous des points pont (condition_ref)
)
# pipeline |= pdal.Filter.assign(value=[f"{dsm_dimension}=0 WHERE (PT_ON_BRIDGE==1 && NOT(Classification==17))"])
pipeline |= pdal.Filter.assign(
value=[f"{dsm_dimension}=0 WHERE PT_ON_BRIDGE==1"]
# value=["dsm_marker=0 WHERE (PT_ON_BRIDGE==1 AND ( " + macro.build_condition("Classification", [2,3,4,5,6,9]) + " ))"]
)
pipeline |= pdal.Filter.assign(value=[f"{dsm_dimension}=0 WHERE PT_ON_BRIDGE==1"])

# 4 - point pour DTM servent au DSM également
pipeline |= pdal.Filter.assign(value=[f"{dsm_dimension}=1 WHERE {dtm_dimension}==1"])

# HOMOGENEISER L UTILISATION DE PTVEGDSM POUR LES POINT SOL SOUS VEGET AVEC PT_ON_VEGET
pipeline |= pdal.Filter.assign(
value=[
f"{dsm_dimension}=1 WHERE ({dtm_dimension}==1 && PT_VEG_DSM==0 && PT_ON_BRIDGE==0 && PT_ON_BUILDING==0 )"
]
)
# ERREUR EN 4!###############################################################################################!
# 5 - export du nuage et des DSM
# TODO: n'ajouter que les dimensions de sortie utiles !
# TODO: n'ajouter que les dimensions de sortie utiles !#

pipeline |= pdal.Writer.las(extra_dims="all", forward="all", filename=output_las)

if output_dtm:
pipeline |= pdal.Writer.gdal(
gdaldriver="GTiff",
output_type="max",
resolution=2.0,
resolution=0.5,
filename=output_dtm,
where=f"{dtm_dimension}==1",
)
Expand All @@ -182,7 +249,7 @@ def main(input_las, output_las, dsm_dimension, dtm_dimension, output_dsm, output
pipeline |= pdal.Writer.gdal(
gdaldriver="GTiff",
output_type="max",
resolution=2.0,
resolution=0.5,
filename=output_dsm,
where=f"{dsm_dimension}==1",
)
Expand All @@ -192,4 +259,4 @@ def main(input_las, output_las, dsm_dimension, dtm_dimension, output_dsm, output

if __name__ == "__main__":
args = parse_args()
main(**args)
main(**vars(args))

0 comments on commit 0b49b82

Please sign in to comment.