source: cpp/frams/util/multimap.cpp @ 438

// This file is a part of Framsticks SDK.  http://www.framsticks.com/
// Copyright (C) 1999-2015  Maciej Komosinski and Szymon Ulatowski.
// See LICENSE.txt for details.

#include "multimap.h"
#include <stdio.h>

int MultiMap::getBegin() const
{
if (isEmpty()) return 0;
return getBegin(0);
}

int MultiMap::getEnd() const
{
if (isEmpty()) return -1;
return getBegin(mappingCount()-1)-1;
}

int MultiMap::findData(int x) const
{
if (isEmpty()) return -1;
if (getBegin(0)>x) return -1;
int a,b,c;
int n=mappingCount()-1;
a=0; b=n; // find c = last range with begin<=x
while(1){
        c=(a+b+1)/2;
        if (getBegin(c)<=x)
                {
                if ((c==n)||(getBegin(c+1)>x)) return c;
                a=c;
                }
        else
                {
                b=c-1;
                }
        }
}

int MultiMap::addRange(int &r,const MultiRange& mr)
{
// precondition: 0 <= r < (mappingCount()-1)
// 1.maybe change mapping in this range
int ret=0;
SingleMapping *sm=getMapping(r);
if (sm->to.contains(mr)) return 0; // do nothing
sm->to.add(mr);
// after adding mr to this mapping it could be the same as the previous/next one
if (r>0) // prev exists
        {
        SingleMapping *prev=getMapping(r-1);
        if (prev->to == sm->to)
                { // splice with prev
                removeData(r);
                delete sm;
                sm=prev;
                r--;
                ret--;
                }
        }
if (r<(mappingCount()-2)) // next exists
        {
        SingleMapping *next=getMapping(r+1);
        if (next->to == sm->to)
                { // splice with next
                removeData(r+1);
                delete next;
                ret--;
                r--;
                }
        }
return ret;
}

int MultiMap::addRangeBeginEnd(int &r,int begin,int end,const MultiRange& mr)
{
// precondition: -1 <= r < mappingCount()
//               begin <= end
//               begin >= mapping.begin
//               end < nextmapping.begin
SingleMapping *m,*m2,*m1;
if (r<0)
        {
        if (mappingCount())
                {
                m=getMapping(0);
                if (end==(m->begin-1))
                        { // adjacent to m
                        if (m->to==mr)
                                { // shift only
                                m->begin=begin;
                                return 0;
                                }
                        // single add
                        SingleMapping *newmap=new SingleMapping(begin,mr);
                        addData(0,newmap);
                        r=0;
                        return 1;
                        }
                }
        // double add
        SingleMapping *newmap=new SingleMapping(begin,mr);
        SingleMapping *newmap2=new SingleMapping(end+1);
        addData(0,newmap);
        addData(1,newmap2);
        r=1;
        return 2;
        }

if (r==(mappingCount()-1))
        {
        m=getMapping(r);
        m1=getMapping(r-1);
        if (begin==m->begin)
                { // adjacent
                if (m1->to==mr)
                        { // shift only
                        m->begin=end+1;
                        return 0;
                        }
                // single add
                m->begin=end+1;
                SingleMapping *newmap=new SingleMapping(begin,mr);
                addData(r,newmap);
                return 1;
                }
        // double add
        SingleMapping *newmap=new SingleMapping(begin,mr);
        SingleMapping *newmap2=new SingleMapping(end+1);
        addData(r+1,newmap);
        addData(r+2,newmap2);
        r+=2;
        return 2;
        }

m=getMapping(r);
if (m->to.contains(mr)) return 0;
if (begin==m->begin) // begin at range boundary
        {
        m2=getMapping(r+1);
        if (end==(m2->begin-1)) return addRange(r,mr);

        SingleMapping *newmap=new SingleMapping(begin,m->to);
        newmap->to.add(mr);
        if (r>0)
                { // join prev possible...
                m1=getMapping(r-1);
                if (m1->to==newmap->to)
                        { // joint prev = shift
                        delete newmap;
                        m->begin=end+1;
                        return 0;
                        }
                }
        // single add:
        m->begin=end+1;
        addData(r,newmap);
        return 1;
        }

m2=getMapping(r+1);
if (end==(m2->begin-1)) // end at range boundary
        {
        SingleMapping *newmap=new SingleMapping(begin,m->to);
        newmap->to.add(mr);
        if (r<(mappingCount()-2))
                { // join next possible...
                if (m2->to==newmap->to)
                        { // joint next = shift
                        m2->begin=begin;
                        delete newmap;
                        return 0;
                        }
                }
        // single add
        r++;
        addData(r,newmap);
        return 1;
        }
// double add:
SingleMapping *newmap=new SingleMapping(begin,m->to);
newmap->to.add(mr);
SingleMapping *newmap2=new SingleMapping(end+1,m->to);
addData(r+1,newmap);
addData(r+2,newmap2);
r+=2;
return 2;
}

