mirror of
https://gitee.com/fasiondog/hikyuu.git
synced 2024-12-03 04:17:58 +08:00
110 lines
2.9 KiB
C++
110 lines
2.9 KiB
C++
/*
|
||
* KData.cpp
|
||
*
|
||
* Created on: 2013-1-20
|
||
* Author: fasiondog
|
||
*/
|
||
|
||
#include "KData.h"
|
||
#include "StockManager.h"
|
||
#include "KDataBufferImp.h"
|
||
#include "indicator/crt/KDATA.h"
|
||
#include <fstream>
|
||
|
||
namespace hku {
|
||
|
||
HKU_API std::ostream& operator<<(std::ostream& os, const KData& kdata) {
|
||
os << "KData{\n size : " << kdata.size() << "\n stock: " << kdata.getStock()
|
||
<< "\n query: " << kdata.getQuery() << "\n}";
|
||
return os;
|
||
}
|
||
|
||
string KData::toString() const {
|
||
std::stringstream os;
|
||
os << "KData{\n size : " << size() << "\n stock: " << getStock().toString()
|
||
<< ",\n query: " << getQuery() << "\n}";
|
||
return os.str();
|
||
}
|
||
|
||
KData::KData(const Stock& stock, const KQuery& query) {
|
||
if (stock.isNull()) {
|
||
return;
|
||
}
|
||
|
||
m_imp = KDataImpPtr(new KDataBufferImp(stock, query));
|
||
return;
|
||
#if 0
|
||
if (stock.isBuffer(query.kType())
|
||
&& query.recoverType() == KQuery::NO_RECOVER) {
|
||
//当Stock已缓存了该类型的K线数据，且不进行复权
|
||
m_imp = KDataImpPtr(new KDataImp(stock, query));
|
||
} else {
|
||
m_imp = KDataImpPtr(new KDataBufferImp(stock, query));
|
||
}
|
||
#endif
|
||
}
|
||
|
||
void KData::tocsv(const string& filename) {
|
||
std::ofstream file(filename.c_str());
|
||
if (!file) {
|
||
HKU_ERROR("Can't open file! ({})", filename);
|
||
return;
|
||
}
|
||
|
||
file << "date, open, high, low, close, amount, count" << std::endl;
|
||
file.setf(std::ios_base::fixed);
|
||
file.precision(4);
|
||
string sep = ",";
|
||
KRecord record;
|
||
for (size_t i = 0; i < size(); ++i) {
|
||
record = getKRecord(i);
|
||
file << record.datetime << sep << record.openPrice << sep << record.highPrice << sep
|
||
<< record.lowPrice << sep << record.closePrice << sep << record.transAmount << sep
|
||
<< record.transCount << std::endl;
|
||
}
|
||
|
||
file.close();
|
||
}
|
||
|
||
Indicator KData::open() const {
|
||
return OPEN(*this);
|
||
}
|
||
|
||
Indicator KData::close() const {
|
||
return CLOSE(*this);
|
||
}
|
||
|
||
Indicator KData::low() const {
|
||
return LOW(*this);
|
||
}
|
||
|
||
Indicator KData::high() const {
|
||
return HIGH(*this);
|
||
}
|
||
|
||
Indicator KData::vol() const {
|
||
return VOL(*this);
|
||
}
|
||
|
||
Indicator KData::amo() const {
|
||
return AMO(*this);
|
||
}
|
||
|
||
KData HKU_API getKData(const string& market_code, const Datetime& start, const Datetime& end,
|
||
KQuery::KType ktype, KQuery::RecoverType recoverType) {
|
||
const StockManager& sm = StockManager::instance();
|
||
Stock stock = sm.getStock(market_code);
|
||
KQuery query(start, end, ktype, recoverType);
|
||
return stock.getKData(query);
|
||
}
|
||
|
||
KData HKU_API getKData(const string& market_code, int64 start, int64 end, KQuery::KType ktype,
|
||
KQuery::RecoverType recoverType) {
|
||
const StockManager& sm = StockManager::instance();
|
||
Stock stock = sm.getStock(market_code);
|
||
KQuery query(start, end, ktype, recoverType);
|
||
return stock.getKData(query);
|
||
}
|
||
|
||
} // namespace hku
|