/*
 * IBarsCount.cpp
 * 
 *  Copyright (c) 2019 hikyuu.org
 *
 *  Created on: 2019-5-12
 *      Author: fasiondog
 */

#include "IBarsCount.h"

#if HKU_SUPPORT_SERIALIZATION
BOOST_CLASS_EXPORT(hku::IBarsCount)
#endif


namespace hku {

IBarsCount::IBarsCount() : IndicatorImp("BARSCOUNT", 1) {

}

IBarsCount::~IBarsCount() {

}

bool IBarsCount::check() {
    return true;
}

void IBarsCount::_calculate(const Indicator& ind) {
    KData k = ind.getContext();
    Stock stk = k.getStock();
    
    size_t total = ind.size();
    m_discard = ind.discard();
    if (m_discard >= total) {
        m_discard = total;
        return;
    }

    //如果没有上下文，直接取该指标的discard进行计算
    if (stk.isNull()) {
        for (size_t i = m_discard; i < total; i++) {
            _set(i + 1 - m_discard, i);
        }

        return;
    }

    //对于1分钟线取得当日交易分钟数
    KQuery q = k.getQuery();
    if (q.kType() == KQuery::MIN) {
        Datetime pre_d = k[m_discard].datetime.startOfDay();
        size_t count = 0;
        for (size_t i = m_discard; i < total; i++) {
            Datetime d = k[i].datetime.startOfDay();
            if (d != pre_d) {
                pre_d = d;
                count = 0;
            }
            _set(++count, i);
        }
        
        return;
    }

    //取得上市以来总交易日数
    size_t k_start_pos = k.startPos();
    if (k_start_pos != Null<size_t>()) {
        for (size_t i = m_discard; i < total; i++) {
            _set(1 + k_start_pos + i, i);
        }
    }

    return;
}


Indicator HKU_API BARSCOUNT() {
    return Indicator(make_shared<IBarsCount>());
}


} /* namespace hku */