mirror of
https://gitee.com/fasiondog/hikyuu.git
synced 2024-12-05 05:17:47 +08:00
111 lines
2.7 KiB
C++
111 lines
2.7 KiB
C++
/*
|
|
* Block.cpp
|
|
*
|
|
* Copyright (c) hikyuu.org
|
|
*
|
|
* Created on: 2015年2月8日
|
|
* Author: fasiondog
|
|
*/
|
|
|
|
#include <boost/algorithm/string.hpp>
|
|
#include "StockManager.h"
|
|
|
|
namespace hku {
|
|
|
|
HKU_API std::ostream& operator<<(std::ostream& os, const Block& blk) {
|
|
string strip(", ");
|
|
os << "Block(" << blk.category() << strip << blk.name() << ")";
|
|
return os;
|
|
}
|
|
|
|
Block::Block() {}
|
|
|
|
Block::~Block() {}
|
|
|
|
Block::Block(const string& category, const string& name) : m_data(make_shared<Data>()) {
|
|
m_data->m_category = category;
|
|
m_data->m_name = name;
|
|
}
|
|
|
|
Block::Block(const Block& block) {
|
|
if (m_data == block.m_data)
|
|
return;
|
|
m_data = block.m_data;
|
|
}
|
|
|
|
Block& Block::operator=(const Block& block) {
|
|
HKU_IF_RETURN(this == &block || m_data == block.m_data, *this);
|
|
m_data = block.m_data;
|
|
return *this;
|
|
}
|
|
|
|
bool Block::have(const string& market_code) const {
|
|
HKU_IF_RETURN(!m_data, false);
|
|
string query_str = market_code;
|
|
to_upper(query_str);
|
|
return m_data->m_stockDict.count(query_str) ? true : false;
|
|
}
|
|
|
|
bool Block::have(const Stock& stock) const {
|
|
HKU_IF_RETURN(!m_data, false);
|
|
return m_data->m_stockDict.count(stock.market_code()) ? true : false;
|
|
}
|
|
|
|
Stock Block::get(const string& market_code) const {
|
|
Stock result;
|
|
HKU_IF_RETURN(!m_data, result);
|
|
string query_str = market_code;
|
|
to_upper(query_str);
|
|
auto iter = m_data->m_stockDict.find(query_str);
|
|
if (iter != m_data->m_stockDict.end()) {
|
|
result = iter->second;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
vector<Stock> Block::getAllStocks() const {
|
|
vector<Stock> ret;
|
|
ret.reserve(size());
|
|
auto iter = m_data->m_stockDict.begin();
|
|
for (; iter != m_data->m_stockDict.end(); ++iter) {
|
|
ret.emplace_back(iter->second);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
bool Block::add(const Stock& stock) {
|
|
HKU_IF_RETURN(stock.isNull() || have(stock), false);
|
|
if (!m_data)
|
|
m_data = shared_ptr<Data>(new Data);
|
|
|
|
m_data->m_stockDict[stock.market_code()] = stock;
|
|
return true;
|
|
}
|
|
|
|
bool Block::add(const string& market_code) {
|
|
const StockManager& sm = StockManager::instance();
|
|
Stock stock = sm.getStock(market_code);
|
|
HKU_IF_RETURN(stock.isNull() || have(stock), false);
|
|
if (!m_data)
|
|
m_data = shared_ptr<Data>(new Data);
|
|
|
|
m_data->m_stockDict[stock.market_code()] = stock;
|
|
return true;
|
|
}
|
|
|
|
bool Block::remove(const string& market_code) {
|
|
HKU_IF_RETURN(!have(market_code), false);
|
|
string query_str = market_code;
|
|
to_upper(query_str);
|
|
m_data->m_stockDict.erase(query_str);
|
|
return true;
|
|
}
|
|
|
|
bool Block::remove(const Stock& stock) {
|
|
HKU_IF_RETURN(!have(stock), false);
|
|
m_data->m_stockDict.erase(stock.market_code());
|
|
return true;
|
|
}
|
|
|
|
} /* namespace hku */
|