xlnt/source/cell.cpp

#include <algorithm>
#include <locale>
#include <sstream>

#include "cell/cell.hpp"
#include "cell/cell_reference.hpp"
#include "common/datetime.hpp"
#include "common/relationship.hpp"
#include "worksheet/worksheet.hpp"
#include "detail/cell_impl.hpp"
#include "common/exceptions.hpp"
#include "workbook/workbook.hpp"

namespace xlnt {
    
const xlnt::color xlnt::color::black(0);
const xlnt::color xlnt::color::white(1);

const std::unordered_map<std::string, int> cell::ErrorCodes =
{
    {"#NULL!", 0},
    {"#DIV/0!", 1},
    {"#VALUE!", 2},
    {"#REF!", 3},
    {"#NAME?", 4},
    {"#NUM!", 5},
    {"#N/A!", 6}
};

cell::cell() : d_(nullptr)
{
}

cell::cell(detail::cell_impl *d) : d_(d)
{
}
    
cell::cell(worksheet worksheet, const cell_reference &reference, const std::string &initial_value) : d_(nullptr)
{
    cell self = worksheet.get_cell(reference);
    d_ = self.d_;

    if(initial_value != "")
    {
        *this = initial_value;
    }
}

std::string cell::get_internal_value_string() const
{
    switch(d_->type_)
    {
    case type::string:
        return d_->string_value;
    case type::formula:
        return d_->string_value;
    case type::error:
        return d_->string_value;
    default:
        throw std::runtime_error("bad enum");
    }
}

bool cell::has_style() const
{
    return d_->style_ != nullptr;
}
    
long double cell::get_internal_value_numeric() const
{
    switch(d_->type_)
    {
    case type::numeric:
        return d_->numeric_value;
    case type::boolean:
        return d_->numeric_value == 0 ? 0 : 1;
    default:
        throw std::runtime_error("bad enum");
    }
}

row_t cell::get_row() const
{
    return d_->row + 1;
}

std::string cell::get_column() const
{
    return cell_reference::column_string_from_index(d_->column + 1);
}

std::vector<std::string> split_string(const std::string &string, char delim = ' ')
{
    std::stringstream ss(string);
    std::string part;
    std::vector<std::string> parts;
    while(std::getline(ss, part, delim))
    {
        parts.push_back(part);
    }
    return parts;
}

cell::type cell::data_type_for_value(const std::string &value)
{
    if(value.empty())
    {
        return type::null;
    }
    
    if(value[0] == '=')
    {
        return type::formula;
    }
    else if(value[0] == '0')
    {
        if(value.length() > 1)
        {
            if(value[1] == '.' || (value.length() > 2 && (value[1] == 'e' || value[1] == 'E')))
            {
                auto first_non_number = std::find_if(value.begin() + 2, value.end(),
                                                     [](char c) { return !std::isdigit(c, std::locale::classic()); });
                if(first_non_number == value.end())
                {
                    return type::numeric;
                }
            }
            auto split = split_string(value, ':');
            if(split.size() == 2 || split.size() == 3)
            {
                for(auto part : split)
                {
                    try
                    {
                        std::stoi(part);
                    }
                    catch(std::invalid_argument)
                    {
                        return type::string;
                    }
                }
                return type::numeric;
            }
            else
            {
                return type::string;
            }
        }
        return type::numeric;
    }
    else if(value[0] == '#')
    {
        return type::error;
    }
    else
    {
        char *p;
        strtod(value.c_str(), &p);
        if(*p != 0)
        {
            static const std::vector<std::string> possible_booleans = {"TRUE", "true", "FALSE", "false"};
            if(std::find(possible_booleans.begin(), possible_booleans.end(), value) != possible_booleans.end())
            {
                return type::boolean;
            }
            if(value.back() == '%')
            {
                strtod(value.substr(0, value.length() - 1).c_str(), &p);
                if(*p == 0)
                {
                    return type::numeric;
                }
            }
            return type::string;
        }
        else
        {
            return type::numeric;
        }
    }
}

void cell::set_explicit_value(const std::string &value, type data_type)
{
    d_->type_ = data_type;

    switch(data_type)
    {
        case type::null:
	  if(value != "")
	    {
	      throw data_type_exception();
	    }
	  return;
        case type::formula: 
	  if(value.length() == 0 || value[0] != '=')
	    {
	      throw data_type_exception();
	    }
	  d_->string_value = value; 
	  return;
        case type::error: d_->string_value = value; return;
        case type::boolean: d_->numeric_value = value == "true"; return;
        case type::numeric: d_->numeric_value = std::stod(value); return;
        case type::string: d_->string_value = value; return;
        default: throw std::runtime_error("bad enum");
    }
}

void cell::set_explicit_value(int value, type data_type)
{
    d_->type_ = data_type;

    switch(data_type)
    {
        case type::numeric: d_->numeric_value = value; return;
        case type::string: d_->string_value = std::to_string(value); return;
        default: throw data_type_exception();
    }
}

void cell::set_explicit_value(double value, type data_type)
{
    d_->type_ = data_type;

    switch(data_type)
    {
        case type::numeric: d_->numeric_value = value; return;
        case type::string: d_->string_value = std::to_string(value); return;
        default: throw data_type_exception();
    }
}

void cell::set_merged(bool merged)
{
    d_->merged = merged;
}

bool cell::is_merged() const
{
    return d_->merged;
}

bool cell::is_date() const
{
    return d_->is_date_;
}

cell_reference cell::get_reference() const
{
    return {d_->column, d_->row};
}

bool cell::operator==(std::nullptr_t) const
{
    return d_ == nullptr;
}

bool cell::operator==(int comparand) const
{
    return d_->type_ == type::numeric && d_->numeric_value == comparand;
}

bool cell::operator==(double comparand) const
{
    return d_->type_ == type::numeric && d_->numeric_value == comparand;
}

bool cell::operator==(const std::string &comparand) const
{
    if(d_->type_ == type::string || d_->type_ == type::formula)
    {
        return d_->string_value == comparand;
    }
    return false;
}

bool cell::operator==(const char *comparand) const
{
    return *this == std::string(comparand);
}

bool cell::operator==(const time &comparand) const
{
    if(d_->type_ != type::numeric)
    {
        return false;
    }
    
    return time::from_number(d_->numeric_value) == comparand;
}

bool cell::operator==(const date &comparand) const
{
    if(d_->type_ != type::numeric)
    {
        return false;
    }
    
    auto base_year = worksheet(d_->parent_).get_parent().get_base_year();
    return date::from_number((int)d_->numeric_value, base_year) == comparand;
}

bool cell::operator==(const datetime &comparand) const
{
    if(d_->type_ != type::numeric)
    {
        return false;
    }
    
    auto base_year = worksheet(d_->parent_).get_parent().get_base_year();
    return datetime::from_number(d_->numeric_value, base_year) == comparand;
}

bool cell::operator==(const cell &comparand) const
{
    if(comparand == nullptr)
    {
        return d_ == nullptr;
    }

    if(comparand.get_data_type() != get_data_type())
    {
        return false;
    }

    switch(get_data_type())
    {
    case type::boolean:
        return d_->numeric_value == comparand.d_->numeric_value;
    case type::error:
        return d_->string_value == comparand.d_->string_value;
    case type::string:
        return d_->string_value == comparand.d_->string_value;
    case type::formula:
        return d_->string_value == comparand.d_->string_value;
    case type::null:
        return true;
    case type::numeric:
        if(is_date() && comparand.is_date())
        {
            auto base_year = worksheet(d_->parent_).get_parent().get_base_year();
            auto other_base_year = worksheet(comparand.d_->parent_).get_parent().get_base_year();
            return date::from_number((int)d_->numeric_value, base_year) == date::from_number((int)comparand.d_->numeric_value, other_base_year);
        }
        return d_->numeric_value == comparand.d_->numeric_value;
    }

    return false;
}

bool operator==(int comparand, const xlnt::cell &cell)
{
    return cell == comparand;
}

bool operator==(const char *comparand, const xlnt::cell &cell)
{
    return cell == comparand;
}

bool operator==(const std::string &comparand, const xlnt::cell &cell)
{
    return cell == comparand;
}

bool operator==(const time &comparand, const xlnt::cell &cell)
{
    return cell == comparand;
}

bool operator==(const date &comparand, const xlnt::cell &cell)
{
    return cell == comparand;
}

bool operator==(const datetime &comparand, const xlnt::cell &cell)
{
    return cell == comparand;
}

style &cell::get_style()
{
    if(d_->style_ == nullptr)
    {
        d_->style_ = new style();
    }
    return *d_->style_;
}

const style &cell::get_style() const
{
    if(d_->style_ == nullptr)
    {
        d_->style_ = new style();
    }
    return *d_->style_;
}

xlnt::cell::type cell::get_data_type() const
{
    return (type)d_->type_;
}

cell &cell::operator=(const cell &rhs)
{
    d_ = rhs.d_;
    return *this;
}

cell &cell::operator=(int value)
{
    d_->is_date_ = false;
    d_->type_ = type::numeric;
    d_->numeric_value = value;
    return *this;
}

cell &cell::operator=(double value)
{
    d_->is_date_ = false;
    d_->type_ = type::numeric;
    d_->numeric_value = value;
    return *this;
}

cell &cell::operator=(long int value)
{
    d_->is_date_ = false;
    d_->type_ = type::numeric;
    d_->numeric_value = value;
    return *this;
}

cell &cell::operator=(long long value)
{
    d_->is_date_ = false;
    d_->type_ = type::numeric;
    d_->numeric_value = (long double)value;
    return *this;
}

cell &cell::operator=(long double value)
{
    d_->is_date_ = false;
    d_->type_ = type::numeric;
    d_->numeric_value = value;
    return *this;
}

cell &cell::operator=(bool value)
{
    d_->is_date_ = false;
    d_->type_ = type::boolean;
    d_->numeric_value = value ? 1 : 0;
    return *this;
}


cell &cell::operator=(const std::string &value)
{
    d_->is_date_ = false;
    d_->type_ = data_type_for_value(value);
    
    switch((type)d_->type_)
    {
        case type::formula:
            d_->string_value = value;
            break;
        case type::numeric:
	  if(value.find(':') != std::string::npos)
	    {
	      d_->is_date_ = true;
	      d_->numeric_value = time(value).to_number();
	    }
	  else if(value.back() == '%')
	    {
            d_->numeric_value = std::stod(value.substr(0, value.length() - 1)) / 100;
            get_style().get_number_format().set_format_code(xlnt::number_format::format::percentage);
        }
        else
        {
	      d_->numeric_value = std::stod(value);
	    }
            break;
        case type::boolean:
            d_->numeric_value = value == "TRUE" || value == "true";
            break;
        case type::error:
            d_->string_value = value;
            break;
        case type::string:
            d_->string_value = value;
            break;
        case type::null:
            break;
        default:
            throw std::runtime_error("bad enum");
    }
    
    return *this;
}

cell &cell::operator=(const char *value)
{
    return *this = std::string(value);
}

cell &cell::operator=(const time &value)
{
    d_->type_ = type::numeric;
    d_->numeric_value = value.to_number();
    d_->is_date_ = true;
    return *this;
}

cell &cell::operator=(const date &value)
{
    d_->type_ = type::numeric;
    auto base_year = worksheet(d_->parent_).get_parent().get_base_year();
    d_->numeric_value = value.to_number(base_year);
    d_->is_date_ = true;
    return *this;
}

cell &cell::operator=(const datetime &value)
{
    d_->type_ = type::numeric;
    auto base_year = worksheet(d_->parent_).get_parent().get_base_year();
    d_->numeric_value = value.to_number(base_year);
    d_->is_date_ = true;
    return *this;
}

std::string cell::to_string() const
{
    return "<Cell " + worksheet(d_->parent_).get_title() + "." + get_reference().to_string() + ">";
}

relationship cell::get_hyperlink() const
{
    if(!d_->has_hyperlink_)
    {
        throw std::runtime_error("no hyperlink set");
    }

    return d_->hyperlink_;
}

bool cell::has_hyperlink() const
{
    return d_->has_hyperlink_;
}

void cell::set_hyperlink(const std::string &hyperlink)
{
    if(hyperlink.length() == 0 || std::find(hyperlink.begin(), hyperlink.end(), ':') == hyperlink.end())
    {
        throw data_type_exception();
    }

    d_->has_hyperlink_ = true;
    d_->hyperlink_ = worksheet(d_->parent_).create_relationship("hyperlink", hyperlink);

    if(d_->type_ == type::null)
    {
        *this = hyperlink;
    }
}

void cell::set_null()
{
    d_->type_ = type::null;
}

void cell::set_formula(const std::string &formula)
{
    if(formula.length() == 0 || formula[0] != '=')
    {
        throw data_type_exception();
    }

    d_->type_ = type::formula;
    d_->string_value = formula;
}

void cell::set_error(const std::string &error)
{
    if(error.length() == 0 || error[0] != '#')
    {
        throw data_type_exception();
    }

    d_->type_ = type::error;
    d_->string_value = error;
}

} // namespace xlnt