Commit 612b6599 authored by Mccaskey, Alex's avatar Mccaskey, Alex
Browse files

uploading initial clang ast modification code 



Signed-off-by: Mccaskey, Alex's avatarAlex McCaskey <mccaskeyaj@ornl.gov>
parent 84c4f2c1

CMakeLists.txt

0 → 100644
+40 −0
Original line number Diff line number Diff line 
cmake_minimum_required(VERSION 3.2 FATAL_ERROR)

if(${CMAKE_VERSION} VERSION_LESS 3.12)

    cmake_policy(VERSION ${CMAKE_MAJOR_VERSION}.${CMAKE_MINOR_VERSION})

endif()



set(CMAKE_STANDARD_REQUIRED ON)

set(CMAKE_CXX_STANDARD 11)

set(CMAKE_DISABLE_IN_SOURCE_BUILDS ON)

set(CMAKE_DISABLE_SOURCE_CHANGES ON)

set(CMAKE_POSITION_INDEPENDENT_CODE TRUE)

set(CMAKE_EXPORT_COMPILE_COMMANDS TRUE)

set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} ${CMAKE_SOURCE_DIR}/cmake/Modules)



message(STATUS "${CMAKE_MODULE_PATH}")

project(qcor LANGUAGES CXX)



option(QCOR_BUILD_TESTS "Build qcor tests" OFF)



include(CTest)



find_package(Clang 8.0.0 REQUIRED)

find_package(XACC REQUIRED)



if (CMAKE_INSTALL_PREFIX_INITIALIZED_TO_DEFAULT)

    set (CMAKE_INSTALL_PREFIX "${XACC_ROOT}" CACHE PATH "default install path" FORCE )

endif()



set(CppMicroServicesConfig_DIR "${XACC_ROOT}/share/cppmicroservices4/cmake")

set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} "${XACC_ROOT}/share/xacc")

set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} "${CppMicroServicesConfig_DIR}")

include(tests)

include(CppMicroServicesConfig)



#include_directories(${XACC_INCLUDE_DIRS})



#link_directories(${XACC_LIBRARY_DIR})



message(STATUS "${CLANG_INCLUDE_DIRS}")



add_subdirectory(compiler)
 
 No newline at end of file

cmake/Modules/FindClang.cmake

0 → 100644
+124 −0
Original line number Diff line number Diff line 
# Detect Clang libraries

#

# Defines the following variables:

#  CLANG_FOUND                 - True if Clang was found

#  CLANG_INCLUDE_DIRS          - Where to find Clang includes

#  CLANG_LIBRARY_DIRS          - Where to find Clang libraries

#

#  CLANG_LIBCLANG_LIB          - Libclang C library

#

#  CLANG_CLANGFRONTEND_LIB     - Clang Frontend (C++) Library

#  CLANG_CLANGDRIVER_LIB       - Clang Driver (C++) Library

#  ...

#

#  CLANG_LIBS                  - All the Clang C++ libraries

#

# Uses the same include and library paths detected by FindLLVM.cmake

#

# See http://clang.llvm.org/docs/InternalsManual.html for full list of libraries



#=============================================================================

# Copyright 2014-2015 Kevin Funk <kfunk@kde.org>

#

# Redistribution and use in source and binary forms, with or without

# modification, are permitted provided that the following conditions

# are met:

#

# 1. Redistributions of source code must retain the above copyright

# notice, this list of conditions and the following disclaimer.

# 2. Redistributions in binary form must reproduce the above copyright

# notice, this list of conditions and the following disclaimer in the

# documentation and/or other materials provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR

# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES

# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.

# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,

# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT

# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

#=============================================================================



if (${Clang_FIND_REQUIRED})

    find_package(LLVM ${Clang_FIND_VERSION} REQUIRED)

else ()

    find_package(LLVM ${Clang_FIND_VERSION})

endif ()

    

set(CLANG_FOUND FALSE)



