/**
 * Copyright (c) 2026 Alexey Gavrilov <alexey.gavrilov@mail.com>
 *
 * This file is part of ObjectiveOS.
 *
 * ObjectiveOS is free software: you can redistribute it and/or modify it under
 * the terms of the GNU General Public License as published by the Free Software
 * Foundation, either version 3 of the License, or (at your option) any later
 * version.
 *
 * ObjectiveOS is distributed in the hope that it will be useful, but WITHOUT ANY
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
 * A PARTICULAR PURPOSE. See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * ObjectiveOS. If not, see <https://www.gnu.org/licenses/>. 
 */

#include "VGADisplay.h"

VGADisplay out::display;

VGADisplay::VGADisplay()
{
    isBright = isBlink = shift = 0;
    xCurr = yCurr = 0;
    color = (char)VGADisplayInfo::colorAttr::gray;
    videoMem = (dchar *)VGADisplayInfo::videoMemAddr;
};

VGADisplay::VGADisplay(uint16 x, uint16 y)
{
    isBright = isBlink = shift = 0;
    xCurr = yCurr = 0;
    color = (char)VGADisplayInfo::colorAttr::gray;
    videoMem = (dchar *)VGADisplayInfo::videoMemAddr;
};

VGADisplay& VGADisplay::operator<<(const char &c)
{
    printChar(c);

    return *this;
};

VGADisplay& VGADisplay::operator<<(const uint32 number)
{
    IntConverter conv;

    (*this) << conv.uintToChar(number);

    return *this;
};

VGADisplay& VGADisplay::operator<<(const uint64 number)
{
    IntConverter conv;

    (*this) << conv.uintToChar(number);

    return *this;
};

VGADisplay& VGADisplay::operator++(int)
{
    if(++xCurr == VGADisplayInfo::width)
    {
        yCurr++; 
        xCurr = 0; 
    }
    if(yCurr == VGADisplayInfo::height)
    {
        if(shift)
        { 
            this->doShift(); 
            xCurr = 0; 
            yCurr--;
        }
        else xCurr = 0, yCurr = 0;
    }
    videoMem = (dchar *)(VGADisplayInfo::videoMemAddr) 
                + yCurr * VGADisplayInfo::width + xCurr;

    return *this;
};

VGADisplay& VGADisplay::operator<<(const char *c)
{
    for(int i = 0; c[i] != '\0'; i++)
    {
        (*this) << c[i];
    }

    return *this;
};

void VGADisplay::doShift()
{
    for(uint16 i = 0; i < VGADisplayInfo::height; i++)
    {
        for(uint16 j = 0; j < VGADisplayInfo::width; j++)
        {
            *((dchar *)(VGADisplayInfo::videoMemAddr) + (i - 1) * VGADisplayInfo::width + j) = *((dchar *)(VGADisplayInfo::videoMemAddr) + i * VGADisplayInfo::width + j);
            *((dchar *)(VGADisplayInfo::videoMemAddr) + i * VGADisplayInfo::width + j) = (*((dchar *)(VGADisplayInfo::videoMemAddr) + i * VGADisplayInfo::width + j) & 0x00) | ' ';
        }
    }
};

void VGADisplay::setShift(bool status)
{
    shift = status;
};

void VGADisplay::setColor(char colorAttr)
{
    color = colorAttr;
}; 

void VGADisplay::setBlink(bool blinkAttr)
{
    isBlink = blinkAttr;
};

void VGADisplay::setBright(bool brigthAttr)
{
    isBright = brigthAttr;
};

void VGADisplay::goNewLine()
{
    uint16 yCurrOld = yCurr;

    while(yCurrOld == yCurr) (*this)++;
};

void VGADisplay::goHorizontalTab()
{
    while(xCurr % 8 != 0) (*this)++;
};

bool VGADisplay::printChar(const char &c)
{
    switch(c)
    {
        case '\0': return true;
        case '\n': this->goNewLine(); return true;
        case '\t': this->goHorizontalTab(); return true;
        default:
                   {
                        char attr = isBlink << 8 | isBright << 4
                                                 | color;
                        *videoMem = c | attr << 8; 
                        (*this)++;
                        return false;
                   }
    }
};

void VGADisplay::clearScreen()
{
    xCurr = yCurr = 0;
    videoMem = (dchar *)VGADisplayInfo::videoMemAddr;
    uint16 maxPos = VGADisplayInfo::height * VGADisplayInfo::width * 2;

    for(uint16 pos = 0; pos < maxPos; pos++) *this << ' ';
};