mirror of
https://gitee.com/fasiondog/hikyuu.git
synced 2024-12-04 04:48:17 +08:00
79 lines
1.6 KiB
C++
79 lines
1.6 KiB
C++
/*
|
||
* 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 */
|