home *** CD-ROM | disk | FTP | other *** search
/ Visual Basic Source Code / Visual Basic Source Code.iso / vbsource / vbdatabs / mtank.cpp < prev    next >
Encoding:
C/C++ Source or Header  |  1999-03-17  |  10.3 KB  |  384 lines
  1. // ------------------------------- //
  2. // -------- Start of File -------- //
  3. // ------------------------------- //
  4. // ----------------------------------------------------------- // 
  5. // C++ Source Code File Name: mtank.cpp 
  6. // Compiler Used: MSVC40, DJGPP 2.7.2.1, GCC 2.7.2.1, HP CPP 10.24
  7. // Produced By: Doug Gaer 
  8. // File Creation Date: 09/17/1997  
  9. // Date Last Modified: 03/18/1999
  10. // Copyright (c) 1997 Douglas M. Gaer
  11. // ----------------------------------------------------------- // 
  12. // ------------- Program Description and Details ------------- // 
  13. // ----------------------------------------------------------- // 
  14. /*
  15. The VBD C++ classes are copyright (c) 1997, by Douglas M. Gaer.
  16. All those who put this code or its derivatives in a commercial
  17. product MUST mention this copyright in their documentation for
  18. users of the products in which this code or its derivative
  19. classes are used. Otherwise, you have the freedom to redistribute
  20. verbatim copies of this source code, adapt it to your specific
  21. needs, or improve the code and release your improvements to the
  22. public provided that the modified files carry prominent notices
  23. stating that you changed the files and the date of any change.
  24.  
  25. THIS SOFTWARE IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND.
  26. THE ENTIRE RISK OF THE QUALITY AND PERFORMANCE OF THIS SOFTWARE
  27. IS WITH YOU. SHOULD ANY ELEMENT OF THIS SOFTWARE PROVE DEFECTIVE,
  28. YOU WILL ASSUME THE COST OF ALL NECESSARY SERVICING, REPAIR, OR
  29. CORRECTION.
  30.  
  31. This is a test program used to test the (P)ersistent base class
  32. in a practical database application.
  33. */
  34. // ----------------------------------------------------------- // 
  35. #include "mtank.h"
  36.  
  37. MarineTank::MarineTank(POD *pod) : Persistent(pod)
  38. {
  39.   specific_gravity = pH = ammonia = nitrite = 0;
  40.   nitrate = 0;
  41.   comments = "\0";
  42. }
  43.  
  44. MarineTank::MarineTank(const POD *pod) : Persistent(pod)
  45. {
  46.   specific_gravity = pH = ammonia = nitrite = 0;
  47.   nitrate = 0;
  48.   comments = "\0";
  49. }
  50.  
  51. MarineTank::MarineTank() 
  52. {
  53.   specific_gravity = pH = ammonia = nitrite = 0;
  54.   nitrate = 0;
  55.   comments = "\0";
  56. }
  57.  
  58. unsigned MarineTank::ObjectLength()
  59. {
  60.   unsigned len = cdate.SizeOf() + \
  61.     sizeof(specific_gravity) + \
  62.     sizeof(pH) + \
  63.     sizeof(ammonia) + \
  64.     sizeof(nitrate) + \
  65.     sizeof(nitrite) + \
  66.     StringFileLength(comments);
  67.   
  68.   return len;
  69. }
  70.  
  71. FAU MarineTank::Write()
  72. {
  73.   ObjectHeader oh;
  74.  
  75.   FAU addr = pod->OpenDatabase()->Alloc(ObjectLength() + sizeof(ObjectHeader));
  76.   if(!addr) return 0;
  77.   
  78.   oh.ClassID = GetClassID();
  79.   oh.ObjectID = addr;
  80.   WriteObjHeader(oh);
  81.  
  82.   pod->OpenDatabase()->Write(&cdate, cdate.SizeOf());
  83.   pod->OpenDatabase()->Write(&specific_gravity, sizeof(specific_gravity));
  84.   pod->OpenDatabase()->Write(&pH, sizeof(pH));
  85.   pod->OpenDatabase()->Write(&ammonia, sizeof(ammonia));
  86.   pod->OpenDatabase()->Write(&nitrate, sizeof(nitrate));
  87.   pod->OpenDatabase()->Write(&nitrite, sizeof(nitrite));
  88.   WriteString(comments);
  89.   objectaddress = addr;
  90.   
  91.   // Add the entry to the Index file
  92.   if (UsingIndex()) {
  93.     EntryKey key(cdate.c_str(), oh.ObjectID, oh.ClassID);
  94.     AddKey(key);
  95.   }
  96.  
  97.   return addr;
  98. }
  99.  
  100. void MarineTank::Read(FAU Address)
  101. {
  102.   ObjectHeader oh;
  103.  
  104.   ReadObjHeader(oh, Address);
  105.   if(oh.ClassID != GetClassID()) return; // Incorrect object type
  106.  
  107.   pod->OpenDatabase()->Read(&cdate, cdate.SizeOf());
  108.   pod->OpenDatabase()->Read(&specific_gravity, sizeof(specific_gravity));
  109.   pod->OpenDatabase()->Read(&pH, sizeof(pH));
  110.   pod->OpenDatabase()->Read(&ammonia, sizeof(ammonia));
  111.   pod->OpenDatabase()->Read(&nitrate, sizeof(nitrate));
  112.   pod->OpenDatabase()->Read(&nitrite, sizeof(nitrite));
  113.   ReadString(comments);
  114.   objectaddress = Address;
  115. }
  116.  
  117. FAU MarineTank::Find()
  118. {
  119.   MarineTank mtank;
  120.   
  121.   mtank.cdate = this->cdate;
  122.   mtank.specific_gravity = this->specific_gravity;
  123.   mtank.pH = this->pH;
  124.   mtank.ammonia = this->ammonia;
  125.   mtank.nitrate = this->nitrate;
  126.   mtank.nitrite = this->nitrite;
  127.   mtank.comments = this->comments;
  128.   
  129.   int rv; // Return value
  130.  
  131.   // Search the index file for this entry
  132.   if(UsingIndex()) {
  133.     EntryKey e(this->cdate.c_str());
  134.     rv = FindKey(e);
  135.     if(rv) { // Found the object in the index file
  136.       Read(e.object_address);
  137.       return e.object_address;
  138.     }
  139.     else
  140.       return 0;
  141.   }
  142.  
  143.   FAU addr = 0;
  144.   ObjectHeader oh;
  145.  
  146.   while(1)
  147.     {
  148.       addr = pod->OpenDatabase()->FindFirstObject(addr);
  149.       if(!addr) break;
  150.       ReadObjHeader(oh, addr);
  151.       if(oh.ClassID == GetClassID()) {
  152.     Read(addr);
  153.     if(mtank.cdate == cdate) {
  154.       objectaddress = addr;
  155.       return addr; // Found unique data member
  156.     }
  157.     else {
  158.       // Reset the objects data
  159.           this->cdate = mtank.cdate;
  160.       this->specific_gravity = mtank.specific_gravity;
  161.       this->pH = mtank.pH;
  162.       this->ammonia = mtank.ammonia;
  163.           this->nitrate = mtank.nitrate;
  164.           this->nitrite = mtank.nitrite;
  165.       this->comments = mtank.comments;
  166.     }
  167.       }
  168.     }
  169.   return 0; // Could not find 
  170. }
  171.  
  172. void MarineTank::Copy(const MarineTank &ob)
  173. {
  174.   cdate = ob.cdate;
  175.   specific_gravity = ob.specific_gravity;
  176.   pH = ob.pH;
  177.   ammonia = ob.ammonia;
  178.   pH = ob.pH;
  179.   nitrate = ob.nitrate;
  180.   nitrite = ob.nitrite;
  181.   comments = ob.comments;
  182. }
  183.  
  184. int MarineTank::FullCompare(const MarineTank &ob)
  185. {
  186.   if(ob.cdate != cdate) return 0;
  187.   if(ob.specific_gravity != specific_gravity) return 0;
  188.   if(ob.pH != pH) return 0;
  189.   if(ob.ammonia != ammonia) return 0;
  190.   if(ob.nitrate != nitrate) return 0;
  191.   if(ob.nitrite != nitrite) return 0;
  192.   if(ob.comments != comments) return 0;
  193.   return 1;
  194. }
  195.  
  196. int operator==(const MarineTank &a, const MarineTank &b)
  197. {
  198.   return a.GetDate() == b.GetDate();
  199. }
  200.  
  201. int operator!=(const MarineTank &a, const MarineTank &b)
  202. {
  203.   return a.GetDate() != b.GetDate();
  204. }
  205.  
  206. int operator<(const MarineTank &a, const MarineTank &b)
  207. {
  208.   return a.GetDate() < b.GetDate();
  209. }
  210.   
  211. FAU MarineTank::Delete()
  212. {
  213.   if(UsingIndex()) {
  214.     EntryKey e(this->cdate.c_str());
  215.     int rv = FindKey(e);
  216.     if(rv) {  // Found the object in the index file
  217.       FAU addr = e.object_address;
  218.       DeleteObject(e.object_address); // Delete from the data file
  219.       RemoveKey(e);                   // Remove from the index file
  220.       return addr;
  221.     }
  222.     else
  223.       return 0; // Could not delete
  224.   }
  225.  
  226.   FAU addr = Find();
  227.   if(!addr) return 0; // Object does not exist
  228.   DeleteObject(addr);
  229.   return addr;
  230. }
  231.  
  232. FAU MarineTank::Remove()
  233. {
  234.   if(UsingIndex()) {
  235.     EntryKey e(this->cdate.c_str());
  236.     int rv = FindKey(e);
  237.     if(rv) {  // Found the object in the index file
  238.       FAU addr = e.object_address;
  239.       RemoveObject(e.object_address); // Remove from the data file
  240.       RemoveKey(e);                   // Remove from the index file
  241.       return addr;
  242.     }
  243.     else
  244.       return 0; // Could not remove
  245.   }
  246.  
  247.   FAU addr = Find();
  248.   if(!addr) return 0; // Object does not exist
  249.   RemoveObject(addr);
  250.   return addr;
  251. }
  252.  
  253. int MarineTank::CompareIndex()
  254. // Compares the data file to the index file.
  255. // Returns true if data and index file match.
  256. {
  257.   if(!UsingIndex()) return 0;
  258.  
  259.   MarineTank mtank(pod);
  260.   EntryKey key;
  261.   
  262.   FAU oa;          // Object Address
  263.   VBHeader vb;     // Variable Block Header
  264.   ObjectHeader oh; // Object Header
  265.   
  266.   int objects = 0; // Keeps track of good variable blocks
  267.   int matches = 0; // Keep track of matches
  268.   
  269.   FAU vbdfileEOF = pod->OpenDatabase()->GetEOF();
  270.   FAU addr = 0;
  271.   addr = pod->OpenDatabase()->FindFirstVB(addr); // Search the entire file
  272.  
  273.   if(addr == 0) { // No variable blocks found in file
  274. #ifdef CPP_EXCEPTIONS
  275.     throw CNoObjectsExist();
  276. #else
  277.     Error->SignalException(EHandler::NoObjectsExist, EHandler::DISPLAY);
  278.     return 0;
  279. #endif
  280.   }
  281.   
  282.   while(1) { 
  283.     if(addr >= vbdfileEOF) break;
  284.     pod->OpenDatabase()->Read(&vb, sizeof(VBHeader), addr);
  285.     if(vb.CkWord == CheckWord) {
  286.       if((__SBYTE__)vb.Status == NormalVB) {
  287.     oa = addr + sizeof(VBHeader);
  288.     ReadObjHeader(oh, oa);
  289.     if(oh.ClassID == GetClassID()) { 
  290.       objects++; // Increment the object count
  291.       mtank.Read(oa);
  292.       key.SetStrKey(mtank.GetDateStr());
  293.       key.SetOA(oa);
  294.       key.SetCID(oh.ClassID);
  295.       int rv = FindKey(key, 1);
  296.       if(rv) matches++; // Index and data file match
  297.     }
  298.       }
  299.       addr = addr + vb.Length; // Goto the next variable block
  300.     }
  301.     else {
  302.       addr = pod->OpenDatabase()->VBSearch(addr); 
  303.       if(!addr) break;
  304.     }
  305.   }
  306.  
  307.   return objects == matches;
  308. }
  309.  
  310. int MarineTank::RebuildIndexFile(const char *fname)
  311. {
  312.   if(!UsingIndex()) return 0;
  313.  
  314.   int rv; // (R)eturn (V)alue
  315.   int CacheSize = 15;
  316.  
  317.   VBDFilePtr f(new VBDFile);
  318.   Btree btx(CacheSize);
  319.   f->Create(fname, sizeof(BtreeHeader));
  320.   btx.Connect(f, 1);
  321.   
  322.   MarineTank mtank(pod);
  323.   EntryKey key;
  324.   
  325.   FAU oa;          // Object Address
  326.   VBHeader vb;     // Variable Block Header
  327.   ObjectHeader oh; // Object Header
  328.   
  329.   int objects = 0; // Keeps track of good variable blocks
  330.   int inserts = 0; // Keep track of inserts
  331.   
  332.   FAU vbdfileEOF = pod->OpenDatabase()->GetEOF();
  333.   FAU addr = 0;
  334.   addr = pod->OpenDatabase()->FindFirstVB(addr); // Search the entire file
  335.  
  336.   if(addr == 0) { // No variable blocks found in file
  337. #ifdef CPP_EXCEPTIONS
  338.     throw CNoObjectsExist();
  339. #else
  340.     Error->SignalException(EHandler::NoObjectsExist, EHandler::DISPLAY);
  341.     return 0;
  342. #endif
  343.   }
  344.   
  345.   while(1) { 
  346.     if(addr >= vbdfileEOF) break;
  347.     pod->OpenDatabase()->Read(&vb, sizeof(VBHeader), addr);
  348.     if(vb.CkWord == CheckWord) {
  349.       if((__SBYTE__)vb.Status == NormalVB) {
  350.     oa = addr + sizeof(VBHeader);
  351.     ReadObjHeader(oh, oa);
  352.     if(oh.ClassID == GetClassID()) { 
  353.       objects++; // Increment the object count
  354.       mtank.Read(oa);
  355.       key.SetStrKey(mtank.GetDateStr());
  356.       key.SetOA(oa);
  357.       key.SetCID(oh.ClassID);
  358.       rv = btx.Add(key);
  359.       if(!rv) { 
  360. #ifdef CPP_EXCEPTIONS
  361.         throw CAssertError();
  362. #else
  363.       Error->SignalException(EHandler::AssertError);
  364. #endif
  365.       }
  366.       else
  367.         inserts++; // Index and data file match
  368.     }
  369.       }
  370.       addr = addr + vb.Length; // Goto the next variable block
  371.     }
  372.     else {
  373.       addr = pod->OpenDatabase()->VBSearch(addr); 
  374.       if(!addr) break;
  375.     }
  376.   }
  377.  
  378.   return objects == inserts;
  379. }
  380. // ----------------------------------------------------------- //
  381. // ------------------------------- //
  382. // --------- End of File --------- //
  383. // ------------------------------- //
  384.