#!/usr/bin/python

# 10.3

import scipy as sp

def imcsplit(B, polynomial):

    """ Splits a polynomial B into good, nonminimum with 

    positive real & with negative real parts.

    All are returned in polynomial form.

    Gain is returned in Kp and delay in k."""

    k = 0

    Kp = 1

    if polynomial:

        roots = sp.roots(B)

        Kp = sum(B)/sum(sp.poly(roots))

    else:

        roots = B

    Bg, Bnmp, Bm = sp.array([1]), sp.array([1]), sp.array([1])

    for root in roots:

        if root == 0:

            k += 1

        elif abs(root)<1 and root.real>=0:

            Bg = sp.convolve(Bg, [1, -root])

        elif abs(root)>=1 and root.real>=0:

            Bnmp = sp.convolve(Bnmp, [1, -root])

        else:

            Bm = sp.convolve(Bm, [1, -root])

    return Kp, k, Bg, Bnmp, Bm