/*

    ......... 2015 Ivan Mahonin

    This program 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.

    This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.

*/

using System;

using System.Collections.Generic;

using System.Drawing;

namespace Diagram {

    public class Block {

        public string id = "";

        public string caption = "";

        public string text = "";

        public Color color = Color.Black;

    }

    public class Link {

        public string id = "";

        public string srcId = "";

        public string dstId = "";

    }

    public class Diagram {

        public readonly Dictionary<string, Block> blocks = new Dictionary<string, Block>();

        public readonly Dictionary<string, Link> links = new Dictionary<string, Link>();

        public Diagram addBlock(Block block) {

            blocks.Add(block.id, block);

            return this;

        }

        public Diagram addBlock(string id, string caption, string text, Color color, string[] links = null) {

            addBlock(new Block() { id = id, caption = caption, text = text, color = color });

            if (links != null)

                foreach(string link in links)

                    addLink(id, link);

            return this;

        }

        public Diagram addBlock(string id, string caption, string text, string[] links = null) {

            addBlock(id, caption, text, Color.Black, links);

            return this;

        }

        public Diagram addLink(Link link, string srcId = "", string dstId = "") {

            if (srcId != "")

                link.srcId = srcId;

            if (dstId != "")

                link.dstId = dstId;

            if (link.id == "")

                link.id = link.srcId + link.dstId;

            links.Add(link.id, link);

            return this;

        }

        public Diagram addLink(Link link) {

            if (link.id == "")

                link.id = link.srcId + link.dstId;

            links.Add(link.id, link);

            return this;

        }

        public Diagram addLink(string srcId, string dstId) {

            addLink(new Link(), srcId, dstId);

            return this;

        }

    }

}