hikyuu 2.6.6__py3-none-win_amd64.whl → 2.6.8__py3-none-win_amd64.whl

hikyuu/include/hikyuu/KDataSharedBufferImp.h

@@ -0,0 +1,69 @@
+/*
+ *  Copyright (c) 2025 hikyuu.org
+ *
+ *  Created on: 2025-08-28
+ *      Author: fasiondog
+ */
+
+#pragma once
+
+#include "KDataImp.h"
+
+namespace hku {
+
+class HKU_API KDataSharedBufferImp : public KDataImp {
+public:
+    KDataSharedBufferImp() = default;
+    KDataSharedBufferImp(const Stock& stock, const KQuery& query);
+    virtual ~KDataSharedBufferImp();
+
+    virtual bool empty() const override {
+        return m_size == 0;
+    }
+
+    virtual size_t size() const override {
+        return m_size;
+    }
+
+    virtual size_t startPos() const override {
+        return m_start;
+    }
+
+    virtual size_t endPos() const override {
+        return m_end;
+    }
+
+    virtual size_t lastPos() const override {
+        return m_end == 0 ? 0 : m_end - 1;
+    }
+
+    virtual size_t getPos(const Datetime& datetime) const override;
+
+    virtual const KRecord& getKRecord(size_t pos) const override;
+
+    virtual const KRecord& front() const override {
+        return m_data[0];
+    }
+
+    virtual const KRecord& back() const override {
+        return m_data[m_size - 1];
+    }
+
+    virtual const KRecord* data() const override {
+        return m_data;
+    }
+
+    virtual KRecord* data() override {
+        return m_data;
+    }
+
+    virtual DatetimeList getDatetimeList() const override;
+
+private:
+    size_t m_start{0};
+    size_t m_end{0};
+    size_t m_size{0};
+    KRecord* m_data{nullptr};
+};
+
+} /* namespace hku */

hikyuu/include/hikyuu/KQuery.h

@@ -26,56 +26,54 @@ public:
         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;
 
+    // 基础K线类型
     static const string MIN;
     static const string MIN5;
     static const string MIN15;
     static const string MIN30;
     static const string MIN60;
+    static const string HOUR2;
     static const string DAY;
     static const string WEEK;
     static const string MONTH;
     static const string QUARTER;
     static const string HALFYEAR;
     static const string YEAR;
+
+    // 扩展K线类型
+    static const string DAY3;
+    static const string DAY5;
+    static const string DAY7;
     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;
+    static const string HOUR4;   // 默认不支持
+    static const string HOUR6;   // 默认不支持
+    static const string HOUR12;  // 默认不支持
 
-    /** 获取所有的 KType */
-    static const vector<KType>& getAllKType();
+    static const string TIMELINE;  // 分时
+    static const string TRANS;     // 分笔
 
-    static int32_t getKTypeInMin(KType);
+    /** 判断指定的K线类型是否有效 */
+    static bool isValidKType(const string& ktype);
 
     /** 判断是否为有效 ktype */
-    static bool isKType(const string& ktype);
+    static bool isBaseKType(const string& ktype);
+
+    /** 判断是否为扩展 ktype */
+    static bool isExtraKType(const string& ktype);
+
+    /** 获取所有的 KType */
+    static vector<KType> getBaseKTypeList();
+
+    /** 获取所有扩展 KType */
+    static vector<KType> getExtraKTypeList();
+
+    static int32_t getKTypeInMin(const KType& ktype);
+
+    static int32_t getBaseKTypeInMin(const KType& ktype);
+
+    static int64_t getKTypeInSeconds(const KType& ktype);
 
     /**
      * 复权类型
@@ -181,6 +179,14 @@ public:
      */
     uint64_t hash() const;
 
+    /** 判断是否为右开区间，即未指定结束时间 */
+    bool isRightOpening() const {
+        if (m_queryType == DATE) {
+            return endDatetime().isNull();
+        }
+        return m_end == Null<int64_t>();
+    }
+
     /** 获取queryType名称，用于显示输出 */
     static string getQueryTypeName(QueryType);
 

hikyuu/include/hikyuu/KRecord.h

