qcor_syntax_handler.hpp

#pragma once
 
#include "clang/Parse/Parser.h"
 
using namespace clang;
 
namespace qcor {
extern std::string qpu_name;
extern int shots;
 
// Add this for internal development, specifically JIT tests
// where I don't want AddPredefines to add qcor.hpp. For example
// where I want to compile a simple c++ code with no dependencies, 
// I don't want to include qcor.hpp bc it makes it much slower.
namespace __internal__developer__flags__ { extern bool add_predefines;}
 
class QCORSyntaxHandler : public SyntaxHandler {
public:
  QCORSyntaxHandler() : SyntaxHandler("qcor") {}
  void GetReplacement(Preprocessor &PP, Declarator &D, CachedTokens &Toks,
                      llvm::raw_string_ostream &OS) override;
  
  // For use with qcor jit
  void GetReplacement(Preprocessor &PP, std::string &kernel_name,
                      std::vector<std::string> program_arg_types,
                      std::vector<std::string> program_parameters,
                      std::vector<std::string> bufferNames, CachedTokens &Toks,
                      llvm::raw_string_ostream &OS, bool add_het_map_ctor = false);
 
  void AddToPredefines(llvm::raw_string_ostream &OS) override;
};
} // namespace qcor