canonical_emission.hpp

Go to the documentation of this file.

/*
 * Copyright © 2025-2026 Akreon LLC
 * SPDX-License-Identifier: Apache-2.0
 */
 
#pragma once
 
#include "akreon/linalg/affine/affine_map.hpp"
#include "akreon/convex/canonical_trace.hpp"
#include "akreon/convex/cone_traits.hpp"
 
namespace conic {
 
using EmptyTrace = convex::EmptyCanonicalTrace;
 
template <affine::AffineMapLike CoordMap, typename ConeTuple,
          affine::AffineMapLike EqMap, typename Trace = EmptyTrace>
struct LocalEmission {
  using Tuple = linalg::remove_cvref_t<ConeTuple>;
 
  static constexpr std::size_t local_coord_rows =
      linalg::mat_rows_v<affine::affine_mat_t<CoordMap>>;
  static constexpr std::size_t local_coord_cols =
      linalg::mat_cols_v<affine::affine_mat_t<CoordMap>>;
  static constexpr std::size_t cone_count = cone_tuple_num_cones_v<Tuple>;
  static constexpr std::size_t local_cone_dim =
      cone_tuple_local_cone_dim_v<Tuple>;
  static constexpr std::size_t local_eq_rows =
      linalg::mat_rows_v<affine::affine_mat_t<EqMap>>;
  static constexpr std::size_t local_eq_cols =
      linalg::mat_cols_v<affine::affine_mat_t<EqMap>>;
 
  constexpr explicit LocalEmission(const CoordMap& coord_map_in,
                                   const Tuple& cones_in,
                                   const EqMap& eq_map_in)
      : local_coord_map(coord_map_in), local_cones(cones_in),
        local_eq_map(eq_map_in) {}
 
  CoordMap local_coord_map; 
  ConeTuple local_cones;    
  EqMap local_eq_map;       
  Trace trace;              
};
 
template <affine::AffineMapLike EqMap, typename Trace = EmptyTrace>
struct LocalEqualityEmission {
  static constexpr std::size_t local_eq_rows =
      linalg::mat_rows_v<affine::affine_mat_t<EqMap>>;
  static constexpr std::size_t local_eq_cols =
      linalg::mat_cols_v<affine::affine_mat_t<EqMap>>;
 
  constexpr explicit LocalEqualityEmission(const EqMap& eq_map_in)
      : local_eq_map(eq_map_in) {}
 
  EqMap local_eq_map; 
  Trace trace;        
};
 
template <typename CoordMap, typename ConeTuple, typename EqMap>
LocalEmission(const CoordMap&, const ConeTuple&, const EqMap&)
    -> LocalEmission<CoordMap, ConeTuple, EqMap>;
 
template <typename CoordMap, typename ConeTuple, typename EqMap, typename Trace>
LocalEmission(const CoordMap&, const ConeTuple&, const EqMap&, const Trace&)
    -> LocalEmission<CoordMap, ConeTuple, EqMap, Trace>;
 
template <typename EqMap>
LocalEqualityEmission(const EqMap&) -> LocalEqualityEmission<EqMap>;
 
template <typename EqMap, typename Trace>
LocalEqualityEmission(const EqMap&, const Trace&)
    -> LocalEqualityEmission<EqMap, Trace>;
 
template <typename Emission>
struct emission_traits;
 
template <affine::AffineMapLike CoordMap, typename ConeTuple,
          affine::AffineMapLike EqMap, typename Trace>
struct emission_traits<LocalEmission<CoordMap, ConeTuple, EqMap, Trace>> {
  using LocalCoordinateMap = CoordMap;
  using Cones = ConeTuple;
  using LocalEqualityMap = EqMap;
  using TraceT = Trace;
 
