unity_sframe_interface.hpp

/* Copyright © 2017 Apple Inc. All rights reserved.
 *
 * Use of this source code is governed by a BSD-3-clause license that can
 * be found in the LICENSE.txt file or at https://opensource.org/licenses/BSD-3-Clause
 */
#ifndef TURI_UNITY_SFRAME_INTERFACE_HPP
#define TURI_UNITY_SFRAME_INTERFACE_HPP
#include <memory>
#include <vector>
#include <string>
#include <core/data/flexible_type/flexible_type.hpp>
#include <core/storage/sframe_data/dataframe.hpp>
#include <model_server/lib/options_map.hpp>
#include <core/system/cppipc/magic_macros.hpp>
#include <model_server/lib/api/unity_sarray_interface.hpp>

namespace turi {

class unity_sframe_base;
typedef std::map<std::string, flex_type_enum> str_flex_type_map;
typedef std::map<std::string, flexible_type> csv_parsing_config_map;
typedef std::map<std::string, std::string> string_map;
typedef std::map<std::string, std::shared_ptr<unity_sarray_base>> csv_parsing_errors;

GENERATE_INTERFACE_AND_PROXY(unity_sframe_base, unity_sframe_proxy,
      (void, construct_from_dataframe, (const dataframe_t&))
      (void, construct_from_sframe_index, (std::string))
      (csv_parsing_errors, construct_from_csvs, (std::string)(csv_parsing_config_map)(str_flex_type_map))
      (void, clear, )
      (size_t, size, )
      (std::shared_ptr<unity_sarray_base>, transform, (const std::string&)(flex_type_enum)(bool)(uint64_t))
      (std::shared_ptr<unity_sarray_base>, transform_native, (const function_closure_info&)(flex_type_enum)(bool)(uint64_t))
      (std::shared_ptr<unity_sframe_base>, flat_map, (const std::string&)(std::vector<std::string>)
                                     (std::vector<flex_type_enum>)(bool)(uint64_t))
      (void, save_frame, (std::string) )
      (void, save_frame_reference, (std::string) )
      (size_t, num_columns, )
      (std::vector<flex_type_enum>, dtype, )
      (std::vector<std::string>, column_names, )
      (size_t, column_index, (const std::string &))
      (std::shared_ptr<unity_sframe_base>, head, (size_t))
      (std::shared_ptr<unity_sframe_base>, tail, (size_t))
      (dataframe_t, _head, (size_t))
      (dataframe_t, _tail, (size_t))
      (std::shared_ptr<unity_sframe_base>, logical_filter, (std::shared_ptr<unity_sarray_base>))
      (std::shared_ptr<unity_sframe_base>, select_columns, (const std::vector<std::string>&))
      (std::shared_ptr<unity_sarray_base>, select_column, (const std::string&))
      (void, add_column, (std::shared_ptr<unity_sarray_base >)(const std::string&))
      (void, add_columns, (std::list<std::shared_ptr<unity_sarray_base >>)(std::vector<std::string>))
      (void, set_column_name, (size_t)(std::string))
      (void, remove_column, (size_t))
      (void, swap_columns, (size_t)(size_t))
      (void, begin_iterator, )
      (std::vector<std::vector<flexible_type>>, iterator_get_next, (size_t))
      (void, save_as_csv, (const std::string&)(csv_parsing_config_map))
      (std::shared_ptr<unity_sframe_base>, sample, (float)(uint64_t)(bool))
      (std::list<std::shared_ptr<unity_sframe_base>>, random_split, (float)(uint64_t)(bool))
      (std::shared_ptr<unity_sframe_base>, shuffle, )
      (std::shared_ptr<unity_sframe_base>, groupby_aggregate, (const std::vector<std::string>&)
                                              (const std::vector<std::vector<std::string>>&)
                                              (const std::vector<std::string>&)
                                              (const std::vector<std::string>&))
      (std::shared_ptr<unity_sframe_base>, append, (std::shared_ptr<unity_sframe_base>))
      (void, materialize, )
      (bool, is_materialized, )
      (bool, has_size, )
      (std::string, query_plan_string, )
      (std::shared_ptr<unity_sframe_base>, join, (std::shared_ptr<unity_sframe_base>)(const std::string)(const string_map&))
      (std::shared_ptr<unity_sframe_base>, join_with_custom_name, (std::shared_ptr<unity_sframe_base>)(const std::string)(const string_map&)(const string_map&))
      (std::shared_ptr<unity_sframe_base>, sort, (const std::vector<std::string>&)(const std::vector<int>&))
      (std::shared_ptr<unity_sarray_base>, pack_columns, (const std::vector<std::string>&)(const std::vector<std::string>&)(flex_type_enum)(const flexible_type&))
      (std::shared_ptr<unity_sframe_base>, stack,  (const std::string&)(const std::vector<std::string>&)(const std::vector<flex_type_enum>&)(bool))
      (std::shared_ptr<unity_sframe_base>, copy_range, (size_t)(size_t)(size_t))
      (std::list<std::shared_ptr<unity_sframe_base>>, drop_missing_values, (const std::vector<std::string>&)(bool)(bool)(bool))
      (dataframe_t, to_dataframe, )
      (void, delete_on_close, )
      (void, explore, (const std::string&)(const std::string&))
      (void, show, (const std::string&))
      (std::shared_ptr<model_base>, plot, )
    )
} // namespace turi
#endif // TURI_UNITY_SFRAME_INTERFACE_HPP