char *readInput (int argc, char **argv) {
// Verify that a file name is provided and that the file exists.
// Use some new C++ stream features.
if (argc <= 1) {
// printf ("Usage: %s <filename>\n", argv[0]) ;
return NULL ;
}
else
return readInputFromFile (argv[1]) ;
}
Matrix Matrix::readMatrix ( std::string filename )
{
char * buf = readInputFromFile(filename.c_str());
char * c = &buf[0];
while(*c != '\0')
{
if(*c == '\n')
{
*c = ' ';
}
c++;
}
const char delim = ' ';
c = strtok(buf, &delim);
int rows = atoi(c);
c = strtok(NULL, &delim);
int cols = atoi(c);
float ** data = (float**)malloc(sizeof(float*)*rows);
for(int k = 0; k<rows; k++)
{
data[k] = (float*)malloc(sizeof(float) * cols);
}
for(int i = 0; i<rows; i++)
{
for(int j = 0; j<cols; j++)
{
c = strtok(NULL, &delim);
data[i][j] = (float)atof(c);
}
}
Matrix * m = new Matrix(rows, cols);
m->data = data;
return *m;
}
int main(int argc, char *argv[]){
if (argc < 2){
puts("Usage: db [FILE]");
return -1;
}
Collection dbCollection = mkCollection();
char buffer[128];
printWelcome();
// Read the input file
readInputFromFile(dbCollection, argv[1], buffer);
// Main loop
int choice = -1;
while(choice != 0){
puts("Please choose an operation");
puts("1. Query a key");
puts("2. Update an entry");
puts("3. New entry");
puts("4. Remove entry");
puts("5. Print database");
puts("0. Exit database");
printf("? ");
scanf("%d", &choice);
while(getchar() != '\n'); // Clear stdin
switch(choice){
case 1:
// Query
printf("Enter key: ");
readline(buffer, 128, stdin);
puts("Searching database...\n");
char *aStr = collectionGetValueByKey(dbCollection,buffer);
if (aStr != NULL){
puts("Found entry:");
printf("key: %s\nvalue: %s\n",buffer, aStr);
}else {
printf("Could not find an entry matching key \"%s\"!\n", buffer);
}
break;
case 2:
// Update
printf("Enter key: ");
readline(buffer, 128, stdin);
puts("Searching database...\n");
char *uValue = collectionGetValueByKey(dbCollection,buffer);
if (uValue != NULL){
puts("Matching entry found:");
printf("key: %s\nvalue: %s\n\n", buffer, uValue);
char *uKey = malloc(strlen(buffer) +1);
strcpy(uKey, buffer);
puts("Enter new value: ");
readline(buffer, 128, stdin);
int success = collectionInsertKeyValuePair(dbCollection,uKey,buffer);
if (success == 1){
printf ("New value: %s Inserted successfully with key: %s",buffer,uKey);
}else{
puts("FAILED!");
}
free(uKey);
}else{
printf("Could not find an entry matching key \"%s\"!\n", buffer);
}
break;
case 3:
// Insert
printf("Enter key: ");
readline(buffer, 128, stdin);
char* iKey = malloc(strlen(buffer) +1);
strcpy(iKey, buffer);
puts("Searching database for duplicate keys...");
int keyExist = collectionDoseKeyExist(dbCollection,iKey);
if (keyExist == 1)
{
printf("key \"%s\" already exists!\n",iKey);
}else{
puts("Key is unique!\n");
printf("Enter value: ");
readline(buffer, 128, stdin);
char *iValue = malloc(strlen(buffer) +1);
strcpy(iValue, buffer);
int success = collectionInsertKeyValuePair(dbCollection,iKey,iValue);
if (success == 1){
printf ("New value: %s Inserted successfully with key: %s",iValue,iKey);
}else{
puts("FAILED!");
}
free(iValue);
}
free(iKey);
break;
case 4:
// Delete
printf("Enter key: ");
readline(buffer, 128, stdin);
char* dKey = malloc(strlen(buffer) +1);
strcpy (dKey, buffer);
puts("Searching database...\n");
char * dValue = malloc(strlen(buffer) +1);
strcpy(dValue,collectionGetValueByKey(dbCollection,dKey));
int sign = collectionRemoveKeyValuePair(dbCollection,dKey);
if (sign != 0){
printf("Deleted the following entry:\nkey: %s\nvalue: %s\n", dKey, dValue);
}
else{
printf("Could not find an entry matching key \"%s\"!\n", dKey);
}
free(dValue);
free(dKey);
break;
case 5:
collectionPrint(dbCollection);
break;
case 0:
// Exit
puts("Cleaning up memory..");
rmCollection(dbCollection);
puts("Good bye!");
break;
default:
// Please try again
puts("Could not parse choice! Please try again");
}
puts("");
}
return 0;
}
