Be Sweet - a set of visual source code browsers

#include "stdafx.h"
#include "resource.h"

#include "HierarchyBrowser.h"

#include <Model/Editor.h>
#include <Model/MetaModel/MetaModelUtilities.h>

#include <Controller/Browsers/TypeBrowser.h>

#include <Renderer/MFC/Browsers/HierarchyBrowserRenderer.h>

using std::max;
using namespace std;

HierarchyBrowser::HierarchyBrowser()
: mTypeSelection(NULL), mHierarchyType(eFullHierarchy), 
  mShowInheritedMembers(false), mIsHierarchyLocked(false)
{

  mOutlineListener = make_smart_ptr(new OutlineListener(this));
  mHierarchySelectionStrategy = make_smart_ptr(new HierarchyListener(this));
  
  mForwarder = make_smart_ptr(new EventForwarder(this));
  mMemberOutline.addSelectionListener(mForwarder);
  
}

bool HierarchyBrowser::show(bool isModal)
{  
  if(isModal) {
    return getRenderer().DoModal() != 0;

  } else {
    getRenderer().show(); 
    return true;
  }
}

const OutlineRenderer& HierarchyBrowser::getOutlineRenderer()
{
  return getMemberOutline().getRenderer();
}

void HierarchyBrowser::setModel(const smart_ptr<Inheritable>& model)
{
  mModel = model;

  if(find_if(mHistory.begin(), mHistory.end(), bind2nd(FQNComperator<Inheritable>(), mModel)) == mHistory.end()) {
    mHistory.push_front(mModel);
    if(mHistory.size() > 10) {
      mHistory.pop_back();
    }
  }
  
  mRoot = smart_ptr<Inheritable>();
  mTypeSelection = model.get();

  updateHierarchyView();
  updateMemberOutline(getModel());
}

void HierarchyBrowser::historyEntrySelectionChanged(size_t id)
{
  History::iterator entry = mHistory.begin();
  advance(entry, id);
  setModel(*entry);
}

void HierarchyBrowser::hierarchyFilterChanged(HierarchyViewType hierarchyType)
{
  mHierarchyType = hierarchyType;
  updateHierarchyView();
}

struct extractor
{
  const smart_ptr<CppTag>& operator()(const std::pair<long, std::list< smart_ptr<CppTag> > > &x) const
  { 
    return x.second.front(); 
  }
};

void HierarchyBrowser::hierarchySelectionChanged(CppTag *tag, const MouseEvent &e)
{
  if(!tag) {
    return;
  }
  
  Inheritable *inheritable = dynamic_cast<Inheritable*>(tag);
  if(inheritable) {
    mTypeSelection = inheritable;
  }

  mHierarchySelectionStrategy->selectionChanged(*tag, e);
}

void HierarchyBrowser::updateHierarchyView()
{
  getRenderer().setHistory(mHistory);
  getRenderer().render();
}

void HierarchyBrowser::updateMemberOutline(const Inheritable &inheritable)
{
  map<long, list<smart_ptr<CppTag> > > members = inheritable.getMembers(mShowInheritedMembers);

  list<smart_ptr<CppTag> > result;
  transform(members.begin(), members.end(), back_inserter(result), extractor());

  getMemberOutline().setModel(result);
  getRenderer().setOutlineDescriptor(inheritable);
}


void HierarchyBrowser::hierarchyLockChanged(bool isLocked)
{
  mIsHierarchyLocked = isLocked;
  swapMemberOutlineListeners(isLocked);
  
  if(isLocked) {
    mHierarchySelectionStrategy = mForwarder;
  } else {
    mMembers.clear();
    mHierarchySelectionStrategy = make_smart_ptr(new HierarchyListener(this));
  }
  updateHierarchyView();
}

void HierarchyBrowser::focusOn(Inheritable *inheritable)
{
  if(inheritable != NULL) {
    TagCloner cloner;
    inheritable->acceptVisitor(&cloner);
    setModel(Dynamic_cast<Inheritable>(cloner.getClone()));

  } else {
    TagBrowser typeBrowser("Type");

    typeBrowser.show(true);
  }
}
  
void HierarchyBrowser::swapMemberOutlineListeners(bool isLocked)
{
  if(isLocked) {
    mMemberOutline.removeSelectionListener(mForwarder);
    mMemberOutline.addSelectionListener(mOutlineListener);
    
  } else {
    mMemberOutline.removeSelectionListener(mOutlineListener);
    mMemberOutline.addSelectionListener(mForwarder);
  }
}

void HierarchyBrowser::memberFilterChanged(bool show)
{
  mShowInheritedMembers = show;
  updateMemberOutline(*mTypeSelection);
}

HierarchyBrowserRenderer& HierarchyBrowser::getRenderer()
{ 
  if(mRenderer.isNull()) {
    mRenderer = make_smart_ptr(new HierarchyBrowserRenderer(this));
    if(!getRenderer().m_hWnd) {
      getRenderer().Create(IDD_HIERARCHY_BROWSER_PANEL,CWnd::GetDesktopWindow());
    }    
  }
 
  return *mRenderer; 
}

