ALidarSQLManager.cc

Go to the documentation of this file.

#include <sstream>
#include <boost/lexical_cast.hpp>

#include <atm/ALidarSQLManager.h>
#include <det/Detector.h>
#include <utl/ErrorLogger.h>
#include <utl/TimeStamp.h>
#include <utl/AugerException.h>
#include <utl/StringCompare.h>

using namespace std;
using namespace atm;
using namespace det;
using namespace utl;

REGISTER_A_MANAGER("ALidarSQLManager", ALidarSQLManager);


ALidarSQLManager::ALidarSQLManager() :
  VSQLManager()
{
  const char* const tables[] = {
    "VAOD", "authors", "cloud", "lidar", "lidar_zone", "software",
    "zone", "zone_slices"
  };
  const int n = sizeof(tables) / sizeof(tables[0]);
  fAvailableComponents.insert(tables, tables + n);
}


VManager::Status
ALidarSQLManager::GetDBResFundamental(const std::string& tableName,
                                      const std::string& columnName,
                                      const IndexMap& componentIndex)
  const
{
  const utl::TimeStamp detTime = Detector::GetInstance().GetTime();

  ostringstream query;

  // lidar table is a special case, as it requires checking software version
  if (tableName == "lidar") {

    if (StringEquivalent(fDatabaseSoftwareVersion, "default")) {
      // do NOT check software version

      query << "SELECT lidar_id "
               "FROM lidar "
               "WHERE start_time <= \"" << detTime.GetGPSSecond() << "\" "
               "AND (end_time > \"" << detTime.GetGPSSecond() << "\" "
               "OR end_time IS NULL) ORDER BY last_modified DESC, start_time DESC LIMIT 1";

    } else {                 // DO check software version

      query << "SELECT lidar_id "
               "FROM lidar, software "
               "WHERE start_time <= \"" << detTime.GetGPSSecond() << "\" "
               "AND (end_time > \"" << detTime.GetGPSSecond() << "\" "
               "OR end_time IS NULL) "
               "AND lidar.software_id = software.software_id "
               "AND software.software_version = \"" << fDatabaseSoftwareVersion << "\" "
               "ORDER BY lidar.last_modified DESC, start_time DESC LIMIT 1";
    }

  } else if (tableName == "zone") {

    if (componentIndex.find("zone_id") == componentIndex.end()) {
      ERROR("request for zone data called with no zone_id");
      throw DataNotFoundInDBException("zone", "zone_id", componentIndex,
                                      boost::lexical_cast<string>(Detector::GetInstance().GetTime()));
    }
    query << "SELECT " << columnName << " "
             "FROM zone "
             "WHERE zone_id = " << componentIndex.find("zone_id")->second;

  } else {

    if (fAvailableComponents.find(tableName) == fAvailableComponents.end())
      return eNotFound;

    // issue queries in form :
    //   SELECT columnName FROM tableName WHERE
    //     componentIndex[0].first = componentIndex[0].second AND
    //     componentIndex[1].first = componentIndex[1].second
    //     (etc)
    //
    query << "SELECT " << columnName << " "
             "FROM " << tableName << " "
             "WHERE " << MergeIndexMap(componentIndex);
  }

  return Query(query);
}


VManager::Status
ALidarSQLManager::GetDBResVector(const std::string& tableName,
                                 const std::string& columnName,
                                 const IndexMap& componentIndex)
  const
{
  if (tableName == "zone") {

    // index map must specify the lidar_id
    if (componentIndex.find("lidar_id") == componentIndex.end()) {
      DEBUGLOG("no lidar_id specified when looking up zone names");
      return VManager::eNotFound;
    }

    ostringstream query;
    query << "SELECT zone." << columnName << " FROM zone,lidar_zone "
             "WHERE lidar_id = " << componentIndex.find("lidar_id")->second << " "
             "AND lidar_zone.zone_id = zone.zone_id;";

    return Query(query);

  } else {

    if (fAvailableComponents.find(tableName) == fAvailableComponents.end())
      return eNotFound;

    ostringstream query;
    query << "SELECT " << columnName << " "
             "FROM " << tableName << " "
             "WHERE " << MergeIndexMap(componentIndex);

    return Query(query);

  }
}


// Configure (x)emacs for this file ...
// Local Variables:
// mode: c++
// End: