Added error checking for Tao python floor_plan command. (#1063)

bmad-sim · Jul 15, 2024 · 98fcb63 · 98fcb63
1 parent fe42daf
commit 98fcb63
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tao/code/tao_python_cmd.f90 b/tao/code/tao_python_cmd.f90
@@ -4268,6 +4268,10 @@ subroutine tao_python_cmd (input_str)
   u => tao_pointer_to_universe(g%ix_universe, .true.)
   lat => u%model%lat
 
+  if (.not. allocated(g%floor_list)) then
+    call invalid ('Floor plan drawing not yet setup for this graph.')
+  endif
+
   do i = 1, size(g%floor_list)
     ele => pointer_to_ele(lat, g%floor_list(i)%ele_loc)
     ashape = g%floor_list(i)%shape