smart_ptr<Inheritable> HierarchyBrowser::getRoot() const
{
  if(mRoot.isNull()) {
    RootFinder::RootMap result;
    result.swap(RootFinder().findRootOf(mModel));

    /* debug code
    stringstream ss;
    ss << "map size = " << result.size() << endl
       << "root-dist = " << result.rbegin()->first << endl
       << "root list size" << result.rbegin()->second.size() << endl << endl;

    for(RootFinder::RootMap::iterator x = result.begin(); x != result.end(); ++x) {
      ss << "dist = " << x->first << endl;
      ss << "list size = " << x->second.size() << endl;
      for(list<smart_ptr<Inheritable> >::const_iterator y = x->second.begin(); y != x->second.end(); ++y) {
        ss << (*y)->getFQN().toString() << endl;
      }
      ss << endl << endl;
    }
    AfxMessageBox(ss.str().c_str());*/
  
    if(result.size() > 1) {
      AfxMessageBox((getModel().getFQN().toString() + " has multiple parents...\nonly the longest path is displayed").c_str());
    }  

    mRoot = result.rbegin()->second.front();
  }
  return mRoot;
}

std::list<smart_ptr<CppTag> > HierarchyBrowser::getHierarchyOf(CppTag *element) const
{
  std::list<smart_ptr<CppTag> > elements;
  Inheritable *inheritable = dynamic_cast<Inheritable*>(element);

  if(element == NULL && mHierarchyType == eFullHierarchy) {
    elements.push_front(getRoot());
    
  } else if(element == NULL) {
    elements.push_front(mModel);

  } else if(inheritable != NULL) {
    elements.swap(getHierarchyOf(*inheritable));
  }
  
  return elements;
}

std::list<smart_ptr<CppTag> > HierarchyBrowser::getHierarchyOf(const Inheritable &inheritable) const
{
  std::list<smart_ptr<CppTag> > elements;

  switch(mHierarchyType) {
    case eFullHierarchy :
      copy_if(inheritable.getExtendors().begin(), inheritable.getExtendors().end(), back_inserter(elements), bind2nd(IsRelativeOf(), mModel)); 
      break;
    
    case eSubtypesOnly:      
      elements.assign(inheritable.getExtendors().begin(), inheritable.getExtendors().end());
      break;
    
    case eSupertypesOnly:
      elements.assign(inheritable.getParents().begin(), inheritable.getParents().end());
      break;
  }
  
  for(list<CppTag*>::const_iterator member = mMembers.begin(); member != mMembers.end(); ++member) {
    list<smart_ptr<CppTag> >::const_iterator x;
    x = find_if(inheritable.getInnerTags().begin(), inheritable.getInnerTags().end(), bind1st(hash_comperator(), *member));
    if(x != inheritable.getInnerTags().end()) {
      elements.push_front(*x);
    }
  } 

  return elements;
}

int HierarchyBrowser::getSelectionState(const CppTag &tag) const
{
  Inheritable const *inheritable = dynamic_cast<Inheritable const *>(&tag);
  
  /* 0 = nothing, 1 = select, 2 = icon*/
  int select = 0;

  if(inheritable == NULL) {
    return select;
  }

  if(inheritable->equals(*mTypeSelection)) {
    select |= 0x01;
  }
  
  if(inheritable->equals(getModel())) {
    select |= 0x02;
  }

  return select;
}

void HierarchyBrowser::fireSelectionChanged(const CppTag& tag, const MouseEvent &e) const
{
  typedef std::list<smart_ptr<SelectionListener> >::const_iterator Iterator;
  for(Iterator listener = mSelectionListeners.begin(); listener != mSelectionListeners.end(); ++listener) {
    (*listener)->selectionChanged(tag, e);
  }  
}


HierarchyBrowser::OutlineListener::OutlineListener(HierarchyBrowser *b) 
: mBrowser(b) 
{
}

void HierarchyBrowser::OutlineListener::selectionChanged(const std::list<CppTag*> &selection) const
{
  mBrowser->mMembers = selection;
  mBrowser->updateHierarchyView();
}

HierarchyBrowser::HierarchyListener::HierarchyListener(HierarchyBrowser *b)
: mBrowser(b)
{
}

void HierarchyBrowser::HierarchyListener::selectionChanged(const CppTag &tag, const MouseEvent &) const
{
  const Inheritable *inheritable = dynamic_cast<const Inheritable*>(&tag);
  if(inheritable) {
    mBrowser->updateMemberOutline(*inheritable);
  }
}

HierarchyBrowser::EventForwarder::EventForwarder(HierarchyBrowser *b)
: mBrowser(b)
{
}

void HierarchyBrowser::EventForwarder::selectionChanged(const CppTag &tag, const MouseEvent &e) const
{
  mBrowser->fireSelectionChanged(tag, e);
}