RuleSetGenerator.h

Go to the documentation of this file.

#pragma once
#include <map>
#include <utility>
 
#include <nlohmann/json.hpp>
 
#include "messages/classical_messages.h"
#include "rules/RuleSet.h"
 
namespace quisp::modules::ruleset_gen {
 
class RuleSetGenerator {
 public:
  RuleSetGenerator(int responder_addr) : responder_addr(responder_addr) {}
 
  std::map<int, nlohmann::json> generateRuleSets(messages::ConnectionSetupRequest* req, unsigned long ruleset_id);
 
  void generateReverseSwapAtHalfRuleSets(int left_node_index, int right_node_index, std::map<int, std::vector<std::unique_ptr<rules::Rule>>>& rules_map, std::vector<int>& path,
                                         int& shared_rule_tag);
 
 protected:
  std::vector<int> collectPath(messages::ConnectionSetupRequest* req);
 
  std::unique_ptr<rules::Rule> tomographyRule(int partner_address, int owner_address, int num_measure, int shared_rule_tag);
 
  std::unique_ptr<rules::Rule> purifyRule(int partner_address, rules::PurType purification_type, int shared_rule_tag);
 
  std::unique_ptr<rules::Rule> purificationCorrelationRule(int partner_address, rules::PurType purification_protocol, int shared_rule_tag);
 
  std::unique_ptr<rules::Rule> swapRule(std::pair<int, int> partner_address, int shared_rule_tag);
 
  std::unique_ptr<rules::Rule> swapCorrectionRule(int swapper_address, int shared_rule_tag);
 
  int responder_addr;
};
}  // namespace quisp::modules::ruleset_gen