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

#include "printconvmap.h"
#include <stdio.h>
#include <frams/util/multimap.h>
#include <frams/model/model.h>

#define GEN1MAX 15

void printN(const char* t, int maxlen)
{
	while (maxlen-- > 0)
		if (*t) putchar(*(t++));
		else putchar(' ');
}

void printN(char t, int maxlen)
{
	while (maxlen-- > 0) putchar(t);
}

void printmapping(const char* gen1, int len1, const MultiRange &mr, const char* gen2, int len2)
{
	printN(' ', GEN1MAX - len1);
	printN(gen1, len1);
	printf(" : ");
	int i;
	for (i = 0; i < len2; i++)
		if (mr.contains(i))
			putchar(gen2[i]);
		else
			putchar('.');
	putchar('\n');
}

void stripstring(SString &str)
{
	char *t = str.directWrite();
	for (; *t; t++)
		if (strchr("\n\r\t", *t)) *t = ' ';
}

void printConvMap(const SString& gen1, const SString& gen2, const MultiMap& map)
{
	int y, y2, len1;
	int id = 0;
	if (map.isEmpty())
	{
		printf("{ empty }\n");
		return;
	}
	len1 = gen1.len();
	SString g1 = gen1;
	stripstring(g1);
	SString g2 = gen2;
	stripstring(g2);
	const char* g = g1.c_str();
	y = 0;
	MultiRange *mr;
	MultiRange emptyrange;
	printN(' ', GEN1MAX);
	printf("   %s\n", g2.c_str());
	int begin = map.getBegin();
	int end = map.getEnd();
	while (y < len1)
	{
		if (y < begin)
		{
			mr = &emptyrange;
			y2 = begin;
		}
		else if (y > end)
		{
			mr = &emptyrange;
			y2 = len1;
		}
		else	{
			id = map.findMappingId(y);
			mr = &map.getMapping(id)->to;
			y2 = map.getMapping(id + 1)->begin;
		}
		if ((y2 - y) > GEN1MAX) y2 = y + GEN1MAX;
		if (y2 > (y + len1)) y2 = y + len1;
		printmapping(g + y, y2 - y, *mr, g2.c_str(), g2.len());
		y = y2;
	}
}

const MultiMap & getModelDisplayMap()
{
	static MultiMap display_map;
	if (display_map.isEmpty())
	{
		for (int i = 0; i < 10; i++)
		{
			display_map.add(Model::partToMap(i), Model::partToMap(i), i, i);
			display_map.add(Model::jointToMap(i), Model::jointToMap(i), 10 + i, 10 + i);
			display_map.add(Model::neuroToMap(i), Model::neuroToMap(i), 20 + i, 20 + i);
		}
	}
	return display_map;
}

void printModelMap(const SString& gen1, const MultiMap& map)
{
	MultiMap combined_map;
	combined_map.addCombined(map, getModelDisplayMap());
	SString g2("012345678901234567890123456789");
	printN(' ', GEN1MAX);
	printf("   Parts     Joints    Neurons\n");
	printConvMap(gen1, g2, combined_map);
}