/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

/* vim: set ts=8 sts=2 et sw=2 tw=80: */

/* This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef nsWindowSizes_h

#define nsWindowSizes_h

#include "mozilla/Assertions.h"

#include "mozilla/PodOperations.h"

#include "mozilla/SizeOfState.h"

class nsTabSizes {

 public:

  enum Kind {

    DOM,    // DOM stuff.

    Style,  // Style stuff.

    Other   // Everything else.

  };

  nsTabSizes() : mDom(0), mStyle(0), mOther(0) {}

  void add(Kind kind, size_t n) {

    switch (kind) {

      case DOM:

        mDom += n;

        break;

      case Style:

        mStyle += n;

        break;

      case Other:

        mOther += n;

        break;

      default:

        MOZ_CRASH("bad nsTabSizes kind");

    }

  }

  size_t mDom;

  size_t mStyle;

  size_t mOther;

};

#define ZERO_SIZE(kind, mSize) mSize(0),

#define ADD_TO_TAB_SIZES(kind, mSize) aSizes->add(nsTabSizes::kind, mSize);

#define ADD_TO_TOTAL_SIZE(kind, mSize) total += mSize;

#define DECL_SIZE(kind, mSize) size_t mSize;

#define NS_STYLE_SIZES_FIELD(name_) mStyle##name_

struct nsStyleSizes {

  nsStyleSizes()

      :

#define STYLE_STRUCT(name_, cb_) NS_STYLE_SIZES_FIELD(name_)(0),

#define STYLE_STRUCT_LIST_IGNORE_VARIABLES

#include "nsStyleStructList.h"

#undef STYLE_STRUCT

#undef STYLE_STRUCT_LIST_IGNORE_VARIABLES

        dummy() {

  }

  void addToTabSizes(nsTabSizes* aSizes) const {

#define STYLE_STRUCT(name_, cb_) \

  aSizes->add(nsTabSizes::Style, NS_STYLE_SIZES_FIELD(name_));

#define STYLE_STRUCT_LIST_IGNORE_VARIABLES

#include "nsStyleStructList.h"

#undef STYLE_STRUCT

#undef STYLE_STRUCT_LIST_IGNORE_VARIABLES

  }

  size_t getTotalSize() const {

    size_t total = 0;

#define STYLE_STRUCT(name_, cb_) total += NS_STYLE_SIZES_FIELD(name_);

#define STYLE_STRUCT_LIST_IGNORE_VARIABLES

#include "nsStyleStructList.h"

#undef STYLE_STRUCT

#undef STYLE_STRUCT_LIST_IGNORE_VARIABLES

    return total;

  }

#define STYLE_STRUCT(name_, cb_) size_t NS_STYLE_SIZES_FIELD(name_);

#define STYLE_STRUCT_LIST_IGNORE_VARIABLES

#include "nsStyleStructList.h"

#undef STYLE_STRUCT

#undef STYLE_STRUCT_LIST_IGNORE_VARIABLES

  // Present just to absorb the trailing comma in the constructor.

  int dummy;

};

#define NS_ARENA_SIZES_FIELD(classname) mArena##classname

struct nsArenaSizes {

#define FOR_EACH_SIZE(MACRO) \

  MACRO(Other, mLineBoxes)   \

  MACRO(Style, mRuleNodes)   \

  MACRO(Style, mStyleContexts)

  nsArenaSizes()

      : FOR_EACH_SIZE(ZERO_SIZE)

#define FRAME_ID(classname, ...) NS_ARENA_SIZES_FIELD(classname)(0),

#define ABSTRACT_FRAME_ID(...)

#include "nsFrameIdList.h"

#undef FRAME_ID

#undef ABSTRACT_FRAME_ID

            mGeckoStyleSizes() {

  }

  void addToTabSizes(nsTabSizes* aSizes) const {

    FOR_EACH_SIZE(ADD_TO_TAB_SIZES)

#define FRAME_ID(classname, ...) \

  aSizes->add(nsTabSizes::Other, NS_ARENA_SIZES_FIELD(classname));

#define ABSTRACT_FRAME_ID(...)

#include "nsFrameIdList.h"

#undef FRAME_ID

#undef ABSTRACT_FRAME_ID

    mGeckoStyleSizes.addToTabSizes(aSizes);

  }

  size_t getTotalSize() const {

    size_t total = 0;

    FOR_EACH_SIZE(ADD_TO_TOTAL_SIZE)

#define FRAME_ID(classname, ...) total += NS_ARENA_SIZES_FIELD(classname);

#define ABSTRACT_FRAME_ID(...)

#include "nsFrameIdList.h"

#undef FRAME_ID

#undef ABSTRACT_FRAME_ID

    total += mGeckoStyleSizes.getTotalSize();

    return total;

  }

  FOR_EACH_SIZE(DECL_SIZE)

#define FRAME_ID(classname, ...) size_t NS_ARENA_SIZES_FIELD(classname);

#define ABSTRACT_FRAME_ID(...)

#include "nsFrameIdList.h"

#undef FRAME_ID

#undef ABSTRACT_FRAME_ID

  // This is Gecko-only because in Stylo these style structs are stored outside

  // the nsPresArena, and so measured elsewhere.

  nsStyleSizes mGeckoStyleSizes;

#undef FOR_EACH_SIZE

};

class nsWindowSizes {

#define FOR_EACH_SIZE(MACRO)                                      \

  MACRO(DOM, mDOMElementNodesSize)                                \

  MACRO(DOM, mDOMTextNodesSize)                                   \

  MACRO(DOM, mDOMCDATANodesSize)                                  \

  MACRO(DOM, mDOMCommentNodesSize)                                \

  MACRO(DOM, mDOMEventTargetsSize)                                \

  MACRO(DOM, mDOMPerformanceUserEntries)                          \

  MACRO(DOM, mDOMPerformanceResourceEntries)                      \

  MACRO(DOM, mDOMOtherSize)                                       \

  MACRO(Style, mLayoutStyleSheetsSize)                            \

  MACRO(Other, mLayoutPresShellSize)                              \

  MACRO(Style, mLayoutGeckoStyleSets)                             \

  MACRO(Style, mLayoutServoStyleSetsStylistRuleTree)              \

  MACRO(Style, mLayoutServoStyleSetsStylistElementAndPseudosMaps) \

  MACRO(Style, mLayoutServoStyleSetsStylistInvalidationMap)       \

  MACRO(Style, mLayoutServoStyleSetsStylistRevalidationSelectors) \

  MACRO(Style, mLayoutServoStyleSetsStylistOther)                 \

  MACRO(Style, mLayoutServoStyleSetsOther)                        \

  MACRO(Style, mLayoutServoElementDataObjects)                    \

  MACRO(Other, mLayoutTextRunsSize)                               \

  MACRO(Other, mLayoutPresContextSize)                            \

  MACRO(Other, mLayoutFramePropertiesSize)                        \

  MACRO(Style, mLayoutComputedValuesDom)                          \

  MACRO(Style, mLayoutComputedValuesNonDom)                       \

  MACRO(Style, mLayoutComputedValuesVisited)                      \

  MACRO(Style, mLayoutComputedValuesStale)                        \

  MACRO(Other, mPropertyTablesSize)                               \

  MACRO(Other, mBindingsSize)

 public:

  explicit nsWindowSizes(mozilla::SizeOfState& aState)

      : FOR_EACH_SIZE(ZERO_SIZE) mDOMEventTargetsCount(0),

        mDOMEventListenersCount(0),

        mArenaSizes(),

        mServoStyleSizes(),

        mState(aState) {}

  void addToTabSizes(nsTabSizes* aSizes) const {

    FOR_EACH_SIZE(ADD_TO_TAB_SIZES)

    mArenaSizes.addToTabSizes(aSizes);

    mServoStyleSizes.addToTabSizes(aSizes);

  }

  size_t getTotalSize() const {

    size_t total = 0;

    FOR_EACH_SIZE(ADD_TO_TOTAL_SIZE)

    total += mArenaSizes.getTotalSize();

    total += mServoStyleSizes.getTotalSize();

    return total;

  }

  FOR_EACH_SIZE(DECL_SIZE);

  uint32_t mDOMEventTargetsCount;

  uint32_t mDOMEventListenersCount;

  nsArenaSizes mArenaSizes;

  // This is Stylo-only because in Gecko these style structs are stored in the

  // nsPresArena, and so are measured as part of that.

  nsStyleSizes mServoStyleSizes;

  mozilla::SizeOfState& mState;

#undef FOR_EACH_SIZE

};

#undef ZERO_SIZE

#undef ADD_TO_TAB_SIZES

#undef ADD_TO_TOTAL_SIZE

#undef DECL_SIZE

#endif  // nsWindowSizes_h