hcp.h

/* 
 * File:   hcp.h
 * Author: gadial
 *
 * Created on December 3, 2009, 11:50 AM
 */

#ifndef _HCP_H
#define	_HCP_H

#include <map>
#include <string>
#include <gmpxx.h>
#include <vector>
#include <iostream>

#include "zp_int.h"

using std::map;
using std::string;
using std::vector;
using std::ostream;

typedef vector<zp_int> NumberArray;

class ModularPolynomial{
public:
    ModularPolynomial():modulus(0),degree(-1){};
    ModularPolynomial(string, mpz_class);
    ModularPolynomial(const ModularPolynomial&);
    ModularPolynomial(const NumberArray&, mpz_class);
    static ModularPolynomial build_from_roots(const NumberArray& roots, mpz_class p);
    static ModularPolynomial build_hcp_from_discriminant(int D, mpz_class p);
    string to_string() const;
    ModularPolynomial& operator=(const ModularPolynomial&);
    ModularPolynomial& operator+=(const ModularPolynomial&);
    ModularPolynomial& operator-=(const ModularPolynomial&);
    ModularPolynomial& operator*=(const ModularPolynomial&);
    ModularPolynomial& operator/=(const ModularPolynomial&);
    ModularPolynomial modular_exponent(mpz_class exp, const ModularPolynomial& mod) const;
    zp_int operator()(zp_int a) const;

    ModularPolynomial operator%(const ModularPolynomial& lhs);
    bool operator==(const ModularPolynomial&) const;
    ModularPolynomial& normalize();
    bool is_zero() const;
    NumberArray find_roots();
    zp_int find_one_root();
    int get_degree(){return degree;}
    mpz_class get_modulus(){return modulus;}
    void full_print(ostream&);
private:
    mutable map<int,zp_int> coefficients; //mutable since otherwise we get const-correctness problem (seems like mpz_class is not very const-correct)
    mpz_class modulus;
    int degree;

    void divide(ModularPolynomial lhs, ModularPolynomial& Q,ModularPolynomial& R);
};

ModularPolynomial operator+(const ModularPolynomial& rhs, const ModularPolynomial& lhs);
ModularPolynomial operator-(const ModularPolynomial& rhs, const ModularPolynomial& lhs);
ModularPolynomial operator*(const ModularPolynomial& rhs, const ModularPolynomial& lhs);
ModularPolynomial operator/(const ModularPolynomial& rhs, const ModularPolynomial& lhs);
ModularPolynomial gcd(const ModularPolynomial& rhs, const ModularPolynomial& lhs);


ostream& operator<<(ostream& o, const ModularPolynomial& lhs);
ostream& operator<<(ostream& o, const NumberArray lhs);

class HCP{ //Hilbert class polynomials
public:
    HCP();
    map<int, string> H;
    int degree(int D);
};


#endif	/* _HCP_H */