hikyuu2/hikyuu_cpp/hikyuu/KDataImp.cpp

/*
 * KDataImp.cpp
 *
 *  Created on: 2013-2-4
 *      Author: fasiondog
 */

#include "StockManager.h"
#include "KDataImp.h"

namespace hku {

KDataImp::KDataImp() : m_start(0), m_end(0) {}

KDataImp::KDataImp(const Stock& stock, const KQuery& query)
: m_query(query), m_stock(stock), m_start(0), m_end(0) {
    if (m_stock.isNull()) {
        return;
    }

    bool sucess = m_stock.getIndexRange(query, m_start, m_end);
    if (!sucess) {
        m_start = 0;
        m_end = 0;
        return;
    }
}

KDataImp::~KDataImp() {}

KRecord KDataImp::getKRecord(size_t pos) const {
    return empty() ? Null<KRecord>() : m_stock.getKRecord(m_start + pos, m_query.kType());
}

size_t KDataImp::getPos(const Datetime& datetime) const {
    KRecord null_record;
    if (empty()) {
        return Null<size_t>();
    }

    size_t mid, low = 0, high = size() - 1;
    while (low <= high) {
        if (datetime > getKRecord(high).datetime) {
            mid = high + 1;
            break;
        }

        if (getKRecord(low).datetime >= datetime) {
            mid = low;
            break;
        }

        mid = (low + high) / 2;
        if (datetime > getKRecord(mid).datetime) {
            low = mid + 1;
        } else {
            high = mid - 1;
        }
    }

    if (mid >= size()) {
        return Null<size_t>();
    }

    KRecord tmp = getKRecord(mid);
    return tmp.datetime == datetime ? mid : Null<size_t>();
}

} /* namespace hku */