| OLD | NEW |
| (Empty) |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "webkit/fileapi/media/picasa/pmp_table_reader.h" | |
| 6 | |
| 7 #include <algorithm> | |
| 8 | |
| 9 #include "base/file_util.h" | |
| 10 #include "base/files/file_path.h" | |
| 11 #include "base/logging.h" | |
| 12 #include "base/utf_string_conversions.h" | |
| 13 #include "webkit/fileapi/media/picasa/pmp_column_reader.h" | |
| 14 #include "webkit/fileapi/media/picasa/pmp_constants.h" | |
| 15 | |
| 16 namespace picasaimport { | |
| 17 | |
| 18 namespace { | |
| 19 | |
| 20 COMPILE_ASSERT(sizeof(double) == 8, double_must_be_8_bytes_long); | |
| 21 | |
| 22 } // namespace | |
| 23 | |
| 24 PmpTableReader::PmpTableReader() : column_readers_(), max_row_count_(0) { } | |
| 25 | |
| 26 PmpTableReader::~PmpTableReader() { } | |
| 27 | |
| 28 bool PmpTableReader::Init(const std::string& table_name, | |
| 29 const base::FilePath& directory_path, | |
| 30 const std::vector<std::string>& columns) { | |
| 31 DCHECK(!columns.empty()); | |
| 32 | |
| 33 if (!column_readers_.empty()) | |
| 34 return false; | |
| 35 | |
| 36 if (!file_util::DirectoryExists(directory_path)) | |
| 37 return false; | |
| 38 | |
| 39 std::string table_prefix = table_name + "_"; | |
| 40 std::string indicator_file_name = table_prefix + "0"; | |
| 41 | |
| 42 base::FilePath indicator_file = directory_path.Append( | |
| 43 base::FilePath::FromUTF8Unsafe(indicator_file_name)); | |
| 44 | |
| 45 // Look for the indicator_file file, indicating table existence. | |
| 46 if (!file_util::PathExists(indicator_file) || | |
| 47 file_util::DirectoryExists(indicator_file)) { | |
| 48 return false; | |
| 49 } | |
| 50 | |
| 51 ScopedVector<PmpColumnReader> column_readers; | |
| 52 uint32 max_row_count = 0; | |
| 53 | |
| 54 for (std::vector<std::string>::const_iterator it = columns.begin(); | |
| 55 it != columns.end(); ++it) { | |
| 56 std::string filename = table_prefix + *it + "." + kPmpExtension; | |
| 57 | |
| 58 base::FilePath column_file_path = directory_path.Append( | |
| 59 base::FilePath::FromUTF8Unsafe(filename)); | |
| 60 | |
| 61 PmpColumnReader* column_reader = new PmpColumnReader(); | |
| 62 column_readers.push_back(column_reader); | |
| 63 | |
| 64 uint32 row_cnt; | |
| 65 | |
| 66 if (!column_reader->Init(column_file_path, &row_cnt)) | |
| 67 return false; | |
| 68 | |
| 69 max_row_count = std::max(max_row_count, row_cnt); | |
| 70 } | |
| 71 | |
| 72 column_readers_ = column_readers.Pass(); | |
| 73 max_row_count_ = max_row_count; | |
| 74 | |
| 75 return true; | |
| 76 } | |
| 77 | |
| 78 uint32 PmpTableReader::RowCount() const { | |
| 79 return max_row_count_; | |
| 80 } | |
| 81 | |
| 82 std::vector<const PmpColumnReader*> PmpTableReader::GetColumns() const { | |
| 83 std::vector<const PmpColumnReader*> readers( | |
| 84 column_readers_.begin(), column_readers_.end()); | |
| 85 return readers; | |
| 86 } | |
| 87 | |
| 88 } // namespace picasaimport | |
| OLD | NEW |