void MultiMap::add(int begin,int end,const MultiRange& to)
{
if (to.isEmpty()) return;
if (end<begin) return;
int r1=findData(begin);
int r2=findData(end);
if (r1==r2)
        {
        addRangeBeginEnd(r1,begin,end,to);
        }
else
        {
        r2+=addRangeBeginEnd(r1,begin,getBegin(r1+1)-1,to);
        r1++;
        for (;r1<r2;r1++)
                r2+=addRange(r1,to);
        addRangeBeginEnd(r1,getBegin(r1),end,to);
        }
}

const MultiRange& MultiMap::map(int x) const
{
static MultiRange empty;
int m=findData(x);
if (m<0) return empty;
SingleMapping *sm=getMapping(m);
return sm->to;
}

MultiRange MultiMap::map(int begin,int end) const
{
MultiRange mr;
int m=findData(begin);
for (;m<rangeCount();m++)
        if (m>=0)
        {
        SingleMapping *sm=getMapping(m);
        if (sm->begin > end) break;
        mr.add(sm->to);
        }
return mr;
}

MultiRange MultiMap::map(const MultiRange& ranges) const
{
MultiRange mr;
for(int i=0;i<ranges.rangeCount();i++)
        mr.add(map(ranges.getRange(i)));
return mr;
}

IRange MultiMap::getRange(int i) const
{
if ((i<0)||(i>(data.size()-1))) return IRange();
return IRange(getBegin(i),getBegin(i+1)-1);
}

void MultiMap::operator=(const MultiMap& mm)
{
clear();
for(int i=0;i<mm.mappingCount();i++)
        addData(i,new SingleMapping(*mm.getMapping(i)));
}

void MultiMap::addCombined(const MultiMap& m1,const MultiMap& m2)
{
for(int i=0;i<m1.rangeCount();i++)
        {
        IRange r=m1.getRange(i);
        add(r,m2.map(m1.getMapping(i)->to));
        }
}

void MultiMap::add(const MultiRange& from,const MultiRange& to)
{
for(int i=0;i<from.rangeCount();i++)
        add(from.getRange(i),to);
}

void MultiMap::addReversed(const MultiMap& m)
{
for(int i=0;i<m.rangeCount();i++)
        {
        IRange r=m.getRange(i);
        const MultiRange& mr=m.getMapping(i)->to;
        for (int j=0;j<mr.rangeCount();j++)
                add(mr.getRange(j),r);
        }
}

MultiMap::~MultiMap()
{ clear();}

void MultiMap::clear()
{
for(int i=0;i<data.size();i++)
        delete getData(i);
data.clear();
}

///////////////////

void MultiMap::print() const
{
printf("{ ");
for (int i=0;i<(mappingCount()-1);i++)
        {
        if (i) printf("  ");
        SingleMapping *sm=getMapping(i);
        SingleMapping *sm2=getMapping(i+1);
        if (sm2->begin==sm->begin+1)
                printf("[%d] -> ",sm->begin);
        else
                printf("[%d-%d] -> ",sm->begin,sm2->begin-1);
        sm->to.print();
        printf("\n");
        }
printf("}\n");
}

void MultiMap::print2() const
{
int y;
int id=0,id2;
if (isEmpty())
        {
        printf("{ empty }\n");
        return;
        }
printf("{\n");
for (y=getBegin();y<=getEnd();y++)
        {
        id2=findMappingId(y+1);
        printf("%2d %c",y,(id2!=id)?'_':' ');
        id=id2;
        map(y).print2();
        printf("\n");
        }
printf("}\n");
}