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/Location.h"
#include "mlir/IR/Types.h"
#include "mlir/IR/Value.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