如何用sed/awk/vim替换MPI函数调用中的变量并添加赋值语句

针对Fortran代码中MPI_SEND、MPI_RECV等调用的参数替换需求（将第2个参数替换为cnt、第5个参数替换为tag，并添加与call行对齐的赋值语句），可以用awk脚本实现，它能很好适配不同缩进场景和Fortran续行（&）格式，比sed更适合处理这类结构化多行代码。

实现脚本 #

#!/usr/bin/awk -f

# 捕获MPI调用起始行，记录缩进和初始参数
/^[[:space:]]*call mpi_(send|recv)/ {
    indent = substr($0, 1, index($0, "call") - 1)
    split(substr($0, index($0, "(")+1), args, /[[:space:]]*,[[:space:]]*/)
    arg2 = args[2]
    orig_line = $0
    next
}

# 处理MPI调用续行，完成替换和赋值输出
/^[[:space:]]*&[[:space:]]*/ {
    split(substr($0, index($0, "&")+1), args, /[[:space:]]*,[[:space:]]*/)
    arg5 = args[2]
    
    # 输出对齐的赋值语句
    print indent "cnt = " arg2
    print indent "tag = " arg5
    
    # 输出修改后的起始行和续行
    print gensub(/,[[:space:]]*[^,]+[[:space:]]*,/, ", cnt,", 1, orig_line)
    print gensub(/,[[:space:]]*[^,]+[[:space:]]*,/, ", tag,", 1)
    
    orig_line = ""
    next
}

# 非MPI调用行直接输出
orig_line == "" { print $0 }

使用步骤 #

1. 将上述脚本保存为mpi_replace.awk
2. 赋予脚本执行权限：chmod +x mpi_replace.awk
3. 处理目标Fortran文件：./mpi_replace.awk input.f90 > output.f90

示例验证 #

原代码 #

call mpi_send( v(1), nx, MPI_DOUBLE_PRECISION,
          &               prev, myid-1, comm, ierr )

处理后结果 #

cnt = nx
           tag = myid-1
           call mpi_send( v(1), cnt, MPI_DOUBLE_PRECISION,
          &               prev, tag, comm, ierr )

脚本会自动识别call行的缩进量，确保赋值语句与调用行对齐，同时正确解析Fortran续行格式，完成参数替换。

内容的提问来源于stack exchange，提问作者fghoussen