|
|
|
@ -31,8 +31,13 @@ |
|
|
|
|
* |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
#include <algorithm> |
|
|
|
|
#include <chrono> |
|
|
|
|
#include <cmath> |
|
|
|
|
#include <fstream> |
|
|
|
|
#include <iostream> |
|
|
|
|
#include <memory> |
|
|
|
|
#include <sstream> |
|
|
|
|
#include <string> |
|
|
|
|
#include <thread> |
|
|
|
|
|
|
|
|
@ -57,29 +62,148 @@ using examples::Rectangle; |
|
|
|
|
using examples::RouteSummary; |
|
|
|
|
using examples::RouteNote; |
|
|
|
|
using examples::RouteGuide; |
|
|
|
|
using std::chrono::system_clock; |
|
|
|
|
|
|
|
|
|
const float kCoordFactor = 10000000.0; |
|
|
|
|
|
|
|
|
|
bool ParseDb(const std::string& stream, std::vector<Feature>* feature_list) { |
|
|
|
|
// TODO
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
float ConvertToRadians(float num) { |
|
|
|
|
return num * 3.1415926 /180; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
float GetDistance(const Point& start, const Point& end) { |
|
|
|
|
|
|
|
|
|
float lat_1 = start.latitude() / kCoordFactor; |
|
|
|
|
float lat_2 = end.latitude() / kCoordFactor; |
|
|
|
|
float lon_1 = start.longitude() / kCoordFactor; |
|
|
|
|
float lon_2 = end.longitude() / kCoordFactor; |
|
|
|
|
float lat_rad_1 = ConvertToRadians(lat_1); |
|
|
|
|
float lat_rad_2 = ConvertToRadians(lat_2); |
|
|
|
|
float delta_lat_rad = ConvertToRadians(lat_2-lat_1); |
|
|
|
|
float delta_lon_rad = ConvertToRadians(lon_2-lon_1); |
|
|
|
|
|
|
|
|
|
float a = pow(sin(delta_lat_rad/2), 2) + cos(lat_rad_1) * cos(lat_rad_2) * |
|
|
|
|
pow(sin(delta_lon_rad/2), 2); |
|
|
|
|
float c = 2 * atan2(sqrt(a), sqrt(1-a)); |
|
|
|
|
int R = 6371000; // metres
|
|
|
|
|
|
|
|
|
|
return R * c; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void FillFeatureList(const std::string& db_path, std::vector<Feature>* feature_list) { |
|
|
|
|
if (db_path.empty()) { |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
std::ifstream db_file(db_path); |
|
|
|
|
if (!db_file.is_open()) { |
|
|
|
|
std::cout << "Failed to open " << db_path << std::endl; |
|
|
|
|
} |
|
|
|
|
std::stringstream db; |
|
|
|
|
db << db_file.rdbuf(); |
|
|
|
|
ParseDb(db.str(), feature_list); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
std::string GetFeatureName(const Point& point, |
|
|
|
|
const std::vector<Feature>& feature_list) { |
|
|
|
|
for (const Feature& f : feature_list) { |
|
|
|
|
if (f.location().latitude() == point.latitude() && |
|
|
|
|
f.location().longitude() == point.longitude()) { |
|
|
|
|
return f.name(); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
return ""; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
class RouteGuideImpl final : public RouteGuide::Service { |
|
|
|
|
public: |
|
|
|
|
RouteGuideImpl(const std::string& db_path) { |
|
|
|
|
FillFeatureList(db_path, &feature_list_); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
Status GetFeature(ServerContext* context, const Point* point, |
|
|
|
|
Feature* feature) override { |
|
|
|
|
feature->set_name(GetFeatureName(*point, feature_list_)); |
|
|
|
|
feature->mutable_location()->CopyFrom(*point); |
|
|
|
|
return Status::OK; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
Status ListFeatures(ServerContext* context, const Rectangle* rectangle, |
|
|
|
|
ServerWriter<Feature>* writer) override { |
|
|
|
|
auto lo = rectangle->lo(); |
|
|
|
|
auto hi = rectangle->hi(); |
|
|
|
|
long left = std::min(lo.longitude(), hi.longitude()); |
|
|
|
|
long right = std::max(lo.longitude(), hi.longitude()); |
|
|
|
|
long top = std::max(lo.latitude(), hi.latitude()); |
|
|
|
|
long bottom = std::min(lo.latitude(), hi.latitude()); |
|
|
|
|
for (const Feature& f : feature_list_) { |
|
|
|
|
if (f.location().longitude() >= left && |
|
|
|
|
f.location().longitude() <= right && |
|
|
|
|
f.location().latitude() >= bottom && |
|
|
|
|
f.location().latitude() <= top) { |
|
|
|
|
writer->Write(f); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
return Status::OK; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
Status RecordRoute(ServerContext* context, ServerReader<Point>* reader, |
|
|
|
|
RouteSummary* summary) override { |
|
|
|
|
Point point; |
|
|
|
|
int point_count = 0; |
|
|
|
|
int feature_count = 0; |
|
|
|
|
float distance = 0.0; |
|
|
|
|
Point previous; |
|
|
|
|
|
|
|
|
|
system_clock::time_point start_time = system_clock::now(); |
|
|
|
|
while (reader->Read(&point)) { |
|
|
|
|
point_count++; |
|
|
|
|
if (!GetFeatureName(point, feature_list_).empty()) { |
|
|
|
|
feature_count++; |
|
|
|
|
} |
|
|
|
|
if (point_count != 1) { |
|
|
|
|
distance += GetDistance(previous, point); |
|
|
|
|
} |
|
|
|
|
previous = point; |
|
|
|
|
} |
|
|
|
|
system_clock::time_point end_time = system_clock::now(); |
|
|
|
|
summary->set_point_count(point_count); |
|
|
|
|
summary->set_feature_count(feature_count); |
|
|
|
|
summary->set_distance(static_cast<long>(distance)); |
|
|
|
|
auto secs = std::chrono::duration_cast<std::chrono::seconds>( |
|
|
|
|
end_time - start_time); |
|
|
|
|
summary->set_elapsed_time(secs.count()); |
|
|
|
|
|
|
|
|
|
return Status::OK; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
Status RouteChat(ServerContext* context, |
|
|
|
|
ServerReaderWriter<RouteNote, RouteNote>* stream) override { |
|
|
|
|
std::vector<RouteNote> received_notes; |
|
|
|
|
RouteNote note; |
|
|
|
|
while (stream->Read(¬e)) { |
|
|
|
|
for (const RouteNote& n : received_notes) { |
|
|
|
|
if (n.location().latitude() == note.location().latitude() && |
|
|
|
|
n.location().longitude() == note.location().longitude()) { |
|
|
|
|
stream->Write(n); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
received_notes.push_back(note); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
return Status::OK; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
private: |
|
|
|
|
|
|
|
|
|
std::vector<Feature> feature_list_; |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
void RunServer() { |
|
|
|
|
void RunServer(const std::string& db_path) { |
|
|
|
|
std::string server_address("0.0.0.0:50051"); |
|
|
|
|
RouteGuideImpl service; |
|
|
|
|
RouteGuideImpl service(db_path); |
|
|
|
|
|
|
|
|
|
ServerBuilder builder; |
|
|
|
|
builder.AddPort(server_address); |
|
|
|
@ -94,7 +218,20 @@ void RunServer() { |
|
|
|
|
int main(int argc, char** argv) { |
|
|
|
|
grpc_init(); |
|
|
|
|
|
|
|
|
|
RunServer(); |
|
|
|
|
std::string db_path; |
|
|
|
|
std::string arg_str("--db_path"); |
|
|
|
|
if (argc > 1) { |
|
|
|
|
std::string argv_1 = argv[1]; |
|
|
|
|
size_t start_position = argv_1.find(arg_str); |
|
|
|
|
if (start_position != std::string::npos) { |
|
|
|
|
start_position += arg_str.size(); |
|
|
|
|
if (argv_1[start_position] == ' ' || |
|
|
|
|
argv_1[start_position] == '=') { |
|
|
|
|
db_path = argv_1.substr(start_position + 1); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
RunServer(db_path); |
|
|
|
|
|
|
|
|
|
grpc_shutdown(); |
|
|
|
|
return 0; |
|
|
|
|