/*
 * Copyright (C) 2018 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef INCLUDE_PERFETTO_EXT_TRACED_SYS_STATS_COUNTERS_H_
#define INCLUDE_PERFETTO_EXT_TRACED_SYS_STATS_COUNTERS_H_

#include "perfetto/ext/base/utils.h"
#include "protos/perfetto/common/sys_stats_counters.pbzero.h"

#include <vector>

namespace perfetto {

struct KeyAndId {
  const char* str;
  int id;
};

constexpr KeyAndId kMeminfoKeys[] = {
    {"MemUnspecified", protos::pbzero::MeminfoCounters::MEMINFO_UNSPECIFIED},
    {"MemTotal", protos::pbzero::MeminfoCounters::MEMINFO_MEM_TOTAL},
    {"MemFree", protos::pbzero::MeminfoCounters::MEMINFO_MEM_FREE},
    {"MemAvailable", protos::pbzero::MeminfoCounters::MEMINFO_MEM_AVAILABLE},
    {"Buffers", protos::pbzero::MeminfoCounters::MEMINFO_BUFFERS},
    {"Cached", protos::pbzero::MeminfoCounters::MEMINFO_CACHED},
    {"SwapCached", protos::pbzero::MeminfoCounters::MEMINFO_SWAP_CACHED},
    {"Active", protos::pbzero::MeminfoCounters::MEMINFO_ACTIVE},
    {"Inactive", protos::pbzero::MeminfoCounters::MEMINFO_INACTIVE},
    {"Active(anon)", protos::pbzero::MeminfoCounters::MEMINFO_ACTIVE_ANON},
    {"Inactive(anon)", protos::pbzero::MeminfoCounters::MEMINFO_INACTIVE_ANON},
    {"Active(file)", protos::pbzero::MeminfoCounters::MEMINFO_ACTIVE_FILE},
    {"Inactive(file)", protos::pbzero::MeminfoCounters::MEMINFO_INACTIVE_FILE},
    {"Unevictable", protos::pbzero::MeminfoCounters::MEMINFO_UNEVICTABLE},
    {"Mlocked", protos::pbzero::MeminfoCounters::MEMINFO_MLOCKED},
    {"SwapTotal", protos::pbzero::MeminfoCounters::MEMINFO_SWAP_TOTAL},
    {"SwapFree", protos::pbzero::MeminfoCounters::MEMINFO_SWAP_FREE},
    {"Dirty", protos::pbzero::MeminfoCounters::MEMINFO_DIRTY},
    {"Writeback", protos::pbzero::MeminfoCounters::MEMINFO_WRITEBACK},
    {"AnonPages", protos::pbzero::MeminfoCounters::MEMINFO_ANON_PAGES},
    {"Mapped", protos::pbzero::MeminfoCounters::MEMINFO_MAPPED},
    {"Shmem", protos::pbzero::MeminfoCounters::MEMINFO_SHMEM},
    {"Slab", protos::pbzero::MeminfoCounters::MEMINFO_SLAB},
    {"SReclaimable", protos::pbzero::MeminfoCounters::MEMINFO_SLAB_RECLAIMABLE},
    {"SUnreclaim", protos::pbzero::MeminfoCounters::MEMINFO_SLAB_UNRECLAIMABLE},
    {"KernelStack", protos::pbzero::MeminfoCounters::MEMINFO_KERNEL_STACK},
    {"PageTables", protos::pbzero::MeminfoCounters::MEMINFO_PAGE_TABLES},
    {"CommitLimit", protos::pbzero::MeminfoCounters::MEMINFO_COMMIT_LIMIT},
    {"Committed_AS", protos::pbzero::MeminfoCounters::MEMINFO_COMMITED_AS},
    {"VmallocTotal", protos::pbzero::MeminfoCounters::MEMINFO_VMALLOC_TOTAL},
    {"VmallocUsed", protos::pbzero::MeminfoCounters::MEMINFO_VMALLOC_USED},
    {"VmallocChunk", protos::pbzero::MeminfoCounters::MEMINFO_VMALLOC_CHUNK},
    {"CmaTotal", protos::pbzero::MeminfoCounters::MEMINFO_CMA_TOTAL},
    {"CmaFree", protos::pbzero::MeminfoCounters::MEMINFO_CMA_FREE},
};

const KeyAndId kVmstatKeys[] = {
    {"VmstatUnspecified", protos::pbzero::VmstatCounters::VMSTAT_UNSPECIFIED},
    {"nr_free_pages", protos::pbzero::VmstatCounters::VMSTAT_NR_FREE_PAGES},
    {"nr_alloc_batch", protos::pbzero::VmstatCounters::VMSTAT_NR_ALLOC_BATCH},
    {"nr_inactive_anon",
     protos::pbzero::VmstatCounters::VMSTAT_NR_INACTIVE_ANON},
    {"nr_active_anon", protos::pbzero::VmstatCounters::VMSTAT_NR_ACTIVE_ANON},
    {"nr_inactive_file",
     protos::pbzero::VmstatCounters::VMSTAT_NR_INACTIVE_FILE},
    {"nr_active_file", protos::pbzero::VmstatCounters::VMSTAT_NR_ACTIVE_FILE},
    {"nr_unevictable", protos::pbzero::VmstatCounters::VMSTAT_NR_UNEVICTABLE},
    {"nr_mlock", protos::pbzero::VmstatCounters::VMSTAT_NR_MLOCK},
    {"nr_anon_pages", protos::pbzero::VmstatCounters::VMSTAT_NR_ANON_PAGES},
    {"nr_mapped", protos::pbzero::VmstatCounters::VMSTAT_NR_MAPPED},
    {"nr_file_pages", protos::pbzero::VmstatCounters::VMSTAT_NR_FILE_PAGES},
    {"nr_dirty", protos::pbzero::VmstatCounters::VMSTAT_NR_DIRTY},
    {"nr_writeback", protos::pbzero::VmstatCounters::VMSTAT_NR_WRITEBACK},
    {"nr_slab_reclaimable",
     protos::pbzero::VmstatCounters::VMSTAT_NR_SLAB_RECLAIMABLE},
    {"nr_slab_unreclaimable",
     protos::pbzero::VmstatCounters::VMSTAT_NR_SLAB_UNRECLAIMABLE},
    {"nr_page_table_pages",
     protos::pbzero::VmstatCounters::VMSTAT_NR_PAGE_TABLE_PAGES},
    {"nr_kernel_stack", protos::pbzero::VmstatCounters::VMSTAT_NR_KERNEL_STACK},
    {"nr_overhead", protos::pbzero::VmstatCounters::VMSTAT_NR_OVERHEAD},
    {"nr_unstable", protos::pbzero::VmstatCounters::VMSTAT_NR_UNSTABLE},
    {"nr_bounce", protos::pbzero::VmstatCounters::VMSTAT_NR_BOUNCE},
    {"nr_vmscan_write", protos::pbzero::VmstatCounters::VMSTAT_NR_VMSCAN_WRITE},
    {"nr_vmscan_immediate_reclaim",
     protos::pbzero::VmstatCounters::VMSTAT_NR_VMSCAN_IMMEDIATE_RECLAIM},
    {"nr_writeback_temp",
     protos::pbzero::VmstatCounters::VMSTAT_NR_WRITEBACK_TEMP},
    {"nr_isolated_anon",
     protos::pbzero::VmstatCounters::VMSTAT_NR_ISOLATED_ANON},
    {"nr_isolated_file",
     protos::pbzero::VmstatCounters::VMSTAT_NR_ISOLATED_FILE},
    {"nr_shmem", protos::pbzero::VmstatCounters::VMSTAT_NR_SHMEM},
    {"nr_dirtied", protos::pbzero::VmstatCounters::VMSTAT_NR_DIRTIED},
    {"nr_written", protos::pbzero::VmstatCounters::VMSTAT_NR_WRITTEN},
    {"nr_pages_scanned",
     protos::pbzero::VmstatCounters::VMSTAT_NR_PAGES_SCANNED},
    {"workingset_refault",
     protos::pbzero::VmstatCounters::VMSTAT_WORKINGSET_REFAULT},
    {"workingset_activate",
     protos::pbzero::VmstatCounters::VMSTAT_WORKINGSET_ACTIVATE},
    {"workingset_nodereclaim",
     protos::pbzero::VmstatCounters::VMSTAT_WORKINGSET_NODERECLAIM},
    {"nr_anon_transparent_hugepages",
     protos::pbzero::VmstatCounters::VMSTAT_NR_ANON_TRANSPARENT_HUGEPAGES},
    {"nr_free_cma", protos::pbzero::VmstatCounters::VMSTAT_NR_FREE_CMA},
    {"nr_swapcache", protos::pbzero::VmstatCounters::VMSTAT_NR_SWAPCACHE},
    {"nr_dirty_threshold",
     protos::pbzero::VmstatCounters::VMSTAT_NR_DIRTY_THRESHOLD},
    {"nr_dirty_background_threshold",
     protos::pbzero::VmstatCounters::VMSTAT_NR_DIRTY_BACKGROUND_THRESHOLD},
    {"pgpgin", protos::pbzero::VmstatCounters::VMSTAT_PGPGIN},
    {"pgpgout", protos::pbzero::VmstatCounters::VMSTAT_PGPGOUT},
    {"pgpgoutclean", protos::pbzero::VmstatCounters::VMSTAT_PGPGOUTCLEAN},
    {"pswpin", protos::pbzero::VmstatCounters::VMSTAT_PSWPIN},
    {"pswpout", protos::pbzero::VmstatCounters::VMSTAT_PSWPOUT},
    {"pgalloc_dma", protos::pbzero::VmstatCounters::VMSTAT_PGALLOC_DMA},
    {"pgalloc_normal", protos::pbzero::VmstatCounters::VMSTAT_PGALLOC_NORMAL},
    {"pgalloc_movable", protos::pbzero::VmstatCounters::VMSTAT_PGALLOC_MOVABLE},
    {"pgfree", protos::pbzero::VmstatCounters::VMSTAT_PGFREE},
    {"pgactivate", protos::pbzero::VmstatCounters::VMSTAT_PGACTIVATE},
    {"pgdeactivate", protos::pbzero::VmstatCounters::VMSTAT_PGDEACTIVATE},
    {"pgfault", protos::pbzero::VmstatCounters::VMSTAT_PGFAULT},
    {"pgmajfault", protos::pbzero::VmstatCounters::VMSTAT_PGMAJFAULT},
    {"pgrefill_dma", protos::pbzero::VmstatCounters::VMSTAT_PGREFILL_DMA},
    {"pgrefill_normal", protos::pbzero::VmstatCounters::VMSTAT_PGREFILL_NORMAL},
    {"pgrefill_movable",
     protos::pbzero::VmstatCounters::VMSTAT_PGREFILL_MOVABLE},
    {"pgsteal_kswapd_dma",
     protos::pbzero::VmstatCounters::VMSTAT_PGSTEAL_KSWAPD_DMA},
    {"pgsteal_kswapd_normal",
     protos::pbzero::VmstatCounters::VMSTAT_PGSTEAL_KSWAPD_NORMAL},
    {"pgsteal_kswapd_movable",
     protos::pbzero::VmstatCounters::VMSTAT_PGSTEAL_KSWAPD_MOVABLE},
    {"pgsteal_direct_dma",
     protos::pbzero::VmstatCounters::VMSTAT_PGSTEAL_DIRECT_DMA},
    {"pgsteal_direct_normal",
     protos::pbzero::VmstatCounters::VMSTAT_PGSTEAL_DIRECT_NORMAL},
    {"pgsteal_direct_movable",
     protos::pbzero::VmstatCounters::VMSTAT_PGSTEAL_DIRECT_MOVABLE},
    {"pgscan_kswapd_dma",
     protos::pbzero::VmstatCounters::VMSTAT_PGSCAN_KSWAPD_DMA},
    {"pgscan_kswapd_normal",
     protos::pbzero::VmstatCounters::VMSTAT_PGSCAN_KSWAPD_NORMAL},
    {"pgscan_kswapd_movable",
     protos::pbzero::VmstatCounters::VMSTAT_PGSCAN_KSWAPD_MOVABLE},
    {"pgscan_direct_dma",
     protos::pbzero::VmstatCounters::VMSTAT_PGSCAN_DIRECT_DMA},
    {"pgscan_direct_normal",
     protos::pbzero::VmstatCounters::VMSTAT_PGSCAN_DIRECT_NORMAL},
    {"pgscan_direct_movable",
     protos::pbzero::VmstatCounters::VMSTAT_PGSCAN_DIRECT_MOVABLE},
    {"pgscan_direct_throttle",
     protos::pbzero::VmstatCounters::VMSTAT_PGSCAN_DIRECT_THROTTLE},
    {"pginodesteal", protos::pbzero::VmstatCounters::VMSTAT_PGINODESTEAL},
    {"slabs_scanned", protos::pbzero::VmstatCounters::VMSTAT_SLABS_SCANNED},
    {"kswapd_inodesteal",
     protos::pbzero::VmstatCounters::VMSTAT_KSWAPD_INODESTEAL},
    {"kswapd_low_wmark_hit_quickly",
     protos::pbzero::VmstatCounters::VMSTAT_KSWAPD_LOW_WMARK_HIT_QUICKLY},
    {"kswapd_high_wmark_hit_quickly",
     protos::pbzero::VmstatCounters::VMSTAT_KSWAPD_HIGH_WMARK_HIT_QUICKLY},
    {"pageoutrun", protos::pbzero::VmstatCounters::VMSTAT_PAGEOUTRUN},
    {"allocstall", protos::pbzero::VmstatCounters::VMSTAT_ALLOCSTALL},
    {"pgrotated", protos::pbzero::VmstatCounters::VMSTAT_PGROTATED},
    {"drop_pagecache", protos::pbzero::VmstatCounters::VMSTAT_DROP_PAGECACHE},
    {"drop_slab", protos::pbzero::VmstatCounters::VMSTAT_DROP_SLAB},
    {"pgmigrate_success",
     protos::pbzero::VmstatCounters::VMSTAT_PGMIGRATE_SUCCESS},
    {"pgmigrate_fail", protos::pbzero::VmstatCounters::VMSTAT_PGMIGRATE_FAIL},
    {"compact_migrate_scanned",
     protos::pbzero::VmstatCounters::VMSTAT_COMPACT_MIGRATE_SCANNED},
    {"compact_free_scanned",
     protos::pbzero::VmstatCounters::VMSTAT_COMPACT_FREE_SCANNED},
    {"compact_isolated",
     protos::pbzero::VmstatCounters::VMSTAT_COMPACT_ISOLATED},
    {"compact_stall", protos::pbzero::VmstatCounters::VMSTAT_COMPACT_STALL},
    {"compact_fail", protos::pbzero::VmstatCounters::VMSTAT_COMPACT_FAIL},
    {"compact_success", protos::pbzero::VmstatCounters::VMSTAT_COMPACT_SUCCESS},
    {"compact_daemon_wake",
     protos::pbzero::VmstatCounters::VMSTAT_COMPACT_DAEMON_WAKE},
    {"unevictable_pgs_culled",
     protos::pbzero::VmstatCounters::VMSTAT_UNEVICTABLE_PGS_CULLED},
    {"unevictable_pgs_scanned",
     protos::pbzero::VmstatCounters::VMSTAT_UNEVICTABLE_PGS_SCANNED},
    {"unevictable_pgs_rescued",
     protos::pbzero::VmstatCounters::VMSTAT_UNEVICTABLE_PGS_RESCUED},
    {"unevictable_pgs_mlocked",
     protos::pbzero::VmstatCounters::VMSTAT_UNEVICTABLE_PGS_MLOCKED},
    {"unevictable_pgs_munlocked",
     protos::pbzero::VmstatCounters::VMSTAT_UNEVICTABLE_PGS_MUNLOCKED},
    {"unevictable_pgs_cleared",
     protos::pbzero::VmstatCounters::VMSTAT_UNEVICTABLE_PGS_CLEARED},
    {"unevictable_pgs_stranded",
     protos::pbzero::VmstatCounters::VMSTAT_UNEVICTABLE_PGS_STRANDED},
    {"nr_zspages", protos::pbzero::VmstatCounters::VMSTAT_NR_ZSPAGES},
    {"nr_ion_heap", protos::pbzero::VmstatCounters::VMSTAT_NR_ION_HEAP},
    {"nr_gpu_heap", protos::pbzero::VmstatCounters::VMSTAT_NR_GPU_HEAP},
    {"allocstall_dma", protos::pbzero::VmstatCounters::VMSTAT_ALLOCSTALL_DMA},
    {"allocstall_movable",
     protos::pbzero::VmstatCounters::VMSTAT_ALLOCSTALL_MOVABLE},
    {"allocstall_normal",
     protos::pbzero::VmstatCounters::VMSTAT_ALLOCSTALL_NORMAL},
    {"compact_daemon_free_scanned",
     protos::pbzero::VmstatCounters::VMSTAT_COMPACT_DAEMON_FREE_SCANNED},
    {"compact_daemon_migrate_scanned",
     protos::pbzero::VmstatCounters::VMSTAT_COMPACT_DAEMON_MIGRATE_SCANNED},
    {"nr_fastrpc", protos::pbzero::VmstatCounters::VMSTAT_NR_FASTRPC},
    {"nr_indirectly_reclaimable",
     protos::pbzero::VmstatCounters::VMSTAT_NR_INDIRECTLY_RECLAIMABLE},
    {"nr_ion_heap_pool",
     protos::pbzero::VmstatCounters::VMSTAT_NR_ION_HEAP_POOL},
    {"nr_kernel_misc_reclaimable",
     protos::pbzero::VmstatCounters::VMSTAT_NR_KERNEL_MISC_RECLAIMABLE},
    {"nr_shadow_call_stack_bytes",
     protos::pbzero::VmstatCounters::VMSTAT_NR_SHADOW_CALL_STACK_BYTES},
    {"nr_shmem_hugepages",
     protos::pbzero::VmstatCounters::VMSTAT_NR_SHMEM_HUGEPAGES},
    {"nr_shmem_pmdmapped",
     protos::pbzero::VmstatCounters::VMSTAT_NR_SHMEM_PMDMAPPED},
    {"nr_unreclaimable_pages",
     protos::pbzero::VmstatCounters::VMSTAT_NR_UNRECLAIMABLE_PAGES},
    {"nr_zone_active_anon",
     protos::pbzero::VmstatCounters::VMSTAT_NR_ZONE_ACTIVE_ANON},
    {"nr_zone_active_file",
     protos::pbzero::VmstatCounters::VMSTAT_NR_ZONE_ACTIVE_FILE},
    {"nr_zone_inactive_anon",
     protos::pbzero::VmstatCounters::VMSTAT_NR_ZONE_INACTIVE_ANON},
    {"nr_zone_inactive_file",
     protos::pbzero::VmstatCounters::VMSTAT_NR_ZONE_INACTIVE_FILE},
    {"nr_zone_unevictable",
     protos::pbzero::VmstatCounters::VMSTAT_NR_ZONE_UNEVICTABLE},
    {"nr_zone_write_pending",
     protos::pbzero::VmstatCounters::VMSTAT_NR_ZONE_WRITE_PENDING},
    {"oom_kill", protos::pbzero::VmstatCounters::VMSTAT_OOM_KILL},
    {"pglazyfree", protos::pbzero::VmstatCounters::VMSTAT_PGLAZYFREE},
    {"pglazyfreed", protos::pbzero::VmstatCounters::VMSTAT_PGLAZYFREED},
    {"pgrefill", protos::pbzero::VmstatCounters::VMSTAT_PGREFILL},
    {"pgscan_direct", protos::pbzero::VmstatCounters::VMSTAT_PGSCAN_DIRECT},
    {"pgscan_kswapd", protos::pbzero::VmstatCounters::VMSTAT_PGSCAN_KSWAPD},
    {"pgskip_dma", protos::pbzero::VmstatCounters::VMSTAT_PGSKIP_DMA},
    {"pgskip_movable", protos::pbzero::VmstatCounters::VMSTAT_PGSKIP_MOVABLE},
    {"pgskip_normal", protos::pbzero::VmstatCounters::VMSTAT_PGSKIP_NORMAL},
    {"pgsteal_direct", protos::pbzero::VmstatCounters::VMSTAT_PGSTEAL_DIRECT},
    {"pgsteal_kswapd", protos::pbzero::VmstatCounters::VMSTAT_PGSTEAL_KSWAPD},
    {"swap_ra_hit", protos::pbzero::VmstatCounters::VMSTAT_SWAP_RA_HIT},
    {"swap_ra", protos::pbzero::VmstatCounters::VMSTAT_SWAP_RA},
    {"workingset_restore",
     protos::pbzero::VmstatCounters::VMSTAT_WORKINGSET_RESTORE},
};

// Returns a lookup table of meminfo counter names addressable by counter id.
inline std::vector<const char*> BuildMeminfoCounterNames() {
  int max_id = 0;
  for (size_t i = 0; i < base::ArraySize(kMeminfoKeys); i++)
    max_id = std::max(max_id, kMeminfoKeys[i].id);
  std::vector<const char*> v;
  v.resize(static_cast<size_t>(max_id) + 1);
  for (size_t i = 0; i < base::ArraySize(kMeminfoKeys); i++)
    v[static_cast<size_t>(kMeminfoKeys[i].id)] = kMeminfoKeys[i].str;
  return v;
}

inline std::vector<const char*> BuildVmstatCounterNames() {
  int max_id = 0;
  for (size_t i = 0; i < base::ArraySize(kVmstatKeys); i++)
    max_id = std::max(max_id, kVmstatKeys[i].id);
  std::vector<const char*> v;
  v.resize(static_cast<size_t>(max_id) + 1);
  for (size_t i = 0; i < base::ArraySize(kVmstatKeys); i++)
    v[static_cast<size_t>(kVmstatKeys[i].id)] = kVmstatKeys[i].str;
  return v;
}

}  // namespace perfetto

#endif  // INCLUDE_PERFETTO_EXT_TRACED_SYS_STATS_COUNTERS_H_