#include "TGo4DrawObjProxy.h"

#include "TList.h"
#include "THStack.h"
#include "TMultiGraph.h"

TGo4DrawObjProxy::TGo4DrawObjProxy(TObject* obj, Bool_t owner, Bool_t doclear) :
   TGo4ObjectProxy(obj, owner),
   fDoClear(doclear)
{
}

TGo4DrawObjProxy::~TGo4DrawObjProxy()
{
  if (fDoClear) TryToClearObject();
}

Bool_t TGo4DrawObjProxy::RemoveRegisteredObject(TObject* obj)
{
   if (obj == fObject) TryToClearObject();
   return TGo4ObjectProxy::RemoveRegisteredObject(obj);
}

void TGo4DrawObjProxy::TryToClearObject()
{
   if (fObject==0) return;
   if (fObject->InheritsFrom(THStack::Class())) {
      THStack* hs = (THStack*) fObject;
      if (hs->GetHists())
          hs->GetHists()->Clear();
   } else
   if (fObject->InheritsFrom(TMultiGraph::Class())) {
      TMultiGraph* mgr = (TMultiGraph*) fObject;
      if (mgr->GetListOfGraphs())
         mgr->GetListOfGraphs()->Clear();
    }
}