if (LLVM_FOUND AND LLVM_LIBRARY_DIRS)

  macro(FIND_AND_ADD_CLANG_LIB _libname_)

    string(TOUPPER ${_libname_} _prettylibname_)

    find_library(CLANG_${_prettylibname_}_LIB NAMES ${_libname_} HINTS ${LLVM_LIBRARY_DIRS})

    if(CLANG_${_prettylibname_}_LIB)

      set(CLANG_LIBS ${CLANG_LIBS} ${CLANG_${_prettylibname_}_LIB})

    endif()

  endmacro(FIND_AND_ADD_CLANG_LIB)



  # note: On Windows there's 'libclang.dll' instead of 'clang.dll' -> search for 'libclang', too

  find_library(CLANG_LIBCLANG_LIB NAMES clang libclang HINTS ${LLVM_LIBRARY_DIRS}) # LibClang: high-level C interface



  FIND_AND_ADD_CLANG_LIB(clangFrontend)

  FIND_AND_ADD_CLANG_LIB(clangDriver)

  FIND_AND_ADD_CLANG_LIB(clangCodeGen)

  FIND_AND_ADD_CLANG_LIB(clangSema)

  FIND_AND_ADD_CLANG_LIB(clangChecker)

  FIND_AND_ADD_CLANG_LIB(clangAnalysis)

  FIND_AND_ADD_CLANG_LIB(clangRewriteFrontend)

  FIND_AND_ADD_CLANG_LIB(clangRewrite)

  FIND_AND_ADD_CLANG_LIB(clangAST)

  FIND_AND_ADD_CLANG_LIB(clangASTMatchers)

  FIND_AND_ADD_CLANG_LIB(clangParse)

  FIND_AND_ADD_CLANG_LIB(clangLex)

  FIND_AND_ADD_CLANG_LIB(clangBasic)

  FIND_AND_ADD_CLANG_LIB(clangARCMigrate)

  FIND_AND_ADD_CLANG_LIB(clangEdit)

  FIND_AND_ADD_CLANG_LIB(clangFrontendTool)

  FIND_AND_ADD_CLANG_LIB(clangRewrite)

  FIND_AND_ADD_CLANG_LIB(clangSerialization)

  FIND_AND_ADD_CLANG_LIB(clangTooling)

  FIND_AND_ADD_CLANG_LIB(clangStaticAnalyzerCheckers)

  FIND_AND_ADD_CLANG_LIB(clangStaticAnalyzerCore)

  FIND_AND_ADD_CLANG_LIB(clangStaticAnalyzerFrontend)

  FIND_AND_ADD_CLANG_LIB(clangSema)

  FIND_AND_ADD_CLANG_LIB(clangRewriteCore)

endif()



if(CLANG_LIBS OR CLANG_LIBCLANG_LIB)

  set(CLANG_FOUND TRUE)

else()

  message(STATUS "Could not find any Clang libraries in ${LLVM_LIBRARY_DIRS}")

endif()



if(CLANG_FOUND)

  set(CLANG_LIBRARY_DIRS ${LLVM_LIBRARY_DIRS})

  set(CLANG_INCLUDE_DIRS ${LLVM_INCLUDE_DIRS})



  # check whether llvm-config comes from an install prefix

  execute_process(

    COMMAND ${LLVM_CONFIG_EXECUTABLE} --src-root

    OUTPUT_VARIABLE _llvmSourceRoot

    OUTPUT_STRIP_TRAILING_WHITESPACE

  )

  string(FIND "${LLVM_INCLUDE_DIRS}" "${_llvmSourceRoot}" _llvmIsInstalled)

  if (NOT _llvmIsInstalled)

    message(STATUS "Detected that llvm-config comes from a build-tree, adding more include directories for Clang")

    list(APPEND CLANG_INCLUDE_DIRS

         "${LLVM_INSTALL_PREFIX}/tools/clang/include" # build dir

         "${_llvmSourceRoot}/tools/clang/include"     # source dir

    )

  endif()



  message(STATUS "Found Clang (LLVM version: ${LLVM_VERSION})")

  message(STATUS "  Include dirs:       ${CLANG_INCLUDE_DIRS}")

  message(STATUS "  Clang libraries:    ${CLANG_LIBS}")

  message(STATUS "  Libclang C library: ${CLANG_LIBCLANG_LIB}")

else()

  if(Clang_FIND_REQUIRED)

    message(FATAL_ERROR "Could NOT find Clang")

  endif()

endif()

cmake/Modules/FindLLVM.cmake

0 → 100644
+162 −0
Original line number Diff line number Diff line 
# Find the native LLVM includes and libraries

#

# Defines the following variables

#  LLVM_INCLUDE_DIRS   - where to find llvm include files

#  LLVM_LIBRARY_DIRS   - where to find llvm libs

#  LLVM_CFLAGS         - llvm compiler flags

#  LLVM_LFLAGS         - llvm linker flags

#  LLVM_MODULE_LIBS    - list of llvm libs for working with modules.

#  LLVM_INSTALL_PREFIX - LLVM installation prefix

#  LLVM_FOUND          - True if llvm found.

#  LLVM_VERSION        - Version string ("llvm-config --version")

#

# This module reads hints about search locations from variables

#  LLVM_ROOT           - Preferred LLVM installation prefix (containing bin/, lib/, ...)

#

#  Note: One may specify these as environment variables if they are not specified as

#   CMake variables or cache entries.



