extract_prec.f90

program extract_prec_with_coords
    !Author: Rodrigues, L.F [LFR]
    !Data: 24Jun2025
    !Função:
    !Este programa extrai dos arquivos FRNaaaammddhh-FRNaaaammddhh.nc os dados de 
    !precipitação em uma área delimitada pelo arquivo de contorno e gera um CSV
    !aLém de um gráfico/figura gerado a partir de um script gnuplot (que ele produz)
    !
    !Para compilar, faça:
    !
    !  gfortran -O3 -o extract_prec mod_out.f90 mod_netcdf.f90  mod_polygon_point_chek.f90 \
    !  extract_prec.f90 -I/home/lufla/apps/include -L/home/lufla/apps/lib -lnetcdff -lnetcdf |
    !  -ffree-form -ffree-line-length-none
    !
    ! OBS: Coloque o diretório com as bibliotecas NetCDF no LD_LIBRARY_PATH caso não estejam
    !
    ! Para rodar:
    !
    ! ~/home/dir_com_o_programa/extract_prec 2025061500 2025063000 totprec corumba.txt 0 0
    !
    ! Obs:
    !
    ! O arquivo de entrada em NetCDF pode ser gerado à partir de todas as saídas NetCDF do modelo BRAMS
    ! que podem ser concatenadas por um comando CDO (exemplo):
    !
    ! cdo mergetime $(ls -v <dir_com_as_saidas_netCDF/FRN*.nc) FRN2025061500-FRN2025063000.nc
    !
    ! Para rodar o código é necessário um arquivo com o contorno da área de contribuição com a lista de longitudes
    ! e latitudes que a compõe separadas por espaços, tendo o último ponto coincidente com o primeiro, ex:
    !
    ! -53.5079174 -33.3531357
    ! -53.5217374 -33.2651499
    ! -53.5215398 -33.2588114
    ! -53.5182526 -33.1533523
    ! -53.4445171 -33.0530946
    ! -53.4444093 -33.0529481
    ! -53.2456220 -32.9354243
    ! ...
    ! -53.5079174 -33.3531357
    !
    ! O nome do arquivo deve ter a extensão ".txt", por exemplo, corumba.txt.
    !
    ! Exemplo de comando: 
    !
    ! ~/dir_onde_esta_o_executavel/extract_prec 2025061500 2025063000 totprec corumba 0 0
    !
    use netcdf
    use mod_polygon_point_check
    use mod_netcdf
    use mod_out

    implicit none
    
    ! Variáveis NetCDF
    integer :: ncid_in, ncid_out
    integer :: varid_in, varid_out, latid, lonid
    integer :: dimids_in(NF90_MAX_VAR_DIMS)
    integer :: ndims_in, dimids_out(NF90_MAX_VAR_DIMS)
    character(len=NF90_MAX_NAME) :: dim_name
    integer :: dim_size, dimid
    integer :: i, j, t, status
    
    ! Informações das variáveis
    integer :: xtype_in, natts_in
    character(len=NF90_MAX_NAME) :: var_name_in
    
    type(point), allocatable :: polygon(:)
    integer :: ierr, icount
    logical :: is_inside
    real :: sum_prec, sum_tot

    character(len=256) :: arg1

    character(len=256) :: input_file
    character(len=256) :: output_dir
    character(len=256) :: var_name
    character(len=256) :: filename
    character(len=256) :: contour_dir
    character(len=256) :: filecontour  
    character(len=256) :: mapa
    character(len=256) :: write_all
    character(len=256) :: csv_file 
    character(len=256) :: png_file 
    character(12) :: dataini,datafin

    integer :: contagem_inicio, contagem_fim, taxa_contagem, contagem_max
    
    ! Obtém a taxa do relógio do sistema
    call system_clock(count_rate=taxa_contagem)
    call system_clock(count_max=contagem_max)
    
    ! Inicia a contagem
    call system_clock(contagem_inicio)

    ! Verifica se há argumentos suficientes
    if (command_argument_count() /= 8) then
        print *, "Uso: ./extract_prec data_inicial data_final variável pasta_contornos nome_contorno pasta_saida cria_mapa plota_tudo"
        print *, "Ex:  ./extract_prec 2025061500 2025063000 totprec ./tables/contours corumba ./ 0 0"
        stop
    end if

    ! Lê os argumentos da linha de comando
    do i = 1, 8
        call get_command_argument(i, arg1)  ! Só usamos arg1 temporariamente
        select case(i)
            case(1); dataini = trim(arg1)
            case(2); datafin = trim(arg1)
            case(3); var_name = trim(arg1)
            case(4); contour_dir = trim(arg1)
            case(5); filecontour = trim(arg1)
            case(6); output_dir = trim(arg1)
            case(7); mapa = trim(arg1)
            case(8); write_all = trim(arg1)
        end select
    end do
    !FRN2025061500-FRN2025063000.nc
    input_file = "FRN"//dataini//"-FRN"//datafin//".nc"
    filename = trim(contour_dir)//'/'//trim(filecontour)
    csv_file = trim(output_dir)//"/"//trim(filecontour(1:len(trim(filecontour))-4)//"-"//dataini//"-"//datafin//".csv")
    png_file = trim(output_dir)//"/"//trim(filecontour(1:len(trim(filecontour))-4)//"-"//dataini//"-"//datafin//".png")
    
    ! Lê o polígono do arquivo
    write(*,*) 'Lendo polígono de entrada...'//trim(filename)
    call read_polygon_from_file(trim(filename), polygon, ierr)
    if (ierr /= 0) then
        write(*,*) 'Erro ao ler o arquivo do polígono!'
        if (ierr == -1) write(*,*) 'O polígono deve ter pelo menos 3 pontos'
        stop
    end if

    ! Inicialização NetCDF
    write(*,*) 'Iniciando extração da variável totprec e leitura de coordenadas...'
    call process_netCDF_in(input_file,var_name)

    write(*,*) 'Checando pontos dentro do polígono...'
    call check_inside(lon_data,lat_data, polygon, size(totprec_data,1), size(totprec_data,2))

    write(*,*) 'Escrevendo CSV...'
    call write_csv(csv_file,totprec_data,totprec_mask_data,points_inside,mapa,dataini,datafin)
    
!    write(*,*) 'Escrevendo saída NetCDF...'
!    call process_netCDF_out(output_file,var_name, write_all)

    write(*,*) 'Escrevendo script gnuplot...'
    call write_script(csv_file, png_file, dataini,datafin,filecontour)

    call system_clock(contagem_fim)

    write(*,*) ''
    write(*,*) 'Processamento concluído com sucesso!'
     ! Calcula e exibe o tempo decorrido
    if (contagem_fim < contagem_inicio) then
        write(*,*) "Tempo decorrido: ", (contagem_max - contagem_inicio + contagem_fim) / real(taxa_contagem), " segundos"
    else
        write(*,*) "Tempo decorrido: ", (contagem_fim - contagem_inicio) / real(taxa_contagem), " segundos"
    end if
    
end program extract_prec_with_coords