begun cpp version

Joel Grunbaum

2022-01-07 bb235393b72f5974a087e48e05c39fcd83e1db57

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#pragma once
 
#include <chrono>
#include <iostream>
#include <queue>
#include <string>
#include <vector>
 
enum OrderSideEnum { Buy, Sell };
enum ProductTypeEnum { TEST, FUTURE, SPREAD, CALL, PUT };
 
struct Order {
  double price;
  OrderSideEnum side;
  int remaining_volume;
  int filled_volume;
  std::chrono::nanoseconds timestamp;
  std::string id;
  Order(double price, OrderSideEnum side, int volume,
        std::chrono::nanoseconds timestamp, std::string id);
};
 
struct Level {
  double price;
  int volume;
  OrderSideEnum side;
  std::chrono::nanoseconds timestamp;
  std::string id;
 
  Level(Order &order);
};
 
bool operator>(const Level &a, const Level &b);
bool operator<(const Level &a, const Level &b);
bool operator>=(const Level &a, const Level &b);
bool operator<=(const Level &a, const Level &b);
bool operator==(const Level &a, const Level &b);
 
template <class T>
struct Side : public std::priority_queue<Level, std::vector<Level>, T> {
public:
  void deleteLevel(std::string orderId) {
    for (auto i = this->c.begin(); i != this->c.end();) {
      if (*i.id == orderId) {
        this->c.erase(i);
        std::make_heap(this->c.begin(), this->c.end(), this->comp);
      }
    }
  }
 
  void topRemoveVolume(int volume) { this->c[0].volume -= volume; }
 
  void printTop(int num = 5) {
    std::sort(this->c.begin(), this->c.end(), this->comp);
    for (int i = 0; i < num && i < this->size(); i++) {
      std::cout << "Price: " << this->c[i].price
                << ", volume: " << this->c[i].volume
                << ", time: " << this->c[i].timestamp.count()
                << ", id: " << this->c[i].id << std::endl;
    }
    std::make_heap(this->c.begin(), this->c.end(), this->comp);
  }
};
 
using AskSide = Side<std::greater<Level>>;
using BidSide = Side<std::less<Level>>;
 
struct Book {
  BidSide bidSide;
  AskSide askSide;
  ProductTypeEnum productType;
  std::string product;
  std::string stationId;
  std::string unit;
  std::chrono::nanoseconds expiry;
  double aggFee;
  double pasFee;
  double broFee;
 
  Book();
  Book(ProductTypeEnum productType, std::string product, std::string stationId,
       std::string unit, std::chrono::nanoseconds expiry, double aggFee,
       double pasFee, double broFee);
  void ask(Order &order);
  void bid(Order &order);
  void printBook();
};
 
Book testBook(int orders = 10, bool printBook = true);