#=============================================================================

# Copyright 2014 Kevin Funk <kfunk@kde.org>

#

# Redistribution and use in source and binary forms, with or without

# modification, are permitted provided that the following conditions

# are met:

#

# 1. Redistributions of source code must retain the above copyright

# notice, this list of conditions and the following disclaimer.

# 2. Redistributions in binary form must reproduce the above copyright

# notice, this list of conditions and the following disclaimer in the

# documentation and/or other materials provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR

# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES

# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.

# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,

# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT

# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

#=============================================================================



if (NOT LLVM_ROOT AND DEFINED ENV{LLVM_ROOT})

    file(TO_CMAKE_PATH "$ENV{LLVM_ROOT}" LLVM_ROOT)

endif()



# if the user specified LLVM_ROOT, use that and fail otherwise

if (LLVM_ROOT)

  find_program(LLVM_CONFIG_EXECUTABLE NAMES llvm-config HINTS ${LLVM_ROOT}/bin DOC "llvm-config executable" NO_DEFAULT_PATH)

else()

  # find llvm-config, prefer the one with a version suffix, e.g. llvm-config-3.5

  # note: FreeBSD installs llvm-config as llvm-config35 and so on

  # note: on some distributions, only 'llvm-config' is shipped, so let's always try to fallback on that

  string(REPLACE "." "" LLVM_FIND_VERSION_CONCAT ${LLVM_FIND_VERSION})

  find_program(LLVM_CONFIG_EXECUTABLE NAMES llvm-config-${LLVM_FIND_VERSION} llvm-config${LLVM_FIND_VERSION_CONCAT} llvm-config DOC "llvm-config executable")



  # other distributions don't ship llvm-config, but only some llvm-config-VERSION binary

  # try to deduce installed LLVM version by looking up llvm-nm in PATH and *then* find llvm-config-VERSION via that

  if (NOT LLVM_CONFIG_EXECUTABLE)

    find_program(_llvmNmExecutable llvm-nm)

    if (_llvmNmExecutable)

      execute_process(COMMAND ${_llvmNmExecutable} --version OUTPUT_VARIABLE _out)

      string(REGEX REPLACE ".*LLVM version ([^ \n]+).*" "\\1" _versionString "${_out}")

      find_program(LLVM_CONFIG_EXECUTABLE NAMES llvm-config-${_versionString} DOC "llvm-config executable")

    endif()

  endif()

endif()



set(LLVM_FOUND FALSE)



if (LLVM_CONFIG_EXECUTABLE)

  # verify that we've found the correct version of llvm-config

  execute_process(COMMAND ${LLVM_CONFIG_EXECUTABLE} --version

    OUTPUT_VARIABLE LLVM_VERSION

    OUTPUT_STRIP_TRAILING_WHITESPACE)



  if (NOT LLVM_VERSION)

    set(_LLVM_ERROR_MESSAGE "Failed to parse version from llvm-config")

  elseif (LLVM_FIND_VERSION VERSION_GREATER LLVM_VERSION)

    set(_LLVM_ERROR_MESSAGE "LLVM version too old: ${LLVM_VERSION}")

  else()

    set(LLVM_FOUND TRUE)

  endif()

else()

  set(_LLVM_ERROR_MESSAGE "Could NOT find 'llvm-config' executable")

endif()



