smart-ptr.cpp
上传用户:qdkongtiao
上传日期:2022-06-29
资源大小:356k
文件大小:4k
源码类别:

书籍源码

开发平台:

Visual C++

  1. /*
  2.  * This file contains code from "C++ Primer, Fourth Edition", by Stanley B.
  3.  * Lippman, Jose Lajoie, and Barbara E. Moo, and is covered under the
  4.  * copyright and warranty notices given in that book:
  5.  * 
  6.  * "Copyright (c) 2005 by Objectwrite, Inc., Jose Lajoie, and Barbara E. Moo."
  7.  * 
  8.  * 
  9.  * "The authors and publisher have taken care in the preparation of this book,
  10.  * but make no expressed or implied warranty of any kind and assume no
  11.  * responsibility for errors or omissions. No liability is assumed for
  12.  * incidental or consequential damages in connection with or arising out of the
  13.  * use of the information or programs contained herein."
  14.  * 
  15.  * Permission is granted for this code to be used for educational purposes in
  16.  * association with the book, given proper citation if and when posted or
  17.  * reproduced.Any commercial use of this code requires the explicit written
  18.  * permission of the publisher, Addison-Wesley Professional, a division of
  19.  * Pearson Education, Inc. Send your request for permission, stating clearly
  20.  * what code you would like to use, and in what specific way, to the following
  21.  * address: 
  22.  * 
  23.  *  Pearson Education, Inc.
  24.  *  Rights and Contracts Department
  25.  *  75 Arlington Street, Suite 300
  26.  *  Boston, MA 02216
  27.  *  Fax: (617) 848-7047
  28. */ 
  29. #include <iostream>
  30. using std::ostream; using std::cout; using std::endl;
  31. #include <string>
  32. #include <cstddef>
  33. using std::size_t;
  34. /* smart pointer class: takes ownership of the dynamically allocated
  35.  *                      object to which it is bound 
  36.  * User code must dynamically allocate an object to initialize a HasPtr
  37.  * and must not delete that object; the HasPtr class will delete it
  38. */
  39. //private class for use by HasPtr only
  40. class U_Ptr {
  41. friend class HasPtr;
  42. int *ip;
  43. size_t use;
  44. U_Ptr(int *p): ip(p), use(1) { }
  45. ~U_Ptr() { delete ip; }
  46. };
  47. class HasPtr {
  48. public:
  49.     // HasPtr owns the pointer; p must have been dynamically allocated
  50.     HasPtr(int *p, int i): ptr(new U_Ptr(p)), val(i) { }
  51.     // copy members and increment the use count
  52.     HasPtr(const HasPtr &orig):
  53.        ptr(orig.ptr), val(orig.val) { ++ptr->use; }
  54.     HasPtr& operator=(const HasPtr&);
  55.     // if use count goes to zero, delete the U_Ptr object
  56.     ~HasPtr() { if (--ptr->use == 0) delete ptr; } 
  57.     friend ostream& operator<<(ostream&, const HasPtr&);
  58.     // copy control and constructors as before
  59.     // accessors must change to fetch value from U_Ptr object
  60.     int *get_ptr() const { return ptr->ip; } 
  61.     int get_int() const { return val; }
  62.     // change the appropriate data member
  63.     void set_ptr(int *p) { ptr->ip = p; }
  64.     void set_int(int i) { val = i; }
  65.     // return or change the value pointed to, so ok for const objects
  66.     // Note: *ptr->ip is equivalent to *(ptr->ip)
  67.     int get_ptr_val() const { return *ptr->ip; } 
  68.     void set_ptr_val(int i) { *ptr->ip = i; }
  69. private:
  70.     U_Ptr *ptr;        // points to use-counted U_Ptr class
  71.     int val;
  72. };
  73. HasPtr& HasPtr::operator=(const HasPtr &rhs)
  74. {
  75.     ++rhs.ptr->use;     // increment use count on rhs first
  76.     if (--ptr->use == 0)
  77.          delete ptr;    // if use count goes to 0 on this object, delete it
  78.     ptr = rhs.ptr;      // copy the U_Ptr object
  79.     val = rhs.val;      // copy the int member
  80.     return *this;
  81. }
  82. ostream& operator<<(ostream &os, const HasPtr &hp)
  83. {
  84.     os << "*ptr: " << hp.get_ptr_val() << "tval: " << hp.get_int() << endl;
  85.     return os;
  86. }
  87. int main()
  88. {
  89.     int obj = 0;
  90.     HasPtr ptr1(&obj, 42);
  91.     HasPtr ptr2(ptr1);
  92.     cout << "(1) ptr1: " << ptr1 << endl << "ptr2: " << ptr2 << endl;
  93.     ptr1.set_ptr_val(42); // sets object to which both ptr1 and ptr2 point
  94.     ptr2.get_ptr_val();   // returns 42
  95.     cout << "(2) ptr1: " << ptr1 << endl << "ptr2: " << ptr2 << endl;
  96.     ptr1.set_int(0);   // changes s member only in ptr1
  97.     ptr2.get_int();    // returns 42
  98.     ptr1.get_int();    // returns 0
  99.     cout << "(3) ptr1: " << ptr1 << endl << "ptr2: " << ptr2 << endl;
  100. }