Fortran中如何判断数组内多态变量是否已分配为扩展类型？

错误原因解析 #

你遇到的编译错误核心在于：ALLOCATED内在函数仅接受可分配变量作为参数，而foo(1)是数组foo的元素——该元素本身并非可分配变量（数组foo整体是可分配的，但元素不是），因此无法用ALLOCATED检查其状态。同时，原代码中的ALLOCATE(Type1::foo(1))也是非法操作，因为非可分配的数组元素无法单独指定动态类型进行分配。

解决方案 #

方案一：使用包装类型实现元素级可分配 #

若需要单独控制每个数组元素的分配状态（可分配/未分配），可以声明一个包装类型，将可分配的多态对象作为其成员，再创建该包装类型的数组：

MODULE test_mod

TYPE BaseType
ENDTYPE BaseType

TYPE, EXTENDS(BaseType) :: Type1
ENDTYPE Type1 

TYPE, EXTENDS(BaseType) :: Type2
ENDTYPE Type2

! 包装类型：包含可分配的多态对象
TYPE :: PolymorphicElement
  CLASS(BaseType), ALLOCATABLE :: obj
END TYPE PolymorphicElement

! 可分配数组，元素为包装类型
TYPE(PolymorphicElement), ALLOCATABLE :: foo(:)

END MODULE test_mod

PROGRAM TEST
USE test_mod

! 分配数组空间
ALLOCATE(foo(2))

! 单独分配第一个元素的多态对象为Type1类型
ALLOCATE(Type1::foo(1)%obj)

! 检查元素的分配状态
IF (ALLOCATED(foo(1)%obj)) PRINT*, 'allocated foo ',1
IF (ALLOCATED(foo(2)%obj)) PRINT*, 'allocated foo ',2

END PROGRAM TEST

方案二：检查多态数组元素的动态类型 #

若数组元素始终处于已分配状态，仅需区分其动态类型是基类还是派生类，可使用EXTENDS_TYPE_OF函数或SELECT TYPE构造进行检查：

MODULE test_mod

TYPE BaseType
ENDTYPE BaseType

TYPE, EXTENDS(BaseType) :: Type1
ENDTYPE Type1 

TYPE, EXTENDS(BaseType) :: Type2
ENDTYPE Type2

! 可分配多态数组（元素不可单独分配）
CLASS(BaseType), ALLOCATABLE :: foo(:)

END MODULE test_mod

PROGRAM TEST
USE test_mod

TYPE(Type1) :: t1

! 分配数组，元素默认动态类型为BaseType
ALLOCATE(foo(2))

! 将派生类型对象赋值给数组元素，改变其动态类型
foo(1) = t1

! 用EXTENDS_TYPE_OF检查类型
IF (EXTENDS_TYPE_OF(foo(1), t1)) PRINT*, 'foo(1) is Type1'

! 用SELECT TYPE进行类型分支处理
SELECT TYPE(obj => foo(1))
TYPE IS(Type1)
  PRINT*, 'foo(1) is Type1'
TYPE IS(BaseType)
  PRINT*, 'foo(1) is BaseType'
END SELECT

SELECT TYPE(obj => foo(2))
TYPE IS(Type1)
  PRINT*, 'foo(2) is Type1'
TYPE IS(BaseType)
  PRINT*, 'foo(2) is BaseType'
END SELECT

END PROGRAM TEST

总结 #

• 若需单独控制元素的分配状态，优先选择包装类型数组的方案；
• 若仅需区分元素的动态类型，使用动态类型检查的方案即可。

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