#pragma once

#include "libSage++.h"
#include <map>
#include <string>
#include <set>

using ResultSet = std::set<std::tuple<std::string, int, std::string>>;

template<typename Iterator>
static void extract_vars_from_reg(std::set<SAPFOR::Argument*>& worklist,
                                  SAPFOR::Argument* reg,
                                  Iterator instr,
                                  Iterator first_instr);


template<typename Iterator>
static void processArgument(std::set<SAPFOR::Argument*>& worklist,
                            SAPFOR::Argument* arg,
                            Iterator instr,
                            Iterator first_instr);

static void lookup_for_vars(std::map<SgStatement*, std::string>& where_to_add,
                            std::set<SAPFOR::Argument*>& worklist,
                            SAPFOR::Instruction* instr,
                            SAPFOR::BasicBlock* bblock,
                            const std::map<FuncInfo*, std::vector<SAPFOR::BasicBlock*>>& fullIR);

void
findParameters(ResultSet& foundParameters,
               std::map<FuncInfo*, std::vector<SAPFOR::BasicBlock*>>& fullIR,
			   const std::map<std::tuple<int, std::string, std::string>, std::pair<DIST::Array*, DIST::ArrayAccessInfo*>>& declaredArrays);