StockManager.h

浏览该文件的文档.

/*
 * StockManager.h
 *
 *  Created on: 2011-11-9
 *      Author: fasiondog
 */

#ifndef STOCKMANAGER_H_
#define STOCKMANAGER_H_

#include <hikyuu_utils/iniparser/IniParser.h>

#include "utilities/Parameter.h"
#include "data_driver/BlockInfoDriver.h"
#include "Block.h"
#include "MarketInfo.h"
#include "StockTypeInfo.h"

namespace hku {

typedef vector<string> MarketList;

Parameter default_preload_param();
Parameter default_other_param();

class HKU_API StockManager {
public:
    static StockManager& instance();
    virtual ~StockManager();

    void init(const Parameter& baseInfoParam,
              const Parameter& blockParam,
              const Parameter& kdataParam,
              const Parameter& preloadParam = default_preload_param(),
              const Parameter& hikyuuParam = default_other_param());

    Parameter getBaseInfoDriverParameter() const;
    Parameter getBlockDriverParameter() const;
    Parameter getKDataDriverParameter() const;
    Parameter getPreloadParameter() const;
    Parameter getHikyuuParameter() const;

    void setKDataDriver(const KDataDriverPtr&);

    string tmpdir() const;

    size_t size() const;

    Stock getStock(const string&) const;

    Stock operator[](const string&) const;

    MarketInfo getMarketInfo(const string& market) const;

    StockTypeInfo getStockTypeInfo(hku_uint32 type) const;

    MarketList getAllMarket() const;

    Block getBlock(const string& category, const string& name);

    BlockList getBlockList(const string&);

    BlockList getBlockList();

    //目前支持"SH"
    DatetimeList getTradingCalendar(const KQuery& query,
            const string& market = "SH");

    bool addStock(const Stock& stock);

    bool addMarketInfo(const MarketInfo& marketInfo);

    bool addStockTypeInfo(const StockTypeInfo& stkTypeInfo);

    Stock addTempCsvStock(const string& code,
            const string& day_filename,
            const string& min_filename,
            price_t tick = 0.01,
            price_t tickValue = 0.01,
            int precision = 2,
            size_t minTradeNumber = 1,
            size_t maxTradeNumber = 1000000);

    void removeTempCsvStock(const string& code);

public:
    typedef StockMapIterator const_iterator;
    const_iterator begin() const { return m_stockDict.begin(); }
    const_iterator end() const { return m_stockDict.end(); }

private:
    StockManager() { }

private:
    static shared_ptr<StockManager> m_sm;
    string m_tmpdir;
    BlockInfoDriverPtr m_blockDriver;

    StockMapIterator::stock_map_t m_stockDict;  // SH000001 -> stock

    typedef unordered_map<string, MarketInfo> MarketInfoMap;
    MarketInfoMap m_marketInfoDict;

    typedef unordered_map<hku_uint32, StockTypeInfo> StockTypeInfoMap;
    StockTypeInfoMap m_stockTypeInfo;

    Parameter m_baseInfoDriverParam;
    Parameter m_blockDriverParam;
    Parameter m_kdataDriverParam;
    Parameter m_preloadParam;
    Parameter m_hikyuuParam;
};


inline size_t StockManager::size() const {
    return m_stockDict.size();
}


inline Stock StockManager::operator[](const string& query) const {
    return getStock(query);
}

inline Parameter StockManager::getBaseInfoDriverParameter() const {
    return m_baseInfoDriverParam;
}

inline Parameter StockManager::getBlockDriverParameter() const {
    return m_blockDriverParam;
}

inline Parameter StockManager::getKDataDriverParameter() const {
    return m_kdataDriverParam;
}

inline Parameter StockManager::getPreloadParameter() const {
    return m_preloadParam;
}

inline Parameter StockManager::getHikyuuParameter() const {
    return m_hikyuuParam;
}

} /* namespace */

#endif /* STOCKMANAGER_H_ */