mirror of
https://gitee.com/fasiondog/hikyuu.git
synced 2024-11-29 18:39:10 +08:00
276 lines
8.1 KiB
C++
276 lines
8.1 KiB
C++
/*
|
||
* KQuery.h
|
||
*
|
||
* Created on: 2009-11-23
|
||
* Author: fasiondog
|
||
*/
|
||
|
||
#pragma once
|
||
#ifndef KQUERY_H_
|
||
#define KQUERY_H_
|
||
|
||
#include "KRecord.h"
|
||
|
||
namespace hku {
|
||
|
||
/**
|
||
* 按索引方式查询K线数据条件
|
||
* @ingroup StockManage
|
||
*/
|
||
class HKU_API KQuery {
|
||
public:
|
||
/// 查询方式：索引或日期
|
||
enum QueryType {
|
||
INDEX = 0, ///< 按索引方式查询
|
||
DATE = 1, ///< 按日期方式查询
|
||
INVALID = 2
|
||
};
|
||
|
||
/// 查询K线类型：日线/周线等
|
||
/*enum KType {
|
||
//notes: 如添加新类型，请注意按大小顺序添加，否则可能出错
|
||
MIN = 0, ///<1分钟线
|
||
MIN5 = 1, ///<5分钟线
|
||
MIN15 = 2, ///<15分钟线
|
||
MIN30 = 3, ///<30分钟线
|
||
MIN60 = 4, ///<60分钟线
|
||
DAY = 5, ///<日线
|
||
WEEK = 6, ///<周线
|
||
MONTH = 7, ///<月线
|
||
QUARTER = 8, ///<季线
|
||
HALFYEAR = 9, ///<半年线
|
||
YEAR = 10, ///<年线
|
||
|
||
//BTC扩展
|
||
MIN3 = 11, ///<3分钟线
|
||
HOUR2 = 12, ///<2小时线
|
||
HOUR4 = 13, ///<4小时线
|
||
HOUR6 = 14, ///<6小时线
|
||
HOUR12 = 15, ///<12小时线
|
||
INVALID_KTYPE = 16
|
||
};*/
|
||
typedef string KType;
|
||
|
||
static const string MIN;
|
||
static const string MIN5;
|
||
static const string MIN15;
|
||
static const string MIN30;
|
||
static const string MIN60;
|
||
static const string DAY;
|
||
static const string WEEK;
|
||
static const string MONTH;
|
||
static const string QUARTER;
|
||
static const string HALFYEAR;
|
||
static const string YEAR;
|
||
static const string MIN3;
|
||
static const string HOUR2;
|
||
static const string HOUR4;
|
||
static const string HOUR6;
|
||
static const string HOUR12;
|
||
// static const string INVALID_KTYPE;
|
||
|
||
/** 获取所有的 KType */
|
||
static const vector<KType>& getAllKType();
|
||
|
||
static int32_t getKTypeInMin(KType);
|
||
|
||
/** 判断是否为有效 ktype */
|
||
static bool isKType(const string& ktype);
|
||
|
||
/**
|
||
* 复权类型
|
||
* @note 日线以上，如周线/月线不支持复权
|
||
*/
|
||
enum RecoverType {
|
||
NO_RECOVER = 0, ///< 不复权
|
||
FORWARD = 1, ///< 前向复权
|
||
BACKWARD = 2, ///< 后向复权
|
||
EQUAL_FORWARD = 3, ///< 等比前向复权
|
||
EQUAL_BACKWARD = 4, ///< 等比后向复权
|
||
INVALID_RECOVER_TYPE = 5
|
||
};
|
||
|
||
/** 默认构造，按索引方式查询全部日线数据，不复权 */
|
||
KQuery()
|
||
: m_start(0),
|
||
m_end(Null<int64_t>()),
|
||
m_queryType(INDEX),
|
||
m_dataType(DAY),
|
||
m_recoverType(NO_RECOVER) {};
|
||
|
||
/**
|
||
* K线查询，范围[start, end)
|
||
* @param start 起始索引，支持负数
|
||
* @param end 结束索引（不包含本身），支持负数
|
||
* @param dataType K线类型
|
||
* @param recoverType 复权类型
|
||
* @param queryType 默认按索引方式查询
|
||
*/
|
||
KQuery(int64_t start, // cppcheck-suppress [noExplicitConstructor]
|
||
int64_t end = Null<int64_t>(), const KType& dataType = DAY,
|
||
RecoverType recoverType = NO_RECOVER, QueryType queryType = INDEX)
|
||
: m_start(start),
|
||
m_end(end),
|
||
m_queryType(queryType),
|
||
m_dataType(dataType),
|
||
m_recoverType(recoverType) {
|
||
to_upper(m_dataType);
|
||
}
|
||
|
||
/**
|
||
* 按指定日期查询 K 线，范围[start, end)
|
||
* @param start 起始日期
|
||
* @param end 结束日期
|
||
* @param ktype K线类型
|
||
* @param recoverType 复权类型
|
||
*/
|
||
KQuery(Datetime start, // cppcheck-suppress [noExplicitConstructor]
|
||
Datetime end = Null<Datetime>(), const KType& ktype = DAY,
|
||
RecoverType recoverType = NO_RECOVER);
|
||
|
||
/**
|
||
* 按索引方式查询时，返回指定的起始索引，否则返回Null<int64_t>()
|
||
*/
|
||
int64_t start() const {
|
||
return m_queryType != INDEX ? Null<int64_t>() : m_start;
|
||
}
|
||
|
||
/**
|
||
* 按索引方式查询时，返回指定的结束索引，否则返回Null<int64_t>()
|
||
*/
|
||
int64_t end() const {
|
||
return m_queryType != INDEX ? Null<int64_t>() : m_end;
|
||
}
|
||
|
||
/**
|
||
* 按日期方式查询时，返回指定的起始日期，否则返回Null<Datetime>()
|
||
*/
|
||
Datetime startDatetime() const;
|
||
|
||
/**
|
||
* 按日期方式查询时，返回指定的结束日期，否则返回Null<Datetime>()
|
||
*/
|
||
Datetime endDatetime() const;
|
||
|
||
/** 获取查询条件类型 */
|
||
QueryType queryType() const {
|
||
return m_queryType;
|
||
}
|
||
|
||
/** 获取K线数据类型 */
|
||
// KType kType() const { return m_dataType; }
|
||
string kType() const {
|
||
return m_dataType;
|
||
}
|
||
|
||
/** 获取复权类型 */
|
||
RecoverType recoverType() const {
|
||
return m_recoverType;
|
||
}
|
||
|
||
/** 设置复权类型 */
|
||
void recoverType(RecoverType recoverType) {
|
||
m_recoverType = recoverType;
|
||
}
|
||
|
||
/** 获取queryType名称，用于显示输出 */
|
||
static string getQueryTypeName(QueryType);
|
||
|
||
/** 获取KType名称，用于显示输出 */
|
||
static string getKTypeName(KType);
|
||
|
||
/** 获取recoverType名称，用于显示输出 */
|
||
static string getRecoverTypeName(RecoverType);
|
||
|
||
/** 根据字符串名称获取相应的queryType枚举值 */
|
||
static QueryType getQueryTypeEnum(const string&);
|
||
|
||
/** 根据字符串名称，获取相应的枚举值 */
|
||
static KType getKTypeEnum(const string&);
|
||
|
||
/** 根据字符串名称，获取相应的枚举值 */
|
||
static RecoverType getRecoverTypeEnum(const string&);
|
||
|
||
private:
|
||
int64_t m_start;
|
||
int64_t m_end;
|
||
QueryType m_queryType;
|
||
KType m_dataType;
|
||
RecoverType m_recoverType;
|
||
};
|
||
|
||
/**
|
||
* 构造按索引方式K线查询，范围[start, end)
|
||
* @param start 起始索引，支持负数
|
||
* @param end 结束索引（不包含本身），支持负数
|
||
* @param dataType K线类型
|
||
* @param recoverType 复权类型
|
||
* @see KQuery
|
||
* @ingroup StockManage*
|
||
*/
|
||
KQuery HKU_API KQueryByIndex(int64_t start = 0, int64_t end = Null<int64_t>(),
|
||
const KQuery::KType& dataType = KQuery::DAY,
|
||
KQuery::RecoverType recoverType = KQuery::NO_RECOVER);
|
||
|
||
inline KQuery KQueryByIndex(int64_t start, int64_t end, const KQuery::KType& dataType,
|
||
KQuery::RecoverType recoverType) {
|
||
return KQuery(start, end, dataType, recoverType, KQuery::INDEX);
|
||
}
|
||
|
||
/**
|
||
* 构造按日期方式K线查询，范围[startDatetime, endDatetime)
|
||
* @param start 起始日期
|
||
* @param end 结束日期（不包含本身）
|
||
* @param dataType K线类型
|
||
* @param recoverType 复权类型
|
||
* @see KQuery
|
||
* @ingroup StockManage
|
||
*/
|
||
KQuery HKU_API KQueryByDate(const Datetime& start = Datetime::min(),
|
||
const Datetime& end = Null<Datetime>(),
|
||
const KQuery::KType& dataType = KQuery::DAY,
|
||
KQuery::RecoverType recoverType = KQuery::NO_RECOVER);
|
||
|
||
inline KQuery KQueryByDate(const Datetime& start, const Datetime& end,
|
||
const KQuery::KType& dataType, KQuery::RecoverType recoverType) {
|
||
return KQuery(start, end, dataType, recoverType);
|
||
}
|
||
|
||
/**
|
||
* 输出KQuery信息，如：KQuery(start, end, queryType, kType, recoverType)
|
||
* @ingroup StockManage
|
||
*/
|
||
HKU_API std::ostream& operator<<(std::ostream& os, const KQuery& query);
|
||
|
||
///////////////////////////////////////////////////////////////////////////////
|
||
//
|
||
// 关系比较函数, 不直接在类中定义是为了支持 Null<>() == d，Null可以放在左边
|
||
//
|
||
///////////////////////////////////////////////////////////////////////////////
|
||
bool HKU_API operator==(const KQuery&, const KQuery&);
|
||
bool HKU_API operator!=(const KQuery&, const KQuery&);
|
||
|
||
/**
|
||
* 提供KQuery的Null值
|
||
* @ingroup StockManage
|
||
*/
|
||
template <>
|
||
class Null<KQuery> {
|
||
public:
|
||
Null() {}
|
||
operator KQuery() {
|
||
return KQuery(Null<int64_t>(), Null<int64_t>(),
|
||
"", // KQuery::INVALID_KTYPE,
|
||
KQuery::INVALID_RECOVER_TYPE, KQuery::INVALID);
|
||
}
|
||
};
|
||
|
||
} // namespace hku
|
||
|
||
#if FMT_VERSION >= 90000
|
||
template <>
|
||
struct fmt::formatter<hku::KQuery> : ostream_formatter {};
|
||
#endif
|
||
|
||
#endif /* KQUERY_H_ */
|