if (LLVM_FOUND)

  execute_process(

    COMMAND ${LLVM_CONFIG_EXECUTABLE} --includedir

    OUTPUT_VARIABLE LLVM_INCLUDE_DIRS

    OUTPUT_STRIP_TRAILING_WHITESPACE

  )



  execute_process(

    COMMAND ${LLVM_CONFIG_EXECUTABLE} --libdir

    OUTPUT_VARIABLE LLVM_LIBRARY_DIRS

    OUTPUT_STRIP_TRAILING_WHITESPACE

  )



  execute_process(

    COMMAND ${LLVM_CONFIG_EXECUTABLE} --cppflags

    OUTPUT_VARIABLE LLVM_CFLAGS

    OUTPUT_STRIP_TRAILING_WHITESPACE

  )



  execute_process(

    COMMAND ${LLVM_CONFIG_EXECUTABLE} --ldflags

    OUTPUT_VARIABLE LLVM_LFLAGS

    OUTPUT_STRIP_TRAILING_WHITESPACE

  )



  execute_process(

    COMMAND ${LLVM_CONFIG_EXECUTABLE} --libs core bitreader asmparser analysis

    OUTPUT_VARIABLE LLVM_MODULE_LIBS

    OUTPUT_STRIP_TRAILING_WHITESPACE

  )



  execute_process(

    COMMAND ${LLVM_CONFIG_EXECUTABLE} --libfiles

    OUTPUT_VARIABLE LLVM_LIBS

    OUTPUT_STRIP_TRAILING_WHITESPACE

  )

  string(REPLACE " " ";" LLVM_LIBS ${LLVM_LIBS}) # Make it consistent with --libs



  execute_process(

    COMMAND ${LLVM_CONFIG_EXECUTABLE} --system-libs

    OUTPUT_VARIABLE LLVM_SYSTEM_LIBS

    OUTPUT_STRIP_TRAILING_WHITESPACE

  )



  execute_process(

    COMMAND ${LLVM_CONFIG_EXECUTABLE} --prefix

    OUTPUT_VARIABLE LLVM_INSTALL_PREFIX

    OUTPUT_STRIP_TRAILING_WHITESPACE

  )



  # potentially add include dir from binary dir for non-installed LLVM

  execute_process(

    COMMAND ${LLVM_CONFIG_EXECUTABLE} --src-root

    OUTPUT_VARIABLE _llvmSourceRoot

    OUTPUT_STRIP_TRAILING_WHITESPACE

  )

  string(FIND "${LLVM_INCLUDE_DIRS}" "${_llvmSourceRoot}" _llvmIsInstalled)

  if (NOT _llvmIsInstalled)

    list(APPEND LLVM_INCLUDE_DIRS "${LLVM_INSTALL_PREFIX}/include")

  endif()

endif()



if (LLVM_FIND_REQUIRED AND NOT LLVM_FOUND)

  message(FATAL_ERROR "Could not find LLVM: ${_LLVM_ERROR_MESSAGE}")

elseif(_LLVM_ERROR_MESSAGE)

  message(STATUS "Could not find LLVM: ${_LLVM_ERROR_MESSAGE}")

endif()



if (LLVM_FOUND)

  message(STATUS "Found LLVM (version: ${LLVM_VERSION}): (using ${LLVM_CONFIG_EXECUTABLE})")

  message(STATUS "  Include dirs:   ${LLVM_INCLUDE_DIRS}")

  message(STATUS "  LLVM libraries: ${LLVM_LIBS}")

  message(STATUS "  LLVM System libraries: ${LLVM_SYSTEM_LIBS}")

endif()

compiler/CMakeLists.txt

0 → 100644
+8 −0
Original line number Diff line number Diff line 
set(LIBRARY_NAME qcor-ast-plugin) 

add_library(${LIBRARY_NAME} SHARED QCorPluginASTAction.cpp)

target_include_directories(${LIBRARY_NAME} PUBLIC .)

target_include_directories(${LIBRARY_NAME} PUBLIC ${CLANG_INCLUDE_DIRS})

target_include_directories(${LIBRARY_NAME} PUBLIC ${LLVM_INCLUDE_DIRS})

target_include_directories(${LIBRARY_NAME} PUBLIC ${XACC_INCLUDE_DIRS})

target_link_libraries(${LIBRARY_NAME} PUBLIC ${CLANG_LIBS} ${XACC_LIBRARIES})

compiler/QCorASTConsumer.hpp

0 → 100644
+41 −0
Original line number Diff line number Diff line 
#ifndef COMPILER_QCORASTCONSUMER_HPP_

#define COMPILER_QCORASTCONSUMER_HPP_



#include "clang/AST/AST.h"

#include "clang/AST/ASTConsumer.h"

#include "clang/AST/RecursiveASTVisitor.h"

#include "clang/Frontend/CompilerInstance.h"

#include "clang/Frontend/MultiplexConsumer.h"



#include "QcorASTVisitor.hpp"

#include "QuantumKernelHandler.hpp"



using namespace clang;



namespace qcor {

class QCorASTConsumer : public ASTConsumer {

public:

  QCorASTConsumer(CompilerInstance &c)

      : ci(c),

        handler(std::make_shared<QuantumKernelHandler>(c.getASTContext())) {}



  // Override the method that gets called for each parsed top-level

  // declaration.

  bool HandleTopLevelDecl(DeclGroupRef DR) override {

    QcorASTVisitor Visitor(ci);

    ci.getSema().addExternalSource(handler.get());

    Visitor.TraverseDecl(ci.getASTContext().getTranslationUnitDecl());

    // for (DeclGroupRef::iterator b = DR.begin(), e = DR.end(); b != e; ++b) {

    //   // Traverse the declaration using our AST visitor.

    //   Visitor.TraverseDecl(*b);

    // }

    // (*DR.begin())->dump();

    return true;

  }



private:

  CompilerInstance &ci;

  std::shared_ptr<QuantumKernelHandler> handler;

};

} // namespace qcor

#endif
 
 No newline at end of file