diff options
| author | Kelly Rauchenberger <fefferburbia@gmail.com> | 2018-03-26 15:33:06 -0400 |
|---|---|---|
| committer | Kelly Rauchenberger <fefferburbia@gmail.com> | 2018-03-26 15:33:06 -0400 |
| commit | 0e6355329f241a3e3c8b38cb4146fda4470e6ed0 (patch) | |
| tree | df5853ba6df148e8eeb0082d090dfbcd9f94f651 /generator | |
| parent | 09acb46a9c32257b0003eb152d32c15b43b95dfe (diff) | |
| download | cadence-0e6355329f241a3e3c8b38cb4146fda4470e6ed0.tar.gz cadence-0e6355329f241a3e3c8b38cb4146fda4470e6ed0.tar.bz2 cadence-0e6355329f241a3e3c8b38cb4146fda4470e6ed0.zip | |
Whitespace changes
Diffstat (limited to 'generator')
| -rw-r--r-- | generator/database.cpp | 14 | ||||
| -rw-r--r-- | generator/database.h | 2 | ||||
| -rw-r--r-- | generator/field.h | 12 | ||||
| -rw-r--r-- | generator/generator.cpp | 56 | ||||
| -rw-r--r-- | generator/generator.h | 36 | ||||
| -rw-r--r-- | generator/main.cpp | 2 | ||||
| -rw-r--r-- | generator/mood.cpp | 42 | ||||
| -rw-r--r-- | generator/mood.h | 22 | ||||
| -rw-r--r-- | generator/progress.h | 16 |
9 files changed, 101 insertions, 101 deletions
| diff --git a/generator/database.cpp b/generator/database.cpp index f326f46..b46a0d1 100644 --- a/generator/database.cpp +++ b/generator/database.cpp | |||
| @@ -73,19 +73,19 @@ namespace cadence { | |||
| 73 | { | 73 | { |
| 74 | sqlite3_close_v2(ppdb_); | 74 | sqlite3_close_v2(ppdb_); |
| 75 | } | 75 | } |
| 76 | 76 | ||
| 77 | void database::runQuery(std::string query) | 77 | void database::runQuery(std::string query) |
| 78 | { | 78 | { |
| 79 | // This can only happen when doing bad things with move semantics. | 79 | // This can only happen when doing bad things with move semantics. |
| 80 | assert(ppdb_ != nullptr); | 80 | assert(ppdb_ != nullptr); |
| 81 | 81 | ||
| 82 | sqlite3_stmt* ppstmt; | 82 | sqlite3_stmt* ppstmt; |
| 83 | 83 | ||
| 84 | if (sqlite3_prepare_v2(ppdb_, query.c_str(), query.length(), &ppstmt, NULL) != SQLITE_OK) | 84 | if (sqlite3_prepare_v2(ppdb_, query.c_str(), query.length(), &ppstmt, NULL) != SQLITE_OK) |
| 85 | { | 85 | { |
| 86 | throw sqlite3_error("Error writing to database", sqlite3_errmsg(ppdb_)); | 86 | throw sqlite3_error("Error writing to database", sqlite3_errmsg(ppdb_)); |
| 87 | } | 87 | } |
| 88 | 88 | ||
| 89 | int result = sqlite3_step(ppstmt); | 89 | int result = sqlite3_step(ppstmt); |
| 90 | sqlite3_finalize(ppstmt); | 90 | sqlite3_finalize(ppstmt); |
| 91 | 91 | ||
| @@ -99,10 +99,10 @@ namespace cadence { | |||
| 99 | { | 99 | { |
| 100 | // This can only happen when doing bad things with move semantics. | 100 | // This can only happen when doing bad things with move semantics. |
| 101 | assert(ppdb_ != nullptr); | 101 | assert(ppdb_ != nullptr); |
| 102 | 102 | ||
| 103 | // This shouldn't happen. | 103 | // This shouldn't happen. |
| 104 | assert(!fields.empty()); | 104 | assert(!fields.empty()); |
| 105 | 105 | ||
| 106 | std::list<std::string> fieldNames; | 106 | std::list<std::string> fieldNames; |
| 107 | std::list<std::string> qs; | 107 | std::list<std::string> qs; |
| 108 | for (field& f : fields) | 108 | for (field& f : fields) |
| @@ -110,7 +110,7 @@ namespace cadence { | |||
| 110 | fieldNames.push_back(f.getName()); | 110 | fieldNames.push_back(f.getName()); |
| 111 | qs.push_back("?"); | 111 | qs.push_back("?"); |
| 112 | } | 112 | } |
| 113 | 113 | ||
| 114 | std::ostringstream query; | 114 | std::ostringstream query; |
| 115 | query << "INSERT INTO "; | 115 | query << "INSERT INTO "; |
| 116 | query << table; | 116 | query << table; |
| @@ -119,7 +119,7 @@ namespace cadence { | |||
| 119 | query << ") VALUES ("; | 119 | query << ") VALUES ("; |
| 120 | query << implode(std::begin(qs), std::end(qs), ", "); | 120 | query << implode(std::begin(qs), std::end(qs), ", "); |
| 121 | query << ")"; | 121 | query << ")"; |
| 122 | 122 | ||
| 123 | std::string query_str = query.str(); | 123 | std::string query_str = query.str(); |
| 124 | 124 | ||
| 125 | sqlite3_stmt* ppstmt; | 125 | sqlite3_stmt* ppstmt; |
| diff --git a/generator/database.h b/generator/database.h index 7aa8dfa..e4f3ba2 100644 --- a/generator/database.h +++ b/generator/database.h | |||
| @@ -52,7 +52,7 @@ namespace cadence { | |||
| 52 | ~database(); | 52 | ~database(); |
| 53 | 53 | ||
| 54 | // Actions | 54 | // Actions |
| 55 | 55 | ||
| 56 | void runQuery(std::string query); | 56 | void runQuery(std::string query); |
| 57 | 57 | ||
| 58 | void insertIntoTable(std::string table, std::list<field> fields); | 58 | void insertIntoTable(std::string table, std::list<field> fields); |
| diff --git a/generator/field.h b/generator/field.h index 9d6ed9a..836c079 100644 --- a/generator/field.h +++ b/generator/field.h | |||
| @@ -32,27 +32,27 @@ namespace cadence { | |||
| 32 | ~field(); | 32 | ~field(); |
| 33 | 33 | ||
| 34 | // Generic accessors | 34 | // Generic accessors |
| 35 | 35 | ||
| 36 | type getType() const | 36 | type getType() const |
| 37 | { | 37 | { |
| 38 | return type_; | 38 | return type_; |
| 39 | } | 39 | } |
| 40 | 40 | ||
| 41 | std::string getName() const | 41 | std::string getName() const |
| 42 | { | 42 | { |
| 43 | return name_; | 43 | return name_; |
| 44 | } | 44 | } |
| 45 | 45 | ||
| 46 | // Integer | 46 | // Integer |
| 47 | 47 | ||
| 48 | field(std::string name, int arg); | 48 | field(std::string name, int arg); |
| 49 | 49 | ||
| 50 | int getInteger() const; | 50 | int getInteger() const; |
| 51 | 51 | ||
| 52 | // String | 52 | // String |
| 53 | 53 | ||
| 54 | field(std::string name, std::string arg); | 54 | field(std::string name, std::string arg); |
| 55 | 55 | ||
| 56 | std::string getString() const; | 56 | std::string getString() const; |
| 57 | 57 | ||
| 58 | private: | 58 | private: |
| diff --git a/generator/generator.cpp b/generator/generator.cpp index eb2750c..54f5d69 100644 --- a/generator/generator.cpp +++ b/generator/generator.cpp | |||
| @@ -11,7 +11,7 @@ | |||
| 11 | 11 | ||
| 12 | namespace cadence { | 12 | namespace cadence { |
| 13 | namespace generator { | 13 | namespace generator { |
| 14 | 14 | ||
| 15 | generator::generator( | 15 | generator::generator( |
| 16 | std::string inputpath, | 16 | std::string inputpath, |
| 17 | std::string outputpath) : | 17 | std::string outputpath) : |
| @@ -23,23 +23,23 @@ namespace cadence { | |||
| 23 | { | 23 | { |
| 24 | inputpath_ += '/'; | 24 | inputpath_ += '/'; |
| 25 | } | 25 | } |
| 26 | 26 | ||
| 27 | inputpath_ += "highlevel/"; | 27 | inputpath_ += "highlevel/"; |
| 28 | } | 28 | } |
| 29 | 29 | ||
| 30 | void generator::run() | 30 | void generator::run() |
| 31 | { | 31 | { |
| 32 | // Creates the datafile. | 32 | // Creates the datafile. |
| 33 | writeSchema(); | 33 | writeSchema(); |
| 34 | 34 | ||
| 35 | // Scans the AcousticBrainz data dump and generates a list of all of the | 35 | // Scans the AcousticBrainz data dump and generates a list of all of the |
| 36 | // files in the dump. | 36 | // files in the dump. |
| 37 | scanDirectories(); | 37 | scanDirectories(); |
| 38 | 38 | ||
| 39 | // Parses each data file and enters it into the database. | 39 | // Parses each data file and enters it into the database. |
| 40 | parseData(); | 40 | parseData(); |
| 41 | } | 41 | } |
| 42 | 42 | ||
| 43 | void generator::writeSchema() | 43 | void generator::writeSchema() |
| 44 | { | 44 | { |
| 45 | std::ifstream file("schema.sql"); | 45 | std::ifstream file("schema.sql"); |
| @@ -47,7 +47,7 @@ namespace cadence { | |||
| 47 | { | 47 | { |
| 48 | throw std::invalid_argument("Could not find database schema"); | 48 | throw std::invalid_argument("Could not find database schema"); |
| 49 | } | 49 | } |
| 50 | 50 | ||
| 51 | std::ostringstream schemaBuilder; | 51 | std::ostringstream schemaBuilder; |
| 52 | std::string line; | 52 | std::string line; |
| 53 | while (std::getline(file, line)) | 53 | while (std::getline(file, line)) |
| @@ -56,10 +56,10 @@ namespace cadence { | |||
| 56 | { | 56 | { |
| 57 | line.pop_back(); | 57 | line.pop_back(); |
| 58 | } | 58 | } |
| 59 | 59 | ||
| 60 | schemaBuilder << line; | 60 | schemaBuilder << line; |
| 61 | } | 61 | } |
| 62 | 62 | ||
| 63 | std::string schema = schemaBuilder.str(); | 63 | std::string schema = schemaBuilder.str(); |
| 64 | auto queries = split<std::list<std::string>>(schema, ";"); | 64 | auto queries = split<std::list<std::string>>(schema, ";"); |
| 65 | progress ppgs("Writing database schema...", queries.size()); | 65 | progress ppgs("Writing database schema...", queries.size()); |
| @@ -69,73 +69,73 @@ namespace cadence { | |||
| 69 | { | 69 | { |
| 70 | db_.runQuery(query); | 70 | db_.runQuery(query); |
| 71 | } | 71 | } |
| 72 | 72 | ||
| 73 | ppgs.update(); | 73 | ppgs.update(); |
| 74 | } | 74 | } |
| 75 | } | 75 | } |
| 76 | 76 | ||
| 77 | void generator::scanDirectories() | 77 | void generator::scanDirectories() |
| 78 | { | 78 | { |
| 79 | std::cout << "Scanning AcousticBrainz dump..." << std::endl; | 79 | std::cout << "Scanning AcousticBrainz dump..." << std::endl; |
| 80 | 80 | ||
| 81 | DIR* topdir; | 81 | DIR* topdir; |
| 82 | if ((topdir = opendir(inputpath_.c_str())) == nullptr) | 82 | if ((topdir = opendir(inputpath_.c_str())) == nullptr) |
| 83 | { | 83 | { |
| 84 | throw std::invalid_argument("Invalid AcousticBrainz data directory"); | 84 | throw std::invalid_argument("Invalid AcousticBrainz data directory"); |
| 85 | } | 85 | } |
| 86 | 86 | ||
| 87 | struct dirent* topent; | 87 | struct dirent* topent; |
| 88 | while ((topent = readdir(topdir)) != nullptr) | 88 | while ((topent = readdir(topdir)) != nullptr) |
| 89 | { | 89 | { |
| 90 | if (topent->d_name[0] != '.') | 90 | if (topent->d_name[0] != '.') |
| 91 | { | 91 | { |
| 92 | std::string directory = inputpath_ + topent->d_name + "/"; | 92 | std::string directory = inputpath_ + topent->d_name + "/"; |
| 93 | 93 | ||
| 94 | DIR* subdir; | 94 | DIR* subdir; |
| 95 | if ((subdir = opendir(directory.c_str())) == nullptr) | 95 | if ((subdir = opendir(directory.c_str())) == nullptr) |
| 96 | { | 96 | { |
| 97 | throw std::invalid_argument("Invalid AcousticBrainz data directory"); | 97 | throw std::invalid_argument("Invalid AcousticBrainz data directory"); |
| 98 | } | 98 | } |
| 99 | 99 | ||
| 100 | struct dirent* subent; | 100 | struct dirent* subent; |
| 101 | while ((subent = readdir(subdir)) != nullptr) | 101 | while ((subent = readdir(subdir)) != nullptr) |
| 102 | { | 102 | { |
| 103 | if (subent->d_name[0] != '.') | 103 | if (subent->d_name[0] != '.') |
| 104 | { | 104 | { |
| 105 | std::string subdirectory = directory + subent->d_name + "/"; | 105 | std::string subdirectory = directory + subent->d_name + "/"; |
| 106 | 106 | ||
| 107 | DIR* subsubdir; | 107 | DIR* subsubdir; |
| 108 | if ((subsubdir = opendir(subdirectory.c_str())) == nullptr) | 108 | if ((subsubdir = opendir(subdirectory.c_str())) == nullptr) |
| 109 | { | 109 | { |
| 110 | throw std::invalid_argument("Invalid AcousticBrainz data directory"); | 110 | throw std::invalid_argument("Invalid AcousticBrainz data directory"); |
| 111 | } | 111 | } |
| 112 | 112 | ||
| 113 | struct dirent* subsubent; | 113 | struct dirent* subsubent; |
| 114 | while ((subsubent = readdir(subsubdir)) != nullptr) | 114 | while ((subsubent = readdir(subsubdir)) != nullptr) |
| 115 | { | 115 | { |
| 116 | if (subsubent->d_name[0] != '.') | 116 | if (subsubent->d_name[0] != '.') |
| 117 | { | 117 | { |
| 118 | std::string datafile = subdirectory + subsubent->d_name; | 118 | std::string datafile = subdirectory + subsubent->d_name; |
| 119 | 119 | ||
| 120 | datafiles_.push_back(datafile); | 120 | datafiles_.push_back(datafile); |
| 121 | } | 121 | } |
| 122 | } | 122 | } |
| 123 | 123 | ||
| 124 | closedir(subsubdir); | 124 | closedir(subsubdir); |
| 125 | } | 125 | } |
| 126 | } | 126 | } |
| 127 | 127 | ||
| 128 | closedir(subdir); | 128 | closedir(subdir); |
| 129 | } | 129 | } |
| 130 | } | 130 | } |
| 131 | 131 | ||
| 132 | closedir(topdir); | 132 | closedir(topdir); |
| 133 | } | 133 | } |
| 134 | 134 | ||
| 135 | void generator::parseData() | 135 | void generator::parseData() |
| 136 | { | 136 | { |
| 137 | progress ppgs("Parsing AcousticBrainz data files...", datafiles_.size()); | 137 | progress ppgs("Parsing AcousticBrainz data files...", datafiles_.size()); |
| 138 | 138 | ||
| 139 | for (std::string datafile : datafiles_) | 139 | for (std::string datafile : datafiles_) |
| 140 | { | 140 | { |
| 141 | ppgs.update(); | 141 | ppgs.update(); |
| @@ -145,7 +145,7 @@ namespace cadence { | |||
| 145 | std::ifstream dataStream(datafile); | 145 | std::ifstream dataStream(datafile); |
| 146 | dataStream >> jsonData; | 146 | dataStream >> jsonData; |
| 147 | } | 147 | } |
| 148 | 148 | ||
| 149 | try | 149 | try |
| 150 | { | 150 | { |
| 151 | std::vector<mood> moods; | 151 | std::vector<mood> moods; |
| @@ -158,16 +158,16 @@ namespace cadence { | |||
| 158 | moods.emplace_back(mood::type::relaxed, jsonData["highlevel"]["mood_relaxed"]["all"]["relaxed"]); | 158 | moods.emplace_back(mood::type::relaxed, jsonData["highlevel"]["mood_relaxed"]["all"]["relaxed"]); |
| 159 | moods.emplace_back(mood::type::sad, jsonData["highlevel"]["mood_sad"]["all"]["sad"]); | 159 | moods.emplace_back(mood::type::sad, jsonData["highlevel"]["mood_sad"]["all"]["sad"]); |
| 160 | moods.emplace_back(mood::type::instrumental, jsonData["highlevel"]["voice_instrumental"]["all"]["instrumental"]); | 160 | moods.emplace_back(mood::type::instrumental, jsonData["highlevel"]["voice_instrumental"]["all"]["instrumental"]); |
| 161 | 161 | ||
| 162 | std::sort(std::begin(moods), std::end(moods), [] (const mood& left, const mood& right) -> bool { | 162 | std::sort(std::begin(moods), std::end(moods), [] (const mood& left, const mood& right) -> bool { |
| 163 | return left.getProbability() > right.getProbability(); | 163 | return left.getProbability() > right.getProbability(); |
| 164 | }); | 164 | }); |
| 165 | 165 | ||
| 166 | std::list<field> fields; | 166 | std::list<field> fields; |
| 167 | fields.emplace_back("title", jsonData["metadata"]["tags"]["title"][0].get<std::string>()); | 167 | fields.emplace_back("title", jsonData["metadata"]["tags"]["title"][0].get<std::string>()); |
| 168 | fields.emplace_back("artist", jsonData["metadata"]["tags"]["artist"][0].get<std::string>()); | 168 | fields.emplace_back("artist", jsonData["metadata"]["tags"]["artist"][0].get<std::string>()); |
| 169 | fields.emplace_back("category", moods.front().getCategory()); | 169 | fields.emplace_back("category", moods.front().getCategory()); |
| 170 | 170 | ||
| 171 | db_.insertIntoTable("songs", std::move(fields)); | 171 | db_.insertIntoTable("songs", std::move(fields)); |
| 172 | } catch (const std::domain_error& ex) | 172 | } catch (const std::domain_error& ex) |
| 173 | { | 173 | { |
| @@ -175,6 +175,6 @@ namespace cadence { | |||
| 175 | } | 175 | } |
| 176 | } | 176 | } |
| 177 | } | 177 | } |
| 178 | 178 | ||
| 179 | }; | 179 | }; |
| 180 | }; | 180 | }; |
| diff --git a/generator/generator.h b/generator/generator.h index d911c61..dff8eeb 100644 --- a/generator/generator.h +++ b/generator/generator.h | |||
| @@ -7,44 +7,44 @@ | |||
| 7 | 7 | ||
| 8 | namespace cadence { | 8 | namespace cadence { |
| 9 | namespace generator { | 9 | namespace generator { |
| 10 | 10 | ||
| 11 | class generator { | 11 | class generator { |
| 12 | public: | 12 | public: |
| 13 | 13 | ||
| 14 | // Constructor | 14 | // Constructor |
| 15 | 15 | ||
| 16 | generator( | 16 | generator( |
| 17 | std::string inputpath, | 17 | std::string inputpath, |
| 18 | std::string outputpath); | 18 | std::string outputpath); |
| 19 | 19 | ||
| 20 | // Action | 20 | // Action |
| 21 | 21 | ||
| 22 | void run(); | 22 | void run(); |
| 23 | 23 | ||
| 24 | // Subroutines | 24 | // Subroutines |
| 25 | 25 | ||
| 26 | void writeSchema(); | 26 | void writeSchema(); |
| 27 | 27 | ||
| 28 | void scanDirectories(); | 28 | void scanDirectories(); |
| 29 | 29 | ||
| 30 | void parseData(); | 30 | void parseData(); |
| 31 | 31 | ||
| 32 | private: | 32 | private: |
| 33 | 33 | ||
| 34 | // Input | 34 | // Input |
| 35 | 35 | ||
| 36 | std::string inputpath_; | 36 | std::string inputpath_; |
| 37 | 37 | ||
| 38 | // Output | 38 | // Output |
| 39 | 39 | ||
| 40 | database db_; | 40 | database db_; |
| 41 | 41 | ||
| 42 | // Cache | 42 | // Cache |
| 43 | 43 | ||
| 44 | std::list<std::string> datafiles_; | 44 | std::list<std::string> datafiles_; |
| 45 | 45 | ||
| 46 | }; | 46 | }; |
| 47 | 47 | ||
| 48 | }; | 48 | }; |
| 49 | }; | 49 | }; |
| 50 | 50 | ||
| diff --git a/generator/main.cpp b/generator/main.cpp index 5a3d2dd..318770e 100644 --- a/generator/main.cpp +++ b/generator/main.cpp | |||
| @@ -16,7 +16,7 @@ int main(int argc, char** argv) | |||
| 16 | try | 16 | try |
| 17 | { | 17 | { |
| 18 | cadence::generator::generator app(argv[1], argv[2]); | 18 | cadence::generator::generator app(argv[1], argv[2]); |
| 19 | 19 | ||
| 20 | try | 20 | try |
| 21 | { | 21 | { |
| 22 | app.run(); | 22 | app.run(); |
| diff --git a/generator/mood.cpp b/generator/mood.cpp index 9f53dce..b74211c 100644 --- a/generator/mood.cpp +++ b/generator/mood.cpp | |||
| @@ -2,7 +2,7 @@ | |||
| 2 | 2 | ||
| 3 | namespace cadence { | 3 | namespace cadence { |
| 4 | namespace generator { | 4 | namespace generator { |
| 5 | 5 | ||
| 6 | // The categories are: | 6 | // The categories are: |
| 7 | // - party (+danceable, -acoustic, +electronic, +party) = ~.21 | 7 | // - party (+danceable, -acoustic, +electronic, +party) = ~.21 |
| 8 | // - chill (-danceable, +acoustic, -aggressive, -electronic, -party, +relaxed) = ~.49 | 8 | // - chill (-danceable, +acoustic, -aggressive, -electronic, -party, +relaxed) = ~.49 |
| @@ -11,7 +11,7 @@ namespace cadence { | |||
| 11 | // - sad (-happy, +sad) = ~.02 | 11 | // - sad (-happy, +sad) = ~.02 |
| 12 | // - instrumental (+instrumental) = ~.12 | 12 | // - instrumental (+instrumental) = ~.12 |
| 13 | // - vocal (-instrumental) = ~.10 | 13 | // - vocal (-instrumental) = ~.10 |
| 14 | 14 | ||
| 15 | mood::mood(type t, double prob) : type_(t) | 15 | mood::mood(type t, double prob) : type_(t) |
| 16 | { | 16 | { |
| 17 | if (prob >= 0.5) | 17 | if (prob >= 0.5) |
| @@ -22,73 +22,73 @@ namespace cadence { | |||
| 22 | probability_ = 1.0 - prob; | 22 | probability_ = 1.0 - prob; |
| 23 | positive_ = false; | 23 | positive_ = false; |
| 24 | } | 24 | } |
| 25 | 25 | ||
| 26 | switch (t) | 26 | switch (t) |
| 27 | { | 27 | { |
| 28 | case type::danceable: | 28 | case type::danceable: |
| 29 | { | 29 | { |
| 30 | category_ = (positive_ ? "party" : "chill"); | 30 | category_ = (positive_ ? "party" : "chill"); |
| 31 | 31 | ||
| 32 | break; | 32 | break; |
| 33 | } | 33 | } |
| 34 | 34 | ||
| 35 | case type::acoustic: | 35 | case type::acoustic: |
| 36 | { | 36 | { |
| 37 | category_ = (positive_ ? "chill" : "party"); | 37 | category_ = (positive_ ? "chill" : "party"); |
| 38 | 38 | ||
| 39 | break; | 39 | break; |
| 40 | } | 40 | } |
| 41 | 41 | ||
| 42 | case type::aggressive: | 42 | case type::aggressive: |
| 43 | { | 43 | { |
| 44 | category_ = (positive_ ? "crazy" : "chill"); | 44 | category_ = (positive_ ? "crazy" : "chill"); |
| 45 | 45 | ||
| 46 | break; | 46 | break; |
| 47 | } | 47 | } |
| 48 | 48 | ||
| 49 | case type::electronic: | 49 | case type::electronic: |
| 50 | { | 50 | { |
| 51 | category_ = (positive_ ? "party" : "chill"); | 51 | category_ = (positive_ ? "party" : "chill"); |
| 52 | 52 | ||
| 53 | break; | 53 | break; |
| 54 | } | 54 | } |
| 55 | 55 | ||
| 56 | case type::happy: | 56 | case type::happy: |
| 57 | { | 57 | { |
| 58 | category_ = (positive_ ? "happy" : "sad"); | 58 | category_ = (positive_ ? "happy" : "sad"); |
| 59 | 59 | ||
| 60 | break; | 60 | break; |
| 61 | } | 61 | } |
| 62 | 62 | ||
| 63 | case type::party: | 63 | case type::party: |
| 64 | { | 64 | { |
| 65 | category_ = (positive_ ? "party" : "chill"); | 65 | category_ = (positive_ ? "party" : "chill"); |
| 66 | 66 | ||
| 67 | break; | 67 | break; |
| 68 | } | 68 | } |
| 69 | 69 | ||
| 70 | case type::relaxed: | 70 | case type::relaxed: |
| 71 | { | 71 | { |
| 72 | category_ = (positive_ ? "chill" : "crazy"); | 72 | category_ = (positive_ ? "chill" : "crazy"); |
| 73 | 73 | ||
| 74 | break; | 74 | break; |
| 75 | } | 75 | } |
| 76 | 76 | ||
| 77 | case type::sad: | 77 | case type::sad: |
| 78 | { | 78 | { |
| 79 | category_ = (positive_ ? "sad" : "happy"); | 79 | category_ = (positive_ ? "sad" : "happy"); |
| 80 | 80 | ||
| 81 | break; | 81 | break; |
| 82 | } | 82 | } |
| 83 | 83 | ||
| 84 | case type::instrumental: | 84 | case type::instrumental: |
| 85 | { | 85 | { |
| 86 | category_ = (positive_ ? "instrumental" : "vocal"); | 86 | category_ = (positive_ ? "instrumental" : "vocal"); |
| 87 | 87 | ||
| 88 | break; | 88 | break; |
| 89 | } | 89 | } |
| 90 | } | 90 | } |
| 91 | } | 91 | } |
| 92 | 92 | ||
| 93 | }; | 93 | }; |
| 94 | }; | 94 | }; |
| diff --git a/generator/mood.h b/generator/mood.h index df5d28f..c36e5ee 100644 --- a/generator/mood.h +++ b/generator/mood.h | |||
| @@ -5,7 +5,7 @@ | |||
| 5 | 5 | ||
| 6 | namespace cadence { | 6 | namespace cadence { |
| 7 | namespace generator { | 7 | namespace generator { |
| 8 | 8 | ||
| 9 | class mood { | 9 | class mood { |
| 10 | public: | 10 | public: |
| 11 | enum class type { | 11 | enum class type { |
| @@ -19,41 +19,41 @@ namespace cadence { | |||
| 19 | sad, | 19 | sad, |
| 20 | instrumental | 20 | instrumental |
| 21 | }; | 21 | }; |
| 22 | 22 | ||
| 23 | // Constructor | 23 | // Constructor |
| 24 | 24 | ||
| 25 | mood(type t, double prob); | 25 | mood(type t, double prob); |
| 26 | 26 | ||
| 27 | // Accessors | 27 | // Accessors |
| 28 | 28 | ||
| 29 | type getType() const | 29 | type getType() const |
| 30 | { | 30 | { |
| 31 | return type_; | 31 | return type_; |
| 32 | } | 32 | } |
| 33 | 33 | ||
| 34 | double getProbability() const | 34 | double getProbability() const |
| 35 | { | 35 | { |
| 36 | return probability_; | 36 | return probability_; |
| 37 | } | 37 | } |
| 38 | 38 | ||
| 39 | bool getPositive() const | 39 | bool getPositive() const |
| 40 | { | 40 | { |
| 41 | return positive_; | 41 | return positive_; |
| 42 | } | 42 | } |
| 43 | 43 | ||
| 44 | std::string getCategory() const | 44 | std::string getCategory() const |
| 45 | { | 45 | { |
| 46 | return category_; | 46 | return category_; |
| 47 | } | 47 | } |
| 48 | 48 | ||
| 49 | private: | 49 | private: |
| 50 | type type_; | 50 | type type_; |
| 51 | double probability_; | 51 | double probability_; |
| 52 | bool positive_; | 52 | bool positive_; |
| 53 | std::string category_; | 53 | std::string category_; |
| 54 | 54 | ||
| 55 | }; | 55 | }; |
| 56 | 56 | ||
| 57 | }; | 57 | }; |
| 58 | }; | 58 | }; |
| 59 | 59 | ||
| diff --git a/generator/progress.h b/generator/progress.h index f42f701..e5cc13d 100644 --- a/generator/progress.h +++ b/generator/progress.h | |||
| @@ -5,20 +5,20 @@ | |||
| 5 | 5 | ||
| 6 | namespace cadence { | 6 | namespace cadence { |
| 7 | namespace generator { | 7 | namespace generator { |
| 8 | 8 | ||
| 9 | class progress { | 9 | class progress { |
| 10 | private: | 10 | private: |
| 11 | std::string message; | 11 | std::string message; |
| 12 | int total; | 12 | int total; |
| 13 | int cur = 0; | 13 | int cur = 0; |
| 14 | int lprint = 0; | 14 | int lprint = 0; |
| 15 | 15 | ||
| 16 | public: | 16 | public: |
| 17 | progress(std::string message, int total) : message(message), total(total) | 17 | progress(std::string message, int total) : message(message), total(total) |
| 18 | { | 18 | { |
| 19 | std::cout << message << " 0%" << std::flush; | 19 | std::cout << message << " 0%" << std::flush; |
| 20 | } | 20 | } |
| 21 | 21 | ||
| 22 | void update(int val) | 22 | void update(int val) |
| 23 | { | 23 | { |
| 24 | if (val <= total) | 24 | if (val <= total) |
| @@ -27,29 +27,29 @@ namespace cadence { | |||
| 27 | } else { | 27 | } else { |
| 28 | cur = total; | 28 | cur = total; |
| 29 | } | 29 | } |
| 30 | 30 | ||
| 31 | int pp = cur * 100 / total; | 31 | int pp = cur * 100 / total; |
| 32 | if (pp != lprint) | 32 | if (pp != lprint) |
| 33 | { | 33 | { |
| 34 | lprint = pp; | 34 | lprint = pp; |
| 35 | 35 | ||
| 36 | std::cout << "\b\b\b\b" << std::right; | 36 | std::cout << "\b\b\b\b" << std::right; |
| 37 | std::cout.width(3); | 37 | std::cout.width(3); |
| 38 | std::cout << pp << "%" << std::flush; | 38 | std::cout << pp << "%" << std::flush; |
| 39 | } | 39 | } |
| 40 | } | 40 | } |
| 41 | 41 | ||
| 42 | void update() | 42 | void update() |
| 43 | { | 43 | { |
| 44 | update(cur+1); | 44 | update(cur+1); |
| 45 | } | 45 | } |
| 46 | 46 | ||
| 47 | ~progress() | 47 | ~progress() |
| 48 | { | 48 | { |
| 49 | std::cout << "\b\b\b\b100%" << std::endl; | 49 | std::cout << "\b\b\b\b100%" << std::endl; |
| 50 | } | 50 | } |
| 51 | }; | 51 | }; |
| 52 | 52 | ||
| 53 | }; | 53 | }; |
| 54 | }; | 54 | }; |
| 55 | 55 | ||
