ast_method.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <arkanjo/methods/method.hpp>
 
#include <vector>
#include <filesystem>
#include <fstream>
#include <mutex>
#include <atomic>
#include <tree_sitter/api.h>
 
namespace fs = std::filesystem;
 
struct ZSNode {
  TSSymbol symbol;
  std::vector<ZSNode> children;
};
 
struct PostOrderTree {
  std::vector<TSSymbol> labels;
 
  std::vector<int> lmd;
 
  fs::path path;
};
 
class ASTMethod : public IMethod {
  private:
    static constexpr const char* SAVING_MESSAGE = "Saving results..."; 
 
    fs::path base_path;  
    double similarity; 
 
    std::vector<PostOrderTree> processed;
 
    std::ofstream output_file;
 
    std::mutex processed_mutex;
    std::mutex output_mutex;
 
    std::atomic<size_t> total_matches{0};
 
    ZSNode from_tsnode(TSNode node);
 
    int tree_distance(const PostOrderTree& a, const PostOrderTree& b);
 
    double similarity_score(const PostOrderTree& a, const PostOrderTree& b);
 
    void compare_range(
      const std::vector<PostOrderTree>& processed,
      size_t begin, size_t end,
      std::vector<DuplicationEntry>& local
    );
 
    void save_duplications(std::vector<DuplicationEntry>& file_duplication_pairs) override;
    void flush_duplications(std::vector<DuplicationEntry>& local);
    void header_duplications();
  public:
    ASTMethod(const fs::path& base_path_, double similarity_);
 
    void on_function(const FunctionData& fd) override;
 
    void execute() override;
};