// ------------------------------- //
// -------- Start of File -------- //
// ------------------------------- //
// ----------------------------------------------------------- //
// C++ Source Code File Name: testprog.cpp
// Compiler Used: MSVC, BCC32, GCC, HPUX aCC, SOLARIS CC
// Produced By: glNET Software
// File Creation Date: 09/18/1997
// Date Last Modified: 05/25/2001
// Copyright (c) 2001 glNET Software
// ----------------------------------------------------------- //
// ------------- Program Description and Details ------------- //
// ----------------------------------------------------------- //
/*
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
USA
Test program for the InfoHog database.
*/
// ----------------------------------------------------------- //
#include <iostream.h>
#include <fstring.h>
#include "infohog.h"
#include "gxfloat.h"
#ifdef __MSVC_DEBUG__
#include "leaktest.h"
#endif
// Type definitions
typedef FString Name_t; // String type
typedef gxFLOAT64 Price_t; // Floating point type
typedef gxINT32 StockNumber_t; // Integer type
void PausePrg()
// Function used to pause the program.
{
cout << endl;
cout << "Press enter to continue..." << endl;
cin.get();
}
void ClearInputStream(istream &s)
// Function used to clear the input stream.
{
char c;
s.clear();
while(s.get(c) && c != '\n') { ; }
}
int Quit()
{
cout << "Exiting..." << endl;
return 0;
}
void InputData(Name_t &key)
// Function used to read a string from the input stream.
{
cout << "Item name: ";
cin >> key;
}
void DisplayItem(InfoHog<Name_t> &infohog, int full = 1)
// Function used to print a infohog list object to the stdout.
{
cout << endl;
if(infohog.GetMember(0)) {
cout << "Item Name = " << (char *)infohog.GetMember(0) << endl;
}
if(full) {
if(infohog.GetMember(1)) {
cout << "Stock Number = "
<< *((StockNumber_t *)infohog.GetMember(1)) << endl;
}
if(infohog.GetMember(2)) {
cout.setf(ios::showpoint | ios::fixed);
cout.precision(2);
cout << "Price = $"
<< *((Price_t *)infohog.GetMember(2)) << endl;
}
}
}
void Menu()
// Console based user menu.
{
cout << "(A, a) Add object to the database" << endl;
cout << "(B, b) Build a test database" << endl;
cout << "(C) Change data members" << endl;
cout << "(c) Change object" << endl;
cout << "(D, d) Delete object from the database" << endl;
cout << "(F, f) Find item by name" << endl;
cout << "(L, l) List without using index file" << endl;
cout << "(H, h, ?) Help (prints this menu)" << endl;
cout << "(I, i) List using the index file" << endl;
cout << "(Q, q) Quit" << endl;
cout << "(X, x) Compare the index file to the data file" << endl;
cout << "(Y, y) Rebuild the index file" << endl;
}
void BuildDatabase(const POD *DB)
// Function used to build a test database.
{
const int NUM_OBJECTS = 10;
int i = 0;
InfoHog<Name_t> infohog(DB);
char name[InfoHogNameLength];
StockNumber_t stock_number = (StockNumber_t)2000;
Price_t price = (Price_t)0.95;
cout << "Adding " << NUM_OBJECTS << " objects to the database..." << endl;
for(i = 0; i < NUM_OBJECTS; i++) {
sprintf(name, "Item-%i", i);
stock_number += i;
price += i;
Name_t key(name);
if(!infohog.SetMember(&key, sizeof(key), 0)) {
cout << "Could not set data member 0" << endl;
return;
}
if(!infohog.SetMember(&stock_number, sizeof(stock_number), 1)) {
cout << "Could not set data member 1" << endl;
return;
}
if(!infohog.SetMember(&price, sizeof(price), 2)) {
cout << "Could not set data member 2" << endl;
return;
}
if(!infohog.WriteObject()) {
cout << "Could not add object number " << i << " to the database"
<< endl;
return;
}
}
}
void ListInOrder(const POD *DB, unsigned index_number)
// List the contents of the database using the index file.
{
InfoHog<Name_t> infohog(DB);
InfoHogKey<Name_t>key, compare_key;
gxBtree *btx = DB->Index(index_number);
// Ensure that the in memory buffers and the file data
// stay in sync during multiple file access.
btx->TestTree(index_number);
// Walk through the tree starting at the first key
if(btx->FindFirst(key)) {
if(!infohog.ReadObject(key.ObjectID())) {
cout << "Error reading the object" << endl;
return;
}
DisplayItem(infohog);
while(btx->FindNext(key, compare_key)) {
if(!infohog.ReadObject(key.ObjectID())) {
cout << "Error reading the object" << endl;
return;
}
DisplayItem(infohog);
}
}
}
void ListInOrder(const POD *DB)
// List contents of the database without using the index file.
{
FAU addr = (FAU)0;
InfoHog<Name_t> infohog(DB);
unsigned count = 0;
ClearInputStream(cin); // Clear input stream
while(1) {
addr = DB->OpenDataFile()->FindFirstObject(addr);
if(!addr) break;
if(!infohog.ReadObject(addr)) {
cout << "Error reading the object" << endl;
return;
}
DisplayItem(infohog);
count++;
if(count == 2) {
PausePrg();
count = 0;
}
}
}
void AddItem(const POD *DB)
// Function used to add an object to the database.
{
Name_t key;
InputData(key);
InfoHog<Name_t> infohog(DB);
infohog.SetMember(&key, sizeof(key), 0);
int exists = infohog.FindObject();
if(exists) {
cout << "Item: " << key << " already exists at address "
<< infohog.ObjectAddress() << endl;
return;
}
int stock_number;
double price;
cout << "Enter the items's stock number: ";
cin >> stock_number;
if(cin) {
StockNumber_t sn = stock_number;
infohog.SetMember(&sn, sizeof(StockNumber_t), 1);
cout << "Enter the items's price: $";
cin >> price;
}
else {
cout << "Invalid entry. Object not added!" << endl;
return;
}
if(cin) {
Price_t p = price;
infohog.SetMember(&p, sizeof(Price_t), 2);
}
else {
cout << "Invalid entry. Object not added!" << endl;
return;
}
ClearInputStream(cin);
if(!infohog.WriteObject()) {
cout << "Could not add object to the database" << endl;
}
}
void DeleteItem(const POD *DB)
// Function used to delete an object from the database.
{
Name_t key;
InputData(key);
InfoHog<Name_t> infohog(DB);
infohog.SetMember(&key, sizeof(key), 0);
if(!infohog.DeleteObject()) {
cout << "Could not find item: " << key << " in the database" << endl;
return;
}
cout << "Deleted item: " << key << " at address "
<< infohog.ObjectAddress() << endl;
}
void FindItem(const POD *DB)
// Function used to find an object in the database.
{
Name_t key;
InputData(key);
InfoHog<Name_t> infohog(DB);
infohog.SetMember(&key, sizeof(key), 0);
if(!infohog.FindObject()) {
cout << "Could not find item: " << key << " in the database"
<< endl;
return;
}
cout << "Found item: " << key << " at address "
<< infohog.ObjectAddress() << endl;
if(!infohog.ReadObject()) {
cout << "Error reading the object" << endl;
return;
}
DisplayItem(infohog);
}
void Compare(const POD *DB)
// Function used to compare the contents of the data file to the
// index file.
{
if(!DB->UsingIndex()) {
cout << "This database is not using the indexing sub-system" << endl;
return;
}
InfoHog<Name_t> infohog(DB);
cout << endl;
cout << "Comparing the index file to the data file..." << endl;
int rv = infohog.CompareIndex(0);
if(!rv) {
cout << "The index file does not match the data file!" << endl;
cout << "The index file needs to be rebuilt." << endl;
cout << endl;
return;
}
cout << "The index file checks good" << endl;
cout << endl;
}
void Rebuild(const POD *DB, const char *fname)
// Function used to rebuild the index file if the index file
// entries no longer match the data file entries.
{
if(!DB->UsingIndex()) return;
InfoHog<Name_t> infohog(DB);
cout << endl;
cout << "Rebuilding the index file..." << endl;
// Rebuild index number 0
int rv = infohog.RebuildIndexFile(fname, 0,
InfoHogNumTrees, InfoHogNodeOrder);
if(!rv) {
cout << "The index file was not rebuilt!" << endl;
cout << endl;
return;
}
cout << "The index file was rebuilt." << endl;
cout << "A new index file named " << fname << " was created.";
cout << endl;
}
void ChangeItem(const POD *DB, int change_object = 1)
{
Name_t key;
InputData(key);
InfoHog<Name_t> infohog(DB);
infohog.SetMember(&key, sizeof(key), 0);
if(!infohog.FindObject()) {
cout << "Could not find item: " << key << " in the database"
<< endl;
return;
}
cout << "Found item: " << key << " at address "
<< infohog.ObjectAddress() << endl;
if(!infohog.ReadObject()) {
cout << "Error reading the object" << endl;
return;
}
DisplayItem(infohog);
ClearInputStream(cin);
cout << "Entering changes to this object..." << endl;
Name_t new_key;
InputData(new_key);
InfoHog<Name_t> infohog_new(DB);
infohog_new.SetMember(&new_key, sizeof(new_key), 0);
int stock_number;
double price;
cout << "Enter the items's stock number: ";
cin >> stock_number;
if(cin) {
StockNumber_t sn = stock_number;
infohog_new.SetMember(&sn, sizeof(StockNumber_t), 1);
cout << "Enter the items's price: $";
cin >> price;
}
else {
cout << "Invalid entry. Object not added!" << endl;
return;
}
if(cin) {
Price_t p = price;
infohog_new.SetMember(&p, sizeof(Price_t), 2);
}
else {
cout << "Invalid entry. Object not added!" << endl;
return;
}
ClearInputStream(cin);
int find = 0;
if(change_object) { // Change the object
if(!infohog.ChangeObject(infohog_new, find)) {
cout << "Could not change this object" << endl;
return;
}
return;
}
// Change the object member by member
if(!infohog.ChangeMember(infohog_new.GetMember(0),
infohog_new.GetMemberLen(0), 0, find)) {
cout << "Could not change key member" << endl;
return;
}
if(!infohog.ChangeMember(infohog_new.GetMember(1),
infohog_new.GetMemberLen(1), 1, find)) {
cout << "Could not change member 1" << endl;
return;
}
if(!infohog.ChangeMember(infohog_new.GetMember(2),
infohog_new.GetMemberLen(2), 2, find)) {
cout << "Could not change member 2" << endl;
return;
}
}
int main()
{
#ifdef __MSVC_DEBUG__
InitLeakTest();
#endif
InfoHogKey<Name_t> key_type;
FStringCaseCompare = 0; // Do not compare the case of FString objects
const char *data_file = "grocery.ihd";
const char *index_file = "grocery.ihx";
// Create or open an existing database using a single index file
POD pod;
gxDatabaseError err = pod.Open(data_file, index_file, key_type,
InfoHogNodeOrder, gxDBASE_READWRITE,
InfoHogUseIndexFile, InfoHogStaticArea,
InfoHogNumTrees);
// Test for any errors
if(err != gxDBASE_NO_ERROR) {
cout << gxDatabaseExceptionMessage(err) << endl;
return 1;
}
char c;
int rv = 1;
Menu();
while(rv) {
if (!cin) {
ClearInputStream(cin);
if (!cin) {
cout << "Input stream error" << endl;
return 0;
}
}
cout << '>';
cin >> c;
if (!cin) continue;
switch(c) {
case 'a' : case 'A' :
ClearInputStream(cin); AddItem(&pod);
break;
case 'b' : case 'B' :
BuildDatabase(&pod);
break;
case 'c' :
ClearInputStream(cin); ChangeItem(&pod);
break;
case 'C' :
ClearInputStream(cin); ChangeItem(&pod, 0);
break;
case 'd' : case 'D' :
ClearInputStream(cin); DeleteItem(&pod);
break;
case 'f' : case 'F' :
ClearInputStream(cin); FindItem(&pod);
break;
case 'l' : case 'L' :
ListInOrder(&pod);
break;
case 'i' : case 'I' :
ListInOrder(&pod, 0);
break;
case 'h' : case 'H' : case '?' : Menu(); break;
case 'q' : case 'Q' : rv = Quit(); break;
case 'x' : case 'X' :
Compare(&pod); break;
case 'y' : case 'Y' :
Rebuild(&pod, "newindex.ihx"); break;
default:
cout << "Unrecognized command" << endl;
}
}
return 0;
}
// ----------------------------------------------------------- //
// ------------------------------- //
// --------- End of File --------- //
// ------------------------------- //