HostAssociations.h

//===-- Lower/HostAssociations.h --------------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
 
#ifndef FORTRAN_LOWER_HOSTASSOCIATIONS_H
#define FORTRAN_LOWER_HOSTASSOCIATIONS_H
 
#include "mlir/IR/Types.h"
#include "llvm/ADT/SetVector.h"
 
namespace Fortran {
namespace semantics {
class Symbol;
class Scope;
} // namespace semantics
 
namespace lower {
class AbstractConverter;
class SymMap;

class HostAssociations {
public:
  bool empty() const { return tupleSymbols.empty() && globalSymbols.empty(); }

  bool hasTupleAssociations() const { return !tupleSymbols.empty(); }

  void addSymbolsToBind(
      const llvm::SetVector<const Fortran::semantics::Symbol *> &symbols,
      const Fortran::semantics::Scope &hostScope);

  void hostProcedureBindings(AbstractConverter &converter, SymMap &symMap);

  void internalProcedureBindings(AbstractConverter &converter, SymMap &symMap);

  mlir::Type getArgumentType(AbstractConverter &convert);

  bool isAssociated(const Fortran::semantics::Symbol &symbol) const {
    return tupleSymbols.contains(&symbol) || globalSymbols.contains(&symbol);
  }
 
private:
  llvm::SetVector<const Fortran::semantics::Symbol *> tupleSymbols;

  llvm::SetVector<const Fortran::semantics::Symbol *> globalSymbols;

  mlir::Type argType;

  const Fortran::semantics::Scope *hostScope;
};
} // namespace lower
} // namespace Fortran
 
#endif // FORTRAN_LOWER_HOSTASSOCIATIONS_H