D:/Uni/8.Semester/InfoVis/Demo/source/Main/TreeMap.h

#pragma once

#include <atlbase.h>

// TreeMap - A TreeMap layout graph
//
// Written by Bjarke Viksoe (bjarke@viksoe.dk)
// Copyright (C) 2008 Bjarke Viksoe.
//
// This code was inspired by the Java TreeMap Layout Comparison
// library. This code may be covered by the Mozilla Public License
// (MPL, http://www.mozilla.org/MPL/).
// The Initial Developers of the Original Code is Ben Bederson
// and Martin Wattinberg.  
// Copyright (C) 2001 University of Maryland, College Park, MD 20742.
// All Rights Reserved.
//
// This file is provided "as is" with no expressed or implied warranty.
// The author accepts no liability if it causes any damage to you or your
// computer whatsoever. It's free, so don't hassle me about it.
//
// Beware of bugs.
//

class TreeMapItem;

struct LayoutRectangle
{
        float    left;
        float    top;
        float    right;
        float    bottom;
};

class TreeMapLayout
{
public:
        virtual ~TreeMapLayout() { };
        virtual void UpdateLayout(TreeMapItem* item, LayoutRectangle rc) = 0;
};

class TreeMapSliceLayout : public TreeMapLayout
{
public:
        void UpdateLayout(TreeMapItem* item, LayoutRectangle rc);

        void UpdateSliceLayout(CSimpleValArray<TreeMapItem*>& items, int startIdx, int endIdx, LayoutRectangle rc);
        float GetChildrenSum(const CSimpleValArray<TreeMapItem*>& items, int startIdx, int endIdx) const;
};

class TreeMapSquaredLayout : public TreeMapSliceLayout
{
public:
        void UpdateLayout(TreeMapItem* item, LayoutRectangle rc);

        void UpdateSquaredLayout(CSimpleValArray<TreeMapItem*>& items, int startIdx, int endIdx, LayoutRectangle rc);
        float GetNormAspect(float big, float smal, float a, float b) const;
};

class TreeMapStripLayout : public TreeMapSliceLayout
{
public:
        void UpdateLayout(TreeMapItem* item, LayoutRectangle rc);

        void UpdateStripLayout(CSimpleValArray<TreeMapItem*>& items, int startIdx, int endIdx, LayoutRectangle rc);
        int UpdateOneLayoutStrip(CSimpleValArray<TreeMapItem*>& items, LayoutRectangle rcBox, int startIdx, int endIdx);
        float ComputeHorizontalBoxLayout(CSimpleValArray<TreeMapItem*>& items, LayoutRectangle rcBox, int startIdx, int nItems);
        float ComputeAverageAspectRatio(CSimpleValArray<TreeMapItem*>& items, int startIdx, int nItems);
};

struct TreeMapItemInfo
{
        TreeMapItemInfo();
        TreeMapItemInfo(LPCTSTR _name, unsigned long _order, unsigned int _depth, float _size);

        TCHAR name[60];
        LayoutRectangle rc;
        unsigned int depth;
        unsigned long order;
        float size;
};

class TreeMapItem
{
public:
        TreeMapItem();
        TreeMapItem(LPCTSTR name, DWORD order, UINT depth, float size);

        ~TreeMapItem();

        TreeMapItemInfo itemInfo;
        CSimpleValArray<TreeMapItem*> children;
};

class TreeMap
{
public:
        TreeMap();
        ~TreeMap();

        TreeMapLayout* layout;
        TreeMapItem* root;

        float layoutSpacing;            

        void Prepare(void);
        void Destroy(void);

        void SetLayoutSpacing(float spacing) { layoutSpacing = spacing; }
        float GetLayoutSpacing(void) const { return layoutSpacing; }

        void SetLayout(TreeMapLayout* _layout);

        void UpdateLayout(LayoutRectangle rc);
        void Render(LayoutRectangle rc);

private:
        void UpdateLayoutItem(TreeMapItem* item, LayoutRectangle rc);
        void RenderItem(const TreeMapItem* item);
        void ClearCalculation(TreeMapItem* item);
        float SizeCalculation(TreeMapItem* item);
};

---