cipherUtils

type Activator struct {
	NumOfActivations int
	// contains filtered or unexported fields
}

type Adder struct {
	// contains filtered or unexported fields
}

type BlockSplit struct {
	Rows, Cols, InnerRows, InnerCols, RowP, ColP int
}

type BlocksOperand interface {
	GetBlock(i, j int) interface{}
	//Returns RowP and ColP. Note that, if this is a weight, actually RowP will represent column partition
	//and viceversa since weights are block transposed
	GetPartitions() (int, int)
	GetInnerDims() (int, int)
	GetRealDims() (int, int)
	GetRotations(params ckks.Parameters) []int
	Level() int
	Scale() float64
}

type Bootstrapper struct {
	// contains filtered or unexported fields
}

type CkksBox struct {
	Params       ckks.Parameters
	BtpParams    *bootstrapping.Parameters
	Encoder      ckks.Encoder
	Evaluator    ckks.Evaluator
	Encryptor    ckks.Encryptor
	Decryptor    ckks.Decryptor
	BootStrapper *bootstrapping.Bootstrapper
	Sk           *rlwe.SecretKey
	// contains filtered or unexported fields
}

type Config struct {
	// contains filtered or unexported fields
}

type DebugStats struct {
	MinPrec  float64
	AvgPrec  float64
	MaxPrec  float64
	MaxValue float64
	L2Dist   float64
}

type DiagMat interface {
	GetDiags() map[int]ckks.Operand
	GetRotations(parameters ckks.Parameters) []int
	IsEncrypted() bool
}

type EncDiagMat struct {
	//store an encrypted weight matrix in diagonal form
	Diags        map[int]*ckks.Ciphertext //enc diagonals
	InnerRows    int                      //rows of sub-matrix
	InnerCols    int
	LeftR, LeftC int //rows cols of left matrix
	Encrypted    bool
}

type EncInput struct {
	Blocks             [][]*ckks.Ciphertext //all the sub-matrixes, encrypted as flatten(A.T)
	RowP, ColP         int                  //num of partitions
	InnerRows          int                  //rows of sub-matrix
	InnerCols          int
	RealRows, RealCols int
}

type EncWeightDiag struct {
	Blocks             [][]*EncDiagMat //blocks of the matrix, each is a sub-matrix in diag form
	RowP, ColP         int
	LeftR, LeftC       int //rows cols of left matrix
	InnerRows          int //rows of matrix
	InnerCols          int
	RealRows, RealCols int
}

type EvalFunc func(X *EncInput, i, j int, poly activationPoly, Box CkksBox)

type IBootstrapper interface {
	Bootstrap(input *EncInput, Box CkksBox)
}

type MulTask struct {
	// contains filtered or unexported fields
}

type Multiplier struct {
	// contains filtered or unexported fields
}

type PlainDiagMat struct {
	//store a plaintext weight matrix in diagonal form
	Diags                map[int]*ckks.Plaintext //diagonals
	InnerRows, InnerCols int
	LeftR, LeftC         int //rows cols of left matrix
	Encrypted            bool
}

type PlainInput struct {
	Blocks             [][]*ckks.Plaintext //all the sub-matrixes, encrypted as flatten(A.T)
	RowP, ColP         int                 //num of partitions
	InnerRows          int                 //rows of sub-matrix
	InnerCols          int
	RealRows, RealCols int
}

type PlainWeightDiag struct {
	Blocks             [][]*PlainDiagMat //blocks of the matrix, each is a sub-matrix in diag form
	RowP, ColP         int
	LeftR, LeftC       int //rows of left matrix
	InnerRows          int //rows of matrix
	InnerCols          int
	RealRows, RealCols int
}

type RotationsSet struct {
	Set map[int]struct{}
}

type Split struct {
	S []BlockSplit
}

type SplitsInfo struct {
	Features          int   `json:"features,omitempty"`
	InputRows         int   `json:"input_rows,omitempty"`
	InputCols         int   `json:"input_cols,omitempty"` //inner dims of input
	InputRowP         int   `json:"input_row_p,omitempty"`
	InputColP         int   `json:"input_col_p,omitempty"` //partition of input
	BatchSize         int   `json:"batch_size,omitempty"`
	NumWeights        int   `json:"num_weights,omitempty"`
	RowsOfWeights     []int `json:"rows_of_weights,omitempty"` //inner rows of weights
	ColsOfWeights     []int `json:"cols_of_weights,omitempty"`
	RowPOfWeights     []int `json:"row_p_of_weights,omitempty"` //row partition of weights
	ColPOfWeights     []int `json:"col_p_of_weights,omitempty"`
	RealRowsOfWeights []int `json:"real_rows_of_weights,omitempty"` //rows of weights before block partition
	RealColsOfWeights []int `json:"real_cols_of_weights,omitempty"`
}

type Splitter struct {
	// contains filtered or unexported fields
}