  static constexpr std::size_t LocalCoordRows =
      LocalEmission<CoordMap, ConeTuple, EqMap, Trace>::local_coord_rows;
  static constexpr std::size_t LocalCoordCols =
      LocalEmission<CoordMap, ConeTuple, EqMap, Trace>::local_coord_cols;
  static constexpr std::size_t NumCones =
      LocalEmission<CoordMap, ConeTuple, EqMap, Trace>::cone_count;
  static constexpr std::size_t LocalConeDim =
      LocalEmission<CoordMap, ConeTuple, EqMap, Trace>::local_cone_dim;
  static constexpr std::size_t LocalEqRows =
      LocalEmission<CoordMap, ConeTuple, EqMap, Trace>::local_eq_rows;
  static constexpr std::size_t LocalEqCols =
      LocalEmission<CoordMap, ConeTuple, EqMap, Trace>::local_eq_cols;
};
 
template <affine::AffineMapLike EqMap, typename Trace>
struct emission_traits<LocalEqualityEmission<EqMap, Trace>> {
  using LocalEqualityMap = EqMap;
  using TraceT = Trace;
 
  static constexpr std::size_t LocalEqRows =
      LocalEqualityEmission<EqMap, Trace>::local_eq_rows;
  static constexpr std::size_t LocalEqCols =
      LocalEqualityEmission<EqMap, Trace>::local_eq_cols;
};
 
template <typename Emission>
concept LocalEmissionLike = requires(const Emission& emission) {
  { emission.local_coord_map };
  { emission.local_cones };
  { emission.local_eq_map };
  { emission.trace };
};
 
template <typename Emission>
concept LocalEqualityEmissionLike =
    !LocalEmissionLike<Emission> && requires(const Emission& emission) {
      { emission.local_eq_map };
      { emission.trace };
    };
 
template <typename Emission>
using emission_local_coord_map_t = typename emission_traits<
    linalg::remove_cvref_t<Emission>>::LocalCoordinateMap;
 
template <typename Emission>
using emission_cones_t =
    typename emission_traits<linalg::remove_cvref_t<Emission>>::Cones;
 
template <typename Emission>
using emission_local_eq_map_t = typename emission_traits<
    linalg::remove_cvref_t<Emission>>::LocalEqualityMap;
 
template <typename Emission>
using emission_trace_t =
    typename emission_traits<linalg::remove_cvref_t<Emission>>::TraceT;
 
template <typename Emission>
inline constexpr std::size_t emission_local_coord_rows_v =
    emission_traits<linalg::remove_cvref_t<Emission>>::LocalCoordRows;
 
template <typename Emission>
inline constexpr std::size_t emission_local_coord_cols_v =
    emission_traits<linalg::remove_cvref_t<Emission>>::LocalCoordCols;
 
template <typename Emission>
inline constexpr std::size_t emission_num_cones_v =
    emission_traits<linalg::remove_cvref_t<Emission>>::NumCones;
 
template <typename Emission>
inline constexpr std::size_t emission_local_cone_dim_v =
    emission_traits<linalg::remove_cvref_t<Emission>>::LocalConeDim;
 
template <typename Emission>
inline constexpr std::size_t emission_local_eq_rows_v =
    emission_traits<linalg::remove_cvref_t<Emission>>::LocalEqRows;
 
template <typename Emission>
inline constexpr std::size_t emission_local_eq_cols_v =
    emission_traits<linalg::remove_cvref_t<Emission>>::LocalEqCols;
 
template <typename Emission>
using emission_coord_map_t = emission_local_coord_map_t<Emission>;
 
template <typename Emission>
using emission_eq_map_t = emission_local_eq_map_t<Emission>;
 
template <typename Emission>
inline constexpr std::size_t emission_coord_rows_v =
    emission_local_coord_rows_v<Emission>;
 
template <typename Emission>
inline constexpr std::size_t emission_coord_cols_v =
    emission_local_coord_cols_v<Emission>;
 
template <typename Emission>
inline constexpr std::size_t emission_cone_dim_v =
    emission_local_cone_dim_v<Emission>;
 
template <typename Emission>
inline constexpr std::size_t emission_eq_rows_v =
    emission_local_eq_rows_v<Emission>;
 
template <typename Emission>
inline constexpr std::size_t emission_eq_cols_v =
    emission_local_eq_cols_v<Emission>;
 
} // namespace conic