#ifndef DRAWVISITOR_H
#define DRAWVISITOR_H
#include "btnodevisitor.h"
#include <sstream>
#include <iomanip>

using namespace std;

class drawVisitor : public btNodeVisitor<int,pair<string,string>, string>
{
public:

  string visit(binTreeNode<int> *p, const pair<string,string>& x)
  {
    stringstream s;
    typedef pair<string,string> prstr;

    string lpad = x.first;
    string rpad = x.second;
    string pad = lpad.substr(0, lpad.size() - 1);
    if (p == 0) {
      return "";
    }
    s << p->right->accept(*this, prstr( rpad + "    |", rpad + "     "));
    s << pad << string("+--") <<  setw(3) << p->data << "\n";
    s << p->left->accept(*this,prstr(lpad + "     ", lpad + "    |"));

    return s.str();
  }
};
#endif