/**
 * Copyright (c) 2016 , Continuum Analytics, Inc.
 * All rights reserved.
 */

#ifndef _ROC_HH
#define _ROC_HH

#include "llvm/IR/Module.h"
#include "llvm/Support/FormattedStream.h"

using namespace std;

namespace librocmlite
{

    class ModuleRef
    {
        public:
            ModuleRef(llvm::Module * module);
            operator bool () const;
            llvm::Module * getModule();
            void destroy();
            std::string to_string();
            static ModuleRef* parseAssembly(const char* Asm);
            static ModuleRef* parseBitcode(const char *Bitcode, size_t Len);
        private:
            llvm::Module * M;
    };

    // Initializes the llvm libary tooling.
    void Initialize();

    // Finalizes the llvm library tooling.
    void Finalize();

    // Optimize a module in place
    void Optimize(llvm::Module * M, int OptLevel, int SizeLevel, int Verify, const char * Cpu);

    // Compile a module
    int CompileModule(std::unique_ptr<llvm::Module> mod, llvm::raw_string_ostream &os, bool emitBRIG,
                      int OptLevel);

}

#ifdef __cplusplus
extern "C"
{
#endif // __cplusplus

using namespace librocmlite;

// ROC_ C/CFFI entry points

void ROC_Initialize();

void ROC_Finalize();

char* ROC_CreateString(const char *str);

void ROC_DisposeString(char *str);

// rename this to ParseIR2Module ?
ModuleRef* ROC_ParseModule(const char *Asm);

ModuleRef* ROC_ParseBitcode(const char *Asm, size_t Len);

void ROC_ModulePrint(ModuleRef *M, char **output);

void ROC_ModuleDestroy(ModuleRef *M);

int ROC_ModuleOptimize(ModuleRef *M, int OptLevel, int SizeLevel, int Verify, const char * Cpu);

int ROC_ModuleLinkIn(ModuleRef * Dst, ModuleRef * Src);

int ROC_ModuleEmitHSAIL(ModuleRef *M, int OptLevel, const char * Cpu, char **output);

size_t ROC_ModuleEmitBRIG(ModuleRef *M, int OptLevel, const char * Cpu, char **output);

void ROC_SetCommandLineOption(int argc, const char * const * argv);



#ifdef __cplusplus
}
#endif // __cplusplus

#endif //ifdef _ROC_HH