luau/Ast/src/Location.cpp

// This file is part of the Luau programming language and is licensed under MIT License; see LICENSE.txt for details
#include "Luau/Location.h"
#include <string>

namespace Luau
{

Position::Position(unsigned int line, unsigned int column)
    : line(line)
    , column(column)
{
}

bool Position::operator==(const Position& rhs) const
{
    return this->column == rhs.column && this->line == rhs.line;
}

bool Position::operator!=(const Position& rhs) const
{
    return !(*this == rhs);
}

bool Position::operator<(const Position& rhs) const
{
    if (line == rhs.line)
        return column < rhs.column;
    else
        return line < rhs.line;
}

bool Position::operator>(const Position& rhs) const
{
    if (line == rhs.line)
        return column > rhs.column;
    else
        return line > rhs.line;
}

bool Position::operator<=(const Position& rhs) const
{
    return *this == rhs || *this < rhs;
}

bool Position::operator>=(const Position& rhs) const
{
    return *this == rhs || *this > rhs;
}

void Position::shift(const Position& start, const Position& oldEnd, const Position& newEnd)
{
    if (*this >= start)
    {
        if (this->line > oldEnd.line)
            this->line += (newEnd.line - oldEnd.line);
        else
        {
            this->line = newEnd.line;
            this->column += (newEnd.column - oldEnd.column);
        }
    }
}

Location::Location()
    : begin(0, 0)
    , end(0, 0)
{
}

Location::Location(const Position& begin, const Position& end)
    : begin(begin)
    , end(end)
{
}

Location::Location(const Position& begin, unsigned int length)
    : begin(begin)
    , end(begin.line, begin.column + length)
{
}

Location::Location(const Location& begin, const Location& end)
    : begin(begin.begin)
    , end(end.end)
{
}

bool Location::operator==(const Location& rhs) const
{
    return this->begin == rhs.begin && this->end == rhs.end;
}

bool Location::operator!=(const Location& rhs) const
{
    return !(*this == rhs);
}

bool Location::encloses(const Location& l) const
{
    return begin <= l.begin && end >= l.end;
}

bool Location::overlaps(const Location& l) const
{
    return (begin <= l.begin && end >= l.begin) || (begin <= l.end && end >= l.end) || (begin >= l.begin && end <= l.end);
}

bool Location::contains(const Position& p) const
{
    return begin <= p && p < end;
}

bool Location::containsClosed(const Position& p) const
{
    return begin <= p && p <= end;
}

void Location::extend(const Location& other)
{
    if (other.begin < begin)
        begin = other.begin;
    if (other.end > end)
        end = other.end;
}

void Location::shift(const Position& start, const Position& oldEnd, const Position& newEnd)
{
    begin.shift(start, oldEnd, newEnd);
    end.shift(start, oldEnd, newEnd);
}

std::string Location::toString(int offset, bool useBegin) const
{
    const Position& pos = useBegin ? this->begin : this->end;
    std::string line{std::to_string(pos.line + offset)};
    std::string column{std::to_string(pos.column + offset)};
    return "(" + line + ", " + column + ")";
}

} // namespace Luau