aboutsummaryrefslogtreecommitdiff
path: root/ink_stroke_modeler/internal/internal_types.h
blob: 1c6b67106b1993a69cb38f952e5c2d04873e5dee (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
/*
 * Copyright 2022 Google LLC
 *
 * 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 INK_STROKE_MODELER_INTERNAL_INTERNAL_TYPES_H_
#define INK_STROKE_MODELER_INTERNAL_INTERNAL_TYPES_H_

#include <ostream>

#include "ink_stroke_modeler/types.h"

namespace ink {
namespace stroke_model {

// This struct contains the position and velocity of the modeled pen tip at
// the indicated time.
struct TipState {
  Vec2 position{0};
  Vec2 velocity{0};
  Time time{0};
};

std::ostream &operator<<(std::ostream &s, const TipState &tip_state);

// This struct contains information about the state of the stylus. See the
// corresponding fields on the Input struct for more info.
struct StylusState {
  float pressure = -1;
  float tilt = -1;
  float orientation = -1;
};

bool operator==(const StylusState &lhs, const StylusState &rhs);
std::ostream &operator<<(std::ostream &s, const StylusState &stylus_state);

////////////////////////////////////////////////////////////////////////////////
// Inline function definitions
////////////////////////////////////////////////////////////////////////////////

inline bool operator==(const StylusState &lhs, const StylusState &rhs) {
  return lhs.pressure == rhs.pressure && lhs.tilt == rhs.tilt &&
         lhs.orientation == rhs.orientation;
}

inline std::ostream &operator<<(std::ostream &s, const TipState &tip_state) {
  return s << "TipState: pos: " << tip_state.position
           << ", velocity: " << tip_state.velocity
           << ", time: " << tip_state.time << ">";
}

inline std::ostream &operator<<(std::ostream &s,
                                const StylusState &stylus_state) {
  return s << "<Result:  pressure: " << stylus_state.pressure
           << ", tilt: " << stylus_state.tilt
           << ", orientation: " << stylus_state.orientation << ">";
}

}  // namespace stroke_model
}  // namespace ink

#endif  // INK_STROKE_MODELER_INTERNAL_INTERNAL_TYPES_H_