DataFrame.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstReport/AstReportGlobal.hpp"
#include "DataSeries.hpp"
 
#include <string>
#include <vector>
#include <map>
#include <stdexcept>
 
AST_NAMESPACE_BEGIN
 
class AST_REPORT_API DataFrame
{
public:
    // ---- 构造 ----
 
    DataFrame();
 
    explicit DataFrame(const std::string& name);
 
    // ---- 基本信息 ----
 
    const std::string& name()  const { return name_; }
    void               setName(const std::string& name) { name_ = name; }
 
    size_t rowCount()    const;
    size_t columnCount() const { return columns_.size(); }
    bool   empty()       const { return columns_.empty(); }
 
    std::vector<std::string> columnNames() const;
 
    // ---- 列管理 ----
 
    DataSeries& addColumn(const std::string& name, EDataType type, size_t size);
 
    DataSeries& addColumn(DataSeries&& series);
 
    bool removeColumn(const std::string& name);
 
    bool hasColumn(const std::string& name) const;
 
    // ---- 列访问 ----
 
    DataSeries& operator[](const std::string& name);
 
    const DataSeries& operator[](const std::string& name) const;
 
    DataSeries& column(size_t index);
 
    const DataSeries& column(size_t index) const;
 
    // ---- 迭代器 ----
 
    using iterator       = std::vector<DataSeries>::iterator;
    using const_iterator = std::vector<DataSeries>::const_iterator;
 
    iterator       begin()       { return columns_.begin(); }
    iterator       end()         { return columns_.end(); }
    const_iterator begin() const { return columns_.begin(); }
    const_iterator end()   const { return columns_.end(); }
 
    void reserve(size_t size){columns_.reserve(size);}
private:
    void rebuildIndex();
 
    std::string                    name_;
    std::vector<DataSeries>        columns_;
    std::map<std::string, size_t>  index_;  
};
 
AST_NAMESPACE_END