// Copyright 2013 Dolphin Emulator Project
// Licensed under GPLv2
// Refer to the license.txt file included.

#include "common/common.h"
#include "common/debug_interface.h"
#include "common/break_points.h"

#include <sstream>
#include <algorithm>

bool BreakPoints::IsAddressBreakPoint(u32 _iAddress)
{
    for (TBreakPoints::iterator i = m_BreakPoints.begin(); i != m_BreakPoints.end(); ++i)
        if (i->iAddress == _iAddress)
            return true;
    return false;
}

bool BreakPoints::IsTempBreakPoint(u32 _iAddress)
{
    for (TBreakPoints::iterator i = m_BreakPoints.begin(); i != m_BreakPoints.end(); ++i)
        if (i->iAddress == _iAddress && i->bTemporary)
            return true;
    return false;
}

BreakPoints::TBreakPointsStr BreakPoints::GetStrings() const
{
    TBreakPointsStr bps;
    for (TBreakPoints::const_iterator i = m_BreakPoints.begin();
        i != m_BreakPoints.end(); ++i)
    {
        if (!i->bTemporary)
        {
            std::stringstream bp;
            bp << std::hex << i->iAddress << " " << (i->bOn ? "n" : "");
            bps.push_back(bp.str());
        }
    }

    return bps;
}

void BreakPoints::AddFromStrings(const TBreakPointsStr& bps)
{
    for (TBreakPointsStr::const_iterator i = bps.begin(); i != bps.end(); ++i)
    {
        TBreakPoint bp;
        std::stringstream bpstr;
        bpstr << std::hex << *i;
        bpstr >> bp.iAddress;
        bp.bOn = i->find("n") != i->npos;
        bp.bTemporary = false;
        Add(bp);
    }
}

void BreakPoints::Add(const TBreakPoint& bp)
{
    if (!IsAddressBreakPoint(bp.iAddress))
    {
        m_BreakPoints.push_back(bp);
        //if (jit)
        //    jit->GetBlockCache()->InvalidateICache(bp.iAddress, 4);
    }
}

void BreakPoints::Add(u32 em_address, bool temp)
{
    if (!IsAddressBreakPoint(em_address)) // only add new addresses
    {
        TBreakPoint pt; // breakpoint settings
        pt.bOn = true;
        pt.bTemporary = temp;
        pt.iAddress = em_address;

        m_BreakPoints.push_back(pt);

        //if (jit)
        //    jit->GetBlockCache()->InvalidateICache(em_address, 4);
    }
}

void BreakPoints::Remove(u32 em_address)
{
    for (TBreakPoints::iterator i = m_BreakPoints.begin(); i != m_BreakPoints.end(); ++i)
    {
        if (i->iAddress == em_address)
        {
            m_BreakPoints.erase(i);
            //if (jit)
            //    jit->GetBlockCache()->InvalidateICache(em_address, 4);
            return;
        }
    }
}

void BreakPoints::Clear()
{
    //if (jit)
    //{
    //    std::for_each(m_BreakPoints.begin(), m_BreakPoints.end(),
    //        [](const TBreakPoint& bp)
    //        {
    //            jit->GetBlockCache()->InvalidateICache(bp.iAddress, 4);
    //        }
    //    );
    //}
    
    m_BreakPoints.clear();
}

MemChecks::TMemChecksStr MemChecks::GetStrings() const
{
    TMemChecksStr mcs;
    for (TMemChecks::const_iterator i = m_MemChecks.begin();
        i != m_MemChecks.end(); ++i)
    {
        std::stringstream mc;
        mc << std::hex << i->StartAddress;
        mc << " " << (i->bRange ? i->EndAddress : i->StartAddress) << " " <<
            (i->bRange ? "n" : "") << (i->OnRead ? "r" : "") <<
            (i->OnWrite ? "w" : "") << (i->Log ? "l" : "") << (i->Break ? "p" : "");
        mcs.push_back(mc.str());
    }

    return mcs;
}

void MemChecks::AddFromStrings(const TMemChecksStr& mcs)
{
    for (TMemChecksStr::const_iterator i = mcs.begin(); i != mcs.end(); ++i)
    {
        TMemCheck mc;
        std::stringstream mcstr;
        mcstr << std::hex << *i;
        mcstr >> mc.StartAddress;
        mc.bRange    = i->find("n") != i->npos;
        mc.OnRead    = i->find("r") != i->npos;
        mc.OnWrite    = i->find("w") != i->npos;
        mc.Log        = i->find("l") != i->npos;
        mc.Break    = i->find("p") != i->npos;
        if (mc.bRange)
            mcstr >> mc.EndAddress;
        else
            mc.EndAddress = mc.StartAddress;
        Add(mc);
    }
}

void MemChecks::Add(const TMemCheck& _rMemoryCheck)
{
    if (GetMemCheck(_rMemoryCheck.StartAddress) == 0)
        m_MemChecks.push_back(_rMemoryCheck);
}

void MemChecks::Remove(u32 _Address)
{
    for (TMemChecks::iterator i = m_MemChecks.begin(); i != m_MemChecks.end(); ++i)
    {
        if (i->StartAddress == _Address)
        {
            m_MemChecks.erase(i);
            return;
        }
    }
}

TMemCheck *MemChecks::GetMemCheck(u32 address)
{
    for (TMemChecks::iterator i = m_MemChecks.begin(); i != m_MemChecks.end(); ++i)
    {
        if (i->bRange)
        {
            if (address >= i->StartAddress && address <= i->EndAddress)
                return &(*i);
        }
        else if (i->StartAddress == address)
            return &(*i);
    }

    // none found
    return 0;
}

void TMemCheck::Action(DebugInterface *debug_interface, u32 iValue, u32 addr,
                        bool write, int size, u32 pc)
{
    if ((write && OnWrite) || (!write && OnRead))
    {
        if (Log)
        {
            INFO_LOG(MEMMAP, "CHK %08x (%s) %s%i %0*x at %08x (%s)",
                pc, debug_interface->getDescription(pc).c_str(),
                write ? "Write" : "Read", size*8, size*2, iValue, addr,
                debug_interface->getDescription(addr).c_str()
                );
        }
        if (Break)
            debug_interface->breakNow();
    }
}
