/*
 * Copyright 2012 SciberQuest Inc.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *  * Redistributions of source code must retain the above copyright notice,
 *    this list of conditions and the following disclaimer.
 *
 *  * Redistributions in binary form must reproduce the above copyright notice,
 *    this list of conditions and the following disclaimer in the documentation
 *    and/or other materials provided with the distribution.
 *
 *  * Neither name of SciberQuest Inc. nor the names of any contributors may be
 *    used to endorse or promote products derived from this software without
 *    specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS''
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
#ifndef RefCountedPointer_h
#define RefCountedPointer_h

#include <iostream> // for ostream

//=============================================================================
class RefCountedPointer
{
public:
  RefCountedPointer() : N(1) {}
  virtual ~RefCountedPointer(){}

  #if defined RefCountedPointerDEBUG
  virtual void Delete();
  #else
  virtual void Delete(){ if ((--this->N)==0){ delete this; } }
  #endif
  virtual void Register(){ ++this->N; }

  virtual int GetRefCount(){ return this->N; }
  virtual void SetRefCount(int rc){ this->N=rc; }

  virtual void Print(std::ostream &){}
  virtual void PrintRefCount(std::ostream &os){ os << "RefCount=" << this->N; }

private:
  int N;
};

//*****************************************************************************
std::ostream &operator<<(std::ostream &os, RefCountedPointer &rcp);

#define SetRefCountedPointer(NAME,TYPE)\
  virtual void Set##NAME(TYPE *v);\
  \
  virtual void SetNew##NAME(TYPE *v);\

#define SetRefCountedPointerImpl(CLASS,NAME,TYPE)\
void CLASS::Set##NAME(TYPE *v)\
{\
  if (v==this->NAME){ return; }\
  if (this->NAME){ this->NAME->Delete(); }\
  this->NAME=v;\
  if (this->NAME){ this->NAME->Register(); }\
}\
\
void CLASS::SetNew##NAME(TYPE *v)\
{\
  if (v==this->NAME){ return; }\
  if (this->NAME){ this->NAME->Delete(); }\
  this->NAME=v;\
}

#endif

// VTK-HeaderTest-Exclude: RefCountedPointer.h
