AGOESSQLManager.cc

Go to the documentation of this file.

#include <atm/AGOESSQLManager.h>

#include <det/Detector.h>

#include <utl/TimeStamp.h>
#include <utl/StringCompare.h>
#include <utl/MoonCycle.h>

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


REGISTER_A_MANAGER("AGOESSQLManager", AGOESSQLManager);


VManager::Status
AGOESSQLManager::GetData(map<unsigned int, int>& returnData,
                          const string& tableName,
                          const string& componentName,
                          const IndexMap& /*componentIndex*/)
  const
{
  if (tableName != "cloud_pixel" || componentName != "cloud_probability") {
    return eNotFound;
  }

  ostringstream query;
  query << "SELECT pixel_id, cloud_probability ";

  // Check for a software version specification...
  if (!utl::StringEquivalent(fDatabaseSoftwareVersion, "default"))
    query << "FROM cloud_pixel, software "
             "WHERE cloud_pixel.software_id = software.software_id "
             "AND software.software_version = \"" << fDatabaseSoftwareVersion
          << "\" "
             "AND ";
  // ...or don't
  else
    query << "FROM cloud_pixel WHERE ";

  const utl::TimeStamp detTime = Detector::GetInstance().GetTime();

  query << "moon_cycle = "
        << utl::MoonCycle(detTime).GetFullMoonCycle() << " "
           "AND start_time <= " << detTime.GetGPSSecond() << " "
           "AND end_time > " << detTime.GetGPSSecond() << " "
           "ORDER BY last_modified ASC, start_time ASC";

  if (Query(query, "what") == eNotFound)
    return eNotFound;

  boost::tuple<unsigned int, int> pixelId_prob;
  returnData.clear();

  while (FetchRowMany(pixelId_prob, false) == eFound)
    returnData[pixelId_prob.get<0>()] = pixelId_prob.get<1>();

  FreeResult();

  if (returnData.empty())
    return eNotFound;

  return eFound;
}


VManager::Status
AGOESSQLManager::GetDBResFundamental(const string& tableName,
                                     const string& columnName,
                                     const IndexMap& index)
  const
{
  if (tableName != "cloud_pixel") {
    return eNotFound;
  }

  const utl::TimeStamp detTime = Detector::GetInstance().GetTime();

  ostringstream query;
  if (StringEquivalent(fDatabaseSoftwareVersion, "default")) {

    query << "SELECT " << columnName
          << " FROM cloud_pixel WHERE"
             " moon_cycle = "
          << utl::MoonCycle(detTime).GetFullMoonCycle()
          << " AND pixel_id = " << index.find("pixel_id")->second
          << " AND start_time <= " << detTime.GetGPSSecond()
          << " AND end_time > " << detTime.GetGPSSecond()
          << " ORDER BY last_modified DESC, start_time DESC LIMIT 1";

  } else {

    query << "SELECT " << columnName
          << " FROM cloud_pixel, software WHERE"
             " moon_cycle = "
          << utl::MoonCycle(detTime).GetFullMoonCycle()
          << " AND pixel_id = " << index.find("pixel_id")->second
          << " AND start_time <= " << detTime.GetGPSSecond()
          << " AND end_time > " << detTime.GetGPSSecond()
          << " and cloud_pixel.software_id = software.software_id"
             " ORDER BY cloud_pixel.last_modified DESC, cloud_pixel.start_time DESC LIMIT 1";
  }

  return Query(query, QueryInfoMessage(tableName, columnName));
}