#!/usr/bin/env python
# coding: utf-8

import quadpy

un_spheres = [
    "dobrodeev_1978",
    "mysovskikh_1",
    "mysovskikh_2",
    "stroud_un_3_1",
    "stroud_un_3_2",
    "stroud_un_5_1",
    "stroud_un_5_2",
    "stroud_un_5_3",
    "stroud_un_5_4",
    "stroud_un_7_2",
    "stroud_un_7_1",
    "stroud_un_11_1",
#    "stroud_1967", # dup of "stroud_un_7_1"
#    "stroud_1969", # dup of "stroud_un_11_1"
]

#č Bacha! Lidstvo nemusí nutně dočkat 
#č ukončení výpočtu této funkce
def get_all_un_sphere_schemes(dim):
    schemes = dict()
    for item in un_spheres:
        Scheme = getattr(quadpy.un, item)
        try:
            scheme = Scheme(dim)
            schemes[scheme.name] = scheme
        except:
            pass
            
    return schemes



# up to degree 7. Ought to be enough for anybody
#č 7 stupňů musí stačit každému
def get_all_tn_simplex_schemes(dim):
    #č současný qt_plot vytahuje schemata ze slovníku
    #č aspoň netřeba v dlouhém seznamu dohledávat 
    schemes = dict()
    # Grundmann, Möller
    for i in range(8):
        scheme = quadpy.tn.grundmann_moeller(dim, i)
        schemes[scheme.name] = scheme
    # Silvester
    for i in range(7):
        scheme = quadpy.tn.silvester(dim, variant="open", n=i+1, symbolic=False)
        schemes[scheme.name] = scheme
        scheme = quadpy.tn.silvester(dim, variant="closed", n=i+1, symbolic=False)
        schemes[scheme.name] = scheme
    
    # Laufer, Hammer and Stroud
    for item in tn_simplices:
        Scheme = getattr(quadpy.tn, item)
        try:
            scheme = Scheme(dim)
            schemes[scheme.name] = scheme
        except:
            pass
        
    return schemes
    
tn_simplices = [
                'stroud_tn_1_1', 
                'stroud_tn_1_2', 
                'stroud_tn_2_1a', 
                'stroud_tn_2_1b', 
                'stroud_tn_2_2', 
                'stroud_tn_3_1', 
                'stroud_tn_3_2', 
                'stroud_tn_3_3', 
                'stroud_tn_3_4', 
                'stroud_tn_3_5', 
                'stroud_tn_3_6a', 
                'stroud_tn_3_6b', 
                'stroud_tn_3_7', 
                'stroud_tn_3_8', 
                'stroud_tn_3_9', 
                'stroud_tn_3_10', 
                'stroud_tn_3_11', 
                'stroud_tn_4_1', 
                'stroud_tn_5_1', 
                'stroud_tn_5_2', 
                'walkington_1', 
                'walkington_2', 
                'walkington_3', 
                'walkington_5', 
                'walkington_7'
                ]