@@ -17,7 +17,7 @@ namespace hku {
  * K线数据记录
  * @ingroup StockManage
  */
-class KRecord {
+class HKU_API KRecord {
 public:
     Datetime datetime;    ///< 日期，格式：YYYYMMDDHHMM 如：200901010930
     price_t openPrice;    ///< 开盘价
@@ -58,6 +58,9 @@ public:
     bool isValid() const {
         return datetime == Null<Datetime>() ? false : true;
     }
+
+public:
+    static const KRecord NullKRecord;
 };
 
 /** @ingroup StockManage */

hikyuu/include/hikyuu/MarketInfo.h

@@ -41,7 +41,7 @@ public:
     MarketInfo& operator=(const MarketInfo&) = default;
 
     MarketInfo(MarketInfo&&);
-    MarketInfo& operator=(MarketInfo&&);
+    MarketInfo& operator=(MarketInfo&&) noexcept;
 
     /** 获取市场简称 */
     const string& market() const {

hikyuu/include/hikyuu/Stock.h

@@ -29,6 +29,7 @@ typedef shared_ptr<KDataDriverConnectPool> KDataDriverConnectPoolPtr;
 class HKU_API KData;
 class HKU_API Parameter;
 class HKU_API Block;
+class HKU_API KDataSharedBufferImp;
 
 /**
  * Stock基类，Application中一般使用StockPtr进行操作
@@ -36,6 +37,7 @@ class HKU_API Block;
  */
 class HKU_API Stock {
     friend class StockManager;
+    friend class KDataSharedBufferImp;
 
 private:
     static const string default_market;
@@ -162,7 +164,7 @@ public:
      */
     bool getIndexRange(const KQuery& query, size_t& out_start, size_t& out_end) const;
 
-    /** 获取指定索引的K线数据记录，pos 无效时返回 Null<KRecord> */
+    /** 获取指定索引的K线数据记录，pos 无效时返回 KRecord::NullRecord */
     KRecord getKRecord(size_t pos, const KQuery::KType& dataType = KQuery::DAY) const;
 
     /** 根据数据类型（日线/周线等），获取指定日期的KRecord */
@@ -230,10 +232,10 @@ public:
      * 将K线数据做自身缓存
      *  @note 一般不主动调用，谨慎
      */
-    void loadKDataToBuffer(KQuery::KType);
+    void loadKDataToBuffer(KQuery::KType) const;
 
     /** 释放对应的K线缓存 */
-    void releaseKDataBuffer(KQuery::KType);
+    void releaseKDataBuffer(KQuery::KType) const;
 
     /** 指定类型的K线数据是否被缓存 */
     bool isBuffer(KQuery::KType) const;
@@ -242,7 +244,10 @@ public:
     bool isNull() const;
 
     /** （临时函数）只用于更新缓存中的K线数据 **/
-    void realtimeUpdate(KRecord, KQuery::KType ktype = KQuery::DAY);
+    void realtimeUpdate(KRecord, const KQuery::KType& ktype = KQuery::DAY);
+
+    /** 获取指定K线数据类型最后更新时间 */
+    Datetime getLastUpdateTime(const KQuery::KType& ktype) const;
 
     /**
      * 部分临时创建的 Stock, 直接设置KRecordList
@@ -258,12 +263,19 @@ private:
     bool _getIndexRangeByIndex(const KQuery&, size_t& out_start, size_t& out_end) const;
 
     // 以下函数属于基础操作添加了读锁
-    size_t _getCountFromBuffer(KQuery::KType ktype) const;
-    KRecord _getKRecordFromBuffer(size_t pos, const KQuery::KType& ktype) const;
+    size_t _getCountFromBuffer(const KQuery::KType& ktype) const;
+    const KRecord& _getKRecordFromBuffer(size_t pos, const KQuery::KType& ktype) const;
     KRecordList _getKRecordListFromBuffer(size_t start_ix, size_t end_ix,
                                           KQuery::KType ktype) const;
     bool _getIndexRangeByDateFromBuffer(const KQuery&, size_t&, size_t&) const;
 
+    KRecordList _getKRecordList(const KQuery& query) const;
+
+    // 仅供 StockManager 初始化时调用
+    void setPreload(const vector<KQuery::KType>& preload_ktypes);
+
+    bool isPreload(KQuery::KType ktype) const;
+
 private:
     struct HKU_API Data;
     shared_ptr<Data> m_data;
@@ -295,8 +307,10 @@ struct HKU_API Stock::Data {
     double m_minTradeNumber;
     double m_maxTradeNumber;
 
+    std::unordered_set<string> m_ktype_preload;  // 记录当前证券的K线数据是否需要预加载
     unordered_map<string, KRecordList*> pKData;
     unordered_map<string, std::shared_mutex*> pMutex;
+    unordered_map<string, Datetime> m_lastUpdate;  // 记录各类型K线数据最后更新时间
 
     Data();
     Data(const string& market, const string& code, const string& name, uint32_t type, bool valid,
@@ -338,6 +352,10 @@ inline bool Stock::operator!=(const Stock& stock) const {
     return !(*this == stock);
 }
 
+inline bool Stock::isNull() const {
+    return !m_data || !m_kdataDriver;
+}
+
 }  // namespace hku
 
 namespace std {

hikyuu/include/hikyuu/StockManager.h

@@ -22,8 +22,6 @@
 
 namespace hku {
 
-typedef vector<string> MarketList;
-
 /**
  * 证券信息统一管理类
  * @ingroup StockManage
@@ -111,7 +109,7 @@ public:
      * @param market 指定的市场标识
      * @return 相应的市场信息，如果相应的市场信息不存在，则返回Null<MarketInfo>()
      */
-    MarketInfo getMarketInfo(const string& market) const;
+    MarketInfo getMarketInfo(const string& market) const noexcept;
 
     /**
      * 获取指定市场代表指数证券
@@ -126,8 +124,11 @@ public:
      */
     StockTypeInfo getStockTypeInfo(uint32_t type) const;
 
+    /** 获取所有证券类型信息 */
+    vector<StockTypeInfo> getStockTypeInfoList() const;
+
     /** 获取市场简称列表 */
-    MarketList getAllMarket() const;
+    StringList getAllMarket() const;
 
     /**
      * 获取预定义的板块
@@ -182,11 +183,19 @@ public:
     const ZhBond10List& getZhBond10() const;
 
     /**
-     * 判断指定日期是否为节假日
-     * @param d 指定日期
+     * 判断指定时间对应的日期是否为节假日(包含周六、周日)
+     * @note 仅支持中国大陆市场, Null<Datetime>() 将抛出异常
+     * @param d 指定时间
      */
     bool isHoliday(const Datetime& d) const;
 
+    /**
+     * 粗略判断指定时间是否为交易时间
+     * @param d 指定时间，Null<Datetime>() 将抛出异常
+     * @param market 指定市场
+     */
+    bool isTradingHours(const Datetime& d, const string& market = "SH") const;
+
     const string& getHistoryFinanceFieldName(size_t ix) const;
     size_t getHistoryFinanceFieldIndex(const string& name) const;
     vector<std::pair<size_t, string>> getHistoryFinanceAllFields() const;

hikyuu/include/hikyuu/StockTypeInfo.h

@@ -38,6 +38,12 @@ public:
     StockTypeInfo();
     StockTypeInfo(uint32_t, const string&, price_t, price_t, int, double, double);
 
+    StockTypeInfo(const StockTypeInfo&) = default;
+    StockTypeInfo& operator=(const StockTypeInfo&) = default;
+
+    StockTypeInfo(StockTypeInfo&&);
+    StockTypeInfo& operator=(StockTypeInfo&&) noexcept;
+
     /** 获取证券类型 */
     uint32_t type() const {
         return m_type;

hikyuu/include/hikyuu/TransRecord.h

@@ -19,19 +19,13 @@ namespace hku {
  */
 class HKU_API TransRecord {
 public:
-    enum DIRECT {
-        BUY = 0,    /**< 买盘 */
-        SELL = 1,   /**< 卖盘 */
-        AUCTION = 2 /**< 集合竞价 */
-    };
-
     Datetime datetime;  // 时间
     price_t price;      // 成交均价
     price_t vol;        // 成交量
-    DIRECT direct;      // 买卖盘性质：1--sell 0--buy 2--集合竞价
+    int32_t direct;     // 买卖盘性质：1--sell 0--buy 2--集合竞价 其他未知
 
     TransRecord();
-    TransRecord(const Datetime& datetime, price_t price, price_t vol, DIRECT);
+    TransRecord(const Datetime& datetime, price_t price, price_t vol, int32_t direct);
 
     bool isValid() const;
 };

hikyuu/include/hikyuu/data_driver/KDataDriver.h

@@ -119,6 +119,11 @@ public:
      */
     virtual TransList getTransList(const string& market, const string& code, const KQuery& query);
 
+protected:
+    virtual bool isPythonObject() const {
+        return false;
+    }
+
 private:
     bool checkType();
 

hikyuu/include/hikyuu/doc.h

@@ -156,6 +156,10 @@
  *
  * @defgroup ThreadPool Thread Pool 线程池
  * @ingroup Utilities
+ *
+ * @defgroup Views 数据视图
+ * 数据视图, 属于用于显示、打印数据
+ * @ingroup Views
  */
 
 /**

hikyuu/include/hikyuu/global/GlobalSpotAgent.h

@@ -30,4 +30,12 @@ HKU_API SpotAgent* getGlobalSpotAgent();
 
 void HKU_API releaseGlobalSpotAgent();
 
+inline bool spotAgentIsRunning() {
+    return getGlobalSpotAgent()->isRunning();
+}
+
+inline bool spotAgentIsConnected() {
+    return getGlobalSpotAgent()->isConnected();
+}
+
 }  // namespace hku

hikyuu/include/hikyuu/global/agent/SpotAgent.h

@@ -68,6 +68,10 @@ public:
         return m_server_addr;
     }
 
+    bool isConnected() const {
+        return m_connected;
+    }
+
     /**
      * 增加收到 Spot 数据时的处理函数
      * @note 仅能在停止状态时执行此操作，否则将抛出异常
@@ -99,9 +103,9 @@ public:
     static void setQuotationServer(const string& server);
 
 private:
-    static string ms_pubUrl;  // 数据发送服务地址
-    static const char* ms_startTag;  // 批次数据接收起始标记，用于判断启动了新的批次数据接收
-    static const char* ms_endTag;  // 批次数据接收接收标记，用于判断该批次数据更新结束
+    static string ms_pubUrl;                 // 数据发送服务地址
+    static const char* ms_startTag;          // 批次数据接收起始标记，用于判断启动了新的批次数据接收
+    static const char* ms_endTag;            // 批次数据接收接收标记，用于判断该批次数据更新结束
     static const char* ms_spotTopic;         // 向数据发送服务订阅的主题
     static const size_t ms_startTagLength;   // 批次数据接收起始标记长度
     static const size_t ms_endTagLength;     // 批次数据接收结束标记长度
@@ -121,9 +125,10 @@ private:
     void work_thread();
 
 private:
-    enum STATUS { WAITING, RECEIVING };  // 等待新的批次数据，正在接收批次数据中
-    enum STATUS m_status = WAITING;      // 当前内部状态
-    std::atomic_bool m_stop = true;      // 结束代理工作标识
+    enum STATUS { WAITING, RECEIVING };    // 等待新的批次数据，正在接收批次数据中
+    enum STATUS m_status = WAITING;        // 当前内部状态
+    std::atomic_bool m_stop = true;        // 结束代理工作标识
+    std::atomic_bool m_connected = false;  // 是否已连接数据服务
 
     int m_revTimeout = 100;                         // 连接数据服务超时时长（毫秒）
     std::thread m_receiveThread;                    // 数据接收线程
@@ -137,7 +142,7 @@ private:
     // 下面属性被修改时需要加锁，以便可以使用多线程方式运行 strategy
     std::mutex m_mutex;
     list<std::function<void(const SpotRecord&)>> m_processList;  // 已注册的 spot 处理函数列表
-    list<std::function<void(Datetime)>> m_postProcessList;  // 已注册的批次后处理函数列表
+    list<std::function<void(Datetime)>> m_postProcessList;       // 已注册的批次后处理函数列表
 };
 
 }  // namespace hku

hikyuu/include/hikyuu/indicator/Indicator.h

@@ -110,6 +110,9 @@ public:
      */
     value_t get(size_t pos, size_t num = 0) const;
 
+    value_t front(size_t num = 0) const;
+    value_t back(size_t num = 0) const;
+
     /**
      * 获取指定位置的日期
      * @param pos 结果集中的位置
@@ -205,6 +208,43 @@ public:
 
     string str() const;
 
+public:
+    class Iterator {
+    private:
+        const Indicator& container;
+        size_t index;
+
+    public:
+        Iterator(const Indicator& cont, size_t idx) : container(cont), index(idx) {}
+
+        Iterator& operator++() {
+            index++;
+            return *this;
+        }
+
+        value_t operator*() const {
+            return container.get(index);
+        }
+
+        bool operator!=(const Iterator& other) const {
+            return index != other.index;
+        }
+
+        bool operator==(const Iterator& other) const {
+            return index == other.index;
+        }
+    };
+
+    Iterator begin() const {
+        return Iterator(*this, 0);
+    }
+
+    Iterator end() const {
+        return Iterator(*this, size());
+    }
+
+    typedef Iterator iterator;
+
 protected:
     IndicatorImpPtr m_imp;
 
@@ -285,6 +325,14 @@ inline Indicator::value_t Indicator::get(size_t pos, size_t num) const {
     return m_imp->get(pos, num);
 }
 
+inline Indicator::value_t Indicator::front(size_t num) const {
+    return m_imp->front(num);
+}
+
+inline Indicator::value_t Indicator::back(size_t num) const {
+    return m_imp->back(num);
+}
+
 inline Datetime Indicator::getDatetime(size_t pos) const {
     return m_imp ? m_imp->getDatetime(pos) : Null<Datetime>();
 }

hikyuu/include/hikyuu/indicator/Indicator2InImp.h

@@ -11,6 +11,7 @@
 
 namespace hku {
 
+/* note: 双输入但同一上下文*/
 class Indicator2InImp : public IndicatorImp {
 public:
     Indicator2InImp();
@@ -21,6 +22,14 @@ public:
 
     virtual IndicatorImpPtr _clone() override;
 
+    virtual bool needSelfAlikeCompare() const noexcept override {
+        return true;
+    }
+
+    virtual bool selfAlike(const IndicatorImp& other) const noexcept override;
+
+    virtual void getSelfInnerNodesWithInputConext(vector<IndicatorImpPtr>& nodes) const override;
+
 protected:
     Indicator prepare(const Indicator& ind);