1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: $RCSfile: itkSmartPointerTest.cxx,v $
Language: C++
Date: $Date: 2007-08-20 13:00:21 $
Version: $Revision: 1.20 $
Copyright (c) Insight Software Consortium. All rights reserved.
See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notices for more information.
=========================================================================*/
#if defined(_MSC_VER)
#pragma warning ( disable : 4786 )
#endif
#include <iostream>
#include "itkSmartPointer.h"
class itkTestObject
{
public:
typedef itk::SmartPointer<itkTestObject> Pointer;
static itkTestObject::Pointer New();
virtual void Register()
{
std::cout << "Register " << *this << " count:"
<< (m_ReferenceCount+1) << " " << std::endl;
m_ReferenceCount++;
}
virtual void UnRegister()
{
std::cout << "UnRegister " << this << " count:"
<< (m_ReferenceCount-1) << " " << std::endl;
m_ReferenceCount--;
if ( m_ReferenceCount == 0 )
{
delete this;
}
}
inline friend std::ostream &operator << (std::ostream &os,
itkTestObject const& o)
{
os << "itkTestObject " << (void const * )&o << " " << o.m_ReferenceCount;
return os;
}
std::ostream& Print(std::ostream& os) const
{
os << "itkTestObject " << (void const * )this << " " << this->m_ReferenceCount;
return os;
}
protected:
itkTestObject()
{
m_ReferenceCount = 0;
std::cout << "construct itkTestObject " << *this << std::endl;
}
virtual ~itkTestObject()
{
std::cout << "destruct itkTestObject " << *this << std::endl;
}
private:
unsigned int m_ReferenceCount;
};
itkTestObject::Pointer itkTestObject::New()
{
return itkTestObject::Pointer(new itkTestObject);
}
class itkTestObjectSubClass : public itkTestObject
{
public:
typedef itk::SmartPointer<itkTestObjectSubClass> Pointer;
static Pointer New();
};
itkTestObjectSubClass::Pointer itkTestObjectSubClass::New()
{
return itkTestObjectSubClass::Pointer(new itkTestObjectSubClass);
}
// This SHOULD NOT be used in ITK, all functions
// should take raw pointers as arguments
void TestUpCastPointer(itkTestObject::Pointer)
{
}
// Test a function that takes an itkTestObject raw pointer
void TestUpCast(itkTestObject*)
{
}
int itkSmartPointerTest(int, char* [] )
{
// Create a base class pointer to a child class
itkTestObject::Pointer to(itkTestObjectSubClass::New());
// test the safe down cast and create a child class Pointer object
itkTestObjectSubClass::Pointer sc
= dynamic_cast<itkTestObjectSubClass*>(to.GetPointer());
// Test the up cast with a function that takes a pointer
TestUpCast(sc);
// Test calling a function that takes a SmartPointer as
// an argument, note, that you have to get the raw pointer
// for this to work
TestUpCastPointer(sc.GetPointer());
// This will not work, but only in construction
//itkTestObject::Pointer p = sc;
// For construction use the constructor call:
// Test casting up the tree, note no explict cast is required
itkTestObject::Pointer p(sc);
// No cast is required for assignment
p = sc;
std::cout <<"second test" << std::endl;
{
itkTestObject::Pointer o1 = itkTestObject::New();
std::cout << "o1 " << &o1 << std::endl;
itkTestObject::Pointer o2 = itkTestObject::New();
std::cout << "o2 " << &o2 << std::endl;
itkTestObject::Pointer o3 = itkTestObject::New();
std::cout << "o3 " << &o3 << std::endl;
itkTestObject::Pointer o4 = itkTestObject::New();
std::cout << "o4 " << &o4 << std::endl;
o1 = o2;
o2 = o3;
o4 = o1;
if ( o1 < o2 )
{
std::cout << "o1 is < o2 " << o1 << " " << o2 << std::endl;
}
else
{
std::cout << "o1 is not < o2 " << &o1 << " " << &o2 << std::endl;
}
}
std::cout <<"end second test" << std::endl << std::endl;
std::cout <<"first test" << std::endl;
{
itkTestObject::Pointer o1 = itkTestObject::New();
}
std::cout <<"end first test" << std::endl << std::endl;
return EXIT_SUCCESS;
}
|