momentum.hpp

// A momentum trading strategy.
// Copyright 2020 Christian Kauten
//
// Author: Christian Kauten (kautenja@auburn.edu)
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
//
// Reference: High frequency trading strategies, market fragility and price
//            spikes: an agent based model perspective
// Reference: TODO
//
 
#ifndef STRATEGIES_MOMENTUM_HPP
#define STRATEGIES_MOMENTUM_HPP
 
#include "data_feed/receiver.hpp"
#include "order_entry/client.hpp"
#include "maths/probability.hpp"
#include "maths/rate_of_change.hpp"
#include <nlohmann/json.hpp>
#include <atomic>
#include <algorithm>
#include <iostream>
 
namespace Strategies {
 
class Momentum {
 private:
    typedef DataFeed::Receiver<Momentum> DataFeedReceiver;
    DataFeedReceiver receiver;
    OrderEntry::Client client;
    asio::steady_timer timer;
    uint32_t sleep_time;
    float P_act;
    std::atomic<bool> is_running;
 
    Maths::RateOfChange<OrderEntry::Price> roc;
    OrderEntry::Quantity max_size;
    double decision_boundary;
 
    inline void start_strategy() {
        timer.expires_after(std::chrono::milliseconds(sleep_time));
        timer.async_wait([this](const std::error_code& error) {
            if (error) {  // TODO: handle error code
                std::cout << "Momentum::start_strategy - " << error << std::endl;
                return;
            }
            if (Maths::Probability::boolean(P_act)) do_strategy();
            start_strategy();
        });
    }
 
    void do_strategy() {
        // update the rate of change based on the current midpoint price
        roc.process(receiver.get_book().last_price());
        // determine the quantity as product of the rate_of_change and the
        // "wealth" of the trader, i.e., their capital
        OrderEntry::Quantity size = std::abs(roc.get_rate_of_change() * client.get_capital());
        static constexpr OrderEntry::Quantity min_size = 0;
        size = std::clamp(size, min_size, max_size);
        // place an order
        if (roc.get_rate_of_change() >= decision_boundary)  // increasing, buy
            client.send<OrderEntry::Messages::OrderRequest>(
                OrderEntry::Messages::ORDER_PRICE_MARKET,
                size,
                OrderEntry::Side::Buy
            );
        else if (roc.get_rate_of_change() <= -decision_boundary)  // decreasing, sell
            client.send<OrderEntry::Messages::OrderRequest>(
                OrderEntry::Messages::ORDER_PRICE_MARKET,
                size,
                OrderEntry::Side::Sell
            );
    }
 
 public:
    Momentum(
        asio::io_context& feed_context,
        asio::io_context& context,
        nlohmann::json options
    ) :
        receiver(
            feed_context,
            asio::ip::make_address(options["data_feed"]["listen"].get<std::string>()),
            asio::ip::make_address(options["data_feed"]["group"].get<std::string>()),
            options["data_feed"]["port"].get<uint16_t>(),
            *this
        ),
        client(
            context,
            options["order_entry"]["host"].get<std::string>(),
            std::to_string(options["order_entry"]["port"].get<uint16_t>())
        ),
        timer(context),
        sleep_time(options["strategy"]["sleep_time"].get<uint32_t>()),
        P_act(options["strategy"]["P_act"].get<double>()),
        is_running(false),
        roc(options["strategy"]["history_size"].get<uint32_t>()),
        max_size(options["strategy"]["max_size"].get<OrderEntry::Quantity>()),
        decision_boundary(options["strategy"]["decision_boundary"].get<double>()) {
            // login to the order entry client
            auto username = OrderEntry::make_username(options["order_entry"]["username"].get<std::string>());
            auto password = OrderEntry::make_password(options["order_entry"]["password"].get<std::string>());
            client.send<OrderEntry::Messages::LoginRequest>(username, password);
            // set the capital and shares for the client
            client.set_capital(options["order_entry"]["capital"].get<OrderEntry::ClientCapital>());
            client.set_shares(options["order_entry"]["shares"].get<OrderEntry::ClientShares>());
        }
 
    inline void did_receive(DataFeedReceiver* receiver, const DataFeed::Messages::StartOfSession& message) {
        if (is_running) {
            std::cerr << "received start of session when already running" << std::endl;
            return;
        }
        // start the strategy
        start_strategy();
        // set the running flag to true
        is_running = true;
    }
 
    inline void did_receive(DataFeedReceiver* receiver, const DataFeed::Messages::EndOfSession& message) {
        if (not is_running) {
            std::cerr << "received end of session when not running" << std::endl;
            return;
        }
        // stop the strategy
        timer.cancel();
        is_running = false;
    }
 
    inline void did_receive(DataFeedReceiver* receiver, const DataFeed::Messages::Clear& message) { }
 
    inline void did_receive(DataFeedReceiver* receiver, const DataFeed::Messages::AddOrder& message) { }
 
    inline void did_receive(DataFeedReceiver* receiver, const DataFeed::Messages::DeleteOrder& message) { }
 
    inline void did_receive(DataFeedReceiver* receiver, const DataFeed::Messages::Trade& message) { }
};
 
}  // namespace Strategies
 
#endif  // STRATEGIES_MOMENTUM_HPP