学步园

// Collection class / traits class example.
// This program demonstrates using a CTraits class
// to create a new comparison operator.
 
#include "stdafx.h"
#include "atlcoll.h"
 
#define MAX_STRING 80
 
// Define our own data type to store in the list.
 
struct MyData {
      int ID;
      TCHAR name[MAX_STRING];
      TCHAR address[MAX_STRING];
   };
 
// Define our own traits class, making use of the
// existing traits and overriding only the comparison
// we need.
 
class MyTraits : public CElementTraits< MyData >
{
public:
 
   // Override the comparison to only compare
   // the ID value.
 
   static bool CompareElements( const MyData& element1, const MyData& element2)
   {
      if ( element1.ID == element2.ID )
            return true;
      else
         return false;
   };
};
 
// The main function
 
int _tmain(int argc, _TCHAR* argv[])
{
   
   // Declare the array, with our data type and traits class 
 
   CAtlList < MyData, MyTraits > MyList;
 
   // Create some variables of our data type
 
   MyData add_item, search_item;
 
   // Add some elements to the list.
   
   add_item.ID = 1;
   wsprintf( add_item.name, "Adam" );
   wsprintf( add_item.address, "One Garden Way" );
 
   MyList.AddHead( add_item );
 
   add_item.ID = 2;
   wsprintf( add_item.name, "Eve" );
   wsprintf( add_item.address, "One Garden Way" );
 
   MyList.AddHead( add_item );
 
   add_item.ID = 3;
   wsprintf( add_item.name, "Cain" );
   wsprintf( add_item.address, "Two Garden Way" );
 
   MyList.AddHead( add_item );
 
   // Create an element which will be used
   // to search the list for a match.
 
   search_item.ID = 2;
   wsprintf( search_item.name, "Don't care" );
   wsprintf( search_item.address, "Don't care" );
 
   // Perform a comparison by searching for a match
   // between any element in the list, and our
   // search item. This operation will use the
   // (overridden) comparison operator and will
   // find a match when the IDs are the same.
 
   POSITION i;
 
   i = MyList.Find( search_item );
 
   if ( i != NULL )
      printf("Item found!\n");
   else
      printf("Item not found.\n");
 
